From b1988daa4e83b7a5a134e3e806199b2ea91a8572 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>
---
 grpc.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/grpc.go b/grpc.go
index 972f4ba..689cd7d 100644
--- 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{