Skip to content

A Deno and Web compatible worker pool implementation

License

Notifications You must be signed in to change notification settings

vicary/deno-workerpool

Repository files navigation

Workerpool

An unopinionated small scale worker pool abstraction which serves as a base interface for more advanced worker managers.

Terminology

  1. Workerpool

    A manager that creates workers on the fly, executing tasks up to defined concurrency.

  2. Runner

    Runners are internal wrappers for user provided runner classes, they maintain internal states such as active/busy and the retry counter.

  3. Executable

    User implementation of task executors, where they all implements the Executable interface.

    Also called workers for the sake of naming convension, but the usage kept minimal to avoid confusion with Web Workers.

  4. Task

    Tasks are named payloads enqueued into a workerpool.

Basic Usage

You need to implement your own enqueue and dequeue logic, see an in-memory implementation in the examples section.

import { Executable, Workerpool } from "https://deno.land/x/workerpool/mod.ts";

class RunnerA implements Executable {...}
class RunnerB implements Executable {...}

const pool = new Workerpool({
  concurrency: 2,
  workers: [RunnerA, RunnerB]
  // enqueue() {...}
  // dequeue() {...}
});

pool
  .enqueue({ name: "RunnerA", payload: {...} })
  .enqueue({ name: "RunnerB", payload: {...} })
  .start();

Examples

In-memory Queue

As a proof of concept, this is a simple implementation of an in-memory queue.

type Payload = any;
type MemoryMutexTask = Task<Payload> & { active?: boolean };

const tasks = new Set<MemoryMutexTask>();
const pool = new Workerpool<Payload>({
  concurrency: 1,
  workers: [RunnerA, RunnerB],
  enqueue(task: MemoryMutexTask) {
    task.active = false;
    tasks.add(task);
  },
  dequeue() {
    // Uncomment the following line for FIFO queues
    // for (const { active } of task) if (active) return;

    for (const task of tasks) {
      if (!task.active) {
        task.active = true;
        return task;
      }
    }
  },
  onTaskFinished(error, result, { task }) {
    tasks.delete(task);

    if (error) {
      console.error(error);
    } else {
      console.log(result);
    }
  },
  onStateChange(state) {
    if (state === "drained") {
      pool.enqueue({...}); // Support immediate restarting of an idle queue.
    }
  }
});

Web Workers

Deno has built-in support for workers, our ExecutableWorker class serves as a simple proxy class via comlink.

import { ExecutableWorker } from "https://deno.land/x/workerpool/mod.ts";

class MyRunner extends ExecutableWorker<string, void> {
  constructor() {
    super(new URL("./worker.ts", import.meta.url).href);
  }
}

You'll also need a separated script file for the worker itself.

// worker.ts
import { initializeWorker } from "https://deno.land/x/workerpool/mod.ts";

initializeWorker({
  execute: async (payload: string) => {
    // Simulate async actions
    await new Promise((resolve) => setTimeout(resolve, 1000));

    return `Worker echo: ${payload}`;
  },
});

Now register the runner into the workerpool:

const pool = new Workerpool<string, void>({
  concurrency: 1,
  workers: [MyRunner],
});

pool
  .enqueue({ name: "MyRunner", payload: "Hello World!" })
  .enqueue({ name: "MyRunner", payload: "Hello Again!" })
  .start();

Sponsorship

If you appreciate my work, or want to see specific features to happen, a coffee would do.

About

A Deno and Web compatible worker pool implementation

Resources

License

Stars

Watchers

Forks

Sponsor this project

 

Packages

No packages published