From b8973df266320d6be98ebcbeba02aa2d70335cad Mon Sep 17 00:00:00 2001
From: Cassandra Heart <cassandra@quilibrium.com>
Date: Sun, 27 Oct 2024 00:51:35 -0500
Subject: [PATCH] check bitmask for peers and keep alive

---
 node/p2p/blossomsub.go | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

diff --git a/node/p2p/blossomsub.go b/node/p2p/blossomsub.go
index b359ff4..b7099e3 100644
--- a/node/p2p/blossomsub.go
+++ b/node/p2p/blossomsub.go
@@ -303,6 +303,17 @@ func NewBlossomSub(
 	bs.h = h
 	bs.signKey = privKey
 
+	go func() {
+		for {
+			time.Sleep(30 * time.Second)
+			for _, b := range bs.bitmaskMap {
+				if len(b.ListPeers()) < 4 {
+					discoverPeers(p2pConfig, bs.ctx, logger, bs.h, routingDiscovery)
+				}
+			}
+		}
+	}()
+
 	return bs
 }
 
@@ -874,16 +885,7 @@ func discoverPeers(
 
 	discover()
 
-	go func() {
-		for {
-			time.Sleep(5 * time.Second)
-			if len(h.Network().Peers()) < 6 {
-				discover()
-			}
-		}
-	}()
-
-	logger.Info("completed initial peer discovery")
+	logger.Info("completed peer discovery")
 }
 
 func mergeDefaults(p2pConfig *config.P2PConfig) blossomsub.BlossomSubParams {