unmarshal.go 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214
  1. package jsonutil
  2. import (
  3. "encoding/base64"
  4. "encoding/json"
  5. "fmt"
  6. "io"
  7. "io/ioutil"
  8. "reflect"
  9. "strings"
  10. "time"
  11. )
  12. // UnmarshalJSON reads a stream and unmarshals the results in object v.
  13. func UnmarshalJSON(v interface{}, stream io.Reader) error {
  14. var out interface{}
  15. b, err := ioutil.ReadAll(stream)
  16. if err != nil {
  17. return err
  18. }
  19. if len(b) == 0 {
  20. return nil
  21. }
  22. if err := json.Unmarshal(b, &out); err != nil {
  23. return err
  24. }
  25. return unmarshalAny(reflect.ValueOf(v), out, "")
  26. }
  27. func unmarshalAny(value reflect.Value, data interface{}, tag reflect.StructTag) error {
  28. vtype := value.Type()
  29. if vtype.Kind() == reflect.Ptr {
  30. vtype = vtype.Elem() // check kind of actual element type
  31. }
  32. t := tag.Get("type")
  33. if t == "" {
  34. switch vtype.Kind() {
  35. case reflect.Struct:
  36. // also it can't be a time object
  37. if _, ok := value.Interface().(*time.Time); !ok {
  38. t = "structure"
  39. }
  40. case reflect.Slice:
  41. // also it can't be a byte slice
  42. if _, ok := value.Interface().([]byte); !ok {
  43. t = "list"
  44. }
  45. case reflect.Map:
  46. t = "map"
  47. }
  48. }
  49. switch t {
  50. case "structure":
  51. if field, ok := vtype.FieldByName("SDKShapeTraits"); ok {
  52. tag = field.Tag
  53. }
  54. return unmarshalStruct(value, data, tag)
  55. case "list":
  56. return unmarshalList(value, data, tag)
  57. case "map":
  58. return unmarshalMap(value, data, tag)
  59. default:
  60. return unmarshalScalar(value, data, tag)
  61. }
  62. }
  63. func unmarshalStruct(value reflect.Value, data interface{}, tag reflect.StructTag) error {
  64. if data == nil {
  65. return nil
  66. }
  67. mapData, ok := data.(map[string]interface{})
  68. if !ok {
  69. return fmt.Errorf("JSON value is not a structure (%#v)", data)
  70. }
  71. t := value.Type()
  72. if value.Kind() == reflect.Ptr {
  73. if value.IsNil() { // create the structure if it's nil
  74. s := reflect.New(value.Type().Elem())
  75. value.Set(s)
  76. value = s
  77. }
  78. value = value.Elem()
  79. t = t.Elem()
  80. }
  81. // unwrap any payloads
  82. if payload := tag.Get("payload"); payload != "" {
  83. field, _ := t.FieldByName(payload)
  84. return unmarshalAny(value.FieldByName(payload), data, field.Tag)
  85. }
  86. for i := 0; i < t.NumField(); i++ {
  87. field := t.Field(i)
  88. if c := field.Name[0:1]; strings.ToLower(c) == c {
  89. continue // ignore unexported fields
  90. }
  91. // figure out what this field is called
  92. name := field.Name
  93. if locName := field.Tag.Get("locationName"); locName != "" {
  94. name = locName
  95. }
  96. member := value.FieldByName(field.Name)
  97. err := unmarshalAny(member, mapData[name], field.Tag)
  98. if err != nil {
  99. return err
  100. }
  101. }
  102. return nil
  103. }
  104. func unmarshalList(value reflect.Value, data interface{}, tag reflect.StructTag) error {
  105. if data == nil {
  106. return nil
  107. }
  108. listData, ok := data.([]interface{})
  109. if !ok {
  110. return fmt.Errorf("JSON value is not a list (%#v)", data)
  111. }
  112. if value.IsNil() {
  113. l := len(listData)
  114. value.Set(reflect.MakeSlice(value.Type(), l, l))
  115. }
  116. for i, c := range listData {
  117. err := unmarshalAny(value.Index(i), c, "")
  118. if err != nil {
  119. return err
  120. }
  121. }
  122. return nil
  123. }
  124. func unmarshalMap(value reflect.Value, data interface{}, tag reflect.StructTag) error {
  125. if data == nil {
  126. return nil
  127. }
  128. mapData, ok := data.(map[string]interface{})
  129. if !ok {
  130. return fmt.Errorf("JSON value is not a map (%#v)", data)
  131. }
  132. if value.IsNil() {
  133. value.Set(reflect.MakeMap(value.Type()))
  134. }
  135. for k, v := range mapData {
  136. kvalue := reflect.ValueOf(k)
  137. vvalue := reflect.New(value.Type().Elem()).Elem()
  138. unmarshalAny(vvalue, v, "")
  139. value.SetMapIndex(kvalue, vvalue)
  140. }
  141. return nil
  142. }
  143. func unmarshalScalar(value reflect.Value, data interface{}, tag reflect.StructTag) error {
  144. errf := func() error {
  145. return fmt.Errorf("unsupported value: %v (%s)", value.Interface(), value.Type())
  146. }
  147. switch d := data.(type) {
  148. case nil:
  149. return nil // nothing to do here
  150. case string:
  151. switch value.Interface().(type) {
  152. case *string:
  153. value.Set(reflect.ValueOf(&d))
  154. case []byte:
  155. b, err := base64.StdEncoding.DecodeString(d)
  156. if err != nil {
  157. return err
  158. }
  159. value.Set(reflect.ValueOf(b))
  160. default:
  161. return errf()
  162. }
  163. case float64:
  164. switch value.Interface().(type) {
  165. case *int64:
  166. di := int64(d)
  167. value.Set(reflect.ValueOf(&di))
  168. case *float64:
  169. value.Set(reflect.ValueOf(&d))
  170. case *time.Time:
  171. t := time.Unix(int64(d), 0).UTC()
  172. value.Set(reflect.ValueOf(&t))
  173. default:
  174. return errf()
  175. }
  176. case bool:
  177. switch value.Interface().(type) {
  178. case *bool:
  179. value.Set(reflect.ValueOf(&d))
  180. default:
  181. return errf()
  182. }
  183. default:
  184. return fmt.Errorf("unsupported JSON value (%v)", data)
  185. }
  186. return nil
  187. }