github/falco
1
0
Fork 0
mirror of https://github.com/falcosecurity/falco.git synced 2025-10-22 12:27:10 +00:00
Code Issues Projects Releases Wiki Activity

Add ability to claim multiple tokens.

Browse Source 
This way you can use it as a form of bandwidth throttling.
This commit is contained in:
Mark Stemm
2017-05-02 11:46:20 -07:00
parent a3c83e7f6e
commit 4d148ce28f
2 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
userspace/engine/token_bucket.cpp
View File

@@ -45,15 +45,15 @@ void token_bucket::init(double rate, double max_tokens, uint64_t now)
m_last_seen = now;
}
bool token_bucket::claim(uint64_t now)
bool token_bucket::claim()
{
// Determine the number of tokens gained. Delta between
// last_seen and now, divided by the rate.
if(now == 0)
{
now = sinsp_utils::get_current_time_ns();
}
uint64_t now = sinsp_utils::get_current_time_ns();
return claim(1, now);
}
bool token_bucket::claim(double tokens, uint64_t now)
{
double tokens_gained = m_rate * ((now - m_last_seen) / (1000000000.0));
m_last_seen = now;
@@ -68,14 +68,14 @@ bool token_bucket::claim(uint64_t now)
}
//
// If tokens is < 1, can't claim.
// If m_tokens is < tokens, can't claim.
//
if(m_tokens < 1)
if(m_tokens < tokens)
{
return false;
}
m_tokens--;
m_tokens -= tokens;
return true;
}