YT: Limit count in viable peer registry

1 files changed, 2 insertions, 0 deletions

diff --git a/yt/yt/core/rpc/viable_peer_registry.cpp b/yt/yt/core/rpc/viable_peer_registry.cpp
index 5e9e0e5d22..118e20efa0 100644
--- a/yt/yt/core/rpc/viable_peer_registry.cpp
+++ b/yt/yt/core/rpc/viable_peer_registry.cpp
@@ -213,6 +213,8 @@ public:
     static THashSet<int> GetRandomIndexes(int max, int count = 1)
     {
         THashSet<int> result;
+        count = std::min(count, max);
+        result.reserve(count);
         while (std::ssize(result) < count) {
             result.insert(static_cast<int>(RandomNumber<unsigned int>(max)));
         }