tx7do/kratos-bootstrap
Public Access
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: api.

Browse Source
This commit is contained in:
Bobo
2025-05-10 15:44:51 +08:00
parent 03c605e139
commit 6099965d89
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rpc/rest.go
View File

@@ -10,6 +10,7 @@ import (
"github.com/go-kratos/aegis/ratelimit/bbr" "github.com/go-kratos/aegis/ratelimit/bbr"
"github.com/go-kratos/kratos/v2/middleware" "github.com/go-kratos/kratos/v2/middleware"
"github.com/go-kratos/kratos/v2/middleware/metadata"
midRateLimit "github.com/go-kratos/kratos/v2/middleware/ratelimit" midRateLimit "github.com/go-kratos/kratos/v2/middleware/ratelimit"
"github.com/go-kratos/kratos/v2/middleware/recovery" "github.com/go-kratos/kratos/v2/middleware/recovery"
"github.com/go-kratos/kratos/v2/middleware/tracing" "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))) 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...)) options = append(options, kratosRest.Middleware(ms...))