bottleneck
Bottleneck is a tiny and efficient Task Scheduler and Rate Limiter for Node.JS and the browser. When dealing with services with limited resources, it's important to ensure that they don't become overloaded.
Bottleneck is the easiest solution as it doesn't add much complexity to the code.
It's battle-hardened, reliable and production-ready. It's used on a large scale in both private companies and open source software.
Install
Node
npm install bottleneck
Browser
bower install bottleneck
or
<script type="text/javascript" src="bottleneck.min.js"></script>
Example
Most APIs have a rate limit. For example, the Reddit.com API limits programs to 1 request every 2 seconds.
var Bottleneck = require("bottleneck");
var limiter = new Bottleneck(1, 2000);
Instead of doing
someAsyncCall(arg1, arg2, argN, callback);
You do
limiter.submit(someAsyncCall, arg1, arg2, argN, callback);
And now you can be assured that someAsyncCall will abide by your rate guidelines!
Promise users can use schedule()
.
Bottleneck builds a queue of requests and executes them as soon as possible. All the requests will be executed in the order that they were received. See priorities if you wish to alter this behavior.
This is sufficient for the vast majority of applications. Read the Gotchas section and you're good to go. Or keep reading to learn about all the fine tuning available for the more complex use cases.
Docs
Constructor
var limiter = new Bottleneck(maxConcurrent, minTime, highWater, strategy, rejectOnDrop);
maxConcurrent
: How many requests can be running at the same time. Default: 0
(unlimited)minTime
: How long to wait after launching a request before launching another one. Default: 0
mshighWater
: How long can the queue get? Default: -1
(unlimited)strategy
: Which strategy to use if the queue gets longer than the high water mark. Default: Bottleneck.strategy.LEAK
.rejectOnDrop
: When true
if a job is dropped its callback will be called with the first argument set to an Error
object. If the job was a promise it will be rejected. Default: false
submit()
Adds a request to the queue.
limiter.submit(someAsyncCall, arg1, arg2, argN, callback);
It returns true
if the strategy was executed.
schedule()
Adds a request to the queue. This is the Promise version of submit
. It uses the bluebird package if installed and otherwise uses the built-in Promise object.
var fn = function(arg1, arg2, argN) {
return httpGet(arg1, arg2, argN);
};
limiter.schedule(fn, arg1, arg2, argN);
In plain language, schedule
takes a function fn and a list of arguments. Fn must return a promise. schedule
returns a promise that will be executed according to the rate limits. It's safe to mix submit
and schedule
in the same limiter.
Here's another example, this time using the ECMAScript 7 syntax:
function fn (url) {
return http.get(url).then(response => console.log(response.body));
}
limiter.schedule(fn, url);
It's also possible to replace the Promise library used:
var Bottleneck = require("bottleneck");
Bottleneck.prototype.Promise = myPromiseLibrary;
var limiter = new Bottleneck(maxConcurrent, minTime, highWater, strategy, rejectOnDrop);
Gotchas
-
When using submit
, if a callback isn't necessary, you must pass null
or an empty function instead. It will not work if you forget to do this.
-
Make sure that all the requests will eventually complete by calling their callback (or resolving/rejecting in the case of promises). Again, even if you submit
ted your request with a null
callback , it still needs to call its callback. This is very important if you are using a maxConcurrent
value that isn't 0
(unlimited), otherwise those uncompleted requests will be clogging up the limiter and no new requests will be getting through. It's safe to call the callback more than once, subsequent calls are ignored.
-
If you want to rate limit a synchronous function (`console.log(), for example), you must wrap it in a closure to make it asynchronous. See this example.
Priorities
Every request has a priority level. It's 5
for every request added with submit
/schedule
. There exists a variant of those functions that lets you set the priority of a request.
Priority 0
is the most important and 9
is the least important.
More important requests will always be executed before less important ones. For requests with the same priority level, the oldest one is executed first.
submitPriority()
limiter.submitPriority(priority, someAsyncCall, arg1, arg2, argN, cb);
schedulePriority()
limiter.schedulePriority(priority, fn, arg1, arg2, argN);
Strategies
A strategy is a simple algorithm that is executed every time adding a request would cause the number of queued requests to exceed highWater
. See Events.
Bottleneck.strategy.LEAK
When submitting a new request, if the queue length reaches highWater
, drop the oldest request with the lowest priority. This is useful when requests that have been waiting for too long are not important anymore. If all the queued up requests are more important than the one being added, it won't be added.
Bottleneck.strategy.OVERFLOW_PRIORITY
Same as LEAK
, except that it will only drop requests that are less important than the one being added. If all the queued up requests are as important or more than the new one, it won't be added.
Bottleneck.strategy.OVERFLOW
When submitting a new request, if the queue length reaches highWater
, do not add the new request. This strategy totally ignores priority levels.
Bottleneck.strategy.BLOCK
When submitting a new request, if the queue length reaches highWater
, the limiter falls into "blocked mode". All queued requests are dropped and no new requests will be accepted until the limiter unblocks. It will unblock after penalty
milliseconds have passed without receiving a new request. penalty
is equal to 15 * minTime
(or 5000
if minTime
is 0
) by default and can be changed by calling changePenalty()
. This strategy is ideal when bruteforce attacks are to be expected. This strategy totally ignores priority levels.
nbQueued()
limiter.nbQueued(priority);
priority
is optional. Without that argument, it'll return the total number of requests waiting to be executed, otherwise it'll only count the number of requests with that specific priority.
nbRunning()
limiter.nbRunning();
Returns the number of requests currently running in the limiter.
check()
limiter.check();
If a request was added right now, would it be run immediately? Returns a boolean.
isBlocked()
limiter.isBlocked();
Is the limiter currently in "blocked mode"? Returns a boolean.
stopAll()
limiter.stopAll(interrupt);
Cancels all queued up requests and every added request will be automatically rejected.
interrupt
: If true, prevent the requests currently running from calling their callback when they're done. Default: false
Events
Event names: empty
, idle
, dropped
.
limiter.on('empty', function () {
})
limiter.on('idle', function () {
})
limiter.on('dropped', function (dropped) {
})
Use removeAllListeners()
with an optional event name as first argument to remove listeners.
Note: It's possible to set multiple callbacks to the same event name.
changeSettings()
limiter.changeSettings(maxConcurrent, minTime, highWater, strategy);
Same parameters as the constructor, pass null
to skip a parameter and keep it to its current value.
Note: Changing maxConcurrent
and minTime
will not affect requests that have already been scheduled for execution.
For example, imagine that three 60-second requests are submitted at time T with maxConcurrent = 0
and minTime = 2000
. The requests will be launched at T seconds, T+2 seconds and T+4 seconds respectively. If right after adding the requests to Bottleneck, you were to call limiter.changeSettings(1);
, it won't change the fact that there will be 3 requests running at the same time for roughly 60 seconds. Once again, changeSettings
only affects requests that have not yet been added.
This is by design, as Bottleneck made a promise to execute those requests according to the settings valid at the time. Changing settings afterwards should not break previous assumptions, as that would make code very error-prone and Bottleneck a tool that cannot be relied upon.
changePenalty()
limiter.changePenalty(penalty);
This changes the penalty
value used by the BLOCK
strategy.
changeReservoir(), incrementReservoir()
limiter.changeReservoir(reservoir);
limiter.incrementReservoir(incrementBy);
reservoir
: How many requests can be executed before the limiter stops executing requests. Default: null
(unlimited)
If reservoir
reaches 0
, no new requests will be executed until it is no more 0
chain()
limiter
: If another limiter is passed, tasks that are ready to be executed will be added to that other limiter. Default: null
(none)
Suppose you have 2 types of tasks, A and B. They both have their own limiter with their own settings, but both must also follow a global limiter C:
var limiterA = new Bottleneck(...some settings...);
var limiterB = new Bottleneck(...some different settings...);
var limiterC = new Bottleneck(...some global settings...);
limiterA.chain(limiterC);
limiterB.chain(limiterC);
Execution guarantee
Bottleneck will execute every request in order of priority first, oldest to youngest within each priority level. You can be certain that they will all eventually be executed as long as:
highWater
is set to -1
(default), which prevents the strategy from ever being run OR you never exceed the highWater
.maxConcurrent
is set to 0
(default) OR all requests call the callback eventually (in the case of promises, they must be resolved or rejected eventually).reservoir
is null
(default).
Cluster
The Cluster
feature of Bottleneck manages many limiters automatically for you. It creates limiters dynamically and transparently.
Let's take a DNS server as an example of how Bottleneck can be used. It's a service that sees a lot of abuse and where incoming DNS requests need to be rate limited. Bottleneck is so tiny, it's perfectly fine to create one limiter for each origin IP, even if it means creating thousands and thousands of limiters. The Cluster
mode is perfect for this use case. We can create one cluster and then use the origin IP to rate limit each IP independently. Each call with the same key will be routed to the same underlying limiter. A cluster is created exactly like a limiter:
var cluster = new Bottleneck.Cluster(maxConcurrent, minTime, highWater, strategy);
The cluster is then used with the .key(str)
method:
cluster.key("77.66.54.32").submit(someAsyncCall, arg1, arg2, cb);
key()
str
: The key to use. All calls submitted with the same key will use the same limiter. Default: ""
The return value of .key(str)
is a limiter. If it doesn't already exist, it is created on the fly. Limiters that have been idle for a long time are deleted to avoid memory leaks.
stopAutoCleanup()
Calling stopAutoCleanup()
on a cluster will turn off its garbage collection, so limiters for keys that have not been used in over 5 minutes will NOT be deleted anymore. It can be reenabled by calling startAutoCleanup()
. The 5 minutes
figure can be modified by calling changeTimeout()
.
startAutoCleanup()
Reactivate the cluster's garbage collection for limiters (in the cluster) that have been inactive for over 5 minutes.
changeTimeout()
timeout
: The expiration time for unused limiters, in milliseconds. By default it is 300000
(5 minutes).
When autocleanup is enabled, limiters having not been used in the last timeout
milliseconds will be deleted to avoid memory leaks.
deleteKey()
str
: The key for the limiter to delete.
Manually deletes the limiter at the specified key. This can be useful when the auto cleanup is turned off.
all()
cb
: A function to be executed on every limiter in the cluster.
For example, this will call stopAll()
on every limiter in the cluster:
cluster.all(function(limiter){
limiter.stopAll();
});
keys()
Returns an array containing all the keys in the cluster.
Rate-limiting synchronous functions
Most of the time, using Bottleneck is as simple as the first example above. However, when Bottleneck is used on a synchronous call, it (obviously) becomes asynchronous, so the returned value of that call can't be used directly. The following example should make it clear why.
This is the original code that we want to rate-limit:
var req = http.request(options, function(res){
});
req.write("some string", "utf8");
req.end();
The following code snippet will NOT work, because http.request
is not executed synchronously therefore req
doesn't contain the expected request object.
var req = limiter.submit(http.request, options, function(res){
});
req.write("some string", "utf8");
req.end();
This is the right way to do it:
limiter.submit(function(cb){
var req = http.request(options, function(res){
cb();
});
req.write("some string", "utf8");
req.end();
}, null);
Contributing
This README file is always in need of better explanations and examples. If things can be clearer and simpler, please consider forking this repo and submitting a Pull Request, or simply open an issue.
Suggestions and bug reports are also welcome.