fix: prevent double node.Shutdown

anycable · Jun 15, 2021 · baedb5a · baedb5a
1 parent 60b0e08
commit baedb5a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/node/node.go b/node/node.go
@@ -1,6 +1,7 @@
 package node
 
 import (
+	"errors"
 	"fmt"
 	"runtime"
 	"time"
@@ -134,7 +135,12 @@ func (n *Node) HandlePubSub(raw []byte) {
 
 // Shutdown stops all services (hub, controller)
 func (n *Node) Shutdown() (err error) {
+	if n.shutdownCh == nil {
+		return errors.New("Already shut down")
+	}
+
 	close(n.shutdownCh)
+	n.shutdownCh = nil
 
 	if n.hub != nil {
 		n.hub.Shutdown()