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

偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《将上下文从 gRPC 端点传递到 goroutine 收到上下文取消错误》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！

我正在尝试将上下文从传入的 grpc 端点传递到 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 中使用从调用函数传入的上下文？

解决方案

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

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

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

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~