go - 将上下文从 gRPC 端点传递到 goroutine 收到上下文取消错误

Question

我正在尝试将上下文从传入的 gRPC 端点传递到 goroutine，该 goroutine 负责向外部服务发送另一个请求，但我收到来自goroutine 内的 ctxhttp.Get 函数调用:

package main

import (
  "fmt"
  "net"
  "net/http"
  "os"
  "sync"

  "golang.org/x/net/context/ctxhttp"

  dummy_service "github.com/myorg/testing-apps/dummy-proto/gogenproto/dummy/service"
  "github.com/myorg/testing-apps/dummy-proto/gogenproto/dummy/service/status"
  "golang.org/x/net/context"

  "google.golang.org/grpc"
  "google.golang.org/grpc/reflection"
)

func main() {
  var err error

  grpcServer := grpc.NewServer()

  server := NewServer()
  dummy_service.RegisterDummyServer(grpcServer, server)
  reflection.Register(grpcServer)

  lis, err := net.Listen("tcp", ":9020")
  if err != nil {
    fmt.Printf("Failed to listen: %+v", err)
    os.Exit(-1)
  }
  defer lis.Close()

  wg := sync.WaitGroup{}

  wg.Add(1)
  go func() {
    defer wg.Done()
    fmt.Println("Starting gRPC Server")
    if err := grpcServer.Serve(lis); err != nil {
      fmt.Printf("Failed to serve gRPC: %+v", err)
      os.Exit(-1)
    }
  }()

  wg.Wait()
}

type server struct{}

func NewServer() server {
  return server{}
}

func (s server) Status(ctx context.Context, in *status.StatusRequest) (*status.StatusResponse, error) {
  go func(ctx context.Context) {
    client := http.Client{}

    // it's important to send the ctx from the parent function here because it contains
    // a correlation-id which was inserted using grpc middleware, and the external service
    // prints this value in the logs to tie everything together
    if _, err := ctxhttp.Get(ctx, &client, "http://localhost:4567"); err != nil {
      fmt.Println("Error encountered:", err)
      return
    }

    fmt.Println("No error encountered")
  }(ctx)

  response := status.StatusResponse{
    Status: status.StatusResponse_SUCCESS,
  }


  // if I enable the following, everything works, and I get "No error encountered"
  // time.Sleep(10 * time.Millisecond)

  return &response, nil
}

如果我在调用函数中添加 time.Sleep() ，goroutine 将按预期成功并且不会收到任何错误。看来父函数的上下文一返回就被取消了，并且由于父函数在 goroutine 之前结束，所以传递给 goroutine 的上下文正在接收 context Canceled 错误。

我意识到我可以通过让调用函数等待 goroutine 完成来解决这个问题，这将防止上下文被取消，但我不想这样做，因为我希望函数立即返回，以便到达端点的客户端会尽快得到响应，而 goroutine 会在后台继续处理。

我也可以通过不使用传入的 ctx 而是在我的 goroutine 中使用 context.Background() 来解决这个问题，但是，我想使用传入的 ctx 因为它包含一个由 grpc 中间件插入的 correlation-id 值，需要作为 goroutine 发出的传出请求的一部分传递，以便下一个服务器可以在其日志消息中打印此 correlation-id 以将请求捆绑在一起。

我最终通过从传入上下文中提取 correlation-id 并将其插入到 goroutine 中的新 context.Background() 中解决了这个问题，但我想避免这种情况，因为它在 goroutine 发出的每个传出请求周围添加了一堆样板代码，而不仅仅是能够传递上下文。

任何人都可以向我解释为什么上下文被取消，并让我知道是否有针对这种情况的“最佳实践”解决方案？是否无法在 gRPC 的 goroutine 中使用从调用函数传入的上下文？

Answer 1

@adamc 如果您还没有找到任何其他方法。

我最终得到了这个解决方案(这也不完美)只是为了复制完整的上下文。但我更喜欢手动将原始上下文中的值添加到 context.Background

md, _ := metadata.FromIncomingContext(ctx)
copiedCtx := metadata.NewOutgoingContext(context.Background(), md)