From 1bac08cc0e6db0b454edf894a77a2e5c8131b5c4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E6=97=AD?= <play89925@gmail.com>
Date: Thu, 9 Jul 2020 01:36:07 +0800
Subject: [PATCH] server.Init(Wait(nil)) update server options context (#1804)

Co-authored-by: wangxu <wangxu@oneniceapp.com>
Co-authored-by: Asim Aslam <asim@aslam.me>
---
 server/grpc/grpc.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server/grpc/grpc.go b/server/grpc/grpc.go
index 972f4ba8..689cd7d3 100644
--- a/server/grpc/grpc.go
+++ b/server/grpc/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{