Use unlimited_rate_sentinel

pwojcikdev · Nov 26, 2024 · a0ca33d · a0ca33d
1 parent fd1afe2
commit a0ca33d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/nano/lib/rate_limiting.cpp b/nano/lib/rate_limiting.cpp
@@ -13,9 +13,9 @@ nano::rate::token_bucket::token_bucket (std::size_t max_token_count_a, std::size
 	reset (max_token_count_a, refill_rate_a);
 }
 
-bool nano::rate::token_bucket::try_consume (unsigned tokens_required)
+bool nano::rate::token_bucket::try_consume (std::size_t tokens_required)
 {
-	debug_assert (tokens_required <= 1e9);
+	debug_assert (tokens_required <= unlimited_rate_sentinel);
 
 	refill ();
 

diff --git a/nano/lib/rate_limiting.hpp b/nano/lib/rate_limiting.hpp
@@ -36,7 +36,7 @@ class token_bucket
 	 * The default cost is 1 token, but resource intensive operations may request
 	 * more tokens to be available.
 	 */
-	bool try_consume (unsigned tokens_required = 1);
+	bool try_consume (std::size_t tokens_required = 1);
 
 	/** Update the max_token_count and/or refill_rate_a parameters */
 	void reset (std::size_t max_token_count, std::size_t refill_rate);