fix waitgroup bug which crashes subscriber

2019-05-29 09:28:04 +01:00 · 2019-05-29 09:28:04 +01:00 · 9f037eafd2
commit 9f037eafd2
parent 58a70562d8
1 changed files with 8 additions and 2 deletions
--- a/server/subscriber.go
+++ b/server/subscriber.go
@ -244,9 +244,15 @@ func (s *rpcServer) createSubHandler(sb *subscriber, opts Options) broker.Handle
 				fn = opts.SubWrappers[i-1](fn)
 			}

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