added missing Close call on the AddrBook member of GossipSubRouter (#568

)
libp2p · Jul 11, 2024 · e508d86 · e508d86
1 parent 093f13c
commit e508d86
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/gossipsub.go b/gossipsub.go
@@ -3,6 +3,7 @@ package pubsub
 import (
 	"context"
 	"fmt"
+	"io"
 	"math/rand"
 	"sort"
 	"time"
@@ -543,6 +544,13 @@ func (gs *GossipSubRouter) manageAddrBook() {
 	for {
 		select {
 		case <-gs.p.ctx.Done():
+			cabCloser, ok := gs.cab.(io.Closer)
+			if ok {
+				errClose := cabCloser.Close()
+				if errClose != nil {
+					log.Warnf("failed to close addr book: %v", errClose)
+				}
+			}
 			return
 		case ev := <-sub.Out():
 			switch ev := ev.(type) {