123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727 |
- package jsoniter
- import (
- "encoding"
- "encoding/json"
- "fmt"
- "reflect"
- "time"
- "unsafe"
- )
- // ValDecoder is an internal type registered to cache as needed.
- // Don't confuse jsoniter.ValDecoder with json.Decoder.
- // For json.Decoder's adapter, refer to jsoniter.AdapterDecoder(todo link).
- //
- // Reflection on type to create decoders, which is then cached
- // Reflection on value is avoided as we can, as the reflect.Value itself will allocate, with following exceptions
- // 1. create instance of new value, for example *int will need a int to be allocated
- // 2. append to slice, if the existing cap is not enough, allocate will be done using Reflect.New
- // 3. assignment to map, both key and value will be reflect.Value
- // For a simple struct binding, it will be reflect.Value free and allocation free
- type ValDecoder interface {
- Decode(ptr unsafe.Pointer, iter *Iterator)
- }
- // ValEncoder is an internal type registered to cache as needed.
- // Don't confuse jsoniter.ValEncoder with json.Encoder.
- // For json.Encoder's adapter, refer to jsoniter.AdapterEncoder(todo godoc link).
- type ValEncoder interface {
- IsEmpty(ptr unsafe.Pointer) bool
- Encode(ptr unsafe.Pointer, stream *Stream)
- EncodeInterface(val interface{}, stream *Stream)
- }
- type checkIsEmpty interface {
- IsEmpty(ptr unsafe.Pointer) bool
- }
- // WriteToStream the default implementation for TypeEncoder method EncodeInterface
- func WriteToStream(val interface{}, stream *Stream, encoder ValEncoder) {
- e := (*emptyInterface)(unsafe.Pointer(&val))
- if e.word == nil {
- stream.WriteNil()
- return
- }
- if reflect.TypeOf(val).Kind() == reflect.Ptr {
- encoder.Encode(unsafe.Pointer(&e.word), stream)
- } else {
- encoder.Encode(e.word, stream)
- }
- }
- var jsonNumberType reflect.Type
- var jsoniterNumberType reflect.Type
- var jsonRawMessageType reflect.Type
- var jsoniterRawMessageType reflect.Type
- var anyType reflect.Type
- var marshalerType reflect.Type
- var unmarshalerType reflect.Type
- var textMarshalerType reflect.Type
- var textUnmarshalerType reflect.Type
- func init() {
- jsonNumberType = reflect.TypeOf((*json.Number)(nil)).Elem()
- jsoniterNumberType = reflect.TypeOf((*Number)(nil)).Elem()
- jsonRawMessageType = reflect.TypeOf((*json.RawMessage)(nil)).Elem()
- jsoniterRawMessageType = reflect.TypeOf((*RawMessage)(nil)).Elem()
- anyType = reflect.TypeOf((*Any)(nil)).Elem()
- marshalerType = reflect.TypeOf((*json.Marshaler)(nil)).Elem()
- unmarshalerType = reflect.TypeOf((*json.Unmarshaler)(nil)).Elem()
- textMarshalerType = reflect.TypeOf((*encoding.TextMarshaler)(nil)).Elem()
- textUnmarshalerType = reflect.TypeOf((*encoding.TextUnmarshaler)(nil)).Elem()
- }
- type OptionalDecoder struct {
- ValueType reflect.Type
- ValueDecoder ValDecoder
- }
- func (decoder *OptionalDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) {
- if iter.ReadNil() {
- *((*unsafe.Pointer)(ptr)) = nil
- } else {
- if *((*unsafe.Pointer)(ptr)) == nil {
- //pointer to null, we have to allocate memory to hold the value
- value := reflect.New(decoder.ValueType)
- newPtr := extractInterface(value.Interface()).word
- decoder.ValueDecoder.Decode(newPtr, iter)
- *((*uintptr)(ptr)) = uintptr(newPtr)
- } else {
- //reuse existing instance
- decoder.ValueDecoder.Decode(*((*unsafe.Pointer)(ptr)), iter)
- }
- }
- }
- type deferenceDecoder struct {
- // only to deference a pointer
- valueType reflect.Type
- valueDecoder ValDecoder
- }
- func (decoder *deferenceDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) {
- if *((*unsafe.Pointer)(ptr)) == nil {
- //pointer to null, we have to allocate memory to hold the value
- value := reflect.New(decoder.valueType)
- newPtr := extractInterface(value.Interface()).word
- decoder.valueDecoder.Decode(newPtr, iter)
- *((*uintptr)(ptr)) = uintptr(newPtr)
- } else {
- //reuse existing instance
- decoder.valueDecoder.Decode(*((*unsafe.Pointer)(ptr)), iter)
- }
- }
- type OptionalEncoder struct {
- ValueEncoder ValEncoder
- }
- func (encoder *OptionalEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
- if *((*unsafe.Pointer)(ptr)) == nil {
- stream.WriteNil()
- } else {
- encoder.ValueEncoder.Encode(*((*unsafe.Pointer)(ptr)), stream)
- }
- }
- func (encoder *OptionalEncoder) EncodeInterface(val interface{}, stream *Stream) {
- WriteToStream(val, stream, encoder)
- }
- func (encoder *OptionalEncoder) IsEmpty(ptr unsafe.Pointer) bool {
- return *((*unsafe.Pointer)(ptr)) == nil
- }
- type optionalMapEncoder struct {
- valueEncoder ValEncoder
- }
- func (encoder *optionalMapEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
- if *((*unsafe.Pointer)(ptr)) == nil {
- stream.WriteNil()
- } else {
- encoder.valueEncoder.Encode(*((*unsafe.Pointer)(ptr)), stream)
- }
- }
- func (encoder *optionalMapEncoder) EncodeInterface(val interface{}, stream *Stream) {
- WriteToStream(val, stream, encoder)
- }
- func (encoder *optionalMapEncoder) IsEmpty(ptr unsafe.Pointer) bool {
- p := *((*unsafe.Pointer)(ptr))
- return p == nil || encoder.valueEncoder.IsEmpty(p)
- }
- type placeholderEncoder struct {
- cfg *frozenConfig
- cacheKey reflect.Type
- }
- func (encoder *placeholderEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
- encoder.getRealEncoder().Encode(ptr, stream)
- }
- func (encoder *placeholderEncoder) EncodeInterface(val interface{}, stream *Stream) {
- encoder.getRealEncoder().EncodeInterface(val, stream)
- }
- func (encoder *placeholderEncoder) IsEmpty(ptr unsafe.Pointer) bool {
- return encoder.getRealEncoder().IsEmpty(ptr)
- }
- func (encoder *placeholderEncoder) getRealEncoder() ValEncoder {
- for i := 0; i < 500; i++ {
- realDecoder := encoder.cfg.getEncoderFromCache(encoder.cacheKey)
- _, isPlaceholder := realDecoder.(*placeholderEncoder)
- if isPlaceholder {
- time.Sleep(10 * time.Millisecond)
- } else {
- return realDecoder
- }
- }
- panic(fmt.Sprintf("real encoder not found for cache key: %v", encoder.cacheKey))
- }
- type placeholderDecoder struct {
- cfg *frozenConfig
- cacheKey reflect.Type
- }
- func (decoder *placeholderDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) {
- for i := 0; i < 500; i++ {
- realDecoder := decoder.cfg.getDecoderFromCache(decoder.cacheKey)
- _, isPlaceholder := realDecoder.(*placeholderDecoder)
- if isPlaceholder {
- time.Sleep(10 * time.Millisecond)
- } else {
- realDecoder.Decode(ptr, iter)
- return
- }
- }
- panic(fmt.Sprintf("real decoder not found for cache key: %v", decoder.cacheKey))
- }
- // emptyInterface is the header for an interface{} value.
- type emptyInterface struct {
- typ unsafe.Pointer
- word unsafe.Pointer
- }
- // emptyInterface is the header for an interface with method (not interface{})
- type nonEmptyInterface struct {
- // see ../runtime/iface.go:/Itab
- itab *struct {
- ityp unsafe.Pointer // static interface type
- typ unsafe.Pointer // dynamic concrete type
- link unsafe.Pointer
- bad int32
- unused int32
- fun [100000]unsafe.Pointer // method table
- }
- word unsafe.Pointer
- }
- // ReadVal copy the underlying JSON into go interface, same as json.Unmarshal
- func (iter *Iterator) ReadVal(obj interface{}) {
- typ := reflect.TypeOf(obj)
- cacheKey := typ.Elem()
- decoder, err := decoderOfType(iter.cfg, cacheKey)
- if err != nil {
- iter.Error = err
- return
- }
- e := (*emptyInterface)(unsafe.Pointer(&obj))
- decoder.Decode(e.word, iter)
- }
- // WriteVal copy the go interface into underlying JSON, same as json.Marshal
- func (stream *Stream) WriteVal(val interface{}) {
- if nil == val {
- stream.WriteNil()
- return
- }
- typ := reflect.TypeOf(val)
- cacheKey := typ
- encoder, err := encoderOfType(stream.cfg, cacheKey)
- if err != nil {
- stream.Error = err
- return
- }
- encoder.EncodeInterface(val, stream)
- }
- type prefix string
- func (p prefix) addToDecoder(decoder ValDecoder, err error) (ValDecoder, error) {
- if err != nil {
- return nil, fmt.Errorf("%s: %s", p, err.Error())
- }
- return decoder, err
- }
- func (p prefix) addToEncoder(encoder ValEncoder, err error) (ValEncoder, error) {
- if err != nil {
- return nil, fmt.Errorf("%s: %s", p, err.Error())
- }
- return encoder, err
- }
- func decoderOfType(cfg *frozenConfig, typ reflect.Type) (ValDecoder, error) {
- cacheKey := typ
- decoder := cfg.getDecoderFromCache(cacheKey)
- if decoder != nil {
- return decoder, nil
- }
- decoder = getTypeDecoderFromExtension(cfg, typ)
- if decoder != nil {
- cfg.addDecoderToCache(cacheKey, decoder)
- return decoder, nil
- }
- decoder = &placeholderDecoder{cfg: cfg, cacheKey: cacheKey}
- cfg.addDecoderToCache(cacheKey, decoder)
- decoder, err := createDecoderOfType(cfg, typ)
- for _, extension := range extensions {
- decoder = extension.DecorateDecoder(typ, decoder)
- }
- for _, extension := range cfg.extensions {
- decoder = extension.DecorateDecoder(typ, decoder)
- }
- cfg.addDecoderToCache(cacheKey, decoder)
- return decoder, err
- }
- func createDecoderOfType(cfg *frozenConfig, typ reflect.Type) (ValDecoder, error) {
- typeName := typ.String()
- if typ == jsonRawMessageType {
- return &jsonRawMessageCodec{}, nil
- }
- if typ == jsoniterRawMessageType {
- return &jsoniterRawMessageCodec{}, nil
- }
- if typ.AssignableTo(jsonNumberType) {
- return &jsonNumberCodec{}, nil
- }
- if typ.AssignableTo(jsoniterNumberType) {
- return &jsoniterNumberCodec{}, nil
- }
- if typ.Implements(unmarshalerType) {
- templateInterface := reflect.New(typ).Elem().Interface()
- var decoder ValDecoder = &unmarshalerDecoder{extractInterface(templateInterface)}
- if typ.Kind() == reflect.Ptr {
- decoder = &OptionalDecoder{typ.Elem(), decoder}
- }
- return decoder, nil
- }
- if reflect.PtrTo(typ).Implements(unmarshalerType) {
- templateInterface := reflect.New(typ).Interface()
- var decoder ValDecoder = &unmarshalerDecoder{extractInterface(templateInterface)}
- return decoder, nil
- }
- if typ.Implements(textUnmarshalerType) {
- templateInterface := reflect.New(typ).Elem().Interface()
- var decoder ValDecoder = &textUnmarshalerDecoder{extractInterface(templateInterface)}
- if typ.Kind() == reflect.Ptr {
- decoder = &OptionalDecoder{typ.Elem(), decoder}
- }
- return decoder, nil
- }
- if reflect.PtrTo(typ).Implements(textUnmarshalerType) {
- templateInterface := reflect.New(typ).Interface()
- var decoder ValDecoder = &textUnmarshalerDecoder{extractInterface(templateInterface)}
- return decoder, nil
- }
- if typ.Kind() == reflect.Slice && typ.Elem().Kind() == reflect.Uint8 {
- sliceDecoder, err := prefix("[slice]").addToDecoder(decoderOfSlice(cfg, typ))
- if err != nil {
- return nil, err
- }
- return &base64Codec{sliceDecoder: sliceDecoder}, nil
- }
- if typ.Implements(anyType) {
- return &anyCodec{}, nil
- }
- switch typ.Kind() {
- case reflect.String:
- if typeName != "string" {
- return decoderOfType(cfg, reflect.TypeOf((*string)(nil)).Elem())
- }
- return &stringCodec{}, nil
- case reflect.Int:
- if typeName != "int" {
- return decoderOfType(cfg, reflect.TypeOf((*int)(nil)).Elem())
- }
- return &intCodec{}, nil
- case reflect.Int8:
- if typeName != "int8" {
- return decoderOfType(cfg, reflect.TypeOf((*int8)(nil)).Elem())
- }
- return &int8Codec{}, nil
- case reflect.Int16:
- if typeName != "int16" {
- return decoderOfType(cfg, reflect.TypeOf((*int16)(nil)).Elem())
- }
- return &int16Codec{}, nil
- case reflect.Int32:
- if typeName != "int32" {
- return decoderOfType(cfg, reflect.TypeOf((*int32)(nil)).Elem())
- }
- return &int32Codec{}, nil
- case reflect.Int64:
- if typeName != "int64" {
- return decoderOfType(cfg, reflect.TypeOf((*int64)(nil)).Elem())
- }
- return &int64Codec{}, nil
- case reflect.Uint:
- if typeName != "uint" {
- return decoderOfType(cfg, reflect.TypeOf((*uint)(nil)).Elem())
- }
- return &uintCodec{}, nil
- case reflect.Uint8:
- if typeName != "uint8" {
- return decoderOfType(cfg, reflect.TypeOf((*uint8)(nil)).Elem())
- }
- return &uint8Codec{}, nil
- case reflect.Uint16:
- if typeName != "uint16" {
- return decoderOfType(cfg, reflect.TypeOf((*uint16)(nil)).Elem())
- }
- return &uint16Codec{}, nil
- case reflect.Uint32:
- if typeName != "uint32" {
- return decoderOfType(cfg, reflect.TypeOf((*uint32)(nil)).Elem())
- }
- return &uint32Codec{}, nil
- case reflect.Uintptr:
- if typeName != "uintptr" {
- return decoderOfType(cfg, reflect.TypeOf((*uintptr)(nil)).Elem())
- }
- return &uintptrCodec{}, nil
- case reflect.Uint64:
- if typeName != "uint64" {
- return decoderOfType(cfg, reflect.TypeOf((*uint64)(nil)).Elem())
- }
- return &uint64Codec{}, nil
- case reflect.Float32:
- if typeName != "float32" {
- return decoderOfType(cfg, reflect.TypeOf((*float32)(nil)).Elem())
- }
- return &float32Codec{}, nil
- case reflect.Float64:
- if typeName != "float64" {
- return decoderOfType(cfg, reflect.TypeOf((*float64)(nil)).Elem())
- }
- return &float64Codec{}, nil
- case reflect.Bool:
- if typeName != "bool" {
- return decoderOfType(cfg, reflect.TypeOf((*bool)(nil)).Elem())
- }
- return &boolCodec{}, nil
- case reflect.Interface:
- if typ.NumMethod() == 0 {
- return &emptyInterfaceCodec{}, nil
- }
- return &nonEmptyInterfaceCodec{}, nil
- case reflect.Struct:
- return prefix(fmt.Sprintf("[%s]", typeName)).addToDecoder(decoderOfStruct(cfg, typ))
- case reflect.Array:
- return prefix("[array]").addToDecoder(decoderOfArray(cfg, typ))
- case reflect.Slice:
- return prefix("[slice]").addToDecoder(decoderOfSlice(cfg, typ))
- case reflect.Map:
- return prefix("[map]").addToDecoder(decoderOfMap(cfg, typ))
- case reflect.Ptr:
- return prefix("[optional]").addToDecoder(decoderOfOptional(cfg, typ))
- default:
- return nil, fmt.Errorf("unsupported type: %v", typ)
- }
- }
- func encoderOfType(cfg *frozenConfig, typ reflect.Type) (ValEncoder, error) {
- cacheKey := typ
- encoder := cfg.getEncoderFromCache(cacheKey)
- if encoder != nil {
- return encoder, nil
- }
- encoder = getTypeEncoderFromExtension(cfg, typ)
- if encoder != nil {
- cfg.addEncoderToCache(cacheKey, encoder)
- return encoder, nil
- }
- encoder = &placeholderEncoder{cfg: cfg, cacheKey: cacheKey}
- cfg.addEncoderToCache(cacheKey, encoder)
- encoder, err := createEncoderOfType(cfg, typ)
- for _, extension := range extensions {
- encoder = extension.DecorateEncoder(typ, encoder)
- }
- for _, extension := range cfg.extensions {
- encoder = extension.DecorateEncoder(typ, encoder)
- }
- cfg.addEncoderToCache(cacheKey, encoder)
- return encoder, err
- }
- func createEncoderOfType(cfg *frozenConfig, typ reflect.Type) (ValEncoder, error) {
- if typ == jsonRawMessageType {
- return &jsonRawMessageCodec{}, nil
- }
- if typ == jsoniterRawMessageType {
- return &jsoniterRawMessageCodec{}, nil
- }
- if typ.AssignableTo(jsonNumberType) {
- return &jsonNumberCodec{}, nil
- }
- if typ.AssignableTo(jsoniterNumberType) {
- return &jsoniterNumberCodec{}, nil
- }
- if typ.Implements(marshalerType) {
- checkIsEmpty, err := createCheckIsEmpty(cfg, typ)
- if err != nil {
- return nil, err
- }
- templateInterface := reflect.New(typ).Elem().Interface()
- var encoder ValEncoder = &marshalerEncoder{
- templateInterface: extractInterface(templateInterface),
- checkIsEmpty: checkIsEmpty,
- }
- if typ.Kind() == reflect.Ptr {
- encoder = &OptionalEncoder{encoder}
- }
- return encoder, nil
- }
- if reflect.PtrTo(typ).Implements(marshalerType) {
- checkIsEmpty, err := createCheckIsEmpty(cfg, reflect.PtrTo(typ))
- if err != nil {
- return nil, err
- }
- templateInterface := reflect.New(typ).Interface()
- var encoder ValEncoder = &marshalerEncoder{
- templateInterface: extractInterface(templateInterface),
- checkIsEmpty: checkIsEmpty,
- }
- return encoder, nil
- }
- if typ.Implements(textMarshalerType) {
- checkIsEmpty, err := createCheckIsEmpty(cfg, typ)
- if err != nil {
- return nil, err
- }
- templateInterface := reflect.New(typ).Elem().Interface()
- var encoder ValEncoder = &textMarshalerEncoder{
- templateInterface: extractInterface(templateInterface),
- checkIsEmpty: checkIsEmpty,
- }
- if typ.Kind() == reflect.Ptr {
- encoder = &OptionalEncoder{encoder}
- }
- return encoder, nil
- }
- if typ.Kind() == reflect.Slice && typ.Elem().Kind() == reflect.Uint8 {
- return &base64Codec{}, nil
- }
- if typ.Implements(anyType) {
- return &anyCodec{}, nil
- }
- return createEncoderOfSimpleType(cfg, typ)
- }
- func createCheckIsEmpty(cfg *frozenConfig, typ reflect.Type) (checkIsEmpty, error) {
- kind := typ.Kind()
- switch kind {
- case reflect.String:
- return &stringCodec{}, nil
- case reflect.Int:
- return &intCodec{}, nil
- case reflect.Int8:
- return &int8Codec{}, nil
- case reflect.Int16:
- return &int16Codec{}, nil
- case reflect.Int32:
- return &int32Codec{}, nil
- case reflect.Int64:
- return &int64Codec{}, nil
- case reflect.Uint:
- return &uintCodec{}, nil
- case reflect.Uint8:
- return &uint8Codec{}, nil
- case reflect.Uint16:
- return &uint16Codec{}, nil
- case reflect.Uint32:
- return &uint32Codec{}, nil
- case reflect.Uintptr:
- return &uintptrCodec{}, nil
- case reflect.Uint64:
- return &uint64Codec{}, nil
- case reflect.Float32:
- return &float32Codec{}, nil
- case reflect.Float64:
- return &float64Codec{}, nil
- case reflect.Bool:
- return &boolCodec{}, nil
- case reflect.Interface:
- if typ.NumMethod() == 0 {
- return &emptyInterfaceCodec{}, nil
- }
- return &nonEmptyInterfaceCodec{}, nil
- case reflect.Struct:
- return &structEncoder{}, nil
- case reflect.Array:
- return &arrayEncoder{}, nil
- case reflect.Slice:
- return &sliceEncoder{}, nil
- case reflect.Map:
- return encoderOfMap(cfg, typ)
- case reflect.Ptr:
- return &OptionalEncoder{}, nil
- default:
- return nil, fmt.Errorf("unsupported type: %v", typ)
- }
- }
- func createEncoderOfSimpleType(cfg *frozenConfig, typ reflect.Type) (ValEncoder, error) {
- typeName := typ.String()
- kind := typ.Kind()
- switch kind {
- case reflect.String:
- if typeName != "string" {
- return encoderOfType(cfg, reflect.TypeOf((*string)(nil)).Elem())
- }
- return &stringCodec{}, nil
- case reflect.Int:
- if typeName != "int" {
- return encoderOfType(cfg, reflect.TypeOf((*int)(nil)).Elem())
- }
- return &intCodec{}, nil
- case reflect.Int8:
- if typeName != "int8" {
- return encoderOfType(cfg, reflect.TypeOf((*int8)(nil)).Elem())
- }
- return &int8Codec{}, nil
- case reflect.Int16:
- if typeName != "int16" {
- return encoderOfType(cfg, reflect.TypeOf((*int16)(nil)).Elem())
- }
- return &int16Codec{}, nil
- case reflect.Int32:
- if typeName != "int32" {
- return encoderOfType(cfg, reflect.TypeOf((*int32)(nil)).Elem())
- }
- return &int32Codec{}, nil
- case reflect.Int64:
- if typeName != "int64" {
- return encoderOfType(cfg, reflect.TypeOf((*int64)(nil)).Elem())
- }
- return &int64Codec{}, nil
- case reflect.Uint:
- if typeName != "uint" {
- return encoderOfType(cfg, reflect.TypeOf((*uint)(nil)).Elem())
- }
- return &uintCodec{}, nil
- case reflect.Uint8:
- if typeName != "uint8" {
- return encoderOfType(cfg, reflect.TypeOf((*uint8)(nil)).Elem())
- }
- return &uint8Codec{}, nil
- case reflect.Uint16:
- if typeName != "uint16" {
- return encoderOfType(cfg, reflect.TypeOf((*uint16)(nil)).Elem())
- }
- return &uint16Codec{}, nil
- case reflect.Uint32:
- if typeName != "uint32" {
- return encoderOfType(cfg, reflect.TypeOf((*uint32)(nil)).Elem())
- }
- return &uint32Codec{}, nil
- case reflect.Uintptr:
- if typeName != "uintptr" {
- return encoderOfType(cfg, reflect.TypeOf((*uintptr)(nil)).Elem())
- }
- return &uintptrCodec{}, nil
- case reflect.Uint64:
- if typeName != "uint64" {
- return encoderOfType(cfg, reflect.TypeOf((*uint64)(nil)).Elem())
- }
- return &uint64Codec{}, nil
- case reflect.Float32:
- if typeName != "float32" {
- return encoderOfType(cfg, reflect.TypeOf((*float32)(nil)).Elem())
- }
- return &float32Codec{}, nil
- case reflect.Float64:
- if typeName != "float64" {
- return encoderOfType(cfg, reflect.TypeOf((*float64)(nil)).Elem())
- }
- return &float64Codec{}, nil
- case reflect.Bool:
- if typeName != "bool" {
- return encoderOfType(cfg, reflect.TypeOf((*bool)(nil)).Elem())
- }
- return &boolCodec{}, nil
- case reflect.Interface:
- if typ.NumMethod() == 0 {
- return &emptyInterfaceCodec{}, nil
- }
- return &nonEmptyInterfaceCodec{}, nil
- case reflect.Struct:
- return prefix(fmt.Sprintf("[%s]", typeName)).addToEncoder(encoderOfStruct(cfg, typ))
- case reflect.Array:
- return prefix("[array]").addToEncoder(encoderOfArray(cfg, typ))
- case reflect.Slice:
- return prefix("[slice]").addToEncoder(encoderOfSlice(cfg, typ))
- case reflect.Map:
- return prefix("[map]").addToEncoder(encoderOfMap(cfg, typ))
- case reflect.Ptr:
- return prefix("[optional]").addToEncoder(encoderOfOptional(cfg, typ))
- default:
- return nil, fmt.Errorf("unsupported type: %v", typ)
- }
- }
- func decoderOfOptional(cfg *frozenConfig, typ reflect.Type) (ValDecoder, error) {
- elemType := typ.Elem()
- decoder, err := decoderOfType(cfg, elemType)
- if err != nil {
- return nil, err
- }
- return &OptionalDecoder{elemType, decoder}, nil
- }
- func encoderOfOptional(cfg *frozenConfig, typ reflect.Type) (ValEncoder, error) {
- elemType := typ.Elem()
- elemEncoder, err := encoderOfType(cfg, elemType)
- if err != nil {
- return nil, err
- }
- encoder := &OptionalEncoder{elemEncoder}
- if elemType.Kind() == reflect.Map {
- encoder = &OptionalEncoder{encoder}
- }
- return encoder, nil
- }
- func decoderOfMap(cfg *frozenConfig, typ reflect.Type) (ValDecoder, error) {
- decoder, err := decoderOfType(cfg, typ.Elem())
- if err != nil {
- return nil, err
- }
- mapInterface := reflect.New(typ).Interface()
- return &mapDecoder{typ, typ.Key(), typ.Elem(), decoder, extractInterface(mapInterface)}, nil
- }
- func extractInterface(val interface{}) emptyInterface {
- return *((*emptyInterface)(unsafe.Pointer(&val)))
- }
- func encoderOfMap(cfg *frozenConfig, typ reflect.Type) (ValEncoder, error) {
- elemType := typ.Elem()
- encoder, err := encoderOfType(cfg, elemType)
- if err != nil {
- return nil, err
- }
- mapInterface := reflect.New(typ).Elem().Interface()
- if cfg.sortMapKeys {
- return &sortKeysMapEncoder{typ, elemType, encoder, *((*emptyInterface)(unsafe.Pointer(&mapInterface)))}, nil
- }
- return &mapEncoder{typ, elemType, encoder, *((*emptyInterface)(unsafe.Pointer(&mapInterface)))}, nil
- }
|