api.go 1.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. package api
  2. import (
  3. "github.com/gin-gonic/gin"
  4. "github.com/gin-gonic/gin/binding"
  5. "github.com/go-playground/locales/zh"
  6. ut "github.com/go-playground/universal-translator"
  7. val "github.com/go-playground/validator/v10"
  8. zhTranslations "github.com/go-playground/validator/v10/translations/zh"
  9. "log"
  10. "net/http"
  11. "reflect"
  12. )
  13. func ErrHandler(c *gin.Context, err error) {
  14. log.Println(err)
  15. c.JSON(http.StatusInternalServerError, gin.H{
  16. "message": err.Error(),
  17. })
  18. }
  19. type ValidError struct {
  20. Key string
  21. Message string
  22. }
  23. func BindAndValid(c *gin.Context, target interface{}) bool {
  24. errs := make(map[string]string)
  25. err := c.ShouldBindJSON(target)
  26. if err != nil {
  27. log.Println("raw err", err)
  28. uni := ut.New(zh.New())
  29. trans, _ := uni.GetTranslator("zh")
  30. v, ok := binding.Validator.Engine().(*val.Validate)
  31. if ok {
  32. _ = zhTranslations.RegisterDefaultTranslations(v, trans)
  33. }
  34. verrs, ok := err.(val.ValidationErrors)
  35. if !ok {
  36. log.Println("verrs", verrs)
  37. c.JSON(http.StatusNotAcceptable, gin.H{
  38. "message": "请求参数错误",
  39. "code": http.StatusNotAcceptable,
  40. })
  41. return false
  42. }
  43. for _, value := range verrs {
  44. t := reflect.ValueOf(target)
  45. realType := t.Type().Elem()
  46. field, _ := realType.FieldByName(value.StructField())
  47. errs[field.Tag.Get("json")] = value.Translate(trans)
  48. }
  49. c.JSON(http.StatusNotAcceptable, gin.H{
  50. "errors": errs,
  51. "message": "请求参数错误",
  52. "code": http.StatusNotAcceptable,
  53. })
  54. return false
  55. }
  56. return true
  57. }