uploader/utils/result/result.go

package result

import (
	syserrors "errors"
	"github.com/gin-gonic/gin"
	"uploader/errors"
)

type Result struct {
	Code    string `json:"code"`
	Message string `json:"message"`
	Data    Data   `json:"data"`
}

type Data map[string]interface{}

func Json(api func(c *gin.Context) (Data, error)) func(c *gin.Context) {
	return func(c *gin.Context) {
		data, err := api(c)
		result := Result{}
		if err != nil {
			result = buildErrorResult(err)
			result.Data = data
		} else {
			result.Code = Success
			result.Message = "成功"
			result.Data = data
		}
		c.JSON(200, result)
	}
}

func Normal(api func(c *gin.Context)) func(c *gin.Context) {
	return api
}

func buildErrorResult(err error) Result {
	result := Result{}
	switch err.(type) {
	case *errors.BizError:
		var bizError *errors.BizError
		syserrors.As(err, &bizError)
		result.Code = bizError.Code
		result.Message = bizError.Message
	default:
		result.Code = Error
		result.Message = "系统异常"
	}
	return result
}