mirror of
https://github.com/ii64/sonic.git
synced 2026-06-21 00:46:43 +08:00
108 lines
2.9 KiB
Go
108 lines
2.9 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 decoder
|
|
|
|
import (
|
|
`encoding/json`
|
|
`reflect`
|
|
|
|
`github.com/bytedance/sonic/internal/rt`
|
|
)
|
|
|
|
const (
|
|
_F_use_int64 = iota
|
|
_F_use_number
|
|
_F_disable_urc
|
|
_F_disable_unknown
|
|
)
|
|
|
|
// Decoder is the decoder context object
|
|
type Decoder struct {
|
|
i int
|
|
f uint64
|
|
s string
|
|
}
|
|
|
|
// NewDecoder creates a new decoder instance.
|
|
func NewDecoder(s string) *Decoder {
|
|
return &Decoder{s: s}
|
|
}
|
|
|
|
// Pos returns the current decoding position.
|
|
func (self *Decoder) Pos() int {
|
|
return self.i
|
|
}
|
|
|
|
// Decode parses the JSON-encoded data from current position and stores the result
|
|
// in the value pointed to by val.
|
|
func (self *Decoder) Decode(val interface{}) error {
|
|
vv := rt.UnpackEface(val)
|
|
vp := vv.Value
|
|
|
|
/* check for nil type */
|
|
if vv.Type == nil {
|
|
return &json.InvalidUnmarshalError{}
|
|
}
|
|
|
|
/* must be a non-nil pointer */
|
|
if vp == nil || vv.Type.Kind() != reflect.Ptr {
|
|
return &json.InvalidUnmarshalError{Type: vv.Type.Pack()}
|
|
}
|
|
|
|
/* create a new stack, and call the decoder */
|
|
sb, etp := newStack(), rt.PtrElem(vv.Type)
|
|
nb, err := decodeTypedPointer(self.s, self.i, etp, vp, sb, self.f)
|
|
|
|
/* return the stack back */
|
|
self.i = nb
|
|
freeStack(sb)
|
|
return err
|
|
}
|
|
|
|
// UseInt64 causes the Decoder to unmarshal an integer into an interface{} as an
|
|
// int64 instead of as a float64.
|
|
func (self *Decoder) UseInt64() {
|
|
self.f |= 1 << _F_use_int64
|
|
self.f &^= 1 << _F_use_number
|
|
}
|
|
|
|
// UseNumber causes the Decoder to unmarshal a number into an interface{} as a
|
|
// json.Number instead of as a float64.
|
|
func (self *Decoder) UseNumber() {
|
|
self.f &^= 1 << _F_use_int64
|
|
self.f |= 1 << _F_use_number
|
|
}
|
|
|
|
// UseUnicodeErrors causes the Decoder to return an error when encounter invalid
|
|
// UTF-8 escape sequences.
|
|
func (self *Decoder) UseUnicodeErrors() {
|
|
self.f |= 1 << _F_disable_urc
|
|
}
|
|
|
|
// DisallowUnknownFields causes the Decoder to return an error when the destination
|
|
// is a struct and the input contains object keys which do not match any
|
|
// non-ignored, exported fields in the destination.
|
|
func (self *Decoder) DisallowUnknownFields() {
|
|
self.f |= 1 << _F_disable_unknown
|
|
}
|
|
|
|
// Pretouch compiles vt ahead-of-time to avoid JIT compilation on-the-fly, in
|
|
// order to reduce the first-hit latency.
|
|
func Pretouch(vt reflect.Type) (err error) {
|
|
_, err = findOrCompile(rt.UnpackType(vt))
|
|
return
|
|
}
|