From f5ce6752be33da2e287c625048bd18c20d282452 Mon Sep 17 00:00:00 2001
From: Mark Stemm <mark.stemm@sysdig.com>
Date: Thu, 27 Apr 2017 11:22:19 -0700
Subject: [PATCH] Add ability to get number of tokens.

Add a method to fetch the current number of available tokens.
---
 userspace/engine/token_bucket.cpp | 5 +++++
 userspace/engine/token_bucket.h   | 4 ++++
 2 files changed, 9 insertions(+)

diff --git a/userspace/engine/token_bucket.cpp b/userspace/engine/token_bucket.cpp
index f40112e7..1c30c649 100644
--- a/userspace/engine/token_bucket.cpp
+++ b/userspace/engine/token_bucket.cpp
@@ -73,3 +73,8 @@ bool token_bucket::claim(uint64_t now)
 
 	return true;
 }
+
+double token_bucket::get_tokens()
+{
+	return m_tokens;
+}
diff --git a/userspace/engine/token_bucket.h b/userspace/engine/token_bucket.h
index b60b5319..2f767ef4 100644
--- a/userspace/engine/token_bucket.h
+++ b/userspace/engine/token_bucket.h
@@ -38,6 +38,10 @@ public:
 	// internal metrics.
 	//
 	bool claim(uint64_t now = 0);
+
+	// Return the current number of tokens available
+	double get_tokens();
+
 private:
 
 	//