当业务需要更复杂的错误码定义时，我们可以自定义实现业务自己的错误码，只需要实现gcode.Code相关的接口即可。

我们来看个例子。

自定义错误码

定义结构体并实现gcode.code接口定义的方法

  1. type BizCode struct {
  2.     code    int
  3.     message string
  4.     detail  BizCodeDetail
  5. }
  6. type BizCodeDetail struct {
  7.     Code     string
  8.     HttpCode int
  9. }
  11. func (c BizCode) BizDetail() BizCodeDetail {
  12.     return c.detail
  13. }
  15. func (c BizCode) Code() int {
  16.     return c.code
  17. }
  19. func (c BizCode) Message() string {
  20.     return c.message
  21. }
  23. func (c BizCode) Detail() interface{} {
  24.     return c.detail
  25. }
  27. func New(httpCode int, code string, message string) gcode.Code {
  28.     return BizCode{
  29.         code:    0,
  30.         message: message,
  31.         detail: BizCodeDetail{
  32.             Code:     code,
  33.             HttpCode: httpCode,
  34.         },
  35.     }
  36. }

定义业务错误码

  1. var (
  2.     CodeNil      = New(200, "OK", "")
  3.     CodeNotFound = New(404, "Not Found", "Resource does not exist")
  4.     CodeInternal = New(500, "Internal Error", "An error occurred internally")
  5.     // ...
  6. )

使用到中间件

  1. func ResponseHandler(r *ghttp.Request) {
  2.     r.Middleware.Next()
  3.     // There's custom buffer content, it then exits current handler.
  4.     if r.Response.BufferLength() > 0 {
  5.         return
  6.     }
  7.     var (
  8.         err  = r.GetError()
  9.         res  = r.GetHandlerResponse()
  10.         code = gerror.Code(err)
  11.     )
  12.     if code == gcode.CodeNil && err != nil {
  13.         code = CodeInternal
  14.     } else {
  15.         code = CodeNil
  16.     }
  17.     if bizCode, ok := code.(BizCode); ok {
  18.         r.Response.WriteStatus(bizCode.BizDetail().HttpCode)
  19.     }
  20.     r.Response.WriteJson(g.Map{
  21.         `code`:    gcode.CodeOK.Code(),
  22.         `message`: gcode.CodeOK.Message(),
  23.         `data`:    res,
  24.     })
  25. }