diff options
Diffstat (limited to 'lib/internal/TokenBucket.js')
-rw-r--r-- | lib/internal/TokenBucket.js | 7 |
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; +} |