mirror of
https://github.com/rocky-linux/peridot.git
synced 2024-11-30 08:36:30 +00:00
268 lines
6.2 KiB
Go
268 lines
6.2 KiB
Go
package jsoniter
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// ReadObject read one field from object.
|
|
// If object ended, returns empty string.
|
|
// Otherwise, returns the field name.
|
|
func (iter *Iterator) ReadObject() (ret string) {
|
|
c := iter.nextToken()
|
|
switch c {
|
|
case 'n':
|
|
iter.skipThreeBytes('u', 'l', 'l')
|
|
return "" // null
|
|
case '{':
|
|
c = iter.nextToken()
|
|
if c == '"' {
|
|
iter.unreadByte()
|
|
field := iter.ReadString()
|
|
c = iter.nextToken()
|
|
if c != ':' {
|
|
iter.ReportError("ReadObject", "expect : after object field, but found "+string([]byte{c}))
|
|
}
|
|
return field
|
|
}
|
|
if c == '}' {
|
|
return "" // end of object
|
|
}
|
|
iter.ReportError("ReadObject", `expect " after {, but found `+string([]byte{c}))
|
|
return
|
|
case ',':
|
|
field := iter.ReadString()
|
|
c = iter.nextToken()
|
|
if c != ':' {
|
|
iter.ReportError("ReadObject", "expect : after object field, but found "+string([]byte{c}))
|
|
}
|
|
return field
|
|
case '}':
|
|
return "" // end of object
|
|
default:
|
|
iter.ReportError("ReadObject", fmt.Sprintf(`expect { or , or } or n, but found %s`, string([]byte{c})))
|
|
return
|
|
}
|
|
}
|
|
|
|
// CaseInsensitive
|
|
func (iter *Iterator) readFieldHash() int64 {
|
|
hash := int64(0x811c9dc5)
|
|
c := iter.nextToken()
|
|
if c != '"' {
|
|
iter.ReportError("readFieldHash", `expect ", but found `+string([]byte{c}))
|
|
return 0
|
|
}
|
|
for {
|
|
for i := iter.head; i < iter.tail; i++ {
|
|
// require ascii string and no escape
|
|
b := iter.buf[i]
|
|
if b == '\\' {
|
|
iter.head = i
|
|
for _, b := range iter.readStringSlowPath() {
|
|
if 'A' <= b && b <= 'Z' && !iter.cfg.caseSensitive {
|
|
b += 'a' - 'A'
|
|
}
|
|
hash ^= int64(b)
|
|
hash *= 0x1000193
|
|
}
|
|
c = iter.nextToken()
|
|
if c != ':' {
|
|
iter.ReportError("readFieldHash", `expect :, but found `+string([]byte{c}))
|
|
return 0
|
|
}
|
|
return hash
|
|
}
|
|
if b == '"' {
|
|
iter.head = i + 1
|
|
c = iter.nextToken()
|
|
if c != ':' {
|
|
iter.ReportError("readFieldHash", `expect :, but found `+string([]byte{c}))
|
|
return 0
|
|
}
|
|
return hash
|
|
}
|
|
if 'A' <= b && b <= 'Z' && !iter.cfg.caseSensitive {
|
|
b += 'a' - 'A'
|
|
}
|
|
hash ^= int64(b)
|
|
hash *= 0x1000193
|
|
}
|
|
if !iter.loadMore() {
|
|
iter.ReportError("readFieldHash", `incomplete field name`)
|
|
return 0
|
|
}
|
|
}
|
|
}
|
|
|
|
func calcHash(str string, caseSensitive bool) int64 {
|
|
if !caseSensitive {
|
|
str = strings.ToLower(str)
|
|
}
|
|
hash := int64(0x811c9dc5)
|
|
for _, b := range []byte(str) {
|
|
hash ^= int64(b)
|
|
hash *= 0x1000193
|
|
}
|
|
return int64(hash)
|
|
}
|
|
|
|
// ReadObjectCB read object with callback, the key is ascii only and field name not copied
|
|
func (iter *Iterator) ReadObjectCB(callback func(*Iterator, string) bool) bool {
|
|
c := iter.nextToken()
|
|
var field string
|
|
if c == '{' {
|
|
if !iter.incrementDepth() {
|
|
return false
|
|
}
|
|
c = iter.nextToken()
|
|
if c == '"' {
|
|
iter.unreadByte()
|
|
field = iter.ReadString()
|
|
c = iter.nextToken()
|
|
if c != ':' {
|
|
iter.ReportError("ReadObject", "expect : after object field, but found "+string([]byte{c}))
|
|
}
|
|
if !callback(iter, field) {
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
c = iter.nextToken()
|
|
for c == ',' {
|
|
field = iter.ReadString()
|
|
c = iter.nextToken()
|
|
if c != ':' {
|
|
iter.ReportError("ReadObject", "expect : after object field, but found "+string([]byte{c}))
|
|
}
|
|
if !callback(iter, field) {
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
c = iter.nextToken()
|
|
}
|
|
if c != '}' {
|
|
iter.ReportError("ReadObjectCB", `object not ended with }`)
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
return iter.decrementDepth()
|
|
}
|
|
if c == '}' {
|
|
return iter.decrementDepth()
|
|
}
|
|
iter.ReportError("ReadObjectCB", `expect " after {, but found `+string([]byte{c}))
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
if c == 'n' {
|
|
iter.skipThreeBytes('u', 'l', 'l')
|
|
return true // null
|
|
}
|
|
iter.ReportError("ReadObjectCB", `expect { or n, but found `+string([]byte{c}))
|
|
return false
|
|
}
|
|
|
|
// ReadMapCB read map with callback, the key can be any string
|
|
func (iter *Iterator) ReadMapCB(callback func(*Iterator, string) bool) bool {
|
|
c := iter.nextToken()
|
|
if c == '{' {
|
|
if !iter.incrementDepth() {
|
|
return false
|
|
}
|
|
c = iter.nextToken()
|
|
if c == '"' {
|
|
iter.unreadByte()
|
|
field := iter.ReadString()
|
|
if iter.nextToken() != ':' {
|
|
iter.ReportError("ReadMapCB", "expect : after object field, but found "+string([]byte{c}))
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
if !callback(iter, field) {
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
c = iter.nextToken()
|
|
for c == ',' {
|
|
field = iter.ReadString()
|
|
if iter.nextToken() != ':' {
|
|
iter.ReportError("ReadMapCB", "expect : after object field, but found "+string([]byte{c}))
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
if !callback(iter, field) {
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
c = iter.nextToken()
|
|
}
|
|
if c != '}' {
|
|
iter.ReportError("ReadMapCB", `object not ended with }`)
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
return iter.decrementDepth()
|
|
}
|
|
if c == '}' {
|
|
return iter.decrementDepth()
|
|
}
|
|
iter.ReportError("ReadMapCB", `expect " after {, but found `+string([]byte{c}))
|
|
iter.decrementDepth()
|
|
return false
|
|
}
|
|
if c == 'n' {
|
|
iter.skipThreeBytes('u', 'l', 'l')
|
|
return true // null
|
|
}
|
|
iter.ReportError("ReadMapCB", `expect { or n, but found `+string([]byte{c}))
|
|
return false
|
|
}
|
|
|
|
func (iter *Iterator) readObjectStart() bool {
|
|
c := iter.nextToken()
|
|
if c == '{' {
|
|
c = iter.nextToken()
|
|
if c == '}' {
|
|
return false
|
|
}
|
|
iter.unreadByte()
|
|
return true
|
|
} else if c == 'n' {
|
|
iter.skipThreeBytes('u', 'l', 'l')
|
|
return false
|
|
}
|
|
iter.ReportError("readObjectStart", "expect { or n, but found "+string([]byte{c}))
|
|
return false
|
|
}
|
|
|
|
func (iter *Iterator) readObjectFieldAsBytes() (ret []byte) {
|
|
str := iter.ReadStringAsSlice()
|
|
if iter.skipWhitespacesWithoutLoadMore() {
|
|
if ret == nil {
|
|
ret = make([]byte, len(str))
|
|
copy(ret, str)
|
|
}
|
|
if !iter.loadMore() {
|
|
return
|
|
}
|
|
}
|
|
if iter.buf[iter.head] != ':' {
|
|
iter.ReportError("readObjectFieldAsBytes", "expect : after object field, but found "+string([]byte{iter.buf[iter.head]}))
|
|
return
|
|
}
|
|
iter.head++
|
|
if iter.skipWhitespacesWithoutLoadMore() {
|
|
if ret == nil {
|
|
ret = make([]byte, len(str))
|
|
copy(ret, str)
|
|
}
|
|
if !iter.loadMore() {
|
|
return
|
|
}
|
|
}
|
|
if ret == nil {
|
|
return str
|
|
}
|
|
return ret
|
|
}
|