Improve fair_queue::next_batch ()

nanocurrency · May 5, 2024 · 02b5c82 · 02b5c82
1 parent 5ac4317
commit 02b5c82
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/nano/node/fair_queue.hpp b/nano/node/fair_queue.hpp
@@ -297,9 +297,10 @@ class fair_queue final
 
 	std::deque<value_type> next_batch (size_t max_count)
 	{
-		// TODO: Naive implementation, could be optimized
+		auto count = std::min (size (), max_count);
+
 		std::deque<value_type> result;
-		while (!empty () && result.size () < max_count)
+		while (result.size () < count)
 		{
 			result.emplace_back (next ());
 		}