fix unlimited growing of addrs in peer info

zeta-chain · Oct 29, 2024 · ba90276 · ba90276
1 parent 694f4b4
commit ba90276
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/p2p/discovery.go b/p2p/discovery.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"io"
+	"slices"
 	"sync"
 	"time"
 
@@ -83,7 +84,11 @@ func (pd *PeerDiscovery) addPeer(pinfo peer.AddrInfo) {
 	}
 	oldPinfo, ok := pd.knownPeers[pinfo.ID]
 	if ok {
-		oldPinfo.Addrs = append(oldPinfo.Addrs, pinfo.Addrs...)
+		for _, addr := range pinfo.Addrs {
+			if !slices.Contains(oldPinfo.Addrs, addr) {
+				oldPinfo.Addrs = append(oldPinfo.Addrs, addr)
+			}
+		}
 	} else {
 		oldPinfo = pinfo
 	}