jsoniter_reflect.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798
  1. package jsoniter
  2. import (
  3. "reflect"
  4. "errors"
  5. "fmt"
  6. "unsafe"
  7. "sync/atomic"
  8. "strings"
  9. "io"
  10. "strconv"
  11. )
  12. type Decoder interface {
  13. decode(ptr unsafe.Pointer, iter *Iterator)
  14. }
  15. type stringDecoder struct {
  16. }
  17. func (decoder *stringDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  18. *((*string)(ptr)) = iter.ReadString()
  19. }
  20. type intDecoder struct {
  21. }
  22. func (decoder *intDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  23. *((*int)(ptr)) = iter.ReadInt()
  24. }
  25. type int8Decoder struct {
  26. }
  27. func (decoder *int8Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  28. *((*int8)(ptr)) = iter.ReadInt8()
  29. }
  30. type int16Decoder struct {
  31. }
  32. func (decoder *int16Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  33. *((*int16)(ptr)) = iter.ReadInt16()
  34. }
  35. type int32Decoder struct {
  36. }
  37. func (decoder *int32Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  38. *((*int32)(ptr)) = iter.ReadInt32()
  39. }
  40. type int64Decoder struct {
  41. }
  42. func (decoder *int64Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  43. *((*int64)(ptr)) = iter.ReadInt64()
  44. }
  45. type uintDecoder struct {
  46. }
  47. func (decoder *uintDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  48. *((*uint)(ptr)) = iter.ReadUint()
  49. }
  50. type uint8Decoder struct {
  51. }
  52. func (decoder *uint8Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  53. *((*uint8)(ptr)) = iter.ReadUint8()
  54. }
  55. type uint16Decoder struct {
  56. }
  57. func (decoder *uint16Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  58. *((*uint16)(ptr)) = iter.ReadUint16()
  59. }
  60. type uint32Decoder struct {
  61. }
  62. func (decoder *uint32Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  63. *((*uint32)(ptr)) = iter.ReadUint32()
  64. }
  65. type uint64Decoder struct {
  66. }
  67. func (decoder *uint64Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  68. *((*uint64)(ptr)) = iter.ReadUint64()
  69. }
  70. type float32Decoder struct {
  71. }
  72. func (decoder *float32Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  73. *((*float32)(ptr)) = iter.ReadFloat32()
  74. }
  75. type float64Decoder struct {
  76. }
  77. func (decoder *float64Decoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  78. *((*float64)(ptr)) = iter.ReadFloat64()
  79. }
  80. type boolDecoder struct {
  81. }
  82. func (decoder *boolDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  83. *((*bool)(ptr)) = iter.ReadBool()
  84. }
  85. type interfaceDecoder struct {
  86. }
  87. func (decoder *interfaceDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  88. *((*interface{})(ptr)) = iter.ReadAny().Get()
  89. }
  90. type anyDecoder struct {
  91. }
  92. func (decoder *anyDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  93. *((*Any)(ptr)) = *iter.ReadAny()
  94. }
  95. type stringNumberDecoder struct {
  96. elemDecoder Decoder
  97. }
  98. func (decoder *stringNumberDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  99. c := iter.readByte()
  100. if c != '"' {
  101. iter.ReportError("stringNumberDecoder", `expect "`)
  102. return
  103. }
  104. decoder.elemDecoder.decode(ptr, iter)
  105. if iter.Error != nil {
  106. return
  107. }
  108. c = iter.readByte()
  109. if c != '"' {
  110. iter.ReportError("stringNumberDecoder", `expect "`)
  111. return
  112. }
  113. }
  114. type optionalDecoder struct {
  115. valueType reflect.Type
  116. valueDecoder Decoder
  117. }
  118. func (decoder *optionalDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  119. if iter.ReadNull() {
  120. *((*unsafe.Pointer)(ptr)) = nil
  121. } else {
  122. value := reflect.New(decoder.valueType)
  123. decoder.valueDecoder.decode(unsafe.Pointer(value.Pointer()), iter)
  124. *((*uintptr)(ptr)) = value.Pointer()
  125. }
  126. }
  127. type generalStructDecoder struct {
  128. type_ reflect.Type
  129. fields map[string]*structFieldDecoder
  130. }
  131. func (decoder *generalStructDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  132. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  133. fieldDecoder := decoder.fields[field]
  134. if fieldDecoder == nil {
  135. iter.Skip()
  136. } else {
  137. fieldDecoder.decode(ptr, iter)
  138. }
  139. }
  140. if iter.Error != nil && iter.Error != io.EOF {
  141. iter.Error = fmt.Errorf("%v: %s", decoder.type_, iter.Error.Error())
  142. }
  143. }
  144. type skipDecoder struct {
  145. type_ reflect.Type
  146. }
  147. func (decoder *skipDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  148. iter.Skip()
  149. if iter.Error != nil && iter.Error != io.EOF {
  150. iter.Error = fmt.Errorf("%v: %s", decoder.type_, iter.Error.Error())
  151. }
  152. }
  153. type oneFieldStructDecoder struct {
  154. type_ reflect.Type
  155. fieldName string
  156. fieldDecoder *structFieldDecoder
  157. }
  158. func (decoder *oneFieldStructDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  159. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  160. if field == decoder.fieldName {
  161. decoder.fieldDecoder.decode(ptr, iter)
  162. } else {
  163. iter.Skip()
  164. }
  165. }
  166. if iter.Error != nil && iter.Error != io.EOF {
  167. iter.Error = fmt.Errorf("%v: %s", decoder.type_, iter.Error.Error())
  168. }
  169. }
  170. type twoFieldsStructDecoder struct {
  171. type_ reflect.Type
  172. fieldName1 string
  173. fieldDecoder1 *structFieldDecoder
  174. fieldName2 string
  175. fieldDecoder2 *structFieldDecoder
  176. }
  177. func (decoder *twoFieldsStructDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  178. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  179. switch field {
  180. case decoder.fieldName1:
  181. decoder.fieldDecoder1.decode(ptr, iter)
  182. case decoder.fieldName2:
  183. decoder.fieldDecoder2.decode(ptr, iter)
  184. default:
  185. iter.Skip()
  186. }
  187. }
  188. if iter.Error != nil && iter.Error != io.EOF {
  189. iter.Error = fmt.Errorf("%v: %s", decoder.type_, iter.Error.Error())
  190. }
  191. }
  192. type threeFieldsStructDecoder struct {
  193. type_ reflect.Type
  194. fieldName1 string
  195. fieldDecoder1 *structFieldDecoder
  196. fieldName2 string
  197. fieldDecoder2 *structFieldDecoder
  198. fieldName3 string
  199. fieldDecoder3 *structFieldDecoder
  200. }
  201. func (decoder *threeFieldsStructDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  202. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  203. switch field {
  204. case decoder.fieldName1:
  205. decoder.fieldDecoder1.decode(ptr, iter)
  206. case decoder.fieldName2:
  207. decoder.fieldDecoder2.decode(ptr, iter)
  208. case decoder.fieldName3:
  209. decoder.fieldDecoder3.decode(ptr, iter)
  210. default:
  211. iter.Skip()
  212. }
  213. }
  214. if iter.Error != nil && iter.Error != io.EOF {
  215. iter.Error = fmt.Errorf("%v: %s", decoder.type_, iter.Error.Error())
  216. }
  217. }
  218. type fourFieldsStructDecoder struct {
  219. type_ reflect.Type
  220. fieldName1 string
  221. fieldDecoder1 *structFieldDecoder
  222. fieldName2 string
  223. fieldDecoder2 *structFieldDecoder
  224. fieldName3 string
  225. fieldDecoder3 *structFieldDecoder
  226. fieldName4 string
  227. fieldDecoder4 *structFieldDecoder
  228. }
  229. func (decoder *fourFieldsStructDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  230. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  231. switch field {
  232. case decoder.fieldName1:
  233. decoder.fieldDecoder1.decode(ptr, iter)
  234. case decoder.fieldName2:
  235. decoder.fieldDecoder2.decode(ptr, iter)
  236. case decoder.fieldName3:
  237. decoder.fieldDecoder3.decode(ptr, iter)
  238. case decoder.fieldName4:
  239. decoder.fieldDecoder4.decode(ptr, iter)
  240. default:
  241. iter.Skip()
  242. }
  243. }
  244. if iter.Error != nil && iter.Error != io.EOF {
  245. iter.Error = fmt.Errorf("%v: %s", decoder.type_, iter.Error.Error())
  246. }
  247. }
  248. type structFieldDecoder struct {
  249. field *reflect.StructField
  250. fieldDecoder Decoder
  251. }
  252. func (decoder *structFieldDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  253. fieldPtr := uintptr(ptr) + decoder.field.Offset
  254. decoder.fieldDecoder.decode(unsafe.Pointer(fieldPtr), iter)
  255. if iter.Error != nil && iter.Error != io.EOF {
  256. iter.Error = fmt.Errorf("%s: %s", decoder.field.Name, iter.Error.Error())
  257. }
  258. }
  259. type mapDecoder struct {
  260. mapType reflect.Type
  261. elemType reflect.Type
  262. elemDecoder Decoder
  263. mapInterface emptyInterface
  264. }
  265. func (decoder *mapDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  266. // dark magic to cast unsafe.Pointer back to interface{} using reflect.Type
  267. mapInterface := decoder.mapInterface
  268. mapInterface.word = ptr
  269. realInterface := (*interface{})(unsafe.Pointer(&mapInterface))
  270. realVal := reflect.ValueOf(*realInterface).Elem()
  271. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  272. elem := reflect.New(decoder.elemType)
  273. decoder.elemDecoder.decode(unsafe.Pointer(elem.Pointer()), iter)
  274. realVal.SetMapIndex(reflect.ValueOf(field), elem.Elem())
  275. }
  276. }
  277. type sliceDecoder struct {
  278. sliceType reflect.Type
  279. elemType reflect.Type
  280. elemDecoder Decoder
  281. }
  282. // sliceHeader is a safe version of SliceHeader used within this package.
  283. type sliceHeader struct {
  284. Data unsafe.Pointer
  285. Len int
  286. Cap int
  287. }
  288. func (decoder *sliceDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  289. decoder.doDecode(ptr, iter)
  290. if iter.Error != nil && iter.Error != io.EOF {
  291. iter.Error = fmt.Errorf("%v: %s", decoder.sliceType, iter.Error.Error())
  292. }
  293. }
  294. func (decoder *sliceDecoder) doDecode(ptr unsafe.Pointer, iter *Iterator) {
  295. slice := (*sliceHeader)(ptr)
  296. reuseSlice(slice, decoder.sliceType, 4)
  297. if !iter.ReadArray() {
  298. return
  299. }
  300. offset := uintptr(0)
  301. decoder.elemDecoder.decode(unsafe.Pointer(uintptr(slice.Data) + offset), iter)
  302. if !iter.ReadArray() {
  303. slice.Len = 1
  304. return
  305. }
  306. offset += decoder.elemType.Size()
  307. decoder.elemDecoder.decode(unsafe.Pointer(uintptr(slice.Data) + offset), iter)
  308. if !iter.ReadArray() {
  309. slice.Len = 2
  310. return
  311. }
  312. offset += decoder.elemType.Size()
  313. decoder.elemDecoder.decode(unsafe.Pointer(uintptr(slice.Data) + offset), iter)
  314. if !iter.ReadArray() {
  315. slice.Len = 3
  316. return
  317. }
  318. offset += decoder.elemType.Size()
  319. decoder.elemDecoder.decode(unsafe.Pointer(uintptr(slice.Data) + offset), iter)
  320. slice.Len = 4
  321. for iter.ReadArray() {
  322. growOne(slice, decoder.sliceType, decoder.elemType)
  323. offset += decoder.elemType.Size()
  324. decoder.elemDecoder.decode(unsafe.Pointer(uintptr(slice.Data) + offset), iter)
  325. }
  326. }
  327. // grow grows the slice s so that it can hold extra more values, allocating
  328. // more capacity if needed. It also returns the old and new slice lengths.
  329. func growOne(slice *sliceHeader, sliceType reflect.Type, elementType reflect.Type) {
  330. newLen := slice.Len + 1
  331. if newLen <= slice.Cap {
  332. slice.Len = newLen
  333. return
  334. }
  335. newCap := slice.Cap
  336. if newCap == 0 {
  337. newCap = 1
  338. } else {
  339. for newCap < newLen {
  340. if slice.Len < 1024 {
  341. newCap += newCap
  342. } else {
  343. newCap += newCap / 4
  344. }
  345. }
  346. }
  347. dst := unsafe.Pointer(reflect.MakeSlice(sliceType, newLen, newCap).Pointer())
  348. originalBytesCount := uintptr(slice.Len) * elementType.Size()
  349. srcPtr := (*[1 << 30]byte)(slice.Data)
  350. dstPtr := (*[1 << 30]byte)(dst)
  351. for i := uintptr(0); i < originalBytesCount; i++ {
  352. dstPtr[i] = srcPtr[i]
  353. }
  354. slice.Len = newLen
  355. slice.Cap = newCap
  356. slice.Data = dst
  357. }
  358. func reuseSlice(slice *sliceHeader, sliceType reflect.Type, expectedCap int) {
  359. if expectedCap <= slice.Cap {
  360. return
  361. }
  362. dst := unsafe.Pointer(reflect.MakeSlice(sliceType, 0, expectedCap).Pointer())
  363. slice.Cap = expectedCap
  364. slice.Data = dst
  365. }
  366. var DECODERS unsafe.Pointer
  367. func addDecoderToCache(cacheKey reflect.Type, decoder Decoder) {
  368. retry := true
  369. for retry {
  370. ptr := atomic.LoadPointer(&DECODERS)
  371. cache := *(*map[reflect.Type]Decoder)(ptr)
  372. copy := map[reflect.Type]Decoder{}
  373. for k, v := range cache {
  374. copy[k] = v
  375. }
  376. copy[cacheKey] = decoder
  377. retry = !atomic.CompareAndSwapPointer(&DECODERS, ptr, unsafe.Pointer(&copy))
  378. }
  379. }
  380. func getDecoderFromCache(cacheKey reflect.Type) Decoder {
  381. ptr := atomic.LoadPointer(&DECODERS)
  382. cache := *(*map[reflect.Type]Decoder)(ptr)
  383. return cache[cacheKey]
  384. }
  385. var typeDecoders map[string]Decoder
  386. var fieldDecoders map[string]Decoder
  387. func init() {
  388. typeDecoders = map[string]Decoder{}
  389. fieldDecoders = map[string]Decoder{}
  390. atomic.StorePointer(&DECODERS, unsafe.Pointer(&map[string]Decoder{}))
  391. }
  392. type DecoderFunc func(ptr unsafe.Pointer, iter *Iterator)
  393. type funcDecoder struct {
  394. func_ DecoderFunc
  395. }
  396. func (decoder *funcDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  397. decoder.func_(ptr, iter)
  398. }
  399. func RegisterTypeDecoder(type_ string, func_ DecoderFunc) {
  400. typeDecoders[type_] = &funcDecoder{func_}
  401. }
  402. func RegisterFieldDecoder(type_ string, field string, func_ DecoderFunc) {
  403. fieldDecoders[fmt.Sprintf("%s/%s", type_, field)] = &funcDecoder{func_}
  404. }
  405. func ClearDecoders() {
  406. typeDecoders = map[string]Decoder{}
  407. fieldDecoders = map[string]Decoder{}
  408. }
  409. // emptyInterface is the header for an interface{} value.
  410. type emptyInterface struct {
  411. typ *struct{}
  412. word unsafe.Pointer
  413. }
  414. func (iter *Iterator) ReadAny() (ret *Any) {
  415. valueType := iter.WhatIsNext()
  416. switch valueType {
  417. case String:
  418. return MakeAny(iter.ReadString())
  419. case Number:
  420. return iter.readNumber()
  421. case Null:
  422. return MakeAny(nil)
  423. case Bool:
  424. return MakeAny(iter.ReadBool())
  425. case Array:
  426. val := []interface{}{}
  427. for (iter.ReadArray()) {
  428. element := iter.ReadAny()
  429. if iter.Error != nil {
  430. return
  431. }
  432. val = append(val, element.val)
  433. }
  434. return MakeAny(val)
  435. case Object:
  436. val := map[string]interface{}{}
  437. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  438. element := iter.ReadAny()
  439. if iter.Error != nil {
  440. return
  441. }
  442. val[field] = element.val
  443. }
  444. return MakeAny(val)
  445. default:
  446. iter.ReportError("ReadAny", fmt.Sprintf("unexpected value type: %v", valueType))
  447. return MakeAny(nil)
  448. }
  449. }
  450. func (iter *Iterator) readNumber() (ret *Any) {
  451. strBuf := [8]byte{}
  452. str := strBuf[0:0]
  453. hasMore := true
  454. foundFloat := false
  455. foundNegative := false
  456. for(hasMore) {
  457. for i := iter.head; i < iter.tail; i++ {
  458. c := iter.buf[i]
  459. switch c {
  460. case '-':
  461. foundNegative = true
  462. str = append(str, c)
  463. continue
  464. case '+', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9':
  465. str = append(str, c)
  466. continue
  467. case '.', 'e', 'E':
  468. foundFloat = true
  469. str = append(str, c)
  470. continue
  471. default:
  472. hasMore = false
  473. break
  474. }
  475. }
  476. if hasMore {
  477. if !iter.loadMore() {
  478. break
  479. }
  480. }
  481. }
  482. if iter.Error != nil && iter.Error != io.EOF {
  483. return
  484. }
  485. number := *(*string)(unsafe.Pointer(&str))
  486. if foundFloat {
  487. val, err := strconv.ParseFloat(number, 64)
  488. if err != nil {
  489. iter.Error = err
  490. return
  491. }
  492. return MakeAny(val)
  493. }
  494. if foundNegative {
  495. val, err := strconv.ParseInt(number, 10, 64)
  496. if err != nil {
  497. iter.Error = err
  498. return
  499. }
  500. return MakeAny(val)
  501. }
  502. val, err := strconv.ParseUint(number, 10, 64)
  503. if err != nil {
  504. iter.Error = err
  505. return
  506. }
  507. return MakeAny(val)
  508. }
  509. func (iter *Iterator) Read(obj interface{}) {
  510. type_ := reflect.TypeOf(obj)
  511. cacheKey := type_.Elem()
  512. cachedDecoder := getDecoderFromCache(cacheKey)
  513. if cachedDecoder == nil {
  514. decoder, err := decoderOfType(type_)
  515. if err != nil {
  516. iter.Error = err
  517. return
  518. }
  519. cachedDecoder = decoder
  520. addDecoderToCache(cacheKey, decoder)
  521. }
  522. e := (*emptyInterface)(unsafe.Pointer(&obj))
  523. cachedDecoder.decode(e.word, iter)
  524. }
  525. type prefix string
  526. func (p prefix) addTo(decoder Decoder, err error) (Decoder, error) {
  527. if err != nil {
  528. return nil, fmt.Errorf("%s: %s", p, err.Error())
  529. }
  530. return decoder, err
  531. }
  532. func decoderOfType(type_ reflect.Type) (Decoder, error) {
  533. switch type_.Kind() {
  534. case reflect.Ptr:
  535. return prefix("ptr").addTo(decoderOfPtr(type_.Elem()))
  536. default:
  537. return nil, errors.New("expect ptr")
  538. }
  539. }
  540. func decoderOfPtr(type_ reflect.Type) (Decoder, error) {
  541. typeName := type_.String()
  542. if typeName == "jsoniter.Any" {
  543. return &anyDecoder{}, nil
  544. }
  545. typeDecoder := typeDecoders[typeName]
  546. if typeDecoder != nil {
  547. return typeDecoder, nil
  548. }
  549. switch type_.Kind() {
  550. case reflect.String:
  551. return &stringDecoder{}, nil
  552. case reflect.Int:
  553. return &intDecoder{}, nil
  554. case reflect.Int8:
  555. return &int8Decoder{}, nil
  556. case reflect.Int16:
  557. return &int16Decoder{}, nil
  558. case reflect.Int32:
  559. return &int32Decoder{}, nil
  560. case reflect.Int64:
  561. return &int64Decoder{}, nil
  562. case reflect.Uint:
  563. return &uintDecoder{}, nil
  564. case reflect.Uint8:
  565. return &uint8Decoder{}, nil
  566. case reflect.Uint16:
  567. return &uint16Decoder{}, nil
  568. case reflect.Uint32:
  569. return &uint32Decoder{}, nil
  570. case reflect.Uint64:
  571. return &uint64Decoder{}, nil
  572. case reflect.Float32:
  573. return &float32Decoder{}, nil
  574. case reflect.Float64:
  575. return &float64Decoder{}, nil
  576. case reflect.Bool:
  577. return &boolDecoder{}, nil
  578. case reflect.Interface:
  579. return &interfaceDecoder{}, nil
  580. case reflect.Struct:
  581. return decoderOfStruct(type_)
  582. case reflect.Slice:
  583. return prefix("[slice]").addTo(decoderOfSlice(type_))
  584. case reflect.Map:
  585. return prefix("[map]").addTo(decoderOfMap(type_))
  586. case reflect.Ptr:
  587. return prefix("[optional]").addTo(decoderOfOptional(type_.Elem()))
  588. default:
  589. return nil, fmt.Errorf("unsupported type: %v", type_)
  590. }
  591. }
  592. func decoderOfOptional(type_ reflect.Type) (Decoder, error) {
  593. decoder, err := decoderOfPtr(type_)
  594. if err != nil {
  595. return nil, err
  596. }
  597. return &optionalDecoder{type_, decoder}, nil
  598. }
  599. func decoderOfStruct(type_ reflect.Type) (Decoder, error) {
  600. fields := map[string]*structFieldDecoder{}
  601. for i := 0; i < type_.NumField(); i++ {
  602. field := type_.Field(i)
  603. fieldDecoderKey := fmt.Sprintf("%s/%s", type_.String(), field.Name)
  604. decoder := fieldDecoders[fieldDecoderKey]
  605. tagParts := strings.Split(field.Tag.Get("json"), ",")
  606. jsonFieldName := tagParts[0]
  607. if jsonFieldName == "" {
  608. jsonFieldName = field.Name
  609. }
  610. if decoder == nil {
  611. var err error
  612. decoder, err = decoderOfPtr(field.Type)
  613. if err != nil {
  614. return prefix(fmt.Sprintf("{%s}", field.Name)).addTo(decoder, err)
  615. }
  616. }
  617. if len(tagParts) > 1 && tagParts[1] == "string" {
  618. decoder = &stringNumberDecoder{decoder}
  619. }
  620. if jsonFieldName != "-" {
  621. fields[jsonFieldName] = &structFieldDecoder{&field, decoder}
  622. }
  623. }
  624. switch len(fields) {
  625. case 0:
  626. return &skipDecoder{type_}, nil
  627. case 1:
  628. for fieldName, fieldDecoder := range fields {
  629. return &oneFieldStructDecoder{type_, fieldName, fieldDecoder}, nil
  630. }
  631. case 2:
  632. var fieldName1 string
  633. var fieldName2 string
  634. var fieldDecoder1 *structFieldDecoder
  635. var fieldDecoder2 *structFieldDecoder
  636. for fieldName, fieldDecoder := range fields {
  637. if fieldName1 == "" {
  638. fieldName1 = fieldName
  639. fieldDecoder1 = fieldDecoder
  640. } else {
  641. fieldName2 = fieldName
  642. fieldDecoder2 = fieldDecoder
  643. }
  644. }
  645. return &twoFieldsStructDecoder{type_, fieldName1, fieldDecoder1, fieldName2, fieldDecoder2}, nil
  646. case 3:
  647. var fieldName1 string
  648. var fieldName2 string
  649. var fieldName3 string
  650. var fieldDecoder1 *structFieldDecoder
  651. var fieldDecoder2 *structFieldDecoder
  652. var fieldDecoder3 *structFieldDecoder
  653. for fieldName, fieldDecoder := range fields {
  654. if fieldName1 == "" {
  655. fieldName1 = fieldName
  656. fieldDecoder1 = fieldDecoder
  657. } else if fieldName2 == "" {
  658. fieldName2 = fieldName
  659. fieldDecoder2 = fieldDecoder
  660. } else {
  661. fieldName3 = fieldName
  662. fieldDecoder3 = fieldDecoder
  663. }
  664. }
  665. return &threeFieldsStructDecoder{type_,
  666. fieldName1, fieldDecoder1, fieldName2, fieldDecoder2, fieldName3, fieldDecoder3}, nil
  667. case 4:
  668. var fieldName1 string
  669. var fieldName2 string
  670. var fieldName3 string
  671. var fieldName4 string
  672. var fieldDecoder1 *structFieldDecoder
  673. var fieldDecoder2 *structFieldDecoder
  674. var fieldDecoder3 *structFieldDecoder
  675. var fieldDecoder4 *structFieldDecoder
  676. for fieldName, fieldDecoder := range fields {
  677. if fieldName1 == "" {
  678. fieldName1 = fieldName
  679. fieldDecoder1 = fieldDecoder
  680. } else if fieldName2 == "" {
  681. fieldName2 = fieldName
  682. fieldDecoder2 = fieldDecoder
  683. } else if fieldName3 == "" {
  684. fieldName3 = fieldName
  685. fieldDecoder3 = fieldDecoder
  686. } else {
  687. fieldName4 = fieldName
  688. fieldDecoder4 = fieldDecoder
  689. }
  690. }
  691. return &fourFieldsStructDecoder{type_,
  692. fieldName1, fieldDecoder1, fieldName2, fieldDecoder2, fieldName3, fieldDecoder3,
  693. fieldName4, fieldDecoder4}, nil
  694. }
  695. return &generalStructDecoder{type_, fields}, nil
  696. }
  697. func decoderOfSlice(type_ reflect.Type) (Decoder, error) {
  698. decoder, err := decoderOfPtr(type_.Elem())
  699. if err != nil {
  700. return nil, err
  701. }
  702. return &sliceDecoder{type_, type_.Elem(), decoder}, nil
  703. }
  704. func decoderOfMap(type_ reflect.Type) (Decoder, error) {
  705. decoder, err := decoderOfPtr(type_.Elem())
  706. if err != nil {
  707. return nil, err
  708. }
  709. mapInterface := reflect.New(type_).Interface()
  710. return &mapDecoder{type_, type_.Elem(), decoder, *((*emptyInterface)(unsafe.Pointer(&mapInterface)))}, nil
  711. }