一、错误处理机制
Go 语言以结果多值返回方式处理错误,函数或者方法最后一个返回值作为错误类型。
func ReadFile(filename string) ([]byte, error)
调用时根据错误值判断是否正常
data, err := ReadFile("abc.txt")
if err != nil {
// 处理错误
} else {
// 处理data
}
1. error 类型及其意义
Go 语言通过内置的 error 类型实现错误处理,它是一个接口
type error interface {
Error() string
}
实现 error 的类型都可以作为错误类型返回,框架会使用 Error()方法打印错误信息。
任意类型只要实现该接口就表示一个错误
type ParseError struct {
Filename string // 文件名
Line int // 错误行号
}
func (e *ParseError) Error() string {
return fmt.Sprintf("%s:%d", e.Filename, e.Line)
}
2. if err != nil 检查错误
if 语句常用于判断函数调用是否发生错误
file, err := os.Open("/tmp/abc.txt")
if err != nil {
log.Fatal(err)
}
3. 自定义错误类型
可自定义错误类型保存更多信息,自定义错误能更好表达错误场景,修正错误也更方便。
type NotFoundError struct {
FileName string
}
func (e *NotFoundError) Error() string {
return fmt.Sprintf("%s not found", e.FileName)
}
func find(filename string) error {
_, err := os.Stat(filename)
if err != nil {
// 创建自定义错误类型并返回
return &NotFoundError{filename}
}
return nil
}
调用时可以转换回来
err := find("hello.txt")
if e, ok := err.(*NotFoundError); ok {
fmt.Println(e.FileName)
}
二、defer 处理资源释放问题
Go 语言中的 defer 语句会将函数执行推迟到外层函数返回之前。
1. defer 的执行规则与特点
defer 延迟调用遵循 FILO 规则。
func main() {
defer fmt.Println(1)
defer fmt.Println(2)
fmt.Println(3)
// 输出顺序是 3,2,1。
}
defer 语句中的变量在定义时求值(输出是 1,不是 2。)
func main() {
x := 1
defer fmt.Println(x)
x++
}
2. defer 释放资源
defer 经常用于释放资源,不管 process 返回与否,都能保证文件关闭。
func main() {
file := openFile("abc.txt")
defer file.Close()
process(file) // 使用文件
}
3. defer 经典案例
读取文件内容的示例,输出完内容后,defer 确保文件正确关闭。
func cat(f string) error {
file, err := os.Open(f)
if err != nil {
return err
}
defer file.Close()
stat, err := file.Stat()
if err != nil {
return err
}
bs := make([]byte, stat.Size())
_, err = file.Read(bs)
if err != nil {
return err
}
fmt.Println(string(bs))
return nil
}
三、panic 和 recover
panic 用于程序未能处理的错误,recover 可以让 panic 跳出的 goroutine 恢复,防止整个程序崩溃。
1. panic、defer 和 recover 的关系
- panic 会导致当前 goroutine 立即中止执行,但 defer 保证执行
- recover 可以捕获 panic,从而影响应用恢复
2. recover 捕获 panic
recover 只在 defer 内调用时有效
func ProtectRun(entry func()) {
defer func() {
// 捕获函数异常
if r := recover(); r != nil {
log.Printf("run time panic: %v", r)
}
}()
entry() // 可能引发panic的函数
}
3. panic 应用场景
不希望看到 panic,但它也有合理应用场景
// 非法状态,不可能恢复
panic("invalid state")
// 参数校验未过,不应继续
// 等于“抛出异常”,让程序停下。
if x