go-zero 统一返回
1、整体目录结构
1、全局处理主入口
package manage
import (
"net/http"
"github.com/zeromicro/go-zero/rest/httpx"
)
type Body struct {
Code int `json:"code"`
Message string `json:"message"`
Result interface{} `json:"data"`
}
// 统一返回入口，
func ResponseHandler(w http.ResponseWriter, resp interface{}, err error) {
if err != nil {
// 如果err不为空的话，走错误处理函数，将err传递过去
ErrHandler(err)
} else {
// 没有错误信息，返回相应内容
httpx.OkJson(w, Body{
Code: OK.Code,
Message: OK.Message,
Result: resp,
})
}
}
2、统一错误处理主入口
package manage
import (
"net/http"
)
type CodeError struct {
Code int `json:"code"`
Message string `json:"message"`
}
// 实现error的接口 然后CodeError继承一下Error方法 CodeError就为error类型的返回值
func (e *CodeError) Error() string {
return e.Message
}
// 返回给前端的数据
func (e *CodeError) ErrorResponse() CodeError {
return CodeError{
Code: e.Code,
Message: e.Message,
}
}
// 用来自定义抛出的异常，支持调用传递
func NewCodeError(code int, msg string) error {
return &CodeError{
Code: code,
Message: msg,
}
}
// 提供new方法，任意地方传递参数返回CodeError类型的数据
func New(code int, msg string) CodeError {
return CodeError{
Code: code,
Message: msg,
}
}
// 自定义错误返回函数 错误函数主入口
func ErrHandler(err error) (int, interface{}) {
switch errorType := err.(type) {
// 如果错误类型为CodeError，就返回错误类型的结构体
case *CodeError:
return http.StatusOK, errorType.ErrorResponse()
default:
// 系统错误，500 错误提示
return http.StatusOK, CodeError{Code: 500, Message: err.Error()}
}
}
3、全局异常状态
package manage
var (
OK = New(0, "操作成功")
ReidsError = New(6379, "redis连接失败")
JwtError = New(500, "JWT解密失败")
TokenError = New(401, "Token已过期")
ServerError = New(500, "服务器内部错误")
)
4、非业务错误的再次封装，使用http.ResponseWriter进行返回给前端，常用在中间件中
package manage
import (
"encoding/json"
"net/http"
)
// codeError 接受统一返回格式，{code:int,message:string}
func SetWrite(w http.ResponseWriter, r *http.Request, codeError CodeError) {
// 接口请求成功 200状态码
w.WriteHeader(http.StatusOK)
// 转换成json
jsonData, _ := json.Marshal(codeError)
// 设置相应格式
w.Header().Set("Content-Type", "application/json")
// 返回给前端结构体
w.Write(jsonData)
}
5、修改路由Handler函数
func CreateMenuHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var req systemMenuType.MenuCreateType
if err := httpx.Parse(r, &req); err != nil {
manage.ResponseHandler(w, nil, err)
return
}
l := systemMenu.NewCreateMenuLogic(r.Context(), svcCtx)
resp, err := l.CreateMenu(&req)
if err != nil {
manage.ResponseHandler(w, nil, err)
} else {
manage.ResponseHandler(w, resp, nil)
}
}
}
httpx.OkJsonCtx和httpx.ErrorCtx() 替换为 manage.ResponseHandler
搞定收工
原文地址：https://blog.csdn.net/qq_37061571/article/details/135853216
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！