jsoniter_reflect.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827
  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(string([]byte(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. var extensions []ExtensionFunc
  388. func init() {
  389. typeDecoders = map[string]Decoder{}
  390. fieldDecoders = map[string]Decoder{}
  391. extensions = []ExtensionFunc{}
  392. atomic.StorePointer(&DECODERS, unsafe.Pointer(&map[string]Decoder{}))
  393. }
  394. type DecoderFunc func(ptr unsafe.Pointer, iter *Iterator)
  395. type ExtensionFunc func(type_ reflect.Type, field *reflect.StructField) ([]string, DecoderFunc)
  396. type funcDecoder struct {
  397. func_ DecoderFunc
  398. }
  399. func (decoder *funcDecoder) decode(ptr unsafe.Pointer, iter *Iterator) {
  400. decoder.func_(ptr, iter)
  401. }
  402. func RegisterTypeDecoder(type_ string, func_ DecoderFunc) {
  403. typeDecoders[type_] = &funcDecoder{func_}
  404. }
  405. func RegisterFieldDecoder(type_ string, field string, func_ DecoderFunc) {
  406. fieldDecoders[fmt.Sprintf("%s/%s", type_, field)] = &funcDecoder{func_}
  407. }
  408. func RegisterExtension(extension ExtensionFunc) {
  409. extensions = append(extensions, extension)
  410. }
  411. func ClearDecoders() {
  412. typeDecoders = map[string]Decoder{}
  413. fieldDecoders = map[string]Decoder{}
  414. }
  415. // emptyInterface is the header for an interface{} value.
  416. type emptyInterface struct {
  417. typ *struct{}
  418. word unsafe.Pointer
  419. }
  420. func (iter *Iterator) ReadAny() (ret *Any) {
  421. valueType := iter.WhatIsNext()
  422. switch valueType {
  423. case String:
  424. return MakeAny(iter.ReadString())
  425. case Number:
  426. return iter.readNumber()
  427. case Null:
  428. return MakeAny(nil)
  429. case Bool:
  430. return MakeAny(iter.ReadBool())
  431. case Array:
  432. val := []interface{}{}
  433. for (iter.ReadArray()) {
  434. element := iter.ReadAny()
  435. if iter.Error != nil {
  436. return
  437. }
  438. val = append(val, element.val)
  439. }
  440. return MakeAny(val)
  441. case Object:
  442. val := map[string]interface{}{}
  443. for field := iter.ReadObject(); field != ""; field = iter.ReadObject() {
  444. element := iter.ReadAny()
  445. if iter.Error != nil {
  446. return
  447. }
  448. val[string([]byte(field))] = element.val
  449. }
  450. return MakeAny(val)
  451. default:
  452. iter.ReportError("ReadAny", fmt.Sprintf("unexpected value type: %v", valueType))
  453. return MakeAny(nil)
  454. }
  455. }
  456. func (iter *Iterator) readNumber() (ret *Any) {
  457. strBuf := [8]byte{}
  458. str := strBuf[0:0]
  459. hasMore := true
  460. foundFloat := false
  461. foundNegative := false
  462. for(hasMore) {
  463. for i := iter.head; i < iter.tail; i++ {
  464. c := iter.buf[i]
  465. switch c {
  466. case '-':
  467. foundNegative = true
  468. str = append(str, c)
  469. continue
  470. case '+', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9':
  471. str = append(str, c)
  472. continue
  473. case '.', 'e', 'E':
  474. foundFloat = true
  475. str = append(str, c)
  476. continue
  477. default:
  478. hasMore = false
  479. break
  480. }
  481. }
  482. if hasMore {
  483. if !iter.loadMore() {
  484. break
  485. }
  486. }
  487. }
  488. if iter.Error != nil && iter.Error != io.EOF {
  489. return
  490. }
  491. number := *(*string)(unsafe.Pointer(&str))
  492. if foundFloat {
  493. val, err := strconv.ParseFloat(number, 64)
  494. if err != nil {
  495. iter.Error = err
  496. return
  497. }
  498. return MakeAny(val)
  499. }
  500. if foundNegative {
  501. val, err := strconv.ParseInt(number, 10, 64)
  502. if err != nil {
  503. iter.Error = err
  504. return
  505. }
  506. return MakeAny(val)
  507. }
  508. val, err := strconv.ParseUint(number, 10, 64)
  509. if err != nil {
  510. iter.Error = err
  511. return
  512. }
  513. return MakeAny(val)
  514. }
  515. func (iter *Iterator) Read(obj interface{}) {
  516. type_ := reflect.TypeOf(obj)
  517. cacheKey := type_.Elem()
  518. cachedDecoder := getDecoderFromCache(cacheKey)
  519. if cachedDecoder == nil {
  520. decoder, err := decoderOfType(type_)
  521. if err != nil {
  522. iter.Error = err
  523. return
  524. }
  525. cachedDecoder = decoder
  526. addDecoderToCache(cacheKey, decoder)
  527. }
  528. e := (*emptyInterface)(unsafe.Pointer(&obj))
  529. cachedDecoder.decode(e.word, iter)
  530. }
  531. type prefix string
  532. func (p prefix) addTo(decoder Decoder, err error) (Decoder, error) {
  533. if err != nil {
  534. return nil, fmt.Errorf("%s: %s", p, err.Error())
  535. }
  536. return decoder, err
  537. }
  538. func decoderOfType(type_ reflect.Type) (Decoder, error) {
  539. switch type_.Kind() {
  540. case reflect.Ptr:
  541. return prefix("ptr").addTo(decoderOfPtr(type_.Elem()))
  542. default:
  543. return nil, errors.New("expect ptr")
  544. }
  545. }
  546. func decoderOfPtr(type_ reflect.Type) (Decoder, error) {
  547. typeName := type_.String()
  548. if typeName == "jsoniter.Any" {
  549. return &anyDecoder{}, nil
  550. }
  551. for _, extension := range extensions {
  552. alternativeFieldNames, func_ := extension(type_, nil)
  553. if alternativeFieldNames != nil {
  554. return nil, fmt.Errorf("%v should not return alternative field names when only type is being passed", extension)
  555. }
  556. typeDecoders[typeName] = &funcDecoder{func_}
  557. }
  558. typeDecoder := typeDecoders[typeName]
  559. if typeDecoder != nil {
  560. return typeDecoder, nil
  561. }
  562. switch type_.Kind() {
  563. case reflect.String:
  564. return &stringDecoder{}, nil
  565. case reflect.Int:
  566. return &intDecoder{}, nil
  567. case reflect.Int8:
  568. return &int8Decoder{}, nil
  569. case reflect.Int16:
  570. return &int16Decoder{}, nil
  571. case reflect.Int32:
  572. return &int32Decoder{}, nil
  573. case reflect.Int64:
  574. return &int64Decoder{}, nil
  575. case reflect.Uint:
  576. return &uintDecoder{}, nil
  577. case reflect.Uint8:
  578. return &uint8Decoder{}, nil
  579. case reflect.Uint16:
  580. return &uint16Decoder{}, nil
  581. case reflect.Uint32:
  582. return &uint32Decoder{}, nil
  583. case reflect.Uint64:
  584. return &uint64Decoder{}, nil
  585. case reflect.Float32:
  586. return &float32Decoder{}, nil
  587. case reflect.Float64:
  588. return &float64Decoder{}, nil
  589. case reflect.Bool:
  590. return &boolDecoder{}, nil
  591. case reflect.Interface:
  592. return &interfaceDecoder{}, nil
  593. case reflect.Struct:
  594. return decoderOfStruct(type_)
  595. case reflect.Slice:
  596. return prefix("[slice]").addTo(decoderOfSlice(type_))
  597. case reflect.Map:
  598. return prefix("[map]").addTo(decoderOfMap(type_))
  599. case reflect.Ptr:
  600. return prefix("[optional]").addTo(decoderOfOptional(type_.Elem()))
  601. default:
  602. return nil, fmt.Errorf("unsupported type: %v", type_)
  603. }
  604. }
  605. func decoderOfOptional(type_ reflect.Type) (Decoder, error) {
  606. decoder, err := decoderOfPtr(type_)
  607. if err != nil {
  608. return nil, err
  609. }
  610. return &optionalDecoder{type_, decoder}, nil
  611. }
  612. func decoderOfStruct(type_ reflect.Type) (Decoder, error) {
  613. fields := map[string]*structFieldDecoder{}
  614. for i := 0; i < type_.NumField(); i++ {
  615. field := type_.Field(i)
  616. fieldDecoderKey := fmt.Sprintf("%s/%s", type_.String(), field.Name)
  617. var fieldNames []string
  618. for _, extension := range extensions {
  619. alternativeFieldNames, func_ := extension(type_, &field)
  620. if alternativeFieldNames != nil {
  621. fieldNames = alternativeFieldNames
  622. }
  623. if func_ != nil {
  624. fieldDecoders[fieldDecoderKey] = &funcDecoder{func_}
  625. }
  626. }
  627. decoder := fieldDecoders[fieldDecoderKey]
  628. tagParts := strings.Split(field.Tag.Get("json"), ",")
  629. if fieldNames == nil {
  630. switch tagParts[0] {
  631. case "":
  632. fieldNames = []string{field.Name}
  633. case "-":
  634. fieldNames = []string{}
  635. }
  636. }
  637. if decoder == nil {
  638. var err error
  639. decoder, err = decoderOfPtr(field.Type)
  640. if err != nil {
  641. return prefix(fmt.Sprintf("{%s}", field.Name)).addTo(decoder, err)
  642. }
  643. }
  644. if len(tagParts) > 1 && tagParts[1] == "string" {
  645. decoder = &stringNumberDecoder{decoder}
  646. }
  647. for _, fieldName := range fieldNames {
  648. fields[fieldName] = &structFieldDecoder{&field, decoder}
  649. }
  650. }
  651. switch len(fields) {
  652. case 0:
  653. return &skipDecoder{type_}, nil
  654. case 1:
  655. for fieldName, fieldDecoder := range fields {
  656. return &oneFieldStructDecoder{type_, fieldName, fieldDecoder}, nil
  657. }
  658. case 2:
  659. var fieldName1 string
  660. var fieldName2 string
  661. var fieldDecoder1 *structFieldDecoder
  662. var fieldDecoder2 *structFieldDecoder
  663. for fieldName, fieldDecoder := range fields {
  664. if fieldName1 == "" {
  665. fieldName1 = fieldName
  666. fieldDecoder1 = fieldDecoder
  667. } else {
  668. fieldName2 = fieldName
  669. fieldDecoder2 = fieldDecoder
  670. }
  671. }
  672. return &twoFieldsStructDecoder{type_, fieldName1, fieldDecoder1, fieldName2, fieldDecoder2}, nil
  673. case 3:
  674. var fieldName1 string
  675. var fieldName2 string
  676. var fieldName3 string
  677. var fieldDecoder1 *structFieldDecoder
  678. var fieldDecoder2 *structFieldDecoder
  679. var fieldDecoder3 *structFieldDecoder
  680. for fieldName, fieldDecoder := range fields {
  681. if fieldName1 == "" {
  682. fieldName1 = fieldName
  683. fieldDecoder1 = fieldDecoder
  684. } else if fieldName2 == "" {
  685. fieldName2 = fieldName
  686. fieldDecoder2 = fieldDecoder
  687. } else {
  688. fieldName3 = fieldName
  689. fieldDecoder3 = fieldDecoder
  690. }
  691. }
  692. return &threeFieldsStructDecoder{type_,
  693. fieldName1, fieldDecoder1, fieldName2, fieldDecoder2, fieldName3, fieldDecoder3}, nil
  694. case 4:
  695. var fieldName1 string
  696. var fieldName2 string
  697. var fieldName3 string
  698. var fieldName4 string
  699. var fieldDecoder1 *structFieldDecoder
  700. var fieldDecoder2 *structFieldDecoder
  701. var fieldDecoder3 *structFieldDecoder
  702. var fieldDecoder4 *structFieldDecoder
  703. for fieldName, fieldDecoder := range fields {
  704. if fieldName1 == "" {
  705. fieldName1 = fieldName
  706. fieldDecoder1 = fieldDecoder
  707. } else if fieldName2 == "" {
  708. fieldName2 = fieldName
  709. fieldDecoder2 = fieldDecoder
  710. } else if fieldName3 == "" {
  711. fieldName3 = fieldName
  712. fieldDecoder3 = fieldDecoder
  713. } else {
  714. fieldName4 = fieldName
  715. fieldDecoder4 = fieldDecoder
  716. }
  717. }
  718. return &fourFieldsStructDecoder{type_,
  719. fieldName1, fieldDecoder1, fieldName2, fieldDecoder2, fieldName3, fieldDecoder3,
  720. fieldName4, fieldDecoder4}, nil
  721. }
  722. return &generalStructDecoder{type_, fields}, nil
  723. }
  724. func decoderOfSlice(type_ reflect.Type) (Decoder, error) {
  725. decoder, err := decoderOfPtr(type_.Elem())
  726. if err != nil {
  727. return nil, err
  728. }
  729. return &sliceDecoder{type_, type_.Elem(), decoder}, nil
  730. }
  731. func decoderOfMap(type_ reflect.Type) (Decoder, error) {
  732. decoder, err := decoderOfPtr(type_.Elem())
  733. if err != nil {
  734. return nil, err
  735. }
  736. mapInterface := reflect.New(type_).Interface()
  737. return &mapDecoder{type_, type_.Elem(), decoder, *((*emptyInterface)(unsafe.Pointer(&mapInterface)))}, nil
  738. }