GGLinnk/bees
1
0
Fork 0
mirror of https://github.com/Zygo/bees.git synced 2025-05-17 21:35:45 +02:00
Code Packages

time: add more methods to support dynamic rate throttling

Browse Source 
* Allow RateLimiter to change rate after construction.
 * Check range of rate argument in constructor.
 * Atomic increment for RateEstimator.

Signed-off-by: Zygo Blaxell <bees@furryterror.org>
This commit is contained in:
Zygo Blaxell 2024-12-10 00:33:22 -05:00
parent 21cedfb13e
commit a02588b16f
2 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
include/crucible/time.h
View File

@ -34,7 +34,7 @@ namespace crucible {
double m_rate; double m_rate;
double m_burst; double m_burst;
double m_tokens = 0.0; double m_tokens = 0.0;
mutex m_mutex; mutable mutex m_mutex;
void update_tokens(); void update_tokens();
RateLimiter() = delete; RateLimiter() = delete;
@ -45,6 +45,8 @@ namespace crucible {
double sleep_time(double cost = 1.0); double sleep_time(double cost = 1.0);
bool is_ready(); bool is_ready();
void borrow(double cost = 1.0); void borrow(double cost = 1.0);
void rate(double new_rate);
double rate() const;
}; };
class RateEstimator { class RateEstimator {
@ -88,6 +90,9 @@ namespace crucible {
// Read count // Read count
uint64_t count() const; uint64_t count() const;
/// Increment count (like update(count() + more), but atomic)
void increment(uint64_t more = 1);
// Convert counts to chrono types // Convert counts to chrono types
chrono::high_resolution_clock::time_point time_point(uint64_t absolute_count) const; chrono::high_resolution_clock::time_point time_point(uint64_t absolute_count) const;
chrono::duration<double> duration(uint64_t relative_count) const; chrono::duration<double> duration(uint64_t relative_count) const;

27
lib/time.cc
View File

@ -98,12 +98,16 @@ namespace crucible {
m_rate(rate), m_rate(rate),
m_burst(burst) m_burst(burst)
{ {
THROW_CHECK1(invalid_argument, m_rate, m_rate > 0);
THROW_CHECK1(invalid_argument, m_burst, m_burst >= 0);
} }
RateLimiter::RateLimiter(double rate) : RateLimiter::RateLimiter(double rate) :
m_rate(rate), m_rate(rate),
m_burst(rate) m_burst(rate)
{ {
THROW_CHECK1(invalid_argument, m_rate, m_rate > 0);
THROW_CHECK1(invalid_argument, m_burst, m_burst >= 0);
} }
void void
@ -119,6 +123,7 @@ namespace crucible {
double double
RateLimiter::sleep_time(double cost) RateLimiter::sleep_time(double cost)
{ {
THROW_CHECK1(invalid_argument, m_rate, m_rate > 0);
borrow(cost); borrow(cost);
unique_lock<mutex> lock(m_mutex); unique_lock<mutex> lock(m_mutex);
update_tokens(); update_tokens();
@ -154,6 +159,21 @@ namespace crucible {
m_tokens -= cost; m_tokens -= cost;
} }
void
RateLimiter::rate(double const new_rate)
{
THROW_CHECK1(invalid_argument, new_rate, new_rate > 0);
unique_lock<mutex> lock(m_mutex);
m_rate = new_rate;
}
double
RateLimiter::rate() const
{
unique_lock<mutex> lock(m_mutex);
return m_rate;
}
RateEstimator::RateEstimator(double min_delay, double max_delay) : RateEstimator::RateEstimator(double min_delay, double max_delay) :
m_min_delay(min_delay), m_min_delay(min_delay),
m_max_delay(max_delay) m_max_delay(max_delay)
@ -202,6 +222,13 @@ namespace crucible {
} }
} }
void
RateEstimator::increment(const uint64_t more)
{
unique_lock<mutex> lock(m_mutex);
return update_unlocked(m_last_count + more);
}
uint64_t uint64_t
RateEstimator::count() const RateEstimator::count() const
{ {