diff --git a/grpc/grpc.go b/grpc/grpc.go index 266732e..9c7e132 100644 --- a/grpc/grpc.go +++ b/grpc/grpc.go @@ -18,6 +18,7 @@ type Config struct { Address string Port int AppName string + opts []grpc.ServerOption } var rpc *grpc.Server @@ -29,12 +30,13 @@ func Init(config *Config) *grpc.Server { Address: "0.0.0.0", Port: 0, AppName: "unknown", + opts: []grpc.ServerOption{grpc.MaxRecvMsgSize(1024 * 1024), grpc.MaxSendMsgSize(1024 * 1024)}, } } conf = config if rpc == nil { - rpc = grpc.NewServer(grpc.MaxRecvMsgSize(1024*1024), grpc.MaxSendMsgSize(1024*1024)) + rpc = grpc.NewServer(conf.opts...) } return rpc }