一 defer 延迟执行

1.1 defer延迟执行修饰符

Go语言提供的defer机制,可以让开发者在创建资源(比如:数据库连接、文件句柄、锁等) 后,能够及时释放资源:

func main() {
    //当执行到defer语句时,暂不执行,会将defer后的语句压入到独立的栈中,当函数执行完毕后,再从该栈按照先入后出的方式出栈执行
    defer fmt.Println("defer1...")
    defer fmt.Println("defer2...")
    fmt.Println("main...")
}

上述代码执行结果:

main...
defer2...
defer1...

defer将语句放入到栈时,也会将相关的值拷贝同时入栈:

func main() {
    num := 0
    defer fmt.Println("defer中:num=", num)
    num = 3
    fmt.Println("main中:num=",num)
}

输出结果:

main中:num= 3
defer中:num= 0

1.2 defer最佳实践

案例一:defer处理资源

没有使用defer时打开文件处理代码:


f,err := os.Open(file)
if err != nil {
    return 0
}

info,err := f.Stat()
if err != nil {
    f.Close()
    return 0
}

f.Close()
return 0;

使用defer优化:


f,err := os.Open(file)
if err != nil {
    return 0
}

defer f.Close()

info,err := f.Stat()

if err != nil {
    // f.Close()            //这句已经不需要了
    return 0
}

//后续一系列文件操作后执行关闭
// f.Close()            //这句已经不需要了
return 0;

案例二:并发使用map的函数。

无defer代码:

var (
    mutex sync.Mutex
    testMap = make(map[string]int)
)
func getMapValue(key string) int {
    mutex.Lock()                        //对共享资源加锁
    value := testMap[key]
    mutex.Unlock()
    return value
}

上述案例是很常见的对并发map执行加锁执行的安全操作,使用defer可以对上述语义进行简化:

var (
    mutex sync.Mutex
    testMap = make(map[string]int)
)
func getMapValue(key string) int {
    mutex.Lock()                        //对共享资源加锁
    defer mutex.Unlock()
    return testMap[key]
}

1.3 defer无法处理全局资源

使用defer语句, 可以方便地组合函数/闭包和资源对象,即使panic时,defer也能保证资源的正确释放。但是上述案例都是在局部使用和释放资源,如果资源的生命周期很长, 而且可能被多个模块共享和随意传递的话,defer语句就不好处理了。

Go的runtime包的func SetFinalize(x, f interface{})函数可以提供类似C++析构函数的机制。

示例:包装一个文件对象,在没有人使用的时候能够自动关闭。

type MyFile struct {
    f *os.File
}

func NewFile(name string) (&MyFile, error){
    file, err := os.Open(name)
    if err != nil {
        return nil, err
    }
    runtime.SetFinalizer(file, file.f.Close)
    return &MyFile{f: file}, nil
}

在使用runtime.SetFinalizer时, 需要注意的地方是尽量要用指针访问内部资源,这样的话, 即使*MyFile对象忘记释放, 或者是被别的对象无意中覆盖, 也可以保证内部的文件资源可以正确释放。

二 Error 错误

2.1 Go自带的错误接口

error是go语言声明的接口类型:

type error interface {
    Error() string
}

所有符合Error()string格式的方法,都能实现错误接口,Error()方法返回错误的具体描述。

2.2 自定义错误

返回错误前,需要定义会产生哪些可能的错误,在Go中,使用errors包进行错误的定义,格式如下:

var err = errors.New("发生了错误")

提示:错误字符串相对固定,一般在包作用于声明,应尽量减少在使用时直接使用errors.New返回。

errors.New使用示例:

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

err常见使用方式:

f, err := os.Open("filename.ext")
if err != nil {
    fmt.Println(err)
}

实现错误接口案例:

package main

import (
    "fmt"
)

//声明一种解析错误
type ParseError struct {
    Filename string
    Line int
}

//实现error接口,返回错误描述
func (e *ParseError) Error() string {
    return fmt.Sprintf("%s:%d", e.Filename, e.Line)
}

//创建一些解析错误
func newParseError(filename string, line int) error {
    return &ParseError{filename, line}
}

func main() {

    var e error

    e = newParseError("main.go", 1)

    fmt.Println(e.Error())

    switch detail := e.(type) {
    case *ParseError:
        fmt.Printf("Filename: %s Line:%d \n", detail.Filename, detail.Line)
    default: 
        fmt.Println("other error")
    }

}

三 panic 宕机

Go语言可以在程序中手动触发宕机,让程序崩溃,这样开发者可以及时发现错误:

package main

func main() {
    defer fmt.Println("before")        // panic()前面已经运行过的defer语句依然会在宕机时发生作用
    panic("crash")                    // panic()参数可以是任意类型
    fmt.Println("after")            // panic()后面的代码将不会执行
}

运行结果是:

before
panic: crash

goroutine 1 [running]:
main.main()

注意:手动触发宕机并不是一种偷懒的方式,反而能迅速报错,终止程序继续运行,防止更大的错误产生,但是如果任何错误都使用宕机处理,也不是一个良好的设计。

四 recover 宕机恢复

Go提供的recover机制:由运行时抛出,或者开发者主动触发的panic,可以使用defer和recover实现错误捕捉和处理,让代码在发生崩溃后允许继续执行。

在其他语言里,宕机往往以异常的形式存在,底层抛出异常,上层逻辑通过try/catch机制捕获异常,没有被捕获的严重异常会导致宕机,捕获的异常可以被忽略,让代码继续执行。Go没有异常系统,使用panic触发宕机类似于其他语言的抛出异常,recover的宕机恢复机制就对应try/catch机制。

panic和defer的组合:

  • 有panic没有recover,程序宕机
  • 有panic也有recover,程序不会宕机,执行完对应的defer后,从宕机点退出当前函数后继续执行

示例:

package main

import "fmt"

func test(num1 int, num2 int){
    defer func(){
        err := recover()                // recover内置函数,可以捕获异常
        if err != nil {
            fmt.Println("err=", err)
        }
    }()
    fmt.Println(num1/num2)
}

func main() {
    test(2,0)
    fmt.Println("after...")            // 该句会输出
}

results matching ""

    No results matching ""