From 6099965d89e22d872cf0c0bae47cd143eb440e7e Mon Sep 17 00:00:00 2001 From: Bobo Date: Sat, 10 May 2025 15:44:51 +0800 Subject: [PATCH] feat: api. --- rpc/rest.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/rpc/rest.go b/rpc/rest.go index e6064e2..d033340 100644 --- a/rpc/rest.go +++ b/rpc/rest.go @@ -10,6 +10,7 @@ import ( "github.com/go-kratos/aegis/ratelimit/bbr" "github.com/go-kratos/kratos/v2/middleware" + "github.com/go-kratos/kratos/v2/middleware/metadata" midRateLimit "github.com/go-kratos/kratos/v2/middleware/ratelimit" "github.com/go-kratos/kratos/v2/middleware/recovery" "github.com/go-kratos/kratos/v2/middleware/tracing" @@ -73,6 +74,9 @@ func initRestConfig(cfg *conf.Bootstrap, mds ...middleware.Middleware) []kratosR } ms = append(ms, midRateLimit.Server(midRateLimit.WithLimiter(limiter))) } + if cfg.Server.Grpc.Middleware.GetEnableMetadata() { + ms = append(ms, metadata.Server()) + } } options = append(options, kratosRest.Middleware(ms...))