文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

golang怎么去除context的deadline

2023-07-05 14:37

关注

今天小编给大家分享一下golang怎么去除context的deadline的相关知识点,内容详细,逻辑清晰,相信大部分人都还太了解这方面的知识,所以分享这篇文章给大家参考一下,希望大家阅读完这篇文章后有所收获,下面我们一起来了解一下吧。

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() errorValue(key any) any}

Context 是通过 Deadline() 这个函数控制整个 ctx 是否超时了的。那么我们就可以通过重写这个函数来规避超时。

// contextWithoutDeadline 伪造了一个没有 deadline 的 contexttype 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 剥离 deadlinefunc 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

符合预期。

以上就是“golang怎么去除context的deadline”这篇文章的所有内容,感谢各位的阅读!相信大家阅读完这篇文章都有很大的收获,小编每天都会为大家更新不同的知识,如果还想学习更多的知识,请关注编程网行业资讯频道。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