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複製上下文的所有值的詳細內容。更多資訊請關注PHP中文網其他相關文章！