From b013106f894902e24549197387e071167d206d10 Mon Sep 17 00:00:00 2001 From: Jimmy Zelinskie Date: Wed, 3 May 2017 21:28:38 -0400 Subject: [PATCH] storage/memorybysubnet: delete empty subnet buckets --- storage/memorybysubnet/peer_store.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/storage/memorybysubnet/peer_store.go b/storage/memorybysubnet/peer_store.go index 417536e..d5f1b47 100644 --- a/storage/memorybysubnet/peer_store.go +++ b/storage/memorybysubnet/peer_store.go @@ -547,6 +547,9 @@ func (s *peerStore) collectGarbage(cutoff time.Time) error { delete(shard.swarms[ih].leechers[subnet], pk) } } + if len(shard.swarms[ih].leechers[subnet]) == 0 { + delete(shard.swarms[ih].leechers, subnet) + } } for subnet := range shard.swarms[ih].seeders { @@ -555,6 +558,9 @@ func (s *peerStore) collectGarbage(cutoff time.Time) error { delete(shard.swarms[ih].seeders[subnet], pk) } } + if len(shard.swarms[ih].seeders[subnet]) == 0 { + delete(shard.swarms[ih].seeders, subnet) + } } // TODO(jzelinskie): fix this to sum all peers in all subnets