summaryrefslogtreecommitdiff
path: root/lib/internal/TokenBucket.js
diff options
context:
space:
mode:
Diffstat (limited to 'lib/internal/TokenBucket.js')
-rw-r--r--lib/internal/TokenBucket.js7
1 files changed, 6 insertions, 1 deletions
diff --git a/lib/internal/TokenBucket.js b/lib/internal/TokenBucket.js
index 011d13e..d31ed9c 100644
--- a/lib/internal/TokenBucket.js
+++ b/lib/internal/TokenBucket.js
@@ -30,9 +30,14 @@ TokenBucket.prototype.enqueue = function(operation) {
}
function onIntervalComplete(bucket) {
- bucket.queued--;
+ if (bucket.queued > 0) bucket.queued--;
if (bucket.queue.length > 0) {
// call first queued operation
(bucket.queue.shift())();
}
}
+
+TokenBucket.prototype.empty = function() {
+ this.queue.empty();
+ this.queued = 0;
+}