Allow rate to be less than 1.

Change all the token-related variables to doubles so the rate can be less than 1.
falcosecurity · Apr 25, 2017 · e183de3 · e183de3
1 parent 87a6c74
commit e183de3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/userspace/engine/token_bucket.cpp b/userspace/engine/token_bucket.cpp
@@ -31,7 +31,7 @@ token_bucket::~token_bucket()
 {
 }
 
-void token_bucket::init(uint32_t rate, uint32_t max_tokens)
+void token_bucket::init(double rate, double max_tokens)
 {
 	m_rate = rate;
 	m_max_tokens = max_tokens;
@@ -48,7 +48,7 @@ bool token_bucket::claim(uint64_t now)
 		now = sinsp_utils::get_current_time_ns();
 	}
 
-	uint64_t tokens_gained = (now - m_last_seen) / (m_rate * 1000000000);
+	double tokens_gained = (now - m_last_seen) / (m_rate * 1000000000);
 	m_last_seen = now;
 
 	m_tokens += tokens_gained;

diff --git a/userspace/engine/token_bucket.h b/userspace/engine/token_bucket.h
@@ -31,7 +31,7 @@ class token_bucket
 	//
 	// Initialize the token bucket and start accumulating tokens
 	//
-	void init(uint32_t rate, uint32_t max_tokens);
+	void init(double rate, double max_tokens);
 
 	//
 	// Returns true if a token can be claimed. Also updates
@@ -43,18 +43,18 @@ class token_bucket
 	//
 	// The number of tokens generated per second.
 	//
-	uint64_t m_rate;
+	double m_rate;
 
 	//
 	// The maximum number of tokens that can be banked for future
 	// claim()s.
 	//
-	uint64_t m_max_tokens;
+	double m_max_tokens;
 
 	//
 	// The current number of tokens
 	//
-	uint64_t m_tokens;
+	double m_tokens;
 
 	//
 	// The last time claim() was called (or the object was created).