nit: cleanup contains check

libp2p · Oct 16, 2024 · 9cb382c · 9cb382c
1 parent a76c6d8
commit 9cb382c
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/p2p/host/autonat/autonat.go b/p2p/host/autonat/autonat.go
@@ -3,6 +3,7 @@ package autonat
 import (
 	"context"
 	"math/rand"
+	"slices"
 	"sync/atomic"
 	"time"
 
@@ -245,15 +246,10 @@ func (as *AmbientAutoNAT) background() {
 
 func (as *AmbientAutoNAT) checkAddrs() (hasNewAddr bool) {
 	currentAddrs := as.addressFunc()
-	for _, a := range currentAddrs {
-		if !manet.IsPublicAddr(a) {
-			continue
-		}
-		if _, ok := as.ourAddrs[string(a.Bytes())]; !ok {
-			hasNewAddr = true
-			break
-		}
-	}
+	hasNewAddr = slices.ContainsFunc(currentAddrs, func(a ma.Multiaddr) bool {
+		_, ok := as.ourAddrs[string(a.Bytes())]
+		return !ok
+	})
 	clear(as.ourAddrs)
 	for _, a := range currentAddrs {
 		if !manet.IsPublicAddr(a) {