Question

我正在尝试将上下文从传入的gRPC端点传递给goroutine，该goroutine负责向外部服务发送另一个请求，但是我从Error occurred: context canceled函数调用中收到了ctxhttp.Get在goroutine中：

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，因为它包含一个{{ 1}}值是由grpc中间件插入的，需要作为goroutine发出的传出请求的一部分传递，以便下一个服务器可以在日志消息中打印此correlation-id，以将请求绑定在一起。

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

任何人都可以向我确切说明为什么取消上下文的情况，并告诉我是否存在针对这种情况的“最佳实践”解决方案？使用gRPC无法在goroutine中使用从调用函数传入的上下文吗？

Answer 1

@adamc（如果您还没有找到其他方法）。

我最终得到了这个解决方案（这也不是很完美）只获取完整的上下文复制。但是我更喜欢这样做，而不是手动将原始上下文中的值添加到context.Background

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

从gRPC端点向goroutine传递上下文会收到上下文取消错误

1 个答案: