2
0
Fork 0
mirror of https://github.com/ii64/sonic.git synced 2026-06-21 00:46:43 +08:00
sonic/ast/node.go
Yi Duan 8c119dd72d
feat(ast): support Node.Unset() and optimize Node.Get() (#80)
Co-authored-by: duanyi.aster <duanyi.aster@bytedance.com>
2021-08-24 13:23:55 +08:00

1197 lines
No EOL
29 KiB
Go

/*
* Copyright 2021 ByteDance Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package ast
import (
`encoding/json`
`unsafe`
`github.com/bytedance/sonic/internal/native/types`
`github.com/bytedance/sonic/internal/rt`
`github.com/bytedance/sonic/unquote`
)
const (
_CAP_BITS = 32
_LEN_MASK = 1 << _CAP_BITS - 1
_NODE_SIZE = unsafe.Sizeof(Node{})
_PAIR_SIZE = unsafe.Sizeof(Pair{})
)
const (
_V_NONE types.ValueType = 0
_V_NODE_BASE types.ValueType = 1<<5
_V_LAZY types.ValueType = 1 << 7
_V_RAW types.ValueType = 1 << 8
_V_NUMBER = _V_NODE_BASE + 1
_V_ARRAY_LAZY = _V_LAZY | types.V_ARRAY
_V_OBJECT_LAZY = _V_LAZY | types.V_OBJECT
_MASK_LAZY = _V_LAZY - 1
_MASK_RAW = _V_RAW - 1
)
const (
V_NONE = 0
V_NULL = 2
V_TRUE = 3
V_FALSE = 4
V_ARRAY = 5
V_OBJECT = 6
V_STRING = 7
V_NUMBER = int(_V_NUMBER)
)
type Node struct {
v int64
t types.ValueType
p unsafe.Pointer
}
/** Node Type Accessor **/
// Type returns json type represented by the node
// It will be one of belows:
// V_NONE = 0
// V_NULL = 2
// V_TRUE = 3
// V_FALSE = 4
// V_ARRAY = 5
// V_OBJECT = 6
// V_STRING = 7
// V_NUMBER = 33
func (self *Node) Type() int {
return int(self.t & _MASK_LAZY & _MASK_RAW)
}
func (self *Node) itype() types.ValueType {
return self.t & _MASK_LAZY & _MASK_RAW
}
// Exists returns false only if the node is nil or got by invalid path
func (self *Node) Exists() bool {
return self != nil && self.t != _V_NONE
}
// IsRaw returns true if node's underlying value is raw json
func (self *Node) IsRaw() bool {
return self.t&_V_RAW != 0
}
func (self *Node) isLazy() bool {
return self.t&_V_LAZY != 0
}
/** Simple Value Methods **/
// Raw returns underlying json string of an raw node,
// which usually created by Search() api
func (self *Node) Raw() string {
if !self.IsRaw() {
panic("value cannot be represented as raw json")
}
return addr2str(self.p, self.v)
}
func (self *Node) checkRaw() {
if !self.IsRaw() {
return
}
*self = self.parseRaw()
}
// Bool returns bool value represented by this node
//
// If node type is not types.V_TRUE or types.V_FALSE, or V_RAW (must be a bool json value)
// it will panic
func (self *Node) Bool() bool {
self.checkRaw()
switch self.t {
case types.V_TRUE : return true
case types.V_FALSE : return false
default : panic("value cannot be represented as a boolean")
}
}
// Int64 as above.
func (self *Node) Int64() int64 {
self.checkRaw()
switch self.t {
case _V_NUMBER : return numberToInt64(self)
case types.V_TRUE : return 1
case types.V_FALSE : return 0
default : panic("value cannot be represented as an integer")
}
}
// Number as above.
func (self *Node) Number() json.Number {
self.checkRaw()
switch self.t {
case _V_NUMBER : return toNumber(self)
default : panic("value cannot be represented as a json.Number")
}
}
// String as above.
func (self *Node) String() string {
self.checkRaw()
switch self.t {
case _V_NUMBER : return toNumber(self).String()
case types.V_NULL : return "null"
case types.V_TRUE : return "true"
case types.V_FALSE : return "false"
case types.V_STRING : return addr2str(self.p, self.v)
default : panic("value cannot be represented as a simple string")
}
}
// Float64 as above.
func (self *Node) Float64() float64 {
self.checkRaw()
switch self.t {
case _V_NUMBER : return numberToFloat64(self)
case types.V_TRUE : return 1.0
case types.V_FALSE : return 0.0
default : panic("value cannot be represented as an integer")
}
}
/** Sequencial Value Methods **/
// Len returns children count of a array|object|string node
// For partially loaded node, it also works but only counts the parsed children
func (self *Node) Len() int {
self.checkRaw()
if self.t == types.V_ARRAY || self.t == types.V_OBJECT || self.t == _V_ARRAY_LAZY || self.t == _V_OBJECT_LAZY {
return int(self.v & _LEN_MASK)
} else if self.t == types.V_STRING {
return int(self.v)
} else {
panic("value does not have a length")
}
}
func (self *Node) len() int {
return int(self.v & _LEN_MASK)
}
// Cap returns malloc capacity of a array|object node for children
func (self *Node) Cap() int {
self.checkRaw()
if self.t == types.V_ARRAY || self.t == types.V_OBJECT || self.t == _V_ARRAY_LAZY || self.t == _V_OBJECT_LAZY {
return int(self.v >> _CAP_BITS)
} else {
panic("value does not have a capacity")
}
}
func (self *Node) cap() int {
return int(self.v >> _CAP_BITS)
}
// Set sets the node of given key under object parent
// If the key doesn't exist, it will be append to the last
func (self *Node) Set(key string, node Node) {
p := self.Get(key)
if !p.Exists() {
l := self.len()
c := self.cap()
if l == c {
// TODO: maybe change append_extra_size in future
c += _DEFAULT_NODE_CAP
mem := unsafe_NewArray(_PAIR_TYPE, c)
memmove(mem, self.p, _PAIR_SIZE * uintptr(l))
self.p = mem
}
v := self.pairAt(l)
v.Key = key
v.Value = node
self.setCapAndLen(c, l+1)
} else {
*p = node
}
}
// Unset remove the node of given key under object parent
func (self *Node) Unset(key string) (exist bool) {
self.must(types.V_OBJECT, "an object")
n, i := self.skipKey(key)
if !n.Exists() {
return false
}
self.removePair(i)
return true
}
// SetByIndex sets the node of given index
//
// The index must within parent array's children
func (self *Node) SetByIndex(index int, node Node) {
p := self.Index(index)
if !p.Exists() {
panic("index to nil value")
} else {
*p = node
}
}
// UnsetByIndex remove the node of given index
func (self *Node) UnsetByIndex(index int) (exist bool) {
var p *Node
it := self.itype()
if it == types.V_ARRAY {
p = self.Index(index)
}else if it == types.V_OBJECT {
pr := self.skipIndexPair(index)
if pr == nil {
return false
}
p = &pr.Value
}else{
panic("value must be object or array type")
}
if !p.Exists() {
return false
}
if it == types.V_ARRAY {
self.removeNode(index)
}else if it == types.V_OBJECT {
self.removePair(index)
}
return true
}
// Add appends the given node under array node
func (self *Node) Add(node Node) {
self.must(types.V_ARRAY, "an array")
self.skipAllIndex()
l := self.len()
c := self.cap()
if l == c {
// TODO: maybe change append_extra_size in future
c += _DEFAULT_NODE_CAP
mem := unsafe_NewArray(_NODE_TYPE, c)
memmove(mem, self.p, _NODE_SIZE * uintptr(l))
self.p = mem
}
v := self.nodeAt(l)
*v = node
self.setCapAndLen(c, l+1)
}
// GetByPath load given path on demands,
// which only ensure nodes before this path got parsed
func (self *Node) GetByPath(path ...interface{}) *Node {
var s = self
for _, p := range path {
switch p.(type) {
case int:
s = s.Index(p.(int))
if !s.Exists() {
return s
}
case string:
s = s.Get(p.(string))
if !s.Exists() {
return s
}
default:
panic("path must be either int or string")
}
}
return s
}
// Get loads given key of an object node on demands
func (self *Node) Get(key string) *Node {
self.must(types.V_OBJECT, "an object")
n, _ := self.skipKey(key)
return n
}
// Index loads given index of an node on demands,
// node type can be either V_OBJECT or V_ARRAY
func (self *Node) Index(idx int) *Node {
self.checkRaw()
it := self.itype()
if it == types.V_ARRAY {
return self.skipIndex(idx)
}else if it == types.V_OBJECT {
pr := self.skipIndexPair(idx)
if pr == nil {
return &Node{}
}
return &pr.Value
}else{
panic("node must be object or array type")
}
}
// Values returns iterator for array's children traversal
func (self *Node) Values() ListIterator {
self.must(types.V_ARRAY, "an array")
self.skipAllIndex()
return ListIterator{Iterator{p: self}}
}
// Properties returns iterator for object's children traversal
func (self *Node) Properties() ObjectIterator {
self.must(types.V_OBJECT, "an object")
self.skipAllKey()
return ObjectIterator{Iterator{p: self}}
}
/** Generic Value Converters **/
// Map loads all keys of an object node
func (self *Node) Map() map[string]interface{} {
self.must(types.V_OBJECT, "an object")
self.loadAllKey()
return self.toGenericObject()
}
// MapUseNumber loads all keys of an object node, with numeric nodes casted to json.Number
func (self *Node) MapUseNumber() map[string]interface{} {
self.must(types.V_OBJECT, "an object")
self.loadAllKey()
return self.toGenericObjectUseNumber()
}
// MapUseNode scans both parsed and non-parsed chidren nodes,
// and map them by their keys
func (self *Node) MapUseNode() map[string]Node {
self.must(types.V_OBJECT, "an object")
self.skipAllKey()
return self.toGenericObjectUseNode()
}
// MapUnsafe exports the underlying pointer to its children map
// WARN: don't use it unless you know what you are doing
func (self *Node) UnsafeMap() []Pair {
self.must(types.V_OBJECT, "an object")
self.skipAllKey()
s := ptr2slice(self.p, int(self.len()), self.cap())
return *(*[]Pair)(s)
}
// Array loads all indexes of an array node
func (self *Node) Array() []interface{} {
self.must(types.V_ARRAY, "an array")
self.loadAllIndex()
return self.toGenericArray()
}
// ArrayUseNumber loads all indexes of an array node, with numeric nodes casted to json.Number
func (self *Node) ArrayUseNumber() []interface{} {
self.must(types.V_ARRAY, "an array")
self.loadAllIndex()
return self.toGenericArrayUseNumber()
}
// ArrayUseNode copys both parsed and non-parsed chidren nodes,
// and indexes them by original order
func (self *Node) ArrayUseNode() []Node {
self.must(types.V_ARRAY, "an array")
self.skipAllIndex()
return self.toGenericArrayUseNode()
}
// ArrayUnsafe exports the underlying pointer to its children array
// WARN: don't use it unless you know what you are doing
func (self *Node) UnsafeArray() []Node {
self.must(types.V_ARRAY, "an array")
self.skipAllIndex()
s := ptr2slice(self.p, self.len(), self.Cap())
return *(*[]Node)(s)
}
// Interface loads all children under all pathes from this node,
// and converts itself as generic go type
// all numberic nodes are casted to float64
func (self *Node) Interface() interface{} {
self.checkRaw()
switch self.t {
case types.V_EOF : panic("invalid value")
case types.V_NULL : return nil
case types.V_TRUE : return true
case types.V_FALSE : return false
case types.V_ARRAY : return self.toGenericArray()
case types.V_OBJECT : return self.toGenericObject()
case types.V_STRING : return addr2str(self.p, self.v)
case _V_NUMBER : return numberToFloat64(self)
case _V_ARRAY_LAZY :
self.loadAllIndex()
return self.toGenericArray()
case _V_OBJECT_LAZY :
self.loadAllKey()
return self.toGenericObject()
default : panic("not gonna happen")
}
}
// InterfaceUseNumber works same with Interface()
// except numberic nodes are casted to json.Number
func (self *Node) InterfaceUseNumber() interface{} {
self.checkRaw()
switch self.t {
case types.V_EOF : panic("invalid value")
case types.V_NULL : return nil
case types.V_TRUE : return true
case types.V_FALSE : return false
case types.V_ARRAY : return self.toGenericArrayUseNumber()
case types.V_OBJECT : return self.toGenericObjectUseNumber()
case types.V_STRING : return addr2str(self.p, self.v)
case _V_NUMBER : return toNumber(self)
case _V_ARRAY_LAZY :
self.loadAllIndex()
return self.toGenericArrayUseNumber()
case _V_OBJECT_LAZY :
self.loadAllKey()
return self.toGenericObjectUseNumber()
default : panic("not gonna happen")
}
}
// InterfaceUseNode clone itself as a new node,
// or its children as map[string]Node (or []Node)
func (self *Node) InterfaceUseNode() interface{} {
self.checkRaw()
switch self.t {
case types.V_ARRAY : return self.toGenericArrayUseNode()
case types.V_OBJECT : return self.toGenericObjectUseNode()
case _V_ARRAY_LAZY:
self.skipAllIndex()
return self.toGenericArrayUseNode()
case _V_OBJECT_LAZY:
self.skipAllKey()
return self.toGenericObjectUseNode()
default : return *self
}
}
/**---------------------------------- Internal Helper Methods ----------------------------------**/
var (
_NODE_TYPE = rt.UnpackEface(Node{}).Type
_PAIR_TYPE = rt.UnpackEface(Pair{}).Type
)
func (self *Node) setCapAndLen(cap int, len int) {
if self.t == types.V_ARRAY || self.t == types.V_OBJECT || self.t == _V_ARRAY_LAZY || self.t == _V_OBJECT_LAZY {
self.v = int64(len&_LEN_MASK | cap<<_CAP_BITS)
} else {
panic("value does not have a length")
}
}
func (self *Node) unsafe_next() *Node {
return (*Node)(unsafe.Pointer(uintptr(unsafe.Pointer(self)) + _NODE_SIZE))
}
func (self *Pair) unsafe_next() *Pair {
return (*Pair)(unsafe.Pointer(uintptr(unsafe.Pointer(self)) + _PAIR_SIZE))
}
func (self *Node) must(t types.ValueType, s string) {
self.checkRaw()
if self.itype() != t {
panic("value cannot be represented as " + s)
}
}
func (self *Node) bound(i int) {
if i < 0 || i >= self.Len() {
panic("list index out of range")
}
}
func (self *Node) nodeAt(i int) *Node {
var p = self.p
if self.isLazy() {
_, stack := self.getParserAndArrayStack()
p = *(*unsafe.Pointer)(unsafe.Pointer(&stack.v))
}
return (*Node)(unsafe.Pointer(uintptr(p) + uintptr(i)*_NODE_SIZE))
}
func (self *Node) pairAt(i int) *Pair {
var p = self.p
if self.isLazy() {
_, stack := self.getParserAndObjectStack()
p = *(*unsafe.Pointer)(unsafe.Pointer(&stack.v))
}
return (*Pair)(unsafe.Pointer(uintptr(p) + uintptr(i)*_PAIR_SIZE))
}
func (self *Node) findKey(key string) (*Node, int) {
nb := self.len()
if nb <= 0 {
return nil, -1
}
var p *Pair
if !self.isLazy() {
p = (*Pair)(self.p)
} else {
s := (*parseObjectStack)(self.p)
p = &s.v[0]
}
if p.Key == key {
return &p.Value, 0
}
for i := 1; i < nb; i++ {
p = p.unsafe_next()
if p.Key == key {
return &p.Value, i
}
}
/* not found */
return nil, -1
}
func (self *Node) getParserAndArrayStack() (*Parser, *parseArrayStack) {
stack := (*parseArrayStack)(self.p)
ret := (*rt.GoSlice)(unsafe.Pointer(&stack.v))
ret.Len = self.len()
ret.Cap = self.cap()
return &stack.parser, stack
}
func (self *Node) getParserAndObjectStack() (*Parser, *parseObjectStack) {
stack := (*parseObjectStack)(self.p)
ret := (*rt.GoSlice)(unsafe.Pointer(&stack.v))
ret.Len = self.len()
ret.Cap = self.cap()
return &stack.parser, stack
}
func (self *Node) skipAllIndex() {
if !self.isLazy() {
return
}
var err types.ParsingError
parser, stack := self.getParserAndArrayStack()
parser.skipValue = true
parser.noLazy = true
*self, err = parser.decodeArray(stack.v)
if err != 0 {
panic(parser.ExportError(err))
}
}
func (self *Node) skipAllKey() {
if !self.isLazy() {
return
}
var err types.ParsingError
parser, stack := self.getParserAndObjectStack()
parser.skipValue = true
parser.noLazy = true
*self, err = parser.decodeObject(stack.v)
if err != 0 {
panic(parser.ExportError(err))
}
}
func (self *Node) skipNextNode() *Node {
if !self.isLazy() {
return nil
}
parser, stack := self.getParserAndArrayStack()
ret := stack.v
sp := parser.p
ns := len(parser.s)
/* check for EOF */
if parser.p = parser.lspace(sp); parser.p >= ns {
panic(parser.ExportError(types.ERR_EOF))
}
/* check for empty array */
if parser.s[parser.p] == ']' {
parser.p++
self.setArray(ret)
return nil
}
var val Node
/* skip the value */
if start, err := parser.skip(); err != 0 {
panic(parser.ExportError(err))
}else{
t := switchRawType(parser.s[start])
if t == _V_NONE {
panic(parser.ExportError(types.ERR_INVALID_CHAR))
}
val = newRawNode(parser.s[start:parser.p], t)
}
/* add the value to result */
ret = append(ret, val)
parser.p = parser.lspace(parser.p)
/* check for EOF */
if parser.p >= ns {
panic(parser.ExportError(types.ERR_EOF))
}
/* check for the next character */
switch parser.s[parser.p] {
case ',':
parser.p++
self.setLazyArray(parser, ret)
return &ret[len(ret)-1]
case ']':
parser.p++
self.setArray(ret)
return &ret[len(ret)-1]
default:
panic(parser.ExportError(types.ERR_INVALID_CHAR))
}
}
func (self *Node) skipNextPair() (*Pair) {
if !self.isLazy() {
return nil
}
parser, stack := self.getParserAndObjectStack()
ret := stack.v
sp := parser.p
ns := len(parser.s)
/* check for EOF */
if parser.p = parser.lspace(sp); parser.p >= ns {
panic(parser.ExportError(types.ERR_EOF))
}
/* check for empty object */
if parser.s[parser.p] == '}' {
parser.p++
self.setObject(ret)
return nil
}
/* decode one pair */
var val Node
var njs types.JsonState
var err types.ParsingError
/* decode the key */
if njs = parser.decodeValue(); njs.Vt != types.V_STRING {
panic(parser.ExportError(types.ERR_INVALID_CHAR))
}
/* extract the key */
idx := parser.p - 1
key := parser.s[njs.Iv:idx]
/* check for escape sequence */
if njs.Ep != -1 {
if key, err = unquote.String(key); err != 0 {
panic(parser.ExportError(err))
}
}
/* expect a ':' delimiter */
if err = parser.delim(); err != 0 {
panic(parser.ExportError(err))
}
/* skip the value */
if start, err := parser.skip(); err != 0 {
panic(parser.ExportError(err))
}else{
t := switchRawType(parser.s[start])
if t == _V_NONE {
panic(parser.ExportError(types.ERR_INVALID_CHAR))
}
val = newRawNode(parser.s[start:parser.p], t)
}
/* add the value to result */
ret = append(ret, Pair{Key: key, Value: val})
parser.p = parser.lspace(parser.p)
/* check for EOF */
if parser.p >= ns {
panic(parser.ExportError(types.ERR_EOF))
}
/* check for the next character */
switch parser.s[parser.p] {
case ',':
parser.p++
self.setLazyObject(parser, ret)
return &ret[len(ret)-1]
case '}':
parser.p++
self.setObject(ret)
return &ret[len(ret)-1]
default:
panic(parser.ExportError(types.ERR_INVALID_CHAR))
}
}
func (self *Node) skipKey(key string) (*Node, int) {
node, pos := self.findKey(key)
if node != nil {
return node, pos
}
if !self.isLazy() {
return &Node{}, -1
}
// lazy load
var i = self.len()
for last := self.skipNextPair(); last != nil; last = self.skipNextPair() {
if last.Key == key {
return &last.Value, i
}
i++
}
return &Node{}, -1
}
func (self *Node) skipIndex(index int) *Node {
nb := self.len()
if nb > index {
v := self.nodeAt(index)
return v
}
if !self.isLazy() {
return &Node{}
}
// lazy load
for last := self.skipNextNode(); last != nil; last = self.skipNextNode(){
if self.len() > index {
return last
}
}
return &Node{}
}
func (self *Node) skipIndexPair(index int) *Pair {
nb := self.len()
if nb > index {
return self.pairAt(index)
}
if !self.isLazy() {
return nil
}
// lazy load
for last := self.skipNextPair(); last != nil; last = self.skipNextPair(){
if self.len() > index {
return last
}
}
return nil
}
func (self *Node) loadAllIndex() {
if !self.isLazy() {
return
}
var err types.ParsingError
parser, stack := self.getParserAndArrayStack()
parser.noLazy = true
*self, err = parser.decodeArray(stack.v)
if err != 0 {
panic(parser.ExportError(err))
}
}
func (self *Node) loadAllKey() {
if !self.isLazy() {
return
}
var err types.ParsingError
parser, stack := self.getParserAndObjectStack()
parser.noLazy = true
*self, err = parser.decodeObject(stack.v)
if err != 0 {
panic(parser.ExportError(err))
}
}
func (self *Node) removeNode(i int) {
nb := self.len() - 1
node := self.nodeAt(i)
if i == nb {
self.setCapAndLen(self.cap(), nb)
*node = Node{}
return
}
from := self.nodeAt(i + 1)
memmove(unsafe.Pointer(node), unsafe.Pointer(from), _NODE_SIZE * uintptr(nb - i))
last := self.nodeAt(nb)
*last = Node{}
self.setCapAndLen(self.cap(), nb)
}
func (self *Node) removePair(i int) {
nb := self.len() - 1
node := self.pairAt(i)
if i == nb {
self.setCapAndLen(self.cap(), nb)
*node = Pair{}
return
}
from := self.pairAt(i + 1)
memmove(unsafe.Pointer(node), unsafe.Pointer(from), _PAIR_SIZE * uintptr(nb - i))
last := self.pairAt(nb)
*last = Pair{}
self.setCapAndLen(self.cap(), nb)
}
func (self *Node) toGenericArray() []interface{} {
nb := self.len()
ret := make([]interface{}, nb)
if nb == 0 {
return ret
}
/* convert each item */
var p = (*Node)(self.p)
ret[0] = p.Interface()
for i := 1; i < nb; i++ {
p = p.unsafe_next()
ret[i] = p.Interface()
}
/* all done */
return ret
}
func (self *Node) toGenericArrayUseNumber() []interface{} {
nb := self.len()
ret := make([]interface{}, nb)
if nb == 0 {
return ret
}
/* convert each item */
var p = (*Node)(self.p)
ret[0] = p.InterfaceUseNumber()
for i := 1; i < nb; i++ {
p = p.unsafe_next()
ret[i] = p.InterfaceUseNumber()
}
/* all done */
return ret
}
func (self *Node) toGenericArrayUseNode() []Node {
var nb = self.len()
var out = make([]Node, nb)
if nb == 0 {
return out
}
var p = (*Node)(self.p)
out[0] = *p
for i := 1; i < nb; i++ {
p = p.unsafe_next()
out[i] = *p
}
return out
}
func (self *Node) toGenericObject() map[string]interface{} {
nb := self.len()
ret := make(map[string]interface{}, nb)
if nb == 0 {
return ret
}
/* convert each item */
var p = (*Pair)(self.p)
ret[p.Key] = p.Value.Interface()
for i := 1; i < nb; i++ {
p = p.unsafe_next()
ret[p.Key] = p.Value.Interface()
}
/* all done */
return ret
}
func (self *Node) toGenericObjectUseNumber() map[string]interface{} {
nb := self.len()
ret := make(map[string]interface{}, nb)
if nb == 0 {
return ret
}
/* convert each item */
var p = (*Pair)(self.p)
ret[p.Key] = p.Value.InterfaceUseNumber()
for i := 1; i < nb; i++ {
p = p.unsafe_next()
ret[p.Key] = p.Value.InterfaceUseNumber()
}
/* all done */
return ret
}
func (self *Node) toGenericObjectUseNode() map[string]Node {
var nb = self.len()
var out = make(map[string]Node, nb)
if nb == 0 {
return out
}
var p = (*Pair)(self.p)
out[p.Key] = p.Value
for i := 1; i < nb; i++ {
p = p.unsafe_next()
out[p.Key] = p.Value
}
/* all done */
return out
}
/**------------------------------------ Factory Methods ------------------------------------**/
var (
nullNode = Node{t: types.V_NULL}
trueNode = Node{t: types.V_TRUE}
falseNode = Node{t: types.V_FALSE}
emptyArrayNode = Node{t: types.V_ARRAY}
emptyObjectNode = Node{t: types.V_OBJECT}
)
// NewNull creates a node of type V_NULL
func NewNull() Node {
return Node{
v: 0,
p: nil,
t: types.V_NULL,
}
}
// NewBool creates a node of type bool:
// If v is true, returns V_TRUE node
// If v is false, returns V_FALSE node
func NewBool(v bool) Node {
var t = types.V_FALSE
if v {
t = types.V_TRUE
}
return Node{
v: 0,
p: nil,
t: t,
}
}
// NewNumber creates a json.Number node
// v must be a decimal string complying with RFC8259
func NewNumber(v string) Node {
return Node{
v: int64(len(v) & _LEN_MASK),
p: str2ptr(v),
t: _V_NUMBER,
}
}
func toNumber(node *Node) json.Number {
return json.Number(addr2str(node.p, node.v))
}
func numberToFloat64(node *Node) float64 {
ret,err := toNumber(node).Float64()
if err != nil {
panic(err)
}
return ret
}
func numberToInt64(node *Node) int64 {
ret,err := toNumber(node).Int64()
if err != nil {
panic(err)
}
return ret
}
func newBytes(v []byte) Node {
return Node{
t: types.V_STRING,
p: mem2ptr(v),
v: int64(len(v) & _LEN_MASK),
}
}
// NewString creates a node of type string
func NewString(v string) Node {
return Node{
t: types.V_STRING,
p: str2ptr(v),
v: int64(len(v) & _LEN_MASK),
}
}
// NewArray creates a node of type V_ARRAY,
// using v as its underlying children
func NewArray(v []Node) Node {
return Node{
t: types.V_ARRAY,
v: int64(len(v)&_LEN_MASK | cap(v)<<_CAP_BITS),
p: *(*unsafe.Pointer)(unsafe.Pointer(&v)),
}
}
func (self *Node) setArray(v []Node) {
self.t = types.V_ARRAY
self.setCapAndLen(cap(v), len(v))
self.p = *(*unsafe.Pointer)(unsafe.Pointer(&v))
}
// NewObject creates a node of type V_OBJECT,
// using v as its underlying children
func NewObject(v []Pair) Node {
return Node{
t: types.V_OBJECT,
v: int64(len(v)&_LEN_MASK | cap(v)<<_CAP_BITS),
p: *(*unsafe.Pointer)(unsafe.Pointer(&v)),
}
}
func (self *Node) setObject(v []Pair) {
self.t = types.V_OBJECT
self.setCapAndLen(cap(v), len(v))
self.p = *(*unsafe.Pointer)(unsafe.Pointer(&v))
}
type parseObjectStack struct {
parser Parser
v []Pair
}
type parseArrayStack struct {
parser Parser
v []Node
}
func newLazyArray(p *Parser, v []Node) Node {
s := new(parseArrayStack)
s.parser = *p
s.v = v
return Node{
t: _V_ARRAY_LAZY,
v: int64(len(v)&_LEN_MASK | cap(v)<<_CAP_BITS),
p: unsafe.Pointer(s),
}
}
func (self *Node) setLazyArray(p *Parser, v []Node) {
s := new(parseArrayStack)
s.parser = *p
s.v = v
self.t = _V_ARRAY_LAZY
self.setCapAndLen(cap(v), len(v))
self.p = (unsafe.Pointer)(s)
}
func newLazyObject(p *Parser, v []Pair) Node {
s := new(parseObjectStack)
s.parser = *p
s.v = v
return Node{
t: _V_OBJECT_LAZY,
v: int64(len(v)&_LEN_MASK | cap(v)<<_CAP_BITS),
p: unsafe.Pointer(s),
}
}
func (self *Node) setLazyObject(p *Parser, v []Pair) {
s := new(parseObjectStack)
s.parser = *p
s.v = v
self.t = _V_OBJECT_LAZY
self.setCapAndLen(cap(v), len(v))
self.p = (unsafe.Pointer)(s)
}
func newRawNode(str string, typ types.ValueType) Node {
return Node{
t: _V_RAW | typ,
p: str2ptr(str),
v: int64(len(str) & _LEN_MASK),
}
}
func (self *Node) parseRaw() Node {
raw := addr2str(self.p, self.v)
parser := NewParser(raw)
n, e := parser.Parse()
if e != 0 {
panic(parser.ExportError(e))
}
return n
}
var typeJumpTable = [256]types.ValueType{
'"' : types.V_STRING,
'-' : _V_NUMBER,
'0' : _V_NUMBER,
'1' : _V_NUMBER,
'2' : _V_NUMBER,
'3' : _V_NUMBER,
'4' : _V_NUMBER,
'5' : _V_NUMBER,
'6' : _V_NUMBER,
'7' : _V_NUMBER,
'8' : _V_NUMBER,
'9' : _V_NUMBER,
'[' : types.V_ARRAY,
'f' : types.V_FALSE,
'n' : types.V_NULL,
't' : types.V_TRUE,
'{' : types.V_OBJECT,
}
func switchRawType(c byte) types.ValueType {
return typeJumpTable[c]
}