api.go 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  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. var trans ut.Translator
  24. func init() {
  25. uni := ut.New(zh.New())
  26. trans, _ = uni.GetTranslator("zh")
  27. v, ok := binding.Validator.Engine().(*val.Validate)
  28. if ok {
  29. _ = zhTranslations.RegisterDefaultTranslations(v, trans)
  30. }
  31. }
  32. func BindAndValid(c *gin.Context, target interface{}) bool {
  33. errs := make(map[string]string)
  34. err := c.ShouldBindJSON(target)
  35. if err != nil {
  36. log.Println("raw err", err)
  37. verrs, ok := err.(val.ValidationErrors)
  38. if !ok {
  39. log.Println("verrs", verrs)
  40. c.JSON(http.StatusNotAcceptable, gin.H{
  41. "message": "请求参数错误",
  42. "code": http.StatusNotAcceptable,
  43. })
  44. return false
  45. }
  46. for _, value := range verrs {
  47. t := reflect.ValueOf(target)
  48. realType := t.Type().Elem()
  49. field, _ := realType.FieldByName(value.StructField())
  50. errs[field.Tag.Get("json")] = value.Translate(trans)
  51. }
  52. c.JSON(http.StatusNotAcceptable, gin.H{
  53. "errors": errs,
  54. "message": "请求参数错误",
  55. "code": http.StatusNotAcceptable,
  56. })
  57. return false
  58. }
  59. return true
  60. }