最佳实践之Golang错误处理

简凡丶 2021-11-05

719

1、原生错误处理

Go 语言通过内置的错误接口提供了非常简单的错误处理机制。error类型是一个接口类型，这是它的定义：type error interface { Error() string }

我们可以在编码中通过实现 error 接口类型来生成错误信息。函数通常在最后的返回值中返回错误信息。使用errors.New 可返回一个错误信息：

func Sqrt(f float64) (float64, error) {
    if f < 0 {
        return 0, errors.New("math: square root of negative number")
    }
    // 实现
}

在下面的例子中，我们在调用Sqrt的时候传递的一个负数，然后就得到了non-nil的error对象，将此对象与nil比较，结果为true，所以fmt.Println(fmt包在处理error时会调用Error方法)被调用，以输出错误，请看下面调用的示例代码：

result, err:= Sqrt(-1)
if err != nil {
   fmt.Println(err)
}

2、开源error包

github.com/pkg/errors包在原生error包基础上增加了以下常用的功能：

可以打印error的堆栈信息：打印错误需要%+v才能详细输出
使用Wrap或Wrapf，初始化一个error
使用errors.WithMessage可以在原来的error基础上再包装一层，包含原有error信息
errors.Is，用于判断error类型，可根据error类型不同做不同处理
errors.As，用于解析error

具体使用案例见全局错误处理一节。

3、工程中错误处理

3.1 需求整理

自定义error信息，并进行编码整理

controller层可以判断自定义error类型，最终判断是按info处理，还是按error处理

可以打印error初始发生的位置（获取error的调用栈）
确认当前系统定位：

用户，获取TagMessage
上游服务，需要错误码映射
日志监控、监控TagMessage

下面在一个工程化的项目中利用github.com/pkg/errors
包，完整实现一套的错误处理机制

3.2 方式一：Map保存错误码与Message的映射

3.2.1 定义错误信息

新建error_handler.go

package error_handle

import (
 "github.com/pkg/errors"
)

// 1、自定义error结构体，并重写Error()方法
// 错误时返回自定义结构
type CustomError struct {
 Code       int    `json:"code"`    // 业务码
 TagMessage string `json:"message"` // 描述信息
}

func (e *CustomError) Error() string {
 return e.TagMessage
}

// 2、定义errorCode
const (
 // 服务级错误码
 ServerError        = 10101
 TooManyRequests    = 10102
 ParamBindError     = 10103
 AuthorizationError = 10104
 CallHTTPError      = 10105
 ResubmitError      = 10106
 ResubmitMsg        = 10107
 HashIdsDecodeError = 10108
 SignatureError     = 10109

 // 业务模块级错误码
 // 用户模块
 IllegalUserName = 20101
 UserCreateError = 20102
 UserUpdateError = 20103
 UserSearchError = 20104

 // 授权调用方
 AuthorizedCreateError    = 20201
 AuthorizedListError      = 20202
 AuthorizedDeleteError    = 20203
 AuthorizedUpdateError    = 20204
 AuthorizedDetailError    = 20205
 AuthorizedCreateAPIError = 20206
 AuthorizedListAPIError   = 20207
 AuthorizedDeleteAPIError = 20208

 // 管理员
 AdminCreateError             = 20301
 AdminListError               = 20302
 AdminDeleteError             = 20303
 AdminUpdateError             = 20304
 AdminResetPasswordError      = 20305
 AdminLoginError              = 20306
 AdminLogOutError             = 20307
 AdminModifyPasswordError     = 20308
 AdminModifyPersonalInfoError = 20309

 // 配置
 ConfigEmailError        = 20401
 ConfigSaveError         = 20402
 ConfigRedisConnectError = 20403
 ConfigMySQLConnectError = 20404
 ConfigMySQLInstallError = 20405
 ConfigGoVersionError    = 20406

 // 实用工具箱
 SearchRedisError = 20501
 ClearRedisError  = 20502
 SearchRedisEmpty = 20503
 SearchMySQLError = 20504

 // 菜单栏
 MenuCreateError = 20601
 MenuUpdateError = 20602
 MenuListError   = 20603
 MenuDeleteError = 20604
 MenuDetailError = 20605

 // 借书
 BookNotFoundError        = 20701
 BookHasBeenBorrowedError = 20702
)

