form_mapping.go 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. // Copyright 2014 Manu Martinez-Almeida. All rights reserved.
  2. // Use of this source code is governed by a MIT style
  3. // license that can be found in the LICENSE file.
  4. package binding
  5. import (
  6. "errors"
  7. "reflect"
  8. "strconv"
  9. )
  10. func mapForm(ptr interface{}, form map[string][]string) error {
  11. typ := reflect.TypeOf(ptr).Elem()
  12. val := reflect.ValueOf(ptr).Elem()
  13. for i := 0; i < typ.NumField(); i++ {
  14. typeField := typ.Field(i)
  15. structField := val.Field(i)
  16. if !structField.CanSet() {
  17. continue
  18. }
  19. inputFieldName := typeField.Tag.Get("form")
  20. if inputFieldName == "" {
  21. inputFieldName = typeField.Name
  22. }
  23. inputValue, exists := form[inputFieldName]
  24. if !exists {
  25. continue
  26. }
  27. numElems := len(inputValue)
  28. if structField.Kind() == reflect.Slice && numElems > 0 {
  29. sliceOf := structField.Type().Elem().Kind()
  30. slice := reflect.MakeSlice(structField.Type(), numElems, numElems)
  31. for i := 0; i < numElems; i++ {
  32. if err := setWithProperType(sliceOf, inputValue[i], slice.Index(i)); err != nil {
  33. return err
  34. }
  35. }
  36. val.Field(i).Set(slice)
  37. } else {
  38. if err := setWithProperType(typeField.Type.Kind(), inputValue[0], structField); err != nil {
  39. return err
  40. }
  41. }
  42. }
  43. return nil
  44. }
  45. func setWithProperType(valueKind reflect.Kind, val string, structField reflect.Value) error {
  46. switch valueKind {
  47. case reflect.Int:
  48. return setIntField(val, 0, structField)
  49. case reflect.Int8:
  50. return setIntField(val, 8, structField)
  51. case reflect.Int16:
  52. return setIntField(val, 16, structField)
  53. case reflect.Int32:
  54. return setIntField(val, 32, structField)
  55. case reflect.Int64:
  56. return setIntField(val, 64, structField)
  57. case reflect.Uint:
  58. return setUintField(val, 0, structField)
  59. case reflect.Uint8:
  60. return setUintField(val, 8, structField)
  61. case reflect.Uint16:
  62. return setUintField(val, 16, structField)
  63. case reflect.Uint32:
  64. return setUintField(val, 32, structField)
  65. case reflect.Uint64:
  66. return setUintField(val, 64, structField)
  67. case reflect.Bool:
  68. return setBoolField(val, structField)
  69. case reflect.Float32:
  70. return setFloatField(val, 32, structField)
  71. case reflect.Float64:
  72. return setFloatField(val, 64, structField)
  73. case reflect.String:
  74. structField.SetString(val)
  75. default:
  76. return errors.New("Unknown type")
  77. }
  78. return nil
  79. }
  80. func setIntField(val string, bitSize int, field reflect.Value) error {
  81. if val == "" {
  82. val = "0"
  83. }
  84. intVal, err := strconv.ParseInt(val, 10, bitSize)
  85. if err == nil {
  86. field.SetInt(intVal)
  87. }
  88. return err
  89. }
  90. func setUintField(val string, bitSize int, field reflect.Value) error {
  91. if val == "" {
  92. val = "0"
  93. }
  94. uintVal, err := strconv.ParseUint(val, 10, bitSize)
  95. if err == nil {
  96. field.SetUint(uintVal)
  97. }
  98. return err
  99. }
  100. func setBoolField(val string, field reflect.Value) error {
  101. if val == "" {
  102. val = "false"
  103. }
  104. boolVal, err := strconv.ParseBool(val)
  105. if err == nil {
  106. field.SetBool(boolVal)
  107. }
  108. return nil
  109. }
  110. func setFloatField(val string, bitSize int, field reflect.Value) error {
  111. if val == "" {
  112. val = "0.0"
  113. }
  114. floatVal, err := strconv.ParseFloat(val, bitSize)
  115. if err == nil {
  116. field.SetFloat(floatVal)
  117. }
  118. return err
  119. }
  120. // Don't pass in pointers to bind to. Can lead to bugs. See:
  121. // https://github.com/codegangsta/martini-contrib/issues/40
  122. // https://github.com/codegangsta/martini-contrib/pull/34#issuecomment-29683659
  123. func ensureNotPointer(obj interface{}) {
  124. if reflect.TypeOf(obj).Kind() == reflect.Ptr {
  125. panic("Pointers are not accepted as binding models")
  126. }
  127. }