Skip to content

compwright/batching-queue

Repository files navigation

batching-queue

Download Status Sponsor on GitHub

A batching queue for items that need to be enqueued one by one, but dequeued in batches

  • Works with multiple storage backends
  • Queues items one by one
  • Dequeues in batches (FIFO)
  • Tracks the number of waiting items
  • Emits an event when the number of waiting items reaches the batch size threshold

Installation

npm install batching-queue

Included Storage Backends

  • MemoryStore - stores items in an array (not for production use)
  • AsyncMemoryStore - same as MemoryStore, but with async methods
  • RedisStore - stores items in Redis

Example Usage

import { BatchingQueue, MemoryStore } from 'batching-queue'

const queue = new BatchingQueue({
  store: new MemoryStore(),
  batchSize: 12
})

const drain = async (batchesWaiting) => {
  const result = await batchesWaiting
  console.log(result, 'batches waiting')

  for (let i = 0; i < result; i++) {
    const batch = await queue.dequeue()
    console.log(batch, 'batch')
  }
}

(async () => {
  // drain remaining batches immediately
  drain(queue.length)

  // drain batches every 12 items
  queue.on('drain', drain)

  for (var i = 0; i < 1000; i++) {
    await queue.enqueue(i)
  }
})()

API

BatchingQueue

constructor(config)

Config:

  • store (require) Storage backend for the queue (see src/stores/memory.js for a reference implementation)
  • batchSize (required) Batch size

enqueue(item)

Adds an item to the queue. Returns true if a batch is full, otherwise returns false.

dequeue()

Dequeues the first [batchSize] number of items. If there are not enough items to satisfy batchSize, all the items available will be returned.

length

Property reports the number of batches waiting in the queue.

Note: if the storage backend is not initialized, this will return null.

Storage Backend Interface

You may to implement your own storage backend to interface your database or cache. See the bundled MemoryStore or AsyncMemoryStore for a reference.

setup()

Initialize and set up the storage backend. Establish connections, create database collections, etc.

This will be called before calling enqueue() or dequeue() when ready === false.

destroy()

Destroy the storage backend (optional).

async enqueue(item)

Store an item in the queue.

The store must return the exact number of items in the store subsequent to the storage operation.

async dequeue(batchSize)

Retrieve and delete [batchSize] items from the storage backend.

Return the set of items retrieved.

length

Return the number of items in the store.

ready

Return true if ready and setup, otherwise, return false.

RedisStore

The RedisStore class has the following additional methods:

constructor(config)

Config:

  • redisClient (required) Instance of node-redis
  • name Queue list key name, will be randomized if omitted

IoredisStore

The RedisStore class has the following additional methods:

constructor(config)

Config:

  • redisClient (required) Instance of ioredis
  • name Queue list key name, will be randomized if omitted

License

MIT License