// 3、定义errorCode对应的文本信息
var codeTag = map[int]string{
 ServerError:        "Internal Server Error",
 TooManyRequests:    "Too Many Requests",
 ParamBindError:     "参数信息有误",
 AuthorizationError: "签名信息有误",
 CallHTTPError:      "调用第三方 HTTP 接口失败",
 ResubmitError:      "Resubmit Error",
 ResubmitMsg:        "请勿重复提交",
 HashIdsDecodeError: "ID参数有误",
 SignatureError:     "SignatureError",

 IllegalUserName: "非法用户名",
 UserCreateError: "创建用户失败",
 UserUpdateError: "更新用户失败",
 UserSearchError: "查询用户失败",

 AuthorizedCreateError:    "创建调用方失败",
 AuthorizedListError:      "获取调用方列表页失败",
 AuthorizedDeleteError:    "删除调用方失败",
 AuthorizedUpdateError:    "更新调用方失败",
 AuthorizedDetailError:    "获取调用方详情失败",
 AuthorizedCreateAPIError: "创建调用方API地址失败",
 AuthorizedListAPIError:   "获取调用方API地址列表失败",
 AuthorizedDeleteAPIError: "删除调用方API地址失败",

 AdminCreateError:             "创建管理员失败",
 AdminListError:               "获取管理员列表页失败",
 AdminDeleteError:             "删除管理员失败",
 AdminUpdateError:             "更新管理员失败",
 AdminResetPasswordError:      "重置密码失败",
 AdminLoginError:              "登录失败",
 AdminLogOutError:             "退出失败",
 AdminModifyPasswordError:     "修改密码失败",
 AdminModifyPersonalInfoError: "修改个人信息失败",

 ConfigEmailError:        "修改邮箱配置失败",
 ConfigSaveError:         "写入配置文件失败",
 ConfigRedisConnectError: "Redis连接失败",
 ConfigMySQLConnectError: "MySQL连接失败",
 ConfigMySQLInstallError: "MySQL初始化数据失败",
 ConfigGoVersionError:    "GoVersion不满足要求",

 SearchRedisError: "查询RedisKey失败",
 ClearRedisError:  "清空RedisKey失败",
 SearchRedisEmpty: "查询的RedisKey不存在",
 SearchMySQLError: "查询mysql失败",

 MenuCreateError: "创建菜单失败",
 MenuUpdateError: "更新菜单失败",
 MenuDeleteError: "删除菜单失败",
 MenuListError:   "获取菜单列表页失败",
 MenuDetailError: "获取菜单详情失败",

 BookNotFoundError:        "书未找到",
 BookHasBeenBorrowedError: "书已经被借走了",
}

func Text(code int) string {
 return codeTag[code]
}

// 4、新建自定义error实例化
func NewCustomError(code int) error {
 // 初次调用得用Wrap方法，进行实例化
 return errors.Wrap(&CustomError{
  Code:       code,
  TagMessage: codeTag[code],
 }, "")
}

3.3 自定义Error使用

新建测试文件：error_handler_test.go

package error_handle

import (
 "fmt"
 "github.com/pkg/errors"
 "testing"
)

func TestText(t *testing.T) {
 books := []string{
  "Book1",
  "Book222222",
  "Book3333333333",
 }

 for _, bookName := range books {
  err := searchBook(bookName)

  // 特殊业务场景：如果发现书被借走了，下次再来就行了，不需要作为错误处理
  if err != nil {
   // 提取error这个interface底层的错误码，一般在API的返回前才提取
   // As - 获取错误的具体实现
   var myError = new(CustomError)
   // As - 解析错误内容
   if errors.As(err, &myError) {
    fmt.Printf("AS中的信息：当前书为: %s ,error code is %d, message is %s\n", bookName, myError.Code, myError.TagMessage)
   }

   // 特殊场景，指定错误(ErrorBookHasBeenBorrowed)时，打印即可，不返回错误
   // Is - 判断错误是否为指定类型
   if errors.Is(err,  NewCustomError(BookHasBeenBorrowedError)) {
    fmt.Printf("IS中的信息：%s 已经被借走了, 只需按Info处理!\n", bookName)
    err = nil
   }else {
    // 如果已有堆栈信息，应调用WithMessage方法
    newErr := errors.WithMessage(err, "WithMessage err")
    fmt.Printf("IS中的信息：%s 未找到，应该按Error处理! ,newErr is %s\n", bookName , newErr)
   }
  }
 }
}

