use BucketContext & record utilization on removal

libp2p · Jul 29, 2019 · 189b505 · 189b505
1 parent 9e2000b
commit 189b505
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/table.go b/table.go
@@ -155,7 +155,9 @@ func (rt *RoutingTable) RemoveCtx(ctx context.Context, p peer.ID) {
 
 	bucket := rt.Buckets[bucketID]
 	if bucket.Remove(p) {
+		ctx = metrics.BucketContext(ctx, rt.local, bucketID)
 		metrics.RecordPeerRemoved(ctx, bucketID)
+		metrics.RecordBucketUtilization(ctx, bucket.Len())
 		rt.PeerRemoved(p)
 	}
 }