missing nil check for grpc WaitGroup

2019-06-17 17:07:55 +08:00 · 2019-06-17 17:07:55 +08:00 · 363fb551af
commit 363fb551af
parent 7a87ae0efa
1 changed files with 6 additions and 2 deletions
--- a/server/grpc/subscriber.go
+++ b/server/grpc/subscriber.go
@ -231,9 +231,13 @@ func (g *grpcServer) createSubHandler(sb *subscriber, opts server.Options) broke
 				fn = opts.SubWrappers[i-1](fn)
 			}

+			if g.wg != nil {
 				g.wg.Add(1)
+			}
 			go func() {
+				if g.wg != nil {
 					defer g.wg.Done()
+				}
 				fn(ctx, &rpcMessage{
 					topic:       sb.topic,
 					contentType: ct,