首页 >后端开发 >Golang >Golang复制上下文的所有值

Golang复制上下文的所有值

WBOY
WBOY转载
2024-02-05 23:39:07885浏览

Golang复制上下文的所有值

问题内容

我有一个提供异步作业服务的 HTTP 服务器应用程序。

-> Request
   --> Do async job with goroutine
<- Response
    -------start goroutine------
       -> Job1
          -> Job1A
          -> Job1B
       -> Job2
       -> Job3

用户可以请求长时间运行的异步作业,应用程序在 make goroutine 后立即响应该请求。

我将请求 ID、经过身份验证的令牌和用户信息放入请求的 context.Context 中。而且,我想把它放在 goroutine 下。但是,对请求 context 使用相同的 context 将导致响应后意外取消,这不是我的预期行为。

如何生成具有所有值的新 context,独立于父请求 context?或者,还有其他方法可以保证 context 带入 goroutine 的响应后不会死吗?

还有一个额外问题:

Job1 ~ Job3 应该被序列化,即 Job2 应该等待 Job1Job3 等待 Job2。并且,Job1AJob1B 可以同时运行。如果我想传播给定 context 的取消,我怎样才能取消它们的路径(?)?我应该检查所有函数的 select 语句吗?

我理解了 context 传播取消和提前退出的概念,而不执行无意义的任务。但是,我还没有明白如何在代码中进行处理。如果有人能帮助理解,我会很高兴。


正确答案


无法发现上下文中的值。它们不存储为地图,而是存储为上下文层,每个级别可能提供值存储方式的不同实现。

但是,如果您知道需要传播哪些值,则可以查询它们并使用这些值创建新上下文。

也就是说,您可以实现一个使用另一个上下文中的值的新上下文类型:

type newContext struct {
  context.Context
  values context.Context
}

func (c newContext) Value(key any) any {
  return c.values.Value(key)
}

...
newCtx:=newContext{
  Context: context.Background(),
  values: ctx,
}

这使用现有的值上下文和其他所有内容的新上下文。

然后,启动一个新的 goroutine 以继续使用新上下文处理请求。

如果你想创建多个并发作业,你可以在该 goroutine 中执行此操作:

go func(ctx context.Context) {
   withCancel, cancel:=context.WithCancel(ctx)
   defer cancel()

   wg:=sync.WaitGroup{}
   wg.Add(2)
   go job1(withCancel,&wg)
   go job2(withCancel,&wg)
   wg.Wait()
}(newCtx)

这样,当上下文被取消时,两个作业都会收到取消通知。如果想分别控制job1和job2的取消:

go func(ctx context.Context) {
   withCancel1, cancel1:=context.WithCancel(ctx)
   defer cancel1()
   withCancel2, cancel2:=context.WithCancel(ctx)
   defer cancel2()

   wg:=sync.WaitGroup{}
   wg.Add(2)
   go job1(withCancel1,&wg)
   go job2(withCancel2,&wg)
   wg.Wait()
}(newCtx)

对于连续作业(即 job3 在 job1 之后完成),只需将它们组合起来,使它们看起来像一个作业。

要检查上下文是否被取消,您可以在上下文的 Done 通道上执行 select ,或者简单地检查:

if ctx.Err()!=nil {
   // Context canceled
}

以上是Golang复制上下文的所有值的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:stackoverflow.com。如有侵权,请联系admin@php.cn删除