go的context包基本使用

Golang 使用树形派生的方式构造 Context，通过在不同过程 [1] 中传递 deadline 和 cancel 信号，来管理处理某个任务所涉及到的一组 goroutine 的生命周期，防止 goroutine 泄露。并且可以通过附加在 Context 上的 Value 来传递/共享一些跨越整个请求间的数据。

Context 最常用来追踪 RPC/HTTP 等耗时的、跨进程的 IO 请求的生命周期，从而让外层调用者可以主动地或者自动地取消该请求，进而告诉子过程回收用到的所有 goroutine 和相关资源。

context是线程安全的。

 

基本使用代码示例:

package main

import (
    "context"
    "fmt"
    "time"
)

// 测试使用context, WithTimeout,WithCancel,WithDeadline,需要在协程内部来检测取消信号，然后退出

func main() {
    // 给一个协程设置超时时间, 5秒后协程自动调用cancel方法
    ctx, cancelFunc := context.WithTimeout(context.Background(), 5 * time.Second)

    // 给一个协程设置取消信号，需要取消时，直接执行取消方法
    ctx2, cancel := context.WithCancel(context.Background())
    // 给一个协程设置过期时间, 3秒后自动调用cancel方法
    ctx3, cancelFunc2 := context.WithDeadline(context.Background(), time.Now().Add(3*time.Second))
    // 设置value
    ctx4 := context.WithValue(context.Background(), "traceId", "abcdefdg")

    go testTimeOut(ctx, "监控1")
    go testTimeOut(ctx, "监控2")
    go testCancel(ctx2)
    go testDeadLine(ctx3, "dead 监控1")
    go testDeadLine(ctx3, "dead 监控2")

    go testValue(ctx4)

    time.Sleep(time.Second * 2)
    // 调用取消
    cancel()

    time.Sleep(10000 * time.Second)
    cancelFunc()  // 最终调用超时
    cancelFunc2() // 最终调用过期时间
}

func testTimeOut(ctx context.Context, name string) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println(name, "执行超时了")
            return
        default:
            fmt.Println(name, "goroutine监控中,time=", time.Now().Unix())
            time.Sleep(1 * time.Second)
        }
    }

}

func testCancel(ctx context.Context) {
    // 检测取消信号
    for {
        select {
          case <-ctx.Done():
              fmt.Println("取消了")
              return
        default:
            fmt.Println("goroutine执行中,time=", time.Now().Unix())
            time.Sleep(1 * time.Second)
        }
    }
}


func testDeadLine(ctx context.Context, name string) {
    for {
        select {
        case <-ctx.Done():
            fmt.Println(name, "收到信号，监控退出,time=", time.Now().Unix())
            return
        default:
            fmt.Println(name, "goroutine监控中,time=", time.Now().Unix())
            time.Sleep(1 * time.Second)
        }
    }
}

// 测试调用value
func testValue(ctx context.Context) {
    traceId, ok := ctx.Value("traceId").(string)
    if ok {
        fmt.Printf("process over. trace_id=%s\n", traceId)
    } else {
        fmt.Printf("process over. no trace_id\n")
    }
    return
}