1package jsoniter 2 3import ( 4 "fmt" 5 "github.com/modern-go/reflect2" 6 "io" 7 "unsafe" 8) 9 10func decoderOfArray(ctx *ctx, typ reflect2.Type) ValDecoder { 11 arrayType := typ.(*reflect2.UnsafeArrayType) 12 decoder := decoderOfType(ctx.append("[arrayElem]"), arrayType.Elem()) 13 return &arrayDecoder{arrayType, decoder} 14} 15 16func encoderOfArray(ctx *ctx, typ reflect2.Type) ValEncoder { 17 arrayType := typ.(*reflect2.UnsafeArrayType) 18 if arrayType.Len() == 0 { 19 return emptyArrayEncoder{} 20 } 21 encoder := encoderOfType(ctx.append("[arrayElem]"), arrayType.Elem()) 22 return &arrayEncoder{arrayType, encoder} 23} 24 25type emptyArrayEncoder struct{} 26 27func (encoder emptyArrayEncoder) Encode(ptr unsafe.Pointer, stream *Stream) { 28 stream.WriteEmptyArray() 29} 30 31func (encoder emptyArrayEncoder) IsEmpty(ptr unsafe.Pointer) bool { 32 return true 33} 34 35type arrayEncoder struct { 36 arrayType *reflect2.UnsafeArrayType 37 elemEncoder ValEncoder 38} 39 40func (encoder *arrayEncoder) Encode(ptr unsafe.Pointer, stream *Stream) { 41 stream.WriteArrayStart() 42 elemPtr := unsafe.Pointer(ptr) 43 encoder.elemEncoder.Encode(elemPtr, stream) 44 for i := 1; i < encoder.arrayType.Len(); i++ { 45 stream.WriteMore() 46 elemPtr = encoder.arrayType.UnsafeGetIndex(ptr, i) 47 encoder.elemEncoder.Encode(elemPtr, stream) 48 } 49 stream.WriteArrayEnd() 50 if stream.Error != nil && stream.Error != io.EOF { 51 stream.Error = fmt.Errorf("%v: %s", encoder.arrayType, stream.Error.Error()) 52 } 53} 54 55func (encoder *arrayEncoder) IsEmpty(ptr unsafe.Pointer) bool { 56 return false 57} 58 59type arrayDecoder struct { 60 arrayType *reflect2.UnsafeArrayType 61 elemDecoder ValDecoder 62} 63 64func (decoder *arrayDecoder) Decode(ptr unsafe.Pointer, iter *Iterator) { 65 decoder.doDecode(ptr, iter) 66 if iter.Error != nil && iter.Error != io.EOF { 67 iter.Error = fmt.Errorf("%v: %s", decoder.arrayType, iter.Error.Error()) 68 } 69} 70 71func (decoder *arrayDecoder) doDecode(ptr unsafe.Pointer, iter *Iterator) { 72 c := iter.nextToken() 73 arrayType := decoder.arrayType 74 if c == 'n' { 75 iter.skipThreeBytes('u', 'l', 'l') 76 return 77 } 78 if c != '[' { 79 iter.ReportError("decode array", "expect [ or n, but found "+string([]byte{c})) 80 return 81 } 82 c = iter.nextToken() 83 if c == ']' { 84 return 85 } 86 iter.unreadByte() 87 elemPtr := arrayType.UnsafeGetIndex(ptr, 0) 88 decoder.elemDecoder.Decode(elemPtr, iter) 89 length := 1 90 for c = iter.nextToken(); c == ','; c = iter.nextToken() { 91 if length >= arrayType.Len() { 92 iter.Skip() 93 continue 94 } 95 idx := length 96 length += 1 97 elemPtr = arrayType.UnsafeGetIndex(ptr, idx) 98 decoder.elemDecoder.Decode(elemPtr, iter) 99 } 100 if c != ']' { 101 iter.ReportError("decode array", "expect ], but found "+string([]byte{c})) 102 return 103 } 104} 105