ZBLOG

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

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

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

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

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

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

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

mux := runtime.NewServeMux(
    runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
        switch header {
        case "Authorization":
            return header, true
        default:
            return "", false
        }
    }),
)

ctx := context.Background()
ctx = metadata.AppendToOutgoingContext(ctx, "Authorization", req.Header.Get("Authorization"))

resp, err := client.MyMethod(ctx, &pb.MyRequest{})

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

mux := runtime.NewServeMux(
    runtime.WithIncomingHeaderMatcher(func(header string) (string, bool) {
        switch header {
        case "X-My-Header":
            return header, true
        default:
            return "", false
        }
    }),
    runtime.WithMetadata(incomingMD),
)

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

注册

已经有帐号?