announce retry should check the pubsub context for cancellation

libp2p · Mar 6, 2018 · 85b3871 · 85b3871
1 parent ace0e2f
commit 85b3871
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pubsub.go b/pubsub.go
@@ -335,12 +335,18 @@ func (p *PubSub) announce(topic string, sub bool) {
 
 func (p *PubSub) announceRetry(topic string, sub bool) {
 	time.Sleep(time.Duration(1+rand.Intn(1000)) * time.Millisecond)
-	p.eval <- func() {
+
+	retry := func() {
 		_, ok := p.myTopics[topic]
 		if (ok && sub) || (!ok && !sub) {
 			p.announce(topic, sub)
 		}
 	}
+
+	select {
+	case p.eval <- retry:
+	case <-p.ctx.Done():
+	}
 }
 
 // notifySubs sends a given message to all corresponding subscribers.