forked from mirror/ledisdb
847 lines
25 KiB
Go
847 lines
25 KiB
Go
// Copyright (c) 2012-2015 Ugorji Nwoke. All rights reserved.
|
|
// Use of this source code is governed by a MIT license found in the LICENSE file.
|
|
|
|
package codec
|
|
|
|
// Contains code shared by both encode and decode.
|
|
|
|
// Some shared ideas around encoding/decoding
|
|
// ------------------------------------------
|
|
//
|
|
// If an interface{} is passed, we first do a type assertion to see if it is
|
|
// a primitive type or a map/slice of primitive types, and use a fastpath to handle it.
|
|
//
|
|
// If we start with a reflect.Value, we are already in reflect.Value land and
|
|
// will try to grab the function for the underlying Type and directly call that function.
|
|
// This is more performant than calling reflect.Value.Interface().
|
|
//
|
|
// This still helps us bypass many layers of reflection, and give best performance.
|
|
//
|
|
// Containers
|
|
// ------------
|
|
// Containers in the stream are either associative arrays (key-value pairs) or
|
|
// regular arrays (indexed by incrementing integers).
|
|
//
|
|
// Some streams support indefinite-length containers, and use a breaking
|
|
// byte-sequence to denote that the container has come to an end.
|
|
//
|
|
// Some streams also are text-based, and use explicit separators to denote the
|
|
// end/beginning of different values.
|
|
//
|
|
// During encode, we use a high-level condition to determine how to iterate through
|
|
// the container. That decision is based on whether the container is text-based (with
|
|
// separators) or binary (without separators). If binary, we do not even call the
|
|
// encoding of separators.
|
|
//
|
|
// During decode, we use a different high-level condition to determine how to iterate
|
|
// through the containers. That decision is based on whether the stream contained
|
|
// a length prefix, or if it used explicit breaks. If length-prefixed, we assume that
|
|
// it has to be binary, and we do not even try to read separators.
|
|
//
|
|
// The only codec that may suffer (slightly) is cbor, and only when decoding indefinite-length.
|
|
// It may suffer because we treat it like a text-based codec, and read separators.
|
|
// However, this read is a no-op and the cost is insignificant.
|
|
//
|
|
// Philosophy
|
|
// ------------
|
|
// On decode, this codec will update containers appropriately:
|
|
// - If struct, update fields from stream into fields of struct.
|
|
// If field in stream not found in struct, handle appropriately (based on option).
|
|
// If a struct field has no corresponding value in the stream, leave it AS IS.
|
|
// If nil in stream, set value to nil/zero value.
|
|
// - If map, update map from stream.
|
|
// If the stream value is NIL, set the map to nil.
|
|
// - if slice, try to update up to length of array in stream.
|
|
// if container len is less than stream array length,
|
|
// and container cannot be expanded, handled (based on option).
|
|
// This means you can decode 4-element stream array into 1-element array.
|
|
//
|
|
// ------------------------------------
|
|
// On encode, user can specify omitEmpty. This means that the value will be omitted
|
|
// if the zero value. The problem may occur during decode, where omitted values do not affect
|
|
// the value being decoded into. This means that if decoding into a struct with an
|
|
// int field with current value=5, and the field is omitted in the stream, then after
|
|
// decoding, the value will still be 5 (not 0).
|
|
// omitEmpty only works if you guarantee that you always decode into zero-values.
|
|
//
|
|
// ------------------------------------
|
|
// We could have truncated a map to remove keys not available in the stream,
|
|
// or set values in the struct which are not in the stream to their zero values.
|
|
// We decided against it because there is no efficient way to do it.
|
|
// We may introduce it as an option later.
|
|
// However, that will require enabling it for both runtime and code generation modes.
|
|
//
|
|
// To support truncate, we need to do 2 passes over the container:
|
|
// map
|
|
// - first collect all keys (e.g. in k1)
|
|
// - for each key in stream, mark k1 that the key should not be removed
|
|
// - after updating map, do second pass and call delete for all keys in k1 which are not marked
|
|
// struct:
|
|
// - for each field, track the *typeInfo s1
|
|
// - iterate through all s1, and for each one not marked, set value to zero
|
|
// - this involves checking the possible anonymous fields which are nil ptrs.
|
|
// too much work.
|
|
//
|
|
// ------------------------------------------
|
|
// Error Handling is done within the library using panic.
|
|
//
|
|
// This way, the code doesn't have to keep checking if an error has happened,
|
|
// and we don't have to keep sending the error value along with each call
|
|
// or storing it in the En|Decoder and checking it constantly along the way.
|
|
//
|
|
// The disadvantage is that small functions which use panics cannot be inlined.
|
|
// The code accounts for that by only using panics behind an interface;
|
|
// since interface calls cannot be inlined, this is irrelevant.
|
|
//
|
|
// We considered storing the error is En|Decoder.
|
|
// - once it has its err field set, it cannot be used again.
|
|
// - panicing will be optional, controlled by const flag.
|
|
// - code should always check error first and return early.
|
|
// We eventually decided against it as it makes the code clumsier to always
|
|
// check for these error conditions.
|
|
|
|
import (
|
|
"encoding"
|
|
"encoding/binary"
|
|
"errors"
|
|
"fmt"
|
|
"math"
|
|
"reflect"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
"unicode"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
const (
|
|
scratchByteArrayLen = 32
|
|
|
|
// Support encoding.(Binary|Text)(Unm|M)arshaler.
|
|
// This constant flag will enable or disable it.
|
|
supportMarshalInterfaces = true
|
|
|
|
// Each Encoder or Decoder uses a cache of functions based on conditionals,
|
|
// so that the conditionals are not run every time.
|
|
//
|
|
// Either a map or a slice is used to keep track of the functions.
|
|
// The map is more natural, but has a higher cost than a slice/array.
|
|
// This flag (useMapForCodecCache) controls which is used.
|
|
//
|
|
// From benchmarks, slices with linear search perform better with < 32 entries.
|
|
// We have typically seen a high threshold of about 24 entries.
|
|
useMapForCodecCache = false
|
|
|
|
// for debugging, set this to false, to catch panic traces.
|
|
// Note that this will always cause rpc tests to fail, since they need io.EOF sent via panic.
|
|
recoverPanicToErr = true
|
|
|
|
// Fast path functions try to create a fast path encode or decode implementation
|
|
// for common maps and slices, by by-passing reflection altogether.
|
|
fastpathEnabled = true
|
|
|
|
// if checkStructForEmptyValue, check structs fields to see if an empty value.
|
|
// This could be an expensive call, so possibly disable it.
|
|
checkStructForEmptyValue = false
|
|
|
|
// if derefForIsEmptyValue, deref pointers and interfaces when checking isEmptyValue
|
|
derefForIsEmptyValue = false
|
|
)
|
|
|
|
var oneByteArr = [1]byte{0}
|
|
var zeroByteSlice = oneByteArr[:0:0]
|
|
|
|
type charEncoding uint8
|
|
|
|
const (
|
|
c_RAW charEncoding = iota
|
|
c_UTF8
|
|
c_UTF16LE
|
|
c_UTF16BE
|
|
c_UTF32LE
|
|
c_UTF32BE
|
|
)
|
|
|
|
// valueType is the stream type
|
|
type valueType uint8
|
|
|
|
const (
|
|
valueTypeUnset valueType = iota
|
|
valueTypeNil
|
|
valueTypeInt
|
|
valueTypeUint
|
|
valueTypeFloat
|
|
valueTypeBool
|
|
valueTypeString
|
|
valueTypeSymbol
|
|
valueTypeBytes
|
|
valueTypeMap
|
|
valueTypeArray
|
|
valueTypeTimestamp
|
|
valueTypeExt
|
|
|
|
// valueTypeInvalid = 0xff
|
|
)
|
|
|
|
type seqType uint8
|
|
|
|
const (
|
|
_ seqType = iota
|
|
seqTypeArray
|
|
seqTypeSlice
|
|
seqTypeChan
|
|
)
|
|
|
|
var (
|
|
bigen = binary.BigEndian
|
|
structInfoFieldName = "_struct"
|
|
|
|
cachedTypeInfo = make(map[uintptr]*typeInfo, 64)
|
|
cachedTypeInfoMutex sync.RWMutex
|
|
|
|
// mapStrIntfTyp = reflect.TypeOf(map[string]interface{}(nil))
|
|
intfSliceTyp = reflect.TypeOf([]interface{}(nil))
|
|
intfTyp = intfSliceTyp.Elem()
|
|
|
|
stringTyp = reflect.TypeOf("")
|
|
timeTyp = reflect.TypeOf(time.Time{})
|
|
rawExtTyp = reflect.TypeOf(RawExt{})
|
|
uint8SliceTyp = reflect.TypeOf([]uint8(nil))
|
|
|
|
mapBySliceTyp = reflect.TypeOf((*MapBySlice)(nil)).Elem()
|
|
|
|
binaryMarshalerTyp = reflect.TypeOf((*encoding.BinaryMarshaler)(nil)).Elem()
|
|
binaryUnmarshalerTyp = reflect.TypeOf((*encoding.BinaryUnmarshaler)(nil)).Elem()
|
|
|
|
textMarshalerTyp = reflect.TypeOf((*encoding.TextMarshaler)(nil)).Elem()
|
|
textUnmarshalerTyp = reflect.TypeOf((*encoding.TextUnmarshaler)(nil)).Elem()
|
|
|
|
selferTyp = reflect.TypeOf((*Selfer)(nil)).Elem()
|
|
|
|
uint8SliceTypId = reflect.ValueOf(uint8SliceTyp).Pointer()
|
|
rawExtTypId = reflect.ValueOf(rawExtTyp).Pointer()
|
|
intfTypId = reflect.ValueOf(intfTyp).Pointer()
|
|
timeTypId = reflect.ValueOf(timeTyp).Pointer()
|
|
stringTypId = reflect.ValueOf(stringTyp).Pointer()
|
|
|
|
// mapBySliceTypId = reflect.ValueOf(mapBySliceTyp).Pointer()
|
|
|
|
intBitsize uint8 = uint8(reflect.TypeOf(int(0)).Bits())
|
|
uintBitsize uint8 = uint8(reflect.TypeOf(uint(0)).Bits())
|
|
|
|
bsAll0x00 = []byte{0, 0, 0, 0, 0, 0, 0, 0}
|
|
bsAll0xff = []byte{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff}
|
|
|
|
chkOvf checkOverflow
|
|
|
|
noFieldNameToStructFieldInfoErr = errors.New("no field name passed to parseStructFieldInfo")
|
|
)
|
|
|
|
// Selfer defines methods by which a value can encode or decode itself.
|
|
//
|
|
// Any type which implements Selfer will be able to encode or decode itself.
|
|
// Consequently, during (en|de)code, this takes precedence over
|
|
// (text|binary)(M|Unm)arshal or extension support.
|
|
type Selfer interface {
|
|
CodecEncodeSelf(*Encoder)
|
|
CodecDecodeSelf(*Decoder)
|
|
}
|
|
|
|
// MapBySlice represents a slice which should be encoded as a map in the stream.
|
|
// The slice contains a sequence of key-value pairs.
|
|
// This affords storing a map in a specific sequence in the stream.
|
|
//
|
|
// The support of MapBySlice affords the following:
|
|
// - A slice type which implements MapBySlice will be encoded as a map
|
|
// - A slice can be decoded from a map in the stream
|
|
type MapBySlice interface {
|
|
MapBySlice()
|
|
}
|
|
|
|
// WARNING: DO NOT USE DIRECTLY. EXPORTED FOR GODOC BENEFIT. WILL BE REMOVED.
|
|
//
|
|
// BasicHandle encapsulates the common options and extension functions.
|
|
type BasicHandle struct {
|
|
extHandle
|
|
EncodeOptions
|
|
DecodeOptions
|
|
}
|
|
|
|
func (x *BasicHandle) getBasicHandle() *BasicHandle {
|
|
return x
|
|
}
|
|
|
|
// Handle is the interface for a specific encoding format.
|
|
//
|
|
// Typically, a Handle is pre-configured before first time use,
|
|
// and not modified while in use. Such a pre-configured Handle
|
|
// is safe for concurrent access.
|
|
type Handle interface {
|
|
getBasicHandle() *BasicHandle
|
|
newEncDriver(w *Encoder) encDriver
|
|
newDecDriver(r *Decoder) decDriver
|
|
isBinary() bool
|
|
}
|
|
|
|
// RawExt represents raw unprocessed extension data.
|
|
// Some codecs will decode extension data as a *RawExt if there is no registered extension for the tag.
|
|
//
|
|
// Only one of Data or Value is nil. If Data is nil, then the content of the RawExt is in the Value.
|
|
type RawExt struct {
|
|
Tag uint64
|
|
// Data is the []byte which represents the raw ext. If Data is nil, ext is exposed in Value.
|
|
// Data is used by codecs (e.g. binc, msgpack, simple) which do custom serialization of the types
|
|
Data []byte
|
|
// Value represents the extension, if Data is nil.
|
|
// Value is used by codecs (e.g. cbor) which use the format to do custom serialization of the types.
|
|
Value interface{}
|
|
}
|
|
|
|
// Ext handles custom (de)serialization of custom types / extensions.
|
|
type Ext interface {
|
|
// WriteExt converts a value to a []byte.
|
|
// It is used by codecs (e.g. binc, msgpack, simple) which do custom serialization of the types.
|
|
WriteExt(v interface{}) []byte
|
|
|
|
// ReadExt updates a value from a []byte.
|
|
// It is used by codecs (e.g. binc, msgpack, simple) which do custom serialization of the types.
|
|
ReadExt(dst interface{}, src []byte)
|
|
|
|
// ConvertExt converts a value into a simpler interface for easy encoding e.g. convert time.Time to int64.
|
|
// It is used by codecs (e.g. cbor) which use the format to do custom serialization of the types.
|
|
ConvertExt(v interface{}) interface{}
|
|
|
|
// UpdateExt updates a value from a simpler interface for easy decoding e.g. convert int64 to time.Time.
|
|
// It is used by codecs (e.g. cbor) which use the format to do custom serialization of the types.
|
|
UpdateExt(dst interface{}, src interface{})
|
|
}
|
|
|
|
// bytesExt is a wrapper implementation to support former AddExt exported method.
|
|
type bytesExt struct {
|
|
encFn func(reflect.Value) ([]byte, error)
|
|
decFn func(reflect.Value, []byte) error
|
|
}
|
|
|
|
func (x bytesExt) WriteExt(v interface{}) []byte {
|
|
// fmt.Printf(">>>>>>>>>> WriteExt: %T, %v\n", v, v)
|
|
bs, err := x.encFn(reflect.ValueOf(v))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return bs
|
|
}
|
|
|
|
func (x bytesExt) ReadExt(v interface{}, bs []byte) {
|
|
// fmt.Printf(">>>>>>>>>> ReadExt: %T, %v\n", v, v)
|
|
if err := x.decFn(reflect.ValueOf(v), bs); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func (x bytesExt) ConvertExt(v interface{}) interface{} {
|
|
return x.WriteExt(v)
|
|
}
|
|
|
|
func (x bytesExt) UpdateExt(dest interface{}, v interface{}) {
|
|
x.ReadExt(dest, v.([]byte))
|
|
}
|
|
|
|
// type errorString string
|
|
// func (x errorString) Error() string { return string(x) }
|
|
|
|
type binaryEncodingType struct{}
|
|
|
|
func (_ binaryEncodingType) isBinary() bool { return true }
|
|
|
|
type textEncodingType struct{}
|
|
|
|
func (_ textEncodingType) isBinary() bool { return false }
|
|
|
|
// noBuiltInTypes is embedded into many types which do not support builtins
|
|
// e.g. msgpack, simple, cbor.
|
|
type noBuiltInTypes struct{}
|
|
|
|
func (_ noBuiltInTypes) IsBuiltinType(rt uintptr) bool { return false }
|
|
func (_ noBuiltInTypes) EncodeBuiltin(rt uintptr, v interface{}) {}
|
|
func (_ noBuiltInTypes) DecodeBuiltin(rt uintptr, v interface{}) {}
|
|
|
|
type noStreamingCodec struct{}
|
|
|
|
func (_ noStreamingCodec) CheckBreak() bool { return false }
|
|
|
|
// bigenHelper.
|
|
// Users must already slice the x completely, because we will not reslice.
|
|
type bigenHelper struct {
|
|
x []byte // must be correctly sliced to appropriate len. slicing is a cost.
|
|
w encWriter
|
|
}
|
|
|
|
func (z bigenHelper) writeUint16(v uint16) {
|
|
bigen.PutUint16(z.x, v)
|
|
z.w.writeb(z.x)
|
|
}
|
|
|
|
func (z bigenHelper) writeUint32(v uint32) {
|
|
bigen.PutUint32(z.x, v)
|
|
z.w.writeb(z.x)
|
|
}
|
|
|
|
func (z bigenHelper) writeUint64(v uint64) {
|
|
bigen.PutUint64(z.x, v)
|
|
z.w.writeb(z.x)
|
|
}
|
|
|
|
type extTypeTagFn struct {
|
|
rtid uintptr
|
|
rt reflect.Type
|
|
tag uint64
|
|
ext Ext
|
|
}
|
|
|
|
type extHandle []*extTypeTagFn
|
|
|
|
// DEPRECATED: AddExt is deprecated in favor of SetExt. It exists for compatibility only.
|
|
//
|
|
// AddExt registes an encode and decode function for a reflect.Type.
|
|
// AddExt internally calls SetExt.
|
|
// To deregister an Ext, call AddExt with nil encfn and/or nil decfn.
|
|
func (o *extHandle) AddExt(
|
|
rt reflect.Type, tag byte,
|
|
encfn func(reflect.Value) ([]byte, error), decfn func(reflect.Value, []byte) error,
|
|
) (err error) {
|
|
if encfn == nil || decfn == nil {
|
|
return o.SetExt(rt, uint64(tag), nil)
|
|
}
|
|
return o.SetExt(rt, uint64(tag), bytesExt{encfn, decfn})
|
|
}
|
|
|
|
// SetExt registers a tag and Ext for a reflect.Type.
|
|
//
|
|
// Note that the type must be a named type, and specifically not
|
|
// a pointer or Interface. An error is returned if that is not honored.
|
|
//
|
|
// To Deregister an ext, call SetExt with nil Ext
|
|
func (o *extHandle) SetExt(rt reflect.Type, tag uint64, ext Ext) (err error) {
|
|
// o is a pointer, because we may need to initialize it
|
|
if rt.PkgPath() == "" || rt.Kind() == reflect.Interface {
|
|
err = fmt.Errorf("codec.Handle.AddExt: Takes named type, especially not a pointer or interface: %T",
|
|
reflect.Zero(rt).Interface())
|
|
return
|
|
}
|
|
|
|
rtid := reflect.ValueOf(rt).Pointer()
|
|
for _, v := range *o {
|
|
if v.rtid == rtid {
|
|
v.tag, v.ext = tag, ext
|
|
return
|
|
}
|
|
}
|
|
|
|
*o = append(*o, &extTypeTagFn{rtid, rt, tag, ext})
|
|
return
|
|
}
|
|
|
|
func (o extHandle) getExt(rtid uintptr) *extTypeTagFn {
|
|
for _, v := range o {
|
|
if v.rtid == rtid {
|
|
return v
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (o extHandle) getExtForTag(tag uint64) *extTypeTagFn {
|
|
for _, v := range o {
|
|
if v.tag == tag {
|
|
return v
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type structFieldInfo struct {
|
|
encName string // encode name
|
|
|
|
// only one of 'i' or 'is' can be set. If 'i' is -1, then 'is' has been set.
|
|
|
|
is []int // (recursive/embedded) field index in struct
|
|
i int16 // field index in struct
|
|
omitEmpty bool
|
|
toArray bool // if field is _struct, is the toArray set?
|
|
}
|
|
|
|
// rv returns the field of the struct.
|
|
// If anonymous, it returns an Invalid
|
|
func (si *structFieldInfo) field(v reflect.Value, update bool) (rv2 reflect.Value) {
|
|
if si.i != -1 {
|
|
v = v.Field(int(si.i))
|
|
return v
|
|
}
|
|
// replicate FieldByIndex
|
|
for _, x := range si.is {
|
|
for v.Kind() == reflect.Ptr {
|
|
if v.IsNil() {
|
|
if !update {
|
|
return
|
|
}
|
|
v.Set(reflect.New(v.Type().Elem()))
|
|
}
|
|
v = v.Elem()
|
|
}
|
|
v = v.Field(x)
|
|
}
|
|
return v
|
|
}
|
|
|
|
func (si *structFieldInfo) setToZeroValue(v reflect.Value) {
|
|
if si.i != -1 {
|
|
v = v.Field(int(si.i))
|
|
v.Set(reflect.Zero(v.Type()))
|
|
// v.Set(reflect.New(v.Type()).Elem())
|
|
// v.Set(reflect.New(v.Type()))
|
|
} else {
|
|
// replicate FieldByIndex
|
|
for _, x := range si.is {
|
|
for v.Kind() == reflect.Ptr {
|
|
if v.IsNil() {
|
|
return
|
|
}
|
|
v = v.Elem()
|
|
}
|
|
v = v.Field(x)
|
|
}
|
|
v.Set(reflect.Zero(v.Type()))
|
|
}
|
|
}
|
|
|
|
func parseStructFieldInfo(fname string, stag string) *structFieldInfo {
|
|
if fname == "" {
|
|
panic(noFieldNameToStructFieldInfoErr)
|
|
}
|
|
si := structFieldInfo{
|
|
encName: fname,
|
|
}
|
|
|
|
if stag != "" {
|
|
for i, s := range strings.Split(stag, ",") {
|
|
if i == 0 {
|
|
if s != "" {
|
|
si.encName = s
|
|
}
|
|
} else {
|
|
if s == "omitempty" {
|
|
si.omitEmpty = true
|
|
} else if s == "toarray" {
|
|
si.toArray = true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
// si.encNameBs = []byte(si.encName)
|
|
return &si
|
|
}
|
|
|
|
type sfiSortedByEncName []*structFieldInfo
|
|
|
|
func (p sfiSortedByEncName) Len() int {
|
|
return len(p)
|
|
}
|
|
|
|
func (p sfiSortedByEncName) Less(i, j int) bool {
|
|
return p[i].encName < p[j].encName
|
|
}
|
|
|
|
func (p sfiSortedByEncName) Swap(i, j int) {
|
|
p[i], p[j] = p[j], p[i]
|
|
}
|
|
|
|
// typeInfo keeps information about each type referenced in the encode/decode sequence.
|
|
//
|
|
// During an encode/decode sequence, we work as below:
|
|
// - If base is a built in type, en/decode base value
|
|
// - If base is registered as an extension, en/decode base value
|
|
// - If type is binary(M/Unm)arshaler, call Binary(M/Unm)arshal method
|
|
// - If type is text(M/Unm)arshaler, call Text(M/Unm)arshal method
|
|
// - Else decode appropriately based on the reflect.Kind
|
|
type typeInfo struct {
|
|
sfi []*structFieldInfo // sorted. Used when enc/dec struct to map.
|
|
sfip []*structFieldInfo // unsorted. Used when enc/dec struct to array.
|
|
|
|
rt reflect.Type
|
|
rtid uintptr
|
|
|
|
// baseId gives pointer to the base reflect.Type, after deferencing
|
|
// the pointers. E.g. base type of ***time.Time is time.Time.
|
|
base reflect.Type
|
|
baseId uintptr
|
|
baseIndir int8 // number of indirections to get to base
|
|
|
|
mbs bool // base type (T or *T) is a MapBySlice
|
|
|
|
bm bool // base type (T or *T) is a binaryMarshaler
|
|
bunm bool // base type (T or *T) is a binaryUnmarshaler
|
|
bmIndir int8 // number of indirections to get to binaryMarshaler type
|
|
bunmIndir int8 // number of indirections to get to binaryUnmarshaler type
|
|
|
|
tm bool // base type (T or *T) is a textMarshaler
|
|
tunm bool // base type (T or *T) is a textUnmarshaler
|
|
tmIndir int8 // number of indirections to get to textMarshaler type
|
|
tunmIndir int8 // number of indirections to get to textUnmarshaler type
|
|
|
|
cs bool // base type (T or *T) is a Selfer
|
|
csIndir int8 // number of indirections to get to Selfer type
|
|
|
|
toArray bool // whether this (struct) type should be encoded as an array
|
|
}
|
|
|
|
func (ti *typeInfo) indexForEncName(name string) int {
|
|
//tisfi := ti.sfi
|
|
const binarySearchThreshold = 16
|
|
if sfilen := len(ti.sfi); sfilen < binarySearchThreshold {
|
|
// linear search. faster than binary search in my testing up to 16-field structs.
|
|
for i, si := range ti.sfi {
|
|
if si.encName == name {
|
|
return i
|
|
}
|
|
}
|
|
} else {
|
|
// binary search. adapted from sort/search.go.
|
|
h, i, j := 0, 0, sfilen
|
|
for i < j {
|
|
h = i + (j-i)/2
|
|
if ti.sfi[h].encName < name {
|
|
i = h + 1
|
|
} else {
|
|
j = h
|
|
}
|
|
}
|
|
if i < sfilen && ti.sfi[i].encName == name {
|
|
return i
|
|
}
|
|
}
|
|
return -1
|
|
}
|
|
|
|
func getStructTag(t reflect.StructTag) (s string) {
|
|
// check for tags: codec, json, in that order.
|
|
// this allows seamless support for many configured structs.
|
|
s = t.Get("codec")
|
|
if s == "" {
|
|
s = t.Get("json")
|
|
}
|
|
return
|
|
}
|
|
|
|
func getTypeInfo(rtid uintptr, rt reflect.Type) (pti *typeInfo) {
|
|
var ok bool
|
|
cachedTypeInfoMutex.RLock()
|
|
pti, ok = cachedTypeInfo[rtid]
|
|
cachedTypeInfoMutex.RUnlock()
|
|
if ok {
|
|
return
|
|
}
|
|
|
|
cachedTypeInfoMutex.Lock()
|
|
defer cachedTypeInfoMutex.Unlock()
|
|
if pti, ok = cachedTypeInfo[rtid]; ok {
|
|
return
|
|
}
|
|
|
|
ti := typeInfo{rt: rt, rtid: rtid}
|
|
pti = &ti
|
|
|
|
var indir int8
|
|
if ok, indir = implementsIntf(rt, binaryMarshalerTyp); ok {
|
|
ti.bm, ti.bmIndir = true, indir
|
|
}
|
|
if ok, indir = implementsIntf(rt, binaryUnmarshalerTyp); ok {
|
|
ti.bunm, ti.bunmIndir = true, indir
|
|
}
|
|
if ok, indir = implementsIntf(rt, textMarshalerTyp); ok {
|
|
ti.tm, ti.tmIndir = true, indir
|
|
}
|
|
if ok, indir = implementsIntf(rt, textUnmarshalerTyp); ok {
|
|
ti.tunm, ti.tunmIndir = true, indir
|
|
}
|
|
if ok, indir = implementsIntf(rt, selferTyp); ok {
|
|
ti.cs, ti.csIndir = true, indir
|
|
}
|
|
if ok, _ = implementsIntf(rt, mapBySliceTyp); ok {
|
|
ti.mbs = true
|
|
}
|
|
|
|
pt := rt
|
|
var ptIndir int8
|
|
// for ; pt.Kind() == reflect.Ptr; pt, ptIndir = pt.Elem(), ptIndir+1 { }
|
|
for pt.Kind() == reflect.Ptr {
|
|
pt = pt.Elem()
|
|
ptIndir++
|
|
}
|
|
if ptIndir == 0 {
|
|
ti.base = rt
|
|
ti.baseId = rtid
|
|
} else {
|
|
ti.base = pt
|
|
ti.baseId = reflect.ValueOf(pt).Pointer()
|
|
ti.baseIndir = ptIndir
|
|
}
|
|
|
|
if rt.Kind() == reflect.Struct {
|
|
var siInfo *structFieldInfo
|
|
if f, ok := rt.FieldByName(structInfoFieldName); ok {
|
|
siInfo = parseStructFieldInfo(structInfoFieldName, getStructTag(f.Tag))
|
|
ti.toArray = siInfo.toArray
|
|
}
|
|
sfip := make([]*structFieldInfo, 0, rt.NumField())
|
|
rgetTypeInfo(rt, nil, make(map[string]bool, 16), &sfip, siInfo)
|
|
|
|
ti.sfip = make([]*structFieldInfo, len(sfip))
|
|
ti.sfi = make([]*structFieldInfo, len(sfip))
|
|
copy(ti.sfip, sfip)
|
|
sort.Sort(sfiSortedByEncName(sfip))
|
|
copy(ti.sfi, sfip)
|
|
}
|
|
// sfi = sfip
|
|
cachedTypeInfo[rtid] = pti
|
|
return
|
|
}
|
|
|
|
func rgetTypeInfo(rt reflect.Type, indexstack []int, fnameToHastag map[string]bool,
|
|
sfi *[]*structFieldInfo, siInfo *structFieldInfo,
|
|
) {
|
|
for j := 0; j < rt.NumField(); j++ {
|
|
f := rt.Field(j)
|
|
// func types are skipped.
|
|
if tk := f.Type.Kind(); tk == reflect.Func {
|
|
continue
|
|
}
|
|
stag := getStructTag(f.Tag)
|
|
if stag == "-" {
|
|
continue
|
|
}
|
|
if r1, _ := utf8.DecodeRuneInString(f.Name); r1 == utf8.RuneError || !unicode.IsUpper(r1) {
|
|
continue
|
|
}
|
|
// if anonymous and there is no struct tag and its a struct (or pointer to struct), inline it.
|
|
if f.Anonymous && stag == "" {
|
|
ft := f.Type
|
|
for ft.Kind() == reflect.Ptr {
|
|
ft = ft.Elem()
|
|
}
|
|
if ft.Kind() == reflect.Struct {
|
|
indexstack2 := make([]int, len(indexstack)+1, len(indexstack)+4)
|
|
copy(indexstack2, indexstack)
|
|
indexstack2[len(indexstack)] = j
|
|
// indexstack2 := append(append(make([]int, 0, len(indexstack)+4), indexstack...), j)
|
|
rgetTypeInfo(ft, indexstack2, fnameToHastag, sfi, siInfo)
|
|
continue
|
|
}
|
|
}
|
|
// do not let fields with same name in embedded structs override field at higher level.
|
|
// this must be done after anonymous check, to allow anonymous field
|
|
// still include their child fields
|
|
if _, ok := fnameToHastag[f.Name]; ok {
|
|
continue
|
|
}
|
|
si := parseStructFieldInfo(f.Name, stag)
|
|
// si.ikind = int(f.Type.Kind())
|
|
if len(indexstack) == 0 {
|
|
si.i = int16(j)
|
|
} else {
|
|
si.i = -1
|
|
si.is = append(append(make([]int, 0, len(indexstack)+4), indexstack...), j)
|
|
}
|
|
|
|
if siInfo != nil {
|
|
if siInfo.omitEmpty {
|
|
si.omitEmpty = true
|
|
}
|
|
}
|
|
*sfi = append(*sfi, si)
|
|
fnameToHastag[f.Name] = stag != ""
|
|
}
|
|
}
|
|
|
|
func panicToErr(err *error) {
|
|
if recoverPanicToErr {
|
|
if x := recover(); x != nil {
|
|
//debug.PrintStack()
|
|
panicValToErr(x, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
// func doPanic(tag string, format string, params ...interface{}) {
|
|
// params2 := make([]interface{}, len(params)+1)
|
|
// params2[0] = tag
|
|
// copy(params2[1:], params)
|
|
// panic(fmt.Errorf("%s: "+format, params2...))
|
|
// }
|
|
|
|
func isMutableKind(k reflect.Kind) (v bool) {
|
|
return k == reflect.Int ||
|
|
k == reflect.Int8 ||
|
|
k == reflect.Int16 ||
|
|
k == reflect.Int32 ||
|
|
k == reflect.Int64 ||
|
|
k == reflect.Uint ||
|
|
k == reflect.Uint8 ||
|
|
k == reflect.Uint16 ||
|
|
k == reflect.Uint32 ||
|
|
k == reflect.Uint64 ||
|
|
k == reflect.Float32 ||
|
|
k == reflect.Float64 ||
|
|
k == reflect.Bool ||
|
|
k == reflect.String
|
|
}
|
|
|
|
// these functions must be inlinable, and not call anybody
|
|
type checkOverflow struct{}
|
|
|
|
func (_ checkOverflow) Float32(f float64) (overflow bool) {
|
|
if f < 0 {
|
|
f = -f
|
|
}
|
|
return math.MaxFloat32 < f && f <= math.MaxFloat64
|
|
}
|
|
|
|
func (_ checkOverflow) Uint(v uint64, bitsize uint8) (overflow bool) {
|
|
if bitsize == 0 || bitsize >= 64 || v == 0 {
|
|
return
|
|
}
|
|
if trunc := (v << (64 - bitsize)) >> (64 - bitsize); v != trunc {
|
|
overflow = true
|
|
}
|
|
return
|
|
}
|
|
|
|
func (_ checkOverflow) Int(v int64, bitsize uint8) (overflow bool) {
|
|
if bitsize == 0 || bitsize >= 64 || v == 0 {
|
|
return
|
|
}
|
|
if trunc := (v << (64 - bitsize)) >> (64 - bitsize); v != trunc {
|
|
overflow = true
|
|
}
|
|
return
|
|
}
|
|
|
|
func (_ checkOverflow) SignedInt(v uint64) (i int64, overflow bool) {
|
|
//e.g. -127 to 128 for int8
|
|
pos := (v >> 63) == 0
|
|
ui2 := v & 0x7fffffffffffffff
|
|
if pos {
|
|
if ui2 > math.MaxInt64 {
|
|
overflow = true
|
|
return
|
|
}
|
|
} else {
|
|
if ui2 > math.MaxInt64-1 {
|
|
overflow = true
|
|
return
|
|
}
|
|
}
|
|
i = int64(v)
|
|
return
|
|
}
|