Golang error处理实践

mazhen

发布于 2023-11-24 14:02:32

2280

发布于 2023-11-24 14:02:32

文章被收录于专栏：mazhen.techmazhen.tech

Go的错误处理机制很简洁，使用errors.New(text)创建 error，方法的调用者一般按照如下模式处理：

if err != nil {
        return err
}

这样做最大的问题是error中没有保存方法调用栈等上下文信息，只能靠创建时传递的string参数来区分error，很难定位错误发生的具体位置。例如：

import (
	"fmt"
	"errors"
)

func f1() error {
	return f2()
}

func f2() error {
	return f3()
}

func f3() error {
	return errors.New("std error")
}

func main() {
	if err := f1(); err != nil {
		fmt.Printf("%+v", err)
	}
}

执行的输出为：

std error

在实际的程序中调用关系复杂，仅凭错误信息很难定位错误源头。TiDB 使用了juju/errors来记录调用栈：

import (
	"github.com/juju/errors"
	"fmt"
)

func jf1() error {
	err := jf2()
	if err != nil {
		return errors.Trace(err)
	}
	return nil
}

func jf2() error {
	err := jf3()
	if err != nil {
		return errors.Trace(err)
	}
	return nil
}

func jf3() error {
	return errors.New("juju error")
}

func main() {
	if err := jf1(); err != nil {
		fmt.Printf("%+v", err)
	}
}

这段代码的输出为：

github.com/mz1999/error/main.go:25: juju error
github.com/mz1999/error/main.go:19: 
github.com/mz1999/error/main.go:11:

可以看到，如果想记录调用栈，每次都需要调用errors.Trace。这样做比较繁琐，而且每次trace时内部都会调用runtime.Caller，性能不佳。TiDB已经调研了新的第三方包pkg/errors准备替换掉juju/errors。

使用pkg/errors会简单很多，和标准库的errors一致，但可以记录调用栈信息：

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

func pf1() error {
	return pf2()
}

func pf2() error {
	return pf3()
}

func pf3() error {
	return errors.New("pkg error")
}

func main() {
	if err := pf1(); err != nil {
		fmt.Printf("%+v", err)
	}
}

这段代码的输出为：

pkg error
main.pf3
	/Users/mazhen/Documents/works/goworkspace/src/github.com/mz1999/error/main.go:17
main.pf2
	/Users/mazhen/Documents/works/goworkspace/src/github.com/mz1999/error/main.go:13
main.pf1
	/Users/mazhen/Documents/works/goworkspace/src/github.com/mz1999/error/main.go:9
main.main
	/Users/mazhen/Documents/works/goworkspace/src/github.com/mz1999/error/main.go:21
runtime.main
	/usr/local/go/src/runtime/proc.go:198
runtime.goexit
	/usr/local/go/src/runtime/asm_amd64.s:2361

这样看，使用pkg/errors替代标准库的errors就可以满足我们的需求。

另外，pkg/errors的作者还给了一些最佳实践的建议：

在你自己的代码中，在错误的发生点使用errors.New 或 errors.Errorf ：

func parseArgs(args []string) error {
        if len(args) < 3 {
                return errors.Errorf("not enough arguments, expected at least 3, got %d", len(args))
        }
        // ...
}

如果你接收到一个error，一般简单的直接返回：

if err != nil {
       return err
}

如果你是调用第三方的包或标准库时接收到error，使用 errors.Wrap or errors.Wrapf 包装这个error，它会记录在这个点的调用栈：

f, err := os.Open(path)
if err != nil {
        return errors.Wrapf(err, "failed to open %q", path)
}

Always return errors to their caller rather than logging them throughout your program.
在程序的top level，或者是worker goroutine，使用 %+v 输出error的详细信息。

func main() {
        err := app.Run()
        if err != nil {
                fmt.Printf("FATAL: %+v\n", err)
                os.Exit(1)
        }
}

如果需要抛出包含MySQL错误码的内部错误，可以使用errors.Wrap包装，附带上报错位置的调用栈信息：

func pf3() error {
	return errors.Wrap(mysql.NewErr(mysql.ErrCantCreateTable, "tablename", 500), "")
}

这样，我们既拿到了完整调用栈，又可以使用errors.Cause获取MySQL的错误码等信息：

if err := pf1(); err != nil {
    fmt.Printf("%+v", err)

    var sqlError *mysql.SQLError
    if m, ok := errors.Cause(err).(*mysql.SQLError); ok {
    	sqlError = m
    } else {
    	sqlError = mysql.NewErrf(mysql.ErrUnknown, "%s", err.Error())
    }
    
    fmt.Printf("\nMySQL error code: %d, state: %s", sqlError.Code, sqlError.State)
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2018-04-06，如有侵权请联系 cloudcommunity@tencent.com 删除

error