支持grpc拦截器加入

This commit is contained in:
suguo.yao 2021-10-04 13:10:37 +08:00
parent e386683768
commit a42b6770c9
1 changed files with 3 additions and 1 deletions

View File

@ -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
}