dahlia-concurrency/src/data-structures/concurrent-queue.js

High-level concurrency primitives and patterns for Node.js using worker_threads (queues, semaphore, mutex, worker pool, scheduler, and more)

const { RwLock } = require("../primitives/rw-lock");

class ListNode {
  constructor(element) {
    this.element = element;
    this.next = null;
  }
}

class ConcurrentQueue {
  #length = 0;
  #head = null;
  #tail = null;

  constructor() {
    this.readerBuffer = new SharedArrayBuffer(4);
    this.writerBuffer = new SharedArrayBuffer(4);
    this.rwLock = new RwLock(this.readerBuffer, this.writerBuffer, true);
  }

  enqueue(item) {
    const node = new ListNode(item);
    this.rwLock.writeLock();
    try {
      if (this.#tail) {
        this.#tail.next = node;
      } else {
        this.#head = node;
      }
      this.#tail = node;
      this.#length++;
    } catch (e) {
      throw e;
    } finally {
      this.rwLock.writeUnlock();
    }
  }

  dequeue() {
    this.rwLock.writeLock();
    try {
      if (this.#length === 0) return null;
      const item = this.#head.element;
      this.#head = this.#head.next;
      if (!this.#head) this.#tail = null;
      this.#length--;
      return item;
    } catch (e) {
      throw e;
    } finally {
      this.rwLock.writeUnlock();
    }
  }

  peek() {
    this.rwLock.readLock();
    try {
      return this.#length > 0 ? this.#head.element : null;
    } catch (e) {
      throw e;
    } finally {
      this.rwLock.readUnlock();
    }
  }

  get size() {
    this.rwLock.readLock();
    try {
      return this.#length;
    } catch (e) {
      throw e;
    } finally {
      this.rwLock.readUnlock();
    }
  }

  isEmpty() {
    this.rwLock.readLock();
    try {
      return this.#length === 0;
    } catch (e) {
      throw e;
    } finally {
      this.rwLock.readUnlock();
    }
  }
}

module.exports = { ConcurrentQueue };