|
|
package stt
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"github.com/modern-go/reflect2"
|
|
|
"reflect"
|
|
|
"unsafe"
|
|
|
)
|
|
|
|
|
|
// ValDecoder is an internal type registered to cache as needed.
|
|
|
// 反射获取类型并缓存
|
|
|
// 尽可能地避免反射获取值，reflect.Value本身具有alloc，但具有如下例外
|
|
|
// 1. 创建新值的实例，如： *int 将需要 int
|
|
|
// 2. slice的append，如果容量(cap)不够，将使用Reflect.New从而alloc
|
|
|
// 3. 将值分配给map，key/value都是reflect.Value
|
|
|
// 对于简单struct的绑定，应该尽量避免reflect.Value和其本身的alloc开销
|
|
|
type ValDecoder interface {
|
|
|
Decode(ptr unsafe.Pointer, iter *Iterator)
|
|
|
}
|
|
|
|
|
|
// ValEncoder is an internal type registered to cache as needed.
|
|
|
type ValEncoder interface {
|
|
|
IsEmpty(ptr unsafe.Pointer) bool
|
|
|
Encode(ptr unsafe.Pointer, stream *Stream)
|
|
|
}
|
|
|
|
|
|
type checkIsEmpty interface {
|
|
|
IsEmpty(ptr unsafe.Pointer) bool
|
|
|
}
|
|
|
|
|
|
type ctx struct {
|
|
|
*frozenConfig
|
|
|
prefix string
|
|
|
encoders map[reflect2.Type]ValEncoder
|
|
|
decoders map[reflect2.Type]ValDecoder
|
|
|
}
|
|
|
|
|
|
func (b *ctx) caseSensitive() bool {
|
|
|
if b.frozenConfig == nil {
|
|
|
// default is case-insensitive
|
|
|
return false
|
|
|
}
|
|
|
return b.frozenConfig.caseSensitive
|
|
|
}
|
|
|
|
|
|
func (b *ctx) append(prefix string) *ctx {
|
|
|
return &ctx{
|
|
|
frozenConfig: b.frozenConfig,
|
|
|
prefix: b.prefix + " " + prefix,
|
|
|
encoders: b.encoders,
|
|
|
decoders: b.decoders,
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// ReadVal copy the underlying STT into go interface
|
|
|
func (iter *Iterator) ReadVal(obj interface{}) {
|
|
|
depth := iter.depth
|
|
|
cacheKey := reflect2.RTypeOf(obj)
|
|
|
decoder := iter.cfg.getDecoderFromCache(cacheKey)
|
|
|
if decoder == nil {
|
|
|
typ := reflect2.TypeOf(obj)
|
|
|
if typ == nil || typ.Kind() != reflect.Ptr {
|
|
|
iter.ReportError("ReadVal", "can only unmarshal into pointer")
|
|
|
return
|
|
|
}
|
|
|
decoder = iter.cfg.DecoderOf(typ)
|
|
|
}
|
|
|
ptr := reflect2.PtrOf(obj)
|
|
|
if ptr == nil {
|
|
|
iter.ReportError("ReadVal", "can not read into nil pointer")
|
|
|
return
|
|
|
}
|
|
|
decoder.Decode(ptr, iter)
|
|
|
if iter.depth != depth {
|
|
|
iter.ReportError("ReadVal", "unexpected mismatched nesting")
|
|
|
return
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// WriteVal copy the go interface into underlying JSON, same as json.Marshal
|
|
|
func (stream *Stream) WriteVal(val interface{}) {
|
|
|
if nil == val {
|
|
|
stream.WriteNil()
|
|
|
return
|
|
|
}
|
|
|
cacheKey := reflect2.RTypeOf(val)
|
|
|
encoder := stream.cfg.getEncoderFromCache(cacheKey)
|
|
|
if encoder == nil {
|
|
|
typ := reflect2.TypeOf(val)
|
|
|
encoder = stream.cfg.EncoderOf(typ)
|
|
|
}
|
|
|
encoder.Encode(reflect2.PtrOf(val), stream)
|
|
|
}
|
|
|
|
|
|
func (cfg *frozenConfig) DecoderOf(typ reflect2.Type) ValDecoder {
|
|
|
cacheKey := typ.RType()
|
|
|
decoder := cfg.getDecoderFromCache(cacheKey)
|
|
|
if decoder != nil {
|
|
|
return decoder
|
|
|
}
|
|
|
ctx := &ctx{
|
|
|
frozenConfig: cfg,
|
|
|
prefix: "",
|
|
|
decoders: map[reflect2.Type]ValDecoder{},
|
|
|
encoders: map[reflect2.Type]ValEncoder{},
|
|
|
}
|
|
|
ptrType := typ.(*reflect2.UnsafePtrType)
|
|
|
decoder = decoderOfType(ctx, ptrType.Elem())
|
|
|
cfg.addDecoderToCache(cacheKey, decoder)
|
|
|
return decoder
|
|
|
}
|
|
|
|
|
|
func decoderOfType(ctx *ctx, typ reflect2.Type) ValDecoder {
|
|
|
decoder := getTypeDecoderFromExtension(ctx, typ)
|
|
|
if decoder != nil {
|
|
|
return decoder
|
|
|
}
|
|
|
decoder = createDecoderOfType(ctx, typ)
|
|
|
for _, extension := range extensions {
|
|
|
decoder = extension.DecorateDecoder(typ, decoder)
|
|
|
}
|
|
|
decoder = ctx.decoderExtension.DecorateDecoder(typ, decoder)
|
|
|
for _, extension := range ctx.extraExtensions {
|
|
|
decoder = extension.DecorateDecoder(typ, decoder)
|
|
|
}
|
|
|
return decoder
|
|
|
}
|
|
|
|
|
|
func createDecoderOfType(ctx *ctx, typ reflect2.Type) ValDecoder {
|
|
|
decoder := ctx.decoders[typ]
|
|
|
if decoder != nil {
|
|
|
return decoder
|
|
|
}
|
|
|
placeholder := &placeholderDecoder{}
|
|
|
ctx.decoders[typ] = placeholder
|
|
|
decoder = _createDecoderOfType(ctx, typ)
|
|
|
placeholder.decoder = decoder
|
|
|
return decoder
|
|
|
}
|
|
|
|
|
|
func _createDecoderOfType(ctx *ctx, typ reflect2.Type) ValDecoder {
|
|
|
decoder := createDecoderOfNative(ctx, typ)
|
|
|
if decoder != nil {
|
|
|
return decoder
|
|
|
}
|
|
|
switch typ.Kind() {
|
|
|
case reflect.Interface:
|
|
|
ifaceType, isIFace := typ.(*reflect2.UnsafeIFaceType)
|
|
|
if isIFace {
|
|
|
return &ifaceDecoder{valType: ifaceType}
|
|
|
}
|
|
|
return &efaceDecoder{}
|
|
|
case reflect.Struct:
|
|
|
return decoderOfStruct(ctx, typ)
|
|
|
case reflect.Array:
|
|
|
return decoderOfArray(ctx, typ)
|
|
|
case reflect.Slice:
|
|
|
return decoderOfSlice(ctx, typ)
|
|
|
case reflect.Map:
|
|
|
return decoderOfMap(ctx, typ)
|
|
|
case reflect.Ptr:
|
|
|
return decoderOfOptional(ctx, typ)
|
|
|
default:
|
|
|
return &lazyErrorDecoder{err: fmt.Errorf("%s%s is unsupported type", ctx.prefix, typ.String())}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (cfg *frozenConfig) EncoderOf(typ reflect2.Type) ValEncoder {
|
|
|
cacheKey := typ.RType()
|
|
|
encoder := cfg.getEncoderFromCache(cacheKey)
|
|
|
if encoder != nil {
|
|
|
return encoder
|
|
|
}
|
|
|
ctx := &ctx{
|
|
|
frozenConfig: cfg,
|
|
|
prefix: "",
|
|
|
decoders: map[reflect2.Type]ValDecoder{},
|
|
|
encoders: map[reflect2.Type]ValEncoder{},
|
|
|
}
|
|
|
encoder = encoderOfType(ctx, typ)
|
|
|
if typ.LikePtr() {
|
|
|
encoder = &onePtrEncoder{encoder}
|
|
|
}
|
|
|
cfg.addEncoderToCache(cacheKey, encoder)
|
|
|
return encoder
|
|
|
}
|
|
|
|
|
|
type onePtrEncoder struct {
|
|
|
encoder ValEncoder
|
|
|
}
|
|
|
|
|
|
func (encoder *onePtrEncoder) IsEmpty(ptr unsafe.Pointer) bool {
|
|
|
return encoder.encoder.IsEmpty(unsafe.Pointer(&ptr))
|
|
|
}
|
|
|
|
|
|
func (encoder *onePtrEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
|
|
|
encoder.encoder.Encode(unsafe.Pointer(&ptr), stream)
|
|
|
}
|
|
|
|
|
|
func encoderOfType(ctx *ctx, typ reflect2.Type) ValEncoder {
|
|
|
encoder := getTypeEncoderFromExtension(ctx, typ)
|
|
|
if encoder != nil {
|
|
|
return encoder
|
|
|
}
|
|
|
encoder = createEncoderOfType(ctx, typ)
|
|
|
for _, extension := range extensions {
|
|
|
encoder = extension.DecorateEncoder(typ, encoder)
|
|
|
}
|
|
|
encoder = ctx.encoderExtension.DecorateEncoder(typ, encoder)
|
|
|
for _, extension := range ctx.extraExtensions {
|
|
|
encoder = extension.DecorateEncoder(typ, encoder)
|
|
|
}
|
|
|
return encoder
|
|
|
}
|
|
|
|
|
|
func createEncoderOfType(ctx *ctx, typ reflect2.Type) ValEncoder {
|
|
|
encoder := ctx.encoders[typ]
|
|
|
if encoder != nil {
|
|
|
return encoder
|
|
|
}
|
|
|
placeholder := &placeholderEncoder{}
|
|
|
ctx.encoders[typ] = placeholder
|
|
|
encoder = _createEncoderOfType(ctx, typ)
|
|
|
placeholder.encoder = encoder
|
|
|
return encoder
|
|
|
}
|
|
|
func _createEncoderOfType(ctx *ctx, typ reflect2.Type) ValEncoder {
|
|
|
encoder := createEncoderOfNative(ctx, typ)
|
|
|
if encoder != nil {
|
|
|
return encoder
|
|
|
}
|
|
|
kind := typ.Kind()
|
|
|
switch kind {
|
|
|
case reflect.Interface:
|
|
|
return &dynamicEncoder{typ}
|
|
|
case reflect.Struct:
|
|
|
return encoderOfStruct(ctx, typ)
|
|
|
case reflect.Array:
|
|
|
return encoderOfArray(ctx, typ)
|
|
|
case reflect.Slice:
|
|
|
return encoderOfSlice(ctx, typ)
|
|
|
case reflect.Map:
|
|
|
return encoderOfMap(ctx, typ)
|
|
|
case reflect.Ptr:
|
|
|
return encoderOfOptional(ctx, typ)
|
|
|
default:
|
|
|
return &lazyErrorEncoder{err: fmt.Errorf("%s%s is unsupported type", ctx.prefix, typ.String())}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
type lazyErrorDecoder struct {
|
|
|
err error
|
|
|
}
|
|
|
|
|
|
func (decoder *lazyErrorDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) {
|
|
|
if iter.WhatIsNext() != NilValue {
|
|
|
if iter.Error == nil {
|
|
|
iter.Error = decoder.err
|
|
|
}
|
|
|
} else {
|
|
|
iter.Skip()
|
|
|
}
|
|
|
}
|
|
|
|
|
|
type lazyErrorEncoder struct {
|
|
|
err error
|
|
|
}
|
|
|
|
|
|
func (encoder *lazyErrorEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
|
|
|
if ptr == nil {
|
|
|
stream.WriteNil()
|
|
|
} else if stream.Error == nil {
|
|
|
stream.Error = encoder.err
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func (encoder *lazyErrorEncoder) IsEmpty(ptr unsafe.Pointer) bool {
|
|
|
return false
|
|
|
}
|
|
|
|
|
|
type placeholderDecoder struct {
|
|
|
decoder ValDecoder
|
|
|
}
|
|
|
|
|
|
func (decoder *placeholderDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) {
|
|
|
decoder.decoder.Decode(ptr, iter)
|
|
|
}
|
|
|
|
|
|
type placeholderEncoder struct {
|
|
|
encoder ValEncoder
|
|
|
}
|
|
|
|
|
|
func (encoder *placeholderEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
|
|
|
encoder.encoder.Encode(ptr, stream)
|
|
|
}
|
|
|
|
|
|
func (encoder *placeholderEncoder) IsEmpty(ptr unsafe.Pointer) bool {
|
|
|
return encoder.encoder.IsEmpty(ptr)
|
|
|
}
|