func searchBook(bookName string) error {
 // 1 发现图书馆不存在这本书 - 认为是错误，需要打印详细的错误信息
 if len(bookName) > 10 {
  return NewCustomError(BookHasBeenBorrowedError)
 } else if len(bookName) > 6 {
  // 2 发现书被借走了 - 打印一下被接走的提示即可，不认为是错误
  return NewCustomError(BookHasBeenBorrowedError)
 }
 // 3 找到书 - 不需要任何处理
 return nil
}

3.3 方式二：借助generate简化代码（建议使用）

方式一维护错误码与错误信息的关系较为复杂，我们可以借助go generate来自动生成代码。

3.3.1 安装stringer

stringer不是Go自带工具，需要手动安装。执行如下命令即可

go get golang.org/x/tools/cmd/stringer

3.3.1 定义错误信息

新建error_handler.go。在error_handler中，增加注释//go:generate stringer -type ErrCode -linecomment。执行go generate，会生成新的文件

package error_handle

import (
 "github.com/pkg/errors"
)

// 1、自定义error结构体，并重写Error()方法
// 错误时返回自定义结构
type CustomError struct {
 Code    ErrCode `json:"code"`    // 业务码
 Message string  `json:"message"` // 业务码
}

func (e *CustomError) Error() string {
 return e.Code.String()
}

type ErrCode int64 //错误码

// 2、定义errorCode
//go:generate stringer -type ErrCode -linecomment
const (
 // 服务级错误码
 ServerError        ErrCode = 10101 // Internal Server Error
 TooManyRequests    ErrCode = 10102 // Too Many Requests
 ParamBindError     ErrCode = 10103 // 参数信息有误
 AuthorizationError ErrCode = 10104 // 签名信息有误
 CallHTTPError      ErrCode = 10105 // 调用第三方HTTP接口失败
 ResubmitError      ErrCode = 10106 // ResubmitError
 ResubmitMsg        ErrCode = 10107 // 请勿重复提交
 HashIdsDecodeError ErrCode = 10108 // ID参数有误
 SignatureError     ErrCode = 10109 // SignatureError

 // 业务模块级错误码
 // 用户模块
 IllegalUserName ErrCode = 20101 // 非法用户名
 UserCreateError ErrCode = 20102 // 创建用户失败
 UserUpdateError ErrCode = 20103 // 更新用户失败
 UserSearchError ErrCode = 20104 // 查询用户失败

 // 配置
 ConfigEmailError        ErrCode = 20401 // 修改邮箱配置失败
 ConfigSaveError         ErrCode = 20402 // 写入配置文件失败
 ConfigRedisConnectError ErrCode = 20403 // Redis连接失败
 ConfigMySQLConnectError ErrCode = 20404 // MySQL连接失败
 ConfigMySQLInstallError ErrCode = 20405 // MySQL初始化数据失败
 ConfigGoVersionError    ErrCode = 20406 // GoVersion不满足要求

 // 实用工具箱
 SearchRedisError ErrCode = 20501 // 查询RedisKey失败
 ClearRedisError  ErrCode = 20502 // 清空RedisKey失败
 SearchRedisEmpty ErrCode = 20503 // 查询的RedisKey不存在
 SearchMySQLError ErrCode = 20504 // 查询mysql失败

 // 菜单栏
 MenuCreateError ErrCode = 20601 // 创建菜单失败
 MenuUpdateError ErrCode = 20602 // 更新菜单失败
 MenuListError   ErrCode = 20603 // 删除菜单失败
 MenuDeleteError ErrCode = 20604 // 获取菜单列表页失败
 MenuDetailError ErrCode = 20605 // 获取菜单详情失败

 // 借书
 BookNotFoundError        ErrCode = 20701 // 书未找到
 BookHasBeenBorrowedError ErrCode = 20702 // 书已经被借走了
)

// 4、新建自定义error实例化
func NewCustomError(code ErrCode) error {
 // 初次调用得用Wrap方法，进行实例化
 return errors.Wrap(&CustomError{
  Code:    code,
  Message: code.String(),
 }, "")
}

3.3.2 自定义Error使用