chore: add tiemout to findpeers (#569)

Signed-off-by: mudler <mudler@localai.io>
mudler · Aug 27, 2024 · 418f7c6 · 418f7c6
1 parent d394cd6
commit 418f7c6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/discovery/dht.go b/pkg/discovery/dht.go
@@ -234,7 +234,9 @@ func (d *DHT) announceAndConnect(l log.StandardLogger, ctx context.Context, kade
 	// This is like your friend telling you the location to meet you.
 	l.Debug("Searching for other peers...")
 
-	peerChan, err := routingDiscovery.FindPeers(ctx, rv)
+	fCtx, cf := context.WithTimeout(ctx, time.Second*120)
+	defer cf()
+	peerChan, err := routingDiscovery.FindPeers(fCtx, rv)
 	if err != nil {
 		return err
 	}