今天小编给大家分享一下golang怎么去除context的deadline的相关知识点,内容详细,逻辑清晰,相信大部分人都还太了解这方面的知识,所以分享这篇文章给大家参考一下,希望大家阅读完这篇文章后有所收获,下面我们一起来了解一下吧。
在使用 context 的时候遇到了开协程处理任务的情况,但是直接在协程里使用主线程的 context 会导致当主线程返回时协程任务也会因为 context cancel 而失败。
本文提供了两种办法可以取消掉 context 里的 timeout 和 deadline,再设置一个新的 timeout 上去。
最简单的方案是通过创建一个新的 context 来解决这个问题。
func main() {
ctx := context.WithValue(context.Background(), "trace", "123")
ctx, cancel := context.WithTimeout(ctx, time.Second)
defer cancel()
go func(ctx context.Context) {
newCtx := context.WithValue(context.Background(), "trace", ctx.Value("trace"))
// do something with newCtx
}(ctx)
fmt.Println("main finished")
}
但是这种方案有一个缺点,当生成一个新的 context 的时候,需要手动把老 context 中的 value 手动拿出来,再设置到新的里面去。 但是在很多情况下,这个 context 是上游传过来的,并不知道 value 里面有哪些具体的 key。或者里面的 value 过多,写起来很麻烦。
通过看 context 的源码,其实可以发现 context 是一个 interface,这就给了我们操作的空间。
type Context interface {
Deadline() (deadline time.Time, ok bool)
Done() <-chan struct{}
Err() error
Value(key any) any
}
Context 是通过 Deadline() 这个函数控制整个 ctx 是否超时了的。那么我们就可以通过重写这个函数来规避超时。
// contextWithoutDeadline 伪造了一个没有 deadline 的 context
type contextWithoutDeadline struct {
ctx context.Context
}
func (*contextWithoutDeadline) Deadline() (time.Time, bool) {
return time.Time{}, false
}
func (*contextWithoutDeadline) Done() <-chan struct{} {
return nil
}
func (*contextWithoutDeadline) Err() error {
return nil
}
func (l *contextWithoutDeadline) Value(key interface{}) interface{} {
return l.ctx.Value(key)
}
// DetachDeadline 从 context 剥离 deadline
func DetachDeadline(ctx context.Context) context.Context {
return &contextWithoutDeadline{ctx: ctx}
}
// SetNewTimeout 设置新的超时时间
func SetNewTimeout(ctx context.Context, timeout time.Duration) (context.Context, context.CancelFunc) {
return context.WithTimeout(DetachDeadline(ctx), timeout)
}
可以通过 DetachDeadline() 方法来将原 ctx 的 deadline 取消掉,或者直接通过 SetNewTimeout 的方法设置一个新的超时时间上去。
演示
func main() {
ctx, _ := context.WithTimeout(context.Background(), time.Duration(10)*time.Second)
fmt.Println(ctx.Deadline())
newCtx := contextwarp.DetachDeadline(ctx)
fmt.Println(newCtx.Deadline())
newCtx2, _ := contextwarp.SetNewTimeout(ctx, time.Duration(1)*time.Second)
fmt.Println(newCtx2.Deadline())
}
符合预期。
以上就是“golang怎么去除context的deadline”这篇文章的所有内容,感谢各位的阅读!相信大家阅读完这篇文章都有很大的收获,小编每天都会为大家更新不同的知识,如果还想学习更多的知识,请关注亿速云行业资讯频道。
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。
原文链接:https://blog.csdn.net/qq_24889575/article/details/129521977