tx7do/kratos-bootstrap
Private
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:43:14 +08:00
parent 00e2881172
commit 03c605e139
4 changed files with 100 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
rpc/grpc.go
View File

@@ -15,6 +15,7 @@ import (
"github.com/go-kratos/kratos/v2/registry"
"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"
@@ -78,6 +79,9 @@ func initGrpcClientConfig(cfg *conf.Bootstrap, mds ...middleware.Middleware) []k
if cfg.Client.Grpc.Middleware.GetEnableValidate() {
ms = append(ms, validate.Validator())
}
if cfg.Client.Grpc.Middleware.GetEnableMetadata() {
ms = append(ms, metadata.Client())
}
}
options = append(options, kratosGrpc.WithMiddleware(ms...))
@@ -137,6 +141,9 @@ func initGrpcServerConfig(cfg *conf.Bootstrap, mds ...middleware.Middleware) []k
}
ms = append(ms, midRateLimit.Server(midRateLimit.WithLimiter(limiter)))
}
if cfg.Server.Grpc.Middleware.GetEnableMetadata() {
ms = append(ms, metadata.Server())
}
}
options = append(options, kratosGrpc.Middleware(ms...))