Miss any of our Open RFC calls?Watch the recordings here! »

ioredis-quota

0.7.0 • Public • Published

Build Status NPM Version Dependency Status

ioredis-quota

General-purpose quota management.

This NPM package is a simple quota manager with a simple, promisified API. It's open-source and it's written with TypeScript.

The source code is available on GitHub where you can also find our issue tracker.

Install

The package depends on ioredis but it should also work with any other Redis library that supports promises.

$ npm install --save ioredis ioredis-quota

Getting Started

To make the code clean, the examples are written in TypeScript.

You start by some initialization code.

import Redis from "ioredis";
import { Quota } from "ioredis-quota";
 
const quota = new Quota({
  redis: new Redis(),
  rates: [
    { key: "github-api", unit: "minute", limit: 10 }, // Allow up to 10 requests per minute.
    { key: "github-api", unit: "hour", limit: 100 },  // Allow up to 100 requests per hour.
  ]
});

Use the grant() method to verify that the request you are making does not exceed the rate limit.

try {
  await quota.grant(); // Grant 10 requests per minute and 100 requests per hour.
  // We have not exceeded the minutely nor daily quota so we can execute a request.
} catch (e) {
  // We reached the limits. Use `e.nextDate` to handle a retry.
}

Or use the schedule() method which uses the grant() method and returns the next appropriate date for execution (e.nextDate) instead of throwing an error.

const nextDate = await quota.schedule(); // => Sat Jun 17 14:58:57 CEST 2017

API

Quota({ redis, prefix, rates })

A core class which is used for checking quota.

Option Type Required Default Description
redis Object Yes - Redis class instance.
prefix String No quota A string which prefix all the keys.
rates Object[] No [] List of quota definitions.
rates.key String Yes - Quota unique name.
rates.unit String Yes - Quota unit (second, minute, hour, day, week, month, quarter or year).
rates.limit Integer Yes - The maximum value of the increment.

QuotaError(nextDate, message)

Quota error class which is thrown when the grant method does not succeed.

Option Type Required Default Description
nextDate Date Yes - A moment when quota is reset.
message String No Quota limit exceeded. Error message.

quota.buildIdentifier({ key, unit }): String

Builds and returns the final Redis key.

Option Type Required Default Description
key String Yes - Quota key name.
unit String Yes - Quota unit (second, minute, hour, day, week, month, quarter or year).

quota.flush([{ key, unit }]): Promise

Atomically removes quota. Note that if no attributes are specified then all identifiers are deleted.

Option Type Required Default Description
key String Yes - Quota key name.
unit String Yes - Quota unit (second, minute, hour, day, week, month, quarter or year).

quota.grant(rates): Promise

Atomically verifies quota for each rate and throws the QuotaError if the rate's value exceeds the specified limit attribute. Please note that this method will increment the counter for each defined rate by one per unit (if you specify two rate objects with the same unit this means that this unit will be incremented by 2).

Option Type Required Default Description
rates Object, Object[] No [] List of quota definitions (appends class rates).
rates.key String Yes - Quota unique name.
rates.unit String Yes - Quota unit (second, minute, hour, day, week, month, quarter or year).
rates.limit Integer Yes - The maximum value of the increment.

quota.parseIdentifier(identifier): String

Parses the identifier string and returns key's data (prefix, timestamp and key).

Option Type Required Default Description
identifier String Yes - Redis key.

quota.schedule(rates): Promise

Atomically verifies quota for each rate and returns the next available date.

Option Type Required Default Description
rates Object, Object[] No [] List of quota definitions (appends class rates).
rates.key String Yes - Quota unique name.
rates.unit String Yes - Quota unit (second, minute, hour, day, week, month, quarter or year).
rates.limit Integer Yes - The maximum value of the increment.

License (MIT)

Copyright (c) 2016 Kristijan Sedlak <xpepermint@gmail.com>

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.

Install

npm i ioredis-quota

DownloadsWeekly Downloads

14

Version

0.7.0

License

MIT

Last publish

Collaborators

  • avatar