blob: be7a0e218801c891a832759d31302b0e49f8af09 [file] [log] [blame]
Matthias Andreas Benkard832a54e2019-01-29 09:27:38 +01001package jsoniter
2
3import (
4 "fmt"
5 "reflect"
6 "unsafe"
7
8 "github.com/modern-go/reflect2"
9)
10
11// ValDecoder is an internal type registered to cache as needed.
12// Don't confuse jsoniter.ValDecoder with json.Decoder.
13// For json.Decoder's adapter, refer to jsoniter.AdapterDecoder(todo link).
14//
15// Reflection on type to create decoders, which is then cached
16// Reflection on value is avoided as we can, as the reflect.Value itself will allocate, with following exceptions
17// 1. create instance of new value, for example *int will need a int to be allocated
18// 2. append to slice, if the existing cap is not enough, allocate will be done using Reflect.New
19// 3. assignment to map, both key and value will be reflect.Value
20// For a simple struct binding, it will be reflect.Value free and allocation free
21type ValDecoder interface {
22 Decode(ptr unsafe.Pointer, iter *Iterator)
23}
24
25// ValEncoder is an internal type registered to cache as needed.
26// Don't confuse jsoniter.ValEncoder with json.Encoder.
27// For json.Encoder's adapter, refer to jsoniter.AdapterEncoder(todo godoc link).
28type ValEncoder interface {
29 IsEmpty(ptr unsafe.Pointer) bool
30 Encode(ptr unsafe.Pointer, stream *Stream)
31}
32
33type checkIsEmpty interface {
34 IsEmpty(ptr unsafe.Pointer) bool
35}
36
37type ctx struct {
38 *frozenConfig
39 prefix string
40 encoders map[reflect2.Type]ValEncoder
41 decoders map[reflect2.Type]ValDecoder
42}
43
44func (b *ctx) caseSensitive() bool {
45 if b.frozenConfig == nil {
46 // default is case-insensitive
47 return false
48 }
49 return b.frozenConfig.caseSensitive
50}
51
52func (b *ctx) append(prefix string) *ctx {
53 return &ctx{
54 frozenConfig: b.frozenConfig,
55 prefix: b.prefix + " " + prefix,
56 encoders: b.encoders,
57 decoders: b.decoders,
58 }
59}
60
61// ReadVal copy the underlying JSON into go interface, same as json.Unmarshal
62func (iter *Iterator) ReadVal(obj interface{}) {
63 cacheKey := reflect2.RTypeOf(obj)
64 decoder := iter.cfg.getDecoderFromCache(cacheKey)
65 if decoder == nil {
66 typ := reflect2.TypeOf(obj)
67 if typ.Kind() != reflect.Ptr {
68 iter.ReportError("ReadVal", "can only unmarshal into pointer")
69 return
70 }
71 decoder = iter.cfg.DecoderOf(typ)
72 }
73 ptr := reflect2.PtrOf(obj)
74 if ptr == nil {
75 iter.ReportError("ReadVal", "can not read into nil pointer")
76 return
77 }
78 decoder.Decode(ptr, iter)
79}
80
81// WriteVal copy the go interface into underlying JSON, same as json.Marshal
82func (stream *Stream) WriteVal(val interface{}) {
83 if nil == val {
84 stream.WriteNil()
85 return
86 }
87 cacheKey := reflect2.RTypeOf(val)
88 encoder := stream.cfg.getEncoderFromCache(cacheKey)
89 if encoder == nil {
90 typ := reflect2.TypeOf(val)
91 encoder = stream.cfg.EncoderOf(typ)
92 }
93 encoder.Encode(reflect2.PtrOf(val), stream)
94}
95
96func (cfg *frozenConfig) DecoderOf(typ reflect2.Type) ValDecoder {
97 cacheKey := typ.RType()
98 decoder := cfg.getDecoderFromCache(cacheKey)
99 if decoder != nil {
100 return decoder
101 }
102 ctx := &ctx{
103 frozenConfig: cfg,
104 prefix: "",
105 decoders: map[reflect2.Type]ValDecoder{},
106 encoders: map[reflect2.Type]ValEncoder{},
107 }
108 ptrType := typ.(*reflect2.UnsafePtrType)
109 decoder = decoderOfType(ctx, ptrType.Elem())
110 cfg.addDecoderToCache(cacheKey, decoder)
111 return decoder
112}
113
114func decoderOfType(ctx *ctx, typ reflect2.Type) ValDecoder {
115 decoder := getTypeDecoderFromExtension(ctx, typ)
116 if decoder != nil {
117 return decoder
118 }
119 decoder = createDecoderOfType(ctx, typ)
120 for _, extension := range extensions {
121 decoder = extension.DecorateDecoder(typ, decoder)
122 }
123 for _, extension := range ctx.extensions {
124 decoder = extension.DecorateDecoder(typ, decoder)
125 }
126 return decoder
127}
128
129func createDecoderOfType(ctx *ctx, typ reflect2.Type) ValDecoder {
130 decoder := ctx.decoders[typ]
131 if decoder != nil {
132 return decoder
133 }
134 placeholder := &placeholderDecoder{}
135 ctx.decoders[typ] = placeholder
136 decoder = _createDecoderOfType(ctx, typ)
137 placeholder.decoder = decoder
138 return decoder
139}
140
141func _createDecoderOfType(ctx *ctx, typ reflect2.Type) ValDecoder {
142 decoder := createDecoderOfJsonRawMessage(ctx, typ)
143 if decoder != nil {
144 return decoder
145 }
146 decoder = createDecoderOfJsonNumber(ctx, typ)
147 if decoder != nil {
148 return decoder
149 }
150 decoder = createDecoderOfMarshaler(ctx, typ)
151 if decoder != nil {
152 return decoder
153 }
154 decoder = createDecoderOfAny(ctx, typ)
155 if decoder != nil {
156 return decoder
157 }
158 decoder = createDecoderOfNative(ctx, typ)
159 if decoder != nil {
160 return decoder
161 }
162 switch typ.Kind() {
163 case reflect.Interface:
164 ifaceType, isIFace := typ.(*reflect2.UnsafeIFaceType)
165 if isIFace {
166 return &ifaceDecoder{valType: ifaceType}
167 }
168 return &efaceDecoder{}
169 case reflect.Struct:
170 return decoderOfStruct(ctx, typ)
171 case reflect.Array:
172 return decoderOfArray(ctx, typ)
173 case reflect.Slice:
174 return decoderOfSlice(ctx, typ)
175 case reflect.Map:
176 return decoderOfMap(ctx, typ)
177 case reflect.Ptr:
178 return decoderOfOptional(ctx, typ)
179 default:
180 return &lazyErrorDecoder{err: fmt.Errorf("%s%s is unsupported type", ctx.prefix, typ.String())}
181 }
182}
183
184func (cfg *frozenConfig) EncoderOf(typ reflect2.Type) ValEncoder {
185 cacheKey := typ.RType()
186 encoder := cfg.getEncoderFromCache(cacheKey)
187 if encoder != nil {
188 return encoder
189 }
190 ctx := &ctx{
191 frozenConfig: cfg,
192 prefix: "",
193 decoders: map[reflect2.Type]ValDecoder{},
194 encoders: map[reflect2.Type]ValEncoder{},
195 }
196 encoder = encoderOfType(ctx, typ)
197 if typ.LikePtr() {
198 encoder = &onePtrEncoder{encoder}
199 }
200 cfg.addEncoderToCache(cacheKey, encoder)
201 return encoder
202}
203
204type onePtrEncoder struct {
205 encoder ValEncoder
206}
207
208func (encoder *onePtrEncoder) IsEmpty(ptr unsafe.Pointer) bool {
209 return encoder.encoder.IsEmpty(unsafe.Pointer(&ptr))
210}
211
212func (encoder *onePtrEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
213 encoder.encoder.Encode(unsafe.Pointer(&ptr), stream)
214}
215
216func encoderOfType(ctx *ctx, typ reflect2.Type) ValEncoder {
217 encoder := getTypeEncoderFromExtension(ctx, typ)
218 if encoder != nil {
219 return encoder
220 }
221 encoder = createEncoderOfType(ctx, typ)
222 for _, extension := range extensions {
223 encoder = extension.DecorateEncoder(typ, encoder)
224 }
225 for _, extension := range ctx.extensions {
226 encoder = extension.DecorateEncoder(typ, encoder)
227 }
228 return encoder
229}
230
231func createEncoderOfType(ctx *ctx, typ reflect2.Type) ValEncoder {
232 encoder := ctx.encoders[typ]
233 if encoder != nil {
234 return encoder
235 }
236 placeholder := &placeholderEncoder{}
237 ctx.encoders[typ] = placeholder
238 encoder = _createEncoderOfType(ctx, typ)
239 placeholder.encoder = encoder
240 return encoder
241}
242func _createEncoderOfType(ctx *ctx, typ reflect2.Type) ValEncoder {
243 encoder := createEncoderOfJsonRawMessage(ctx, typ)
244 if encoder != nil {
245 return encoder
246 }
247 encoder = createEncoderOfJsonNumber(ctx, typ)
248 if encoder != nil {
249 return encoder
250 }
251 encoder = createEncoderOfMarshaler(ctx, typ)
252 if encoder != nil {
253 return encoder
254 }
255 encoder = createEncoderOfAny(ctx, typ)
256 if encoder != nil {
257 return encoder
258 }
259 encoder = createEncoderOfNative(ctx, typ)
260 if encoder != nil {
261 return encoder
262 }
263 kind := typ.Kind()
264 switch kind {
265 case reflect.Interface:
266 return &dynamicEncoder{typ}
267 case reflect.Struct:
268 return encoderOfStruct(ctx, typ)
269 case reflect.Array:
270 return encoderOfArray(ctx, typ)
271 case reflect.Slice:
272 return encoderOfSlice(ctx, typ)
273 case reflect.Map:
274 return encoderOfMap(ctx, typ)
275 case reflect.Ptr:
276 return encoderOfOptional(ctx, typ)
277 default:
278 return &lazyErrorEncoder{err: fmt.Errorf("%s%s is unsupported type", ctx.prefix, typ.String())}
279 }
280}
281
282type lazyErrorDecoder struct {
283 err error
284}
285
286func (decoder *lazyErrorDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) {
287 if iter.WhatIsNext() != NilValue {
288 if iter.Error == nil {
289 iter.Error = decoder.err
290 }
291 } else {
292 iter.Skip()
293 }
294}
295
296type lazyErrorEncoder struct {
297 err error
298}
299
300func (encoder *lazyErrorEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
301 if ptr == nil {
302 stream.WriteNil()
303 } else if stream.Error == nil {
304 stream.Error = encoder.err
305 }
306}
307
308func (encoder *lazyErrorEncoder) IsEmpty(ptr unsafe.Pointer) bool {
309 return false
310}
311
312type placeholderDecoder struct {
313 decoder ValDecoder
314}
315
316func (decoder *placeholderDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) {
317 decoder.decoder.Decode(ptr, iter)
318}
319
320type placeholderEncoder struct {
321 encoder ValEncoder
322}
323
324func (encoder *placeholderEncoder) Encode(ptr unsafe.Pointer, stream *Stream) {
325 encoder.encoder.Encode(ptr, stream)
326}
327
328func (encoder *placeholderEncoder) IsEmpty(ptr unsafe.Pointer) bool {
329 return encoder.encoder.IsEmpty(ptr)
330}