Guarantee minimum when selecting peers to broadcast

planetarium · Jan 22, 2020 · e3b19de · e3b19de
1 parent c4785f6
commit e3b19de
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/Libplanet/Net/Protocols/RoutingTable.cs b/Libplanet/Net/Protocols/RoutingTable.cs
@@ -7,6 +7,9 @@ namespace Libplanet.Net.Protocols
 {
     internal class RoutingTable
     {
+        // FIXME: This would be configurable.
+        private const int _minPeersToBroadcast = 10;
+
         private readonly Address _address;
         private readonly int _tableSize;
         private readonly int _bucketSize;
@@ -80,9 +83,18 @@ private IEnumerable<KBucket> NonEmptyBuckets
 
         public IEnumerable<BoundPeer> PeersToBroadcast(Address? except)
         {
-            return NonEmptyBuckets
+            var peers = NonEmptyBuckets
                 .Select(bucket => bucket.GetRandomPeer(except))
-                .Where(peer => !(peer is null));
+                .Where(peer => !(peer is null)).ToList();
+            var count = peers.Count;
+            if (count < _minPeersToBroadcast)
+            {
+                peers.AddRange(Peers
+                    .Where(peer => except is null || !peer.Address.Equals(except.Value))
+                    .Take(_minPeersToBroadcast - count));
+            }
+
+            return peers;
         }
 
         public IEnumerable<BoundPeer> PeersToRefresh(TimeSpan maxAge)