golang通过context控制并发的应用场景实现

2020-01-28 14:26:58王旭

最开始的 context.WithCancel(context.Background()) 中 context.Background() 就是一个新建的 context, 利用 context 能继承的特性, 可以将自己的程序构建出一个 context 树, context 执行 cancel() 将影响到当前 context 和子 context, 不会影响到父级.

同时 context.WithValue 也会给 context 带上自定义的值, 这样 uniqueID 就能轻松的传递了下去, 而不是一层层的传递参数, 改func什么的

对于 context 很值得参考的应用有:

Gin logrus

Context 相关 func 和接口

继承 context 需要实现如下四个接口


type Context interface {
 Deadline() (deadline time.Time, ok bool)

 Done() <-chan struct{}

 Err() error

 Value(key interface{}) interface{}
}

当使用的时候不需要实现接口, 因为官方包里已经基于 emptyCtx 实现了一个, 调用方法有


var (
 background = new(emptyCtx)
 todo  = new(emptyCtx)
)

// 这个是最初始的ctx, 之后的子ctx都是继承自它
func Background() Context {
 return background
}

// 不清楚context要干嘛, 但是就得有一个ctx的用这个
func TODO() Context {
 return todo
}

继承用的函数


func WithCancel(parent Context) (ctx Context, cancel CancelFunc)
func WithDeadline(parent Context, deadline time.Time) (Context, CancelFunc)
func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc)
func WithValue(parent Context, key, val interface{}) Context
WithCancel 返回一个带 cancel 函数的ctx, WithDeadline 在到达指定时间时自动执行 cancel() WithTimeout 是 WithDeadline的壳子, 区别就是这个函数是多少时间过后执行 cancel

func WithTimeout(parent Context, timeout time.Duration) (Context, CancelFunc) {
 return WithDeadline(parent, time.Now().Add(timeout))
}

WithValue 继承父类ctx时顺便带上一个值

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持易采站长站。