go-json/decode_int.go

89 lines
1.5 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-07 08:21:29 +03:00
var (
numTable = [256]bool{
'0': true,
'1': true,
'2': true,
'3': true,
'4': true,
'5': true,
'6': true,
'7': true,
'8': true,
'9': true,
}
)
2020-05-06 20:37:29 +03:00
func (d *intDecoder) decodeByte(buf []byte, cursor int) ([]byte, int, error) {
2020-05-07 08:21:29 +03:00
for {
2020-04-24 08:07:33 +03:00
switch buf[cursor] {
case ' ', '\n', '\t', '\r':
2020-05-07 08:21:29 +03:00
cursor++
2020-04-24 08:07:33 +03:00
continue
case '-', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9':
start := cursor
cursor++
2020-05-07 08:21:29 +03:00
LOOP:
if numTable[buf[cursor]] {
cursor++
goto LOOP
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-05-07 08:21:29 +03:00
default:
goto ERROR
2020-04-23 19:39:20 +03:00
}
}
2020-05-07 08:21:29 +03:00
ERROR:
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
}