binding.go 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. package binding
  2. import (
  3. "encoding/json"
  4. "encoding/xml"
  5. "errors"
  6. "net/http"
  7. "reflect"
  8. "strings"
  9. )
  10. type (
  11. Binding interface {
  12. Bind(*http.Request, interface{}) error
  13. }
  14. // JSON binding
  15. jsonBinding struct{}
  16. // XML binding
  17. xmlBinding struct{}
  18. // // form binding
  19. formBinding struct{}
  20. )
  21. var (
  22. JSON = jsonBinding{}
  23. XML = xmlBinding{}
  24. Form = formBinding{} // todo
  25. )
  26. func (_ jsonBinding) Bind(req *http.Request, obj interface{}) error {
  27. decoder := json.NewDecoder(req.Body)
  28. if err := decoder.Decode(obj); err == nil {
  29. return Validate(obj)
  30. } else {
  31. return err
  32. }
  33. }
  34. func (_ xmlBinding) Bind(req *http.Request, obj interface{}) error {
  35. decoder := xml.NewDecoder(req.Body)
  36. if err := decoder.Decode(obj); err == nil {
  37. return Validate(obj)
  38. } else {
  39. return err
  40. }
  41. }
  42. func (_ formBinding) Bind(req *http.Request, obj interface{}) error {
  43. return nil
  44. }
  45. func Validate(obj interface{}) error {
  46. typ := reflect.TypeOf(obj)
  47. val := reflect.ValueOf(obj)
  48. if typ.Kind() == reflect.Ptr {
  49. typ = typ.Elem()
  50. val = val.Elem()
  51. }
  52. for i := 0; i < typ.NumField(); i++ {
  53. field := typ.Field(i)
  54. fieldValue := val.Field(i).Interface()
  55. zero := reflect.Zero(field.Type).Interface()
  56. // Validate nested and embedded structs (if pointer, only do so if not nil)
  57. if field.Type.Kind() == reflect.Struct ||
  58. (field.Type.Kind() == reflect.Ptr && !reflect.DeepEqual(zero, fieldValue)) {
  59. if err := Validate(fieldValue); err != nil {
  60. return err
  61. }
  62. }
  63. if strings.Index(field.Tag.Get("binding"), "required") > -1 {
  64. if reflect.DeepEqual(zero, fieldValue) {
  65. name := field.Name
  66. if j := field.Tag.Get("json"); j != "" {
  67. name = j
  68. } else if f := field.Tag.Get("form"); f != "" {
  69. name = f
  70. }
  71. return errors.New("Required " + name)
  72. }
  73. }
  74. }
  75. return nil
  76. }