ZBLOG

grpc-gateway 传递http header 数据到grpc 元数据

要将HTTP头数据传递到gRPC元数据中,需要使用grpc-gateway的以下两个功能:

  1. RequestHeaderMatcher:用于将HTTP标头映射到gRPC元数据键。
  2. IncomingHeaderMatcher:用于从gRPC响应的元数据中提取值并添加到HTTP响应头。

下面是一个示例,说明如何在grpc-gateway和gRPC之间传递HTTP标头数据:

首先,在proto文件中定义消息头:

  1. message HeaderMessage {
  2. map<string, string> headers = 1;
  3. }

然后,使用RequestHeaderMatcher将HTTP标头映射到gRPC元数据键:

  1. mux := runtime.NewServeMux(
  2. runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
  3. switch header {
  4. case "Authorization":
  5. return header, true
  6. default:
  7. return "", false
  8. }
  9. }),
  10. )
  11. ctx := context.Background()
  12. ctx = metadata.AppendToOutgoingContext(ctx, "Authorization", req.Header.Get("Authorization"))
  13. resp, err := client.MyMethod(ctx, &pb.MyRequest{})

最后,在IncomingHeaderMatcher中检索元数据,并将其添加到HTTP响应头中:

  1. mux := runtime.NewServeMux(
  2. runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
  3. switch header {
  4. case "X-My-Header":
  5. return header, true
  6. default:
  7. return "", false
  8. }
  9. }),
  10. runtime.WithMetadata(incomingMD),
  11. )
  12. w.Header().Set("X-My-Header", incomingMD.Get("X-My-Header")[0])

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://golang.0voice.com/?id=5919

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?