golang如何去除 context 的 deadline
更新时间：2023年03月21日 11:04:45 作者：Immortal_s
在使用 context 的时候遇到了开协程处理任务的情况，但是直接在协程里使用主线程的 context 会导致当主线程返回时协程任务也会因为 context cancel 而失败，本文提供了两种办法可以取消掉 context 里的 timeout 和 deadline，再设置一个新的 timeout 上去
golang 去除 context 的 deadline
背景
在使用 context 的时候遇到了开协程处理任务的情况，但是直接在协程里使用主线程的 context 会导致当主线程返回时协程任务也会因为 context cancel 而失败。
本文提供了两种办法可以取消掉 context 里的 timeout 和 deadline，再设置一个新的 timeout 上去。
方法一，创建一个新 context
最简单的方案是通过创建一个新的 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的文章就介绍到这了,更多相关go去除 context 的 deadline内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要为大家详细介绍了Go语言中io/ioutil工具的使用，从而简化文件操作。文中是示例代码讲解详细，感兴趣的小伙伴可以了解一下2022-05-05
本文主要介绍了Go 语言的 :=的具体使用，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-12-12
go1.21中,slog这一被Go语言团队精心设计的结构化日志包正式落地,本文将带领读者上手slog,体会其与传统log的差异,感兴趣的小伙伴快跟随小编一起学习一下吧2023-09-09
这篇文章主要为大家详细介绍了golang中select关键字的常用用法,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2023-10-10
本文主要介绍了Go语言的互斥锁的详细使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-06-06
这篇文章主要为大家介绍了Golang如何快速构建一个CLI小工具详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-11-11
原子操作就是不可中断的操作，本文主要介绍了Go语言原子操作及互斥锁的区别，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-12-12
这篇文章主要介绍了golang switch语句的灵活写法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-05-05
工厂方法是一种创建型设计模式， 解决了在不指定具体类的情况下创建产品对象的问题，本文将通过代码示例详细给大家介绍一下Golang工厂方法模式，感兴趣的同学可以参考一下2023-06-06
这篇文章主要为大家详细介绍了Go语言中各种newreader和newbuffer的使用的相关资料,文中的示例代码讲解详细,具有一定的参考价值,感兴趣的小伙伴可以了解下2023-11-11
最新评论