go-json/decode_uint.go

104 lines
2.3 KiB
Go
Raw Permalink Normal View History

2020-04-23 19:39:20 +03:00
package json
type uintDecoder struct {
op func(uintptr, uint64)
}
func newUintDecoder(op func(uintptr, uint64)) *uintDecoder {
return &uintDecoder{op: op}
}
2020-04-24 08:07:33 +03:00
var pow10u64 = [...]uint64{
1e00, 1e01, 1e02, 1e03, 1e04, 1e05, 1e06, 1e07, 1e08, 1e09,
1e10, 1e11, 1e12, 1e13, 1e14, 1e15, 1e16, 1e17, 1e18, 1e19,
}
2020-04-23 19:39:20 +03:00
func (d *uintDecoder) parseUint(b []byte) uint64 {
maxDigit := len(b)
sum := uint64(0)
for i := 0; i < maxDigit; i++ {
c := uint64(b[i]) - 48
2020-04-24 08:07:33 +03:00
digitValue := pow10u64[maxDigit-i-1]
2020-04-23 19:39:20 +03:00
sum += c * digitValue
}
return sum
}
2020-08-11 13:05:20 +03:00
func (d *uintDecoder) setDisallowUnknownFields(_ bool) {}
2020-07-30 16:41:53 +03:00
func (d *uintDecoder) decodeStreamByte(s *stream) ([]byte, error) {
for {
switch s.char() {
case ' ', '\n', '\t', '\r':
2020-07-31 11:10:03 +03:00
s.cursor++
2020-07-30 16:41:53 +03:00
continue
case '0', '1', '2', '3', '4', '5', '6', '7', '8', '9':
start := s.cursor
2020-07-31 11:10:03 +03:00
for {
s.cursor++
if numTable[s.char()] {
2020-07-30 16:41:53 +03:00
continue
2020-07-31 11:10:03 +03:00
} else if s.char() == nul {
if s.read() {
2020-07-31 12:07:11 +03:00
s.cursor-- // for retry current character
2020-07-31 11:10:03 +03:00
continue
}
2020-07-30 16:41:53 +03:00
}
break
}
num := s.buf[start:s.cursor]
return num, nil
2020-07-31 11:10:03 +03:00
case nul:
if s.read() {
continue
}
2020-07-30 16:41:53 +03:00
}
2020-07-31 11:10:03 +03:00
break
2020-07-30 16:41:53 +03:00
}
return nil, errUnexpectedEndOfJSON("number(unsigned integer)", s.totalOffset())
}
2020-05-23 06:51:09 +03:00
func (d *uintDecoder) decodeByte(buf []byte, cursor int64) ([]byte, int64, error) {
buflen := int64(len(buf))
2020-04-24 08:07:33 +03:00
for ; cursor < buflen; cursor++ {
switch buf[cursor] {
case ' ', '\n', '\t', '\r':
continue
case '0', '1', '2', '3', '4', '5', '6', '7', '8', '9':
start := cursor
cursor++
for ; cursor < buflen; cursor++ {
tk := int(buf[cursor])
if int('0') <= tk && tk <= int('9') {
continue
}
break
2020-04-23 19:39:20 +03:00
}
2020-04-24 08:07:33 +03:00
num := buf[start:cursor]
2020-05-06 20:37:29 +03:00
return num, cursor, nil
2020-05-23 06:51:09 +03:00
default:
return nil, 0, errInvalidCharacter(buf[cursor], "number(unsigned integer)", cursor)
2020-04-23 19:39:20 +03:00
}
}
2020-05-23 06:51:09 +03:00
return nil, 0, errUnexpectedEndOfJSON("number(unsigned integer)", cursor)
2020-04-23 19:39:20 +03:00
}
2020-07-30 16:41:53 +03:00
func (d *uintDecoder) decodeStream(s *stream, p uintptr) error {
bytes, err := d.decodeStreamByte(s)
if err != nil {
return err
}
d.op(p, d.parseUint(bytes))
return nil
}
2020-05-23 06:51:09 +03:00
func (d *uintDecoder) decode(buf []byte, cursor int64, p uintptr) (int64, error) {
2020-05-06 20:37:29 +03:00
bytes, c, err := d.decodeByte(buf, cursor)
2020-04-23 19:39:20 +03:00
if err != nil {
2020-05-06 20:37:29 +03:00
return 0, err
2020-04-23 19:39:20 +03:00
}
2020-05-06 20:37:29 +03:00
cursor = c
2020-04-23 19:39:20 +03:00
d.op(p, d.parseUint(bytes))
2020-05-06 20:37:29 +03:00
return cursor, nil
2020-04-23 19:39:20 +03:00
}