Add AddrCollection Copy()

miguelangel-nubla · Dec 30, 2024 · a62e2f0 · a62e2f0
1 parent 81c3224
commit a62e2f0
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/addr_collection.go b/addr_collection.go
@@ -69,7 +69,7 @@ func (c *AddrCollection) Equal(other *AddrCollection) bool {
 		return false
 	}
 
-	for addrKey, _ := range c.addresses {
+	for addrKey := range c.addresses {
 		if _, ok := other.addresses[addrKey]; !ok {
 			return false
 		}
@@ -78,6 +78,17 @@ func (c *AddrCollection) Equal(other *AddrCollection) bool {
 	return true
 }
 
+func (c *AddrCollection) Copy() *AddrCollection {
+	c.addressesMutex.RLock()
+	defer c.addressesMutex.RUnlock()
+
+	result := NewAddrCollection()
+	for _, addr := range c.addresses {
+		result.Enlist(addr)
+	}
+	return result
+}
+
 func (c *AddrCollection) FilterPrefix(prefix netip.Prefix) *AddrCollection {
 	c.addressesMutex.RLock()
 	defer c.addressesMutex.RUnlock()