add MaxCallRecvMsgSize

gotomicro · Jan 8, 2024 · a2143e0 · a2143e0
1 parent 643c2e7
commit a2143e0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/client/egrpc/config.go b/client/egrpc/config.go
@@ -31,7 +31,7 @@ type Config struct {
 	// EnableCPUUsage               bool          // 是否开启CPU利用率，默认开启
 	EnableServiceConfig          bool // 是否开启服务配置，默认开启
 	EnableFailOnNonTempDialError bool
-	MaxCallRecvMsgSize           int64 // 最大接收消息大小，默认4MB
+	MaxCallRecvMsgSize           int // 最大接收消息大小，默认4MB
 
 	keepAlive   *keepalive.ClientParameters
 	dialOptions []grpc.DialOption

diff --git a/client/egrpc/container.go b/client/egrpc/container.go
@@ -74,6 +74,7 @@ func (c *Container) Build(options ...Option) *Component {
 			WithDialOption(grpc.WithChainUnaryInterceptor(c.metricUnaryClientInterceptor())),
 		)
 	}
+	options = append(options, WithDialOption(grpc.WithDefaultCallOptions(grpc.MaxCallRecvMsgSize(c.config.MaxCallRecvMsgSize))))
 
 	for _, option := range options {
 		option(c)

diff --git a/client/egrpc/option.go b/client/egrpc/option.go
@@ -97,7 +97,7 @@ func WithName(name string) Option {
 }
 
 // WithMaxRecvMsgSize 设置最大接收消息大小
-func WithMaxRecvMsgSize(maxRecvMsgSize int64) Option {
+func WithMaxRecvMsgSize(maxRecvMsgSize int) Option {
 	return func(c *Container) {
 		c.config.MaxCallRecvMsgSize = maxRecvMsgSize
 	}