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 应该等待 Job1 和 Job3 等待 Job2。并且，Job1A 和 Job1B 可以同时运行。如果我想传播给定 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复制上下文的所有值的详细内容，更多请关注编程网其它相关文章！