This commit is contained in:
wangchuxiao 2022-09-15 19:55:24 +08:00
parent 5d32627553
commit eeb5ba4c7b

View File

@ -68,11 +68,10 @@ func (s *groupServer) Run() {
//grpc server //grpc server
recvSize := 1024 * 1024 * constant.GroupRPCRecvSize recvSize := 1024 * 1024 * constant.GroupRPCRecvSize
sendSize := 1024 * 1024 * constant.GroupRPCSendSize sendSize := 1024 * 1024 * constant.GroupRPCSendSize
var options = []grpc.ServerOption{ var grpcOpts = []grpc.ServerOption{
grpc.MaxRecvMsgSize(recvSize), grpc.MaxRecvMsgSize(recvSize),
grpc.MaxSendMsgSize(sendSize), grpc.MaxSendMsgSize(sendSize),
} }
var grpcOpts []grpc.ServerOption
if config.Config.Prometheus.Enable { if config.Config.Prometheus.Enable {
promePkg.NewGrpcRequestCounter() promePkg.NewGrpcRequestCounter()
promePkg.NewGrpcRequestFailedCounter() promePkg.NewGrpcRequestFailedCounter()
@ -83,7 +82,7 @@ func (s *groupServer) Run() {
grpc.UnaryInterceptor(grpcPrometheus.UnaryServerInterceptor), grpc.UnaryInterceptor(grpcPrometheus.UnaryServerInterceptor),
}...) }...)
} }
srv := grpc.NewServer(options...) srv := grpc.NewServer(grpcOpts...)
defer srv.GracefulStop() defer srv.GracefulStop()
//Service registers with etcd //Service registers with etcd
pbGroup.RegisterGroupServer(srv, s) pbGroup.RegisterGroupServer(srv, s)