Delete key when count==0

libp2p · May 14, 2024 · a1d135b · a1d135b
1 parent ccdf0e1
commit a1d135b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/p2p/host/resource-manager/conn_limiter.go b/p2p/host/resource-manager/conn_limiter.go
@@ -134,5 +134,8 @@ func (cl *connLimiter) rmConn(ip netip.Addr) {
 			continue
 		}
 		countsPerLimit[i][masked]--
+		if countsPerLimit[i][masked] == 0 {
+			delete(countsPerLimit[i], masked)
+		}
 	}
 }