feat: api.

This commit is contained in:
Bobo
2025-05-10 15:44:51 +08:00
parent 03c605e139
commit 6099965d89

View File

@@ -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...))