go-json/decode_int.go

73 lines
1.4 KiB
Go
Raw Normal View History

2020-04-23 19:39:20 +03:00
package json
import (
"errors"
)
type intDecoder struct {
op func(uintptr, int64)
}
func newIntDecoder(op func(uintptr, int64)) *intDecoder {
return &intDecoder{op: op}
}
2020-04-24 08:07:33 +03:00
var (
pow10i64 = [...]int64{
1e00, 1e01, 1e02, 1e03, 1e04, 1e05, 1e06, 1e07, 1e08, 1e09,
1e10, 1e11, 1e12, 1e13, 1e14, 1e15, 1e16, 1e17, 1e18,
}
)
2020-04-23 19:39:20 +03:00
func (d *intDecoder) parseInt(b []byte) int64 {
isNegative := false
if b[0] == '-' {
b = b[1:]
isNegative = true
}
maxDigit := len(b)
sum := int64(0)
for i := 0; i < maxDigit; i++ {
c := int64(b[i]) - 48
2020-04-24 08:07:33 +03:00
digitValue := pow10i64[maxDigit-i-1]
2020-04-23 19:39:20 +03:00
sum += c * digitValue
}
if isNegative {
return -1 * sum
}
return sum
}
2020-05-06 20:37:29 +03:00
func (d *intDecoder) decodeByte(buf []byte, cursor int) ([]byte, int, error) {
buflen := 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-05-06 20:37:29 +03:00
num := buf[start:cursor]
return num, cursor, nil
2020-04-23 19:39:20 +03:00
}
}
2020-05-06 20:37:29 +03:00
return nil, 0, errors.New("unexpected error number")
2020-04-23 19:39:20 +03:00
}
2020-05-06 20:37:29 +03:00
func (d *intDecoder) decode(buf []byte, cursor int, p uintptr) (int, error) {
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.parseInt(bytes))
2020-05-06 20:37:29 +03:00
return cursor, nil
2020-04-23 19:39:20 +03:00
}