MaysWind
124 lines
3.5 KiB
Go
124 lines
3.5 KiB
Go
package utils
|
|
|
|
import (
|
|
"net/http"
|
|
"reflect"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/go-playground/validator/v10"
|
|
|
|
"github.com/mayswind/lab/pkg/core"
|
|
"github.com/mayswind/lab/pkg/errs"
|
|
)
|
|
|
|
// PrintJsonSuccessResult writes success response in json format to current http context
|
|
func PrintJsonSuccessResult(c *core.Context, result interface{}) {
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"success": true,
|
|
"result": result,
|
|
})
|
|
}
|
|
|
|
// PrintDataSuccessResult writes success response in custom content type to current http context
|
|
func PrintDataSuccessResult(c *core.Context, contentType string, fileName string, result []byte) {
|
|
if fileName != "" {
|
|
c.Header("Content-Disposition", "attachment;filename="+fileName)
|
|
}
|
|
|
|
c.Data(http.StatusOK, contentType, result)
|
|
}
|
|
|
|
// PrintJsonErrorResult writes error response in json format to current http context
|
|
func PrintJsonErrorResult(c *core.Context, err *errs.Error) {
|
|
c.SetResponseError(err)
|
|
|
|
errorMessage := err.Error()
|
|
|
|
if err.Code() == errs.ErrIncompleteOrIncorrectSubmission.Code() && len(err.BaseError) > 0 {
|
|
validationErrors, ok := err.BaseError[0].(validator.ValidationErrors)
|
|
|
|
if ok {
|
|
for _, err := range validationErrors {
|
|
errorMessage = getValidationErrorText(err)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
c.AbortWithStatusJSON(err.HttpStatusCode, gin.H{
|
|
"success": false,
|
|
"errorCode": err.Code(),
|
|
"errorMessage": errorMessage,
|
|
"path": c.Request.URL.Path,
|
|
})
|
|
}
|
|
|
|
// PrintDataErrorResult writes error response in custom content type to current http context
|
|
func PrintDataErrorResult(c *core.Context, contentType string, err *errs.Error) {
|
|
c.SetResponseError(err)
|
|
|
|
errorMessage := err.Error()
|
|
|
|
if err.Code() == errs.ErrIncompleteOrIncorrectSubmission.Code() && len(err.BaseError) > 0 {
|
|
validationErrors, ok := err.BaseError[0].(validator.ValidationErrors)
|
|
|
|
if ok {
|
|
for _, err := range validationErrors {
|
|
errorMessage = getValidationErrorText(err)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
c.Data(err.HttpStatusCode, contentType, []byte(errorMessage))
|
|
c.Abort()
|
|
}
|
|
|
|
func getValidationErrorText(err validator.FieldError) string {
|
|
fieldName := GetFirstLowerCharString(err.Field())
|
|
|
|
switch err.Tag() {
|
|
case "required":
|
|
return errs.GetParameterIsRequiredMessage(fieldName)
|
|
case "max":
|
|
if isIntegerParameter(err.Kind()) {
|
|
return errs.GetParameterMustLessThanMessage(fieldName, err.Param())
|
|
} else if isStringParameter(err.Kind()) {
|
|
return errs.GetParameterMustLessThanCharsMessage(fieldName, err.Param())
|
|
}
|
|
case "min":
|
|
if isIntegerParameter(err.Kind()) {
|
|
return errs.GetParameterMustMoreThanMessage(fieldName, err.Param())
|
|
} else if isStringParameter(err.Kind()) {
|
|
return errs.GetParameterMustMoreThanCharsMessage(fieldName, err.Param())
|
|
}
|
|
case "len":
|
|
return errs.GetParameterLengthNotEqualMessage(fieldName, err.Param())
|
|
case "notBlank":
|
|
return errs.GetParameterNotBeBlankMessage(fieldName)
|
|
case "validUsername":
|
|
return errs.GetParameterInvalidUsernameMessage(fieldName)
|
|
case "validEmail":
|
|
return errs.GetParameterInvalidEmailMessage(fieldName)
|
|
case "validCurrency":
|
|
return errs.GetParameterInvalidCurrencyMessage(fieldName)
|
|
case "validHexRGBColor":
|
|
return errs.GetParameterInvalidHexRGBColorMessage(fieldName)
|
|
}
|
|
|
|
return errs.GetParameterInvalidMessage(fieldName)
|
|
}
|
|
|
|
func isIntegerParameter(kind reflect.Kind) bool {
|
|
switch kind {
|
|
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}
|
|
|
|
func isStringParameter(kind reflect.Kind) bool {
|
|
return kind == reflect.String
|
|
}
|