最新的gateway
This commit is contained in:
@@ -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
|
||||
|
||||
15
server/logic/gateway_meta.go
Normal file
15
server/logic/gateway_meta.go
Normal 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
|
||||
}
|
||||
@@ -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{}),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user