错误处理升级：让你的 Go 代码告别混乱和痛苦

后端

2023-06-03 03:52:46

错误处理：二次封装 error，让 Go 代码更清晰易维护

错误类型：一个令人沮丧的存在

Go 中的 error 类型可谓是简单粗暴，它就是一个接口，没有任何具体实现。这就导致了以下后果：

错误信息不够详细： 当你遇到一个错误时，你只能看到一个简单的字符串信息，根本无法知道错误的具体原因和位置。
错误传递不够方便： 当你需要将一个错误从一个函数传递到另一个函数时，你只能使用 return，这样很容易导致代码混乱和难以维护。
错误处理不够灵活： 当你需要对不同的错误类型进行不同的处理时，你只能使用一系列的 if-else 语句，这使得代码难以阅读和维护。

二次封装 error：拯救代码的利器

为了解决以上问题，我们可以对 error 类型进行二次封装，创建一个新的 error 类型，并在这个类型中加入更多信息，如错误码、错误原因、错误位置等。这样，我们就可以更轻松地处理和管理错误。

如何二次封装 error

创建一个新的 error 类型非常简单，只需要实现 error 接口即可。error 接口只有一个方法：Error() string，它返回错误的字符串表示。

type MyError struct {
  code    int
  message string
}

func (e MyError) Error() string {
  return fmt.Sprintf("code: %d, message: %s", e.code, e.message)
}

现在，我们就可以使用这个新的 error 类型来捕获和处理错误了。

func foo() error {
  return MyError{code: 100, message: "something went wrong"}
}

func main() {
  if err := foo(); err != nil {
    fmt.Println(err)
  }
}

输出结果：

code: 100, message: something went wrong

进阶技巧：错误码和错误原因分离

在实际开发中，我们经常会遇到这样的情况：一个错误可能有多种原因，或者一个错误可能有多种表现形式。为了更灵活地处理这些情况，我们可以将错误码和错误原因分离。

type ErrorCode int

const (
  ErrCodeUnknown ErrorCode = iota
  ErrCodeNotFound
  ErrCodeInvalidInput
)

type ErrorMessage string

const (
  ErrMsgUnknown ErrorMessage = "unknown error"
  ErrMsgNotFound ErrorMessage = "resource not found"
  ErrMsgInvalidInput ErrorMessage = "invalid input"
)

type MyError struct {
  code    ErrorCode
  message ErrorMessage
}

func (e MyError) Error() string {
  return fmt.Sprintf("code: %d, message: %s", e.code, e.message)
}

现在，我们可以根据不同的错误码和错误原因来对错误进行不同的处理。

func foo() error {
  return MyError{code: ErrCodeNotFound, message: ErrMsgResourceNotFound}
}

func main() {
  if err := foo(); err != nil {
    switch err.code {
    case ErrCodeNotFound:
      fmt.Println("resource not found")
    case ErrCodeInvalidInput:
      fmt.Println("invalid input")
    default:
      fmt.Println("unknown error")
    }
  }
}