Add Custom Encoder/Decoder in Config

config.go

@@ -25,6 +25,9 @@ type Config struct {
 ValidateJsonRawMessage bool
 ObjectFieldMustBeSimpleString bool
 CaseSensitive bool
+
+TypeEncoders map[string]ValEncoder
+TypeDecoders map[string]ValDecoder
 }
 
 // API the public interface of this package.
@@ -80,6 +83,9 @@ type frozenConfig struct {
 streamPool *sync.Pool
 iteratorPool *sync.Pool
 caseSensitive bool
+
+typeEncoders map[string]ValEncoder
+typeDecoders map[string]ValDecoder
 }
 
 func (cfg *frozenConfig) initCache() {
@@ -127,13 +133,23 @@ func addFrozenConfigToCache(cfg Config, frozenConfig *frozenConfig) {
 
 // Froze forge API from config
 func (cfg Config) Froze() API {
+typeDecoders := make(map[string]ValDecoder)
+typeEncoders := make(map[string]ValEncoder)
+if cfg.TypeDecoders != nil {
+typeDecoders = cfg.TypeDecoders
+}
+if cfg.TypeEncoders != nil {
+typeEncoders = cfg.TypeEncoders
+}
 api := &frozenConfig{
 sortMapKeys: cfg.SortMapKeys,
 indentionStep: cfg.IndentionStep,
 objectFieldMustBeSimpleString: cfg.ObjectFieldMustBeSimpleString,
 onlyTaggedField: cfg.OnlyTaggedField,
 disallowUnknownFields: cfg.DisallowUnknownFields,
 caseSensitive: cfg.CaseSensitive,
+typeDecoders: typeDecoders,
+typeEncoders: typeEncoders,
 }
 api.streamPool = &sync.Pool{
 New: func() interface{} {
@@ -272,13 +288,11 @@ func (cfg *frozenConfig) escapeHTML(encoderExtension EncoderExtension) {
 }
 
 func (cfg *frozenConfig) cleanDecoders() {
-typeDecoders = map[string]ValDecoder{}
 fieldDecoders = map[string]ValDecoder{}
 *cfg = *(cfg.configBeforeFrozen.Froze().(*frozenConfig))
 }
 
 func (cfg *frozenConfig) cleanEncoders() {
-typeEncoders = map[string]ValEncoder{}
 fieldEncoders = map[string]ValEncoder{}
 *cfg = *(cfg.configBeforeFrozen.Froze().(*frozenConfig))
 }

go.sum

@@ -9,7 +9,6 @@ github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742 h1:Esafd1046DLD
 github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0=
 github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/stretchr/objx v0.1.0 h1:4G4v2dO3VZwixGIRoQ5Lfboy6nUhCyYzaqnIAPPhYs4=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.3.0 h1:TivCn/peBQ7UY8ooIcPgZFpTNSz0Q2U6UrFlUfqbe0Q=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=

reflect_extension.go

@@ -2,17 +2,16 @@ package jsoniter
 
 import (
 "fmt"
-"github.com/modern-go/reflect2"
 "reflect"
 "sort"
 "strings"
 "unicode"
 "unsafe"
+
+"github.com/modern-go/reflect2"
 )
 
-var typeDecoders = map[string]ValDecoder{}
 var fieldDecoders = map[string]ValDecoder{}
-var typeEncoders = map[string]ValEncoder{}
 var fieldEncoders = map[string]ValEncoder{}
 var extensions = []Extension{}
 
@@ -195,16 +194,6 @@ type DecoderFunc func(ptr unsafe.Pointer, iter *Iterator)
 // EncoderFunc the function form of TypeEncoder
 type EncoderFunc func(ptr unsafe.Pointer, stream *Stream)
 
-// RegisterTypeDecoderFunc register TypeDecoder for a type with function
-func RegisterTypeDecoderFunc(typ string, fun DecoderFunc) {
-typeDecoders[typ] = &funcDecoder{fun}
-}
-
-// RegisterTypeDecoder register TypeDecoder for a typ
-func RegisterTypeDecoder(typ string, decoder ValDecoder) {
-typeDecoders[typ] = decoder
-}
-
 // RegisterFieldDecoderFunc register TypeDecoder for a struct field with function
 func RegisterFieldDecoderFunc(typ string, field string, fun DecoderFunc) {
 RegisterFieldDecoder(typ, field, &funcDecoder{fun})
@@ -215,16 +204,6 @@ func RegisterFieldDecoder(typ string, field string, decoder ValDecoder) {
 fieldDecoders[fmt.Sprintf("%s/%s", typ, field)] = decoder
 }
 
-// RegisterTypeEncoderFunc register TypeEncoder for a type with encode/isEmpty function
-func RegisterTypeEncoderFunc(typ string, fun EncoderFunc, isEmptyFunc func(unsafe.Pointer) bool) {
-typeEncoders[typ] = &funcEncoder{fun, isEmptyFunc}
-}
-
-// RegisterTypeEncoder register TypeEncoder for a type
-func RegisterTypeEncoder(typ string, encoder ValEncoder) {
-typeEncoders[typ] = encoder
-}
-
 // RegisterFieldEncoderFunc register TypeEncoder for a struct field with encode/isEmpty function
 func RegisterFieldEncoderFunc(typ string, field string, fun EncoderFunc, isEmptyFunc func(unsafe.Pointer) bool) {
 RegisterFieldEncoder(typ, field, &funcEncoder{fun, isEmptyFunc})
@@ -270,21 +249,28 @@ func _getTypeDecoderFromExtension(ctx *ctx, typ reflect2.Type) ValDecoder {
 return decoder
 }
 }
-typeName := typ.String()
-decoder = typeDecoders[typeName]
+decoder = ctx.typeDecoders[GetKind(typ.Type1())]
 if decoder != nil {
 return decoder
 }
 if typ.Kind() == reflect.Ptr {
 ptrType := typ.(*reflect2.UnsafePtrType)
-decoder := typeDecoders[ptrType.Elem().String()]
+decoder := ctx.typeDecoders[GetKind(ptrType.Type1())]
 if decoder != nil {
 return &OptionalDecoder{ptrType.Elem(), decoder}
 }
 }
 return nil
 }
 
+func GetKind(p reflect.Type) string {
+if p.PkgPath() == "" {
+return p.String()
+}
+
+return p.PkgPath() + "." + p.Name()
+}
+
 func getTypeEncoderFromExtension(ctx *ctx, typ reflect2.Type) ValEncoder {
 encoder := _getTypeEncoderFromExtension(ctx, typ)
 if encoder != nil {
@@ -316,14 +302,13 @@ func _getTypeEncoderFromExtension(ctx *ctx, typ reflect2.Type) ValEncoder {
 return encoder
 }
 }
-typeName := typ.String()
-encoder = typeEncoders[typeName]
+encoder = ctx.typeEncoders[GetKind(typ.Type1())]
 if encoder != nil {
 return encoder
 }
 if typ.Kind() == reflect.Ptr {
 typePtr := typ.(*reflect2.UnsafePtrType)
-encoder := typeEncoders[typePtr.Elem().String()]
+encoder := ctx.typeEncoders[GetKind(typePtr.Type1())]
 if encoder != nil {
 return &OptionalEncoder{encoder}
 }