Golang error处理实践

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

1
2
3
if err != nil {
        return err
}

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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
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)
	}
}

执行的输出为:

1
std error

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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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)
	}
}

这段代码的输出为:

1
2
3
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一致,但可以记录调用栈信息:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
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)
	}
}

这段代码的输出为:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
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.Newerrors.Errorf
1
2
3
4
5
6
func parseArgs(args []string) error {
        if len(args) < 3 {
                return errors.Errorf("not enough arguments, expected at least 3, got %d", len(args))
        }
        // ...
}
  • 如果你接收到一个error,一般简单的直接返回:
1
2
3
if err != nil {
       return err
}
  • 如果你是调用第三方的包或标准库时接收到error,使用 errors.Wrap or errors.Wrapf 包装这个error,它会记录在这个点的调用栈:
1
2
3
4
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的详细信息。

1
2
3
4
5
6
7
func main() {
        err := app.Run()
        if err != nil {
                fmt.Printf("FATAL: %+v\n", err)
                os.Exit(1)
        }
}
  • 如果需要抛出包含MySQL错误码的内部错误,可以使用errors.Wrap包装,附带上报错位置的调用栈信息:
1
2
3
func pf3() error {
	return errors.Wrap(mysql.NewErr(mysql.ErrCantCreateTable, "tablename", 500), "")
}

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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
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)
}
Licensed under CC BY-NC-SA 4.0
comments powered by Disqus