Question

如何将元数据从grpc-gateway发送到grpc服务器？

在客户端（grpc-gateway）上：

func (c *Client) MiddlewareAuth(h http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        ctx := r.Context()

        ...

        ctxOut := metadata.NewOutgoingContext(ctx, metadata.New(map[string]string{
            "key":    "value",
        }))

        r = r.WithContext(ctxOut)

        h.ServeHTTP(w, r)
    })
}

在服务器上：

func (s *Server) List(ctx context.Context, request *pb.Request) (*pb.Response, error) {
    md, _ := metadata.FromIncomingContext(ctx)
    fmt.Println(md)

    return &pb.Response{
        Ok: true
    }, nil
}

Answer 1

可以将HTTP标头映射到gRPC元数据，如here

所述

这应该有效：

// in Client.MiddlewareAuth
r.Header.Set("Grpc-Metadata-My-Data", "...")

// in Server.List
md.Get("grpcgateway-My-Data")

Answer 2

您可能不喜欢 the default mapping rule 并且可能想要传递所有 HTTP 标头，例如：

写一个HeaderMatcherFunc。
使用 WithIncomingHeaderMatcher 注册函数

例如

func CustomMatcher(key string) (string, bool) {
 switch key {
 case "X-Custom-Header1":
 return key, true
 case "X-Custom-Header2":
 return "custom-header2", true
 default:
 return key, false
 }
}

mux := runtime.NewServeMux(
 runtime.WithIncomingHeaderMatcher(CustomMatcher),
)

要使 the default mapping rule 与您自己的规则保持一致，请编写：

func CustomMatcher(key string) (string, bool) {
 switch key {
 case "X-User-Id":
 return key, true
 default:
 return runtime.DefaultHeaderMatcher(key)
 }
}

它适用于两者：

$ curl --header "x-user-id: 100d9f38-2777-4ee2-ac3b-b3a108f81a30" ...

和

$ curl --header "X-USER-ID: 100d9f38-2777-4ee2-ac3b-b3a108f81a30" ...

要在 gRPC 服务器端访问此标头，请使用：

userID := ""
if md, ok := metadata.FromIncomingContext(ctx); ok {
 if uID, ok := md["x-user-id"]; ok {
 userID = strings.Join(uID, ",")
 }
}

此外，您应该查看关于 gRPC-Gateway 的系列教程，即 https://grpc-ecosystem.github.io/grpc-gateway/docs/tutorials/。

如何从grpc-gateway发送元数据到grpc服务器？

2 个答案: