server.Init(Wait(nil)) update server options context (#1804)

Co-authored-by: wangxu <wangxu@oneniceapp.com> Co-authored-by: Asim Aslam <asim@aslam.me>
2020-07-09 01:36:07 +08:00 · 2020-07-09 01:36:07 +08:00 · b1988daa4e
commit b1988daa4e
parent f6ff36f501
1 changed files with 2 additions and 0 deletions
--- a/grpc.go
+++ b/grpc.go
@ -118,6 +118,8 @@ func (g *grpcServer) configure(opts ...server.Option) {
 		o(&g.opts)
 	}

+	g.wg = wait(g.opts.Context)
+
 	maxMsgSize := g.getMaxMsgSize()

 	gopts := []grpc.ServerOption{