form_mapping.go 3.5 KB

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