fix: calling message queue Shutdown twice causes panic (because close…

… is called twice on done channel) (#414)
ipfs · Mar 21, 2023 · 975c0f7 · 975c0f7
1 parent 18ee3cb
commit 975c0f7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/messagequeue/messagequeue.go b/messagequeue/messagequeue.go
@@ -68,6 +68,7 @@ type MessageQueue struct {
 
 	outgoingWork chan struct{}
 	done         chan struct{}
+	doneOnce     sync.Once
 
 	// internal do not touch outside go routines
 	sender             gsnet.MessageSender
@@ -144,7 +145,9 @@ func (mq *MessageQueue) Startup() {
 
 // Shutdown stops the processing of messages for a message queue.
 func (mq *MessageQueue) Shutdown() {
-	close(mq.done)
+	mq.doneOnce.Do(func() {
+		close(mq.done)
+	})
 }
 
 func (mq *MessageQueue) runQueue() {