最新的gateway

This commit is contained in:
eson
2023-11-16 16:24:52 +08:00
parent 309b18a2b7
commit f7efb6faf3
6 changed files with 25 additions and 4 deletions

View File

@@ -16,6 +16,11 @@ func AutoRegisterHandler(ctx context.Context, mux *runtime.ServeMux, opts ...grp
return err
}
err = service.RegisterFsserviceHandlerClientNacos(ctx, mux, opts...)
if err != nil {
return err
}
err = service.RegisterInfoHandlerClientNacos(ctx, mux, opts...)
if err != nil {
return err

View File

@@ -0,0 +1,15 @@
package logic
import (
"context"
"net/http"
"google.golang.org/grpc/metadata"
)
func PassMetadata(ctx context.Context, r *http.Request) metadata.MD {
headerValue := r.Header.Get("Debug-Token")
// 将值添加到 gRPC 元数据
md := metadata.Pairs("debug-token", headerValue)
return md
}

View File

@@ -27,6 +27,7 @@ func main() {
ctx, cancel := context.WithCancel(ctx)
defer cancel()
mux := runtime.NewServeMux(
runtime.WithMetadata(logic.PassMetadata),
runtime.WithForwardResponseOption(logic.ResponseHeaderMatcher),
runtime.WithMarshalerOption("*", &logic.EmptyMarshaler{}),
)