@nxtedition/scheduler 1.0.1 → 1.0.2

package/lib/index.d.ts

@@ -3,8 +3,10 @@ export declare class Scheduler {
     static LOW: 0;
     static NORMAL: 1;
     static HIGH: 2;
-    constructor({ concurrency }?: {
+    static makeState(concurrency?: number): SharedArrayBuffer;
+    constructor({ concurrency, stateBuffer }?: {
         concurrency?: number | undefined;
+        stateBuffer?: SharedArrayBuffer | undefined;
     });
-    schedule(fn: ((next: Function) => any), priority?: 1): any;
+    schedule(fn: (next: Function) => any, priority?: 1): any;
 }

package/lib/index.js

@@ -6,18 +6,34 @@ export class Scheduler {
   static HIGH    = 2
 
   #concurrency
+  #stateView
+
+  static #RUNNING_INDEX = 0
+  static #COUNTER_INDEX = 1
+  static #CONCURRENCY_INDEX = 2
+
   #running = 0
-  #counter = 0
+  #pending = 0
 
   #lowQueue = new FixedQueue()
   #normalQueue = new FixedQueue()
   #highQueue = new FixedQueue()
 
-  constructor({ concurrency = Infinity } = {}) {
-    this.#concurrency = concurrency
+  static makeState(concurrency         = 0) {
+    const stateBuffer = new SharedArrayBuffer(64)
+    const stateView = new Uint32Array(stateBuffer)
+    Atomics.store(stateView, Scheduler.#CONCURRENCY_INDEX, concurrency ?? 0)
+    return stateBuffer
+  }
+
+  constructor({ concurrency = Infinity, stateBuffer = Scheduler.makeState(concurrency) } = {}) {
+    this.#stateView = new Uint32Array(stateBuffer)
+    this.#concurrency =
+      concurrency ??
+      (Atomics.load(new Uint32Array(stateBuffer), Scheduler.#CONCURRENCY_INDEX) || Infinity)
   }
 
-  schedule(fn                           , priority = Scheduler.NORMAL) {
+  schedule(fn                         , priority = Scheduler.NORMAL) {
     if (typeof fn !== 'function') {
       throw new TypeError('First argument must be a function')
     }
@@ -30,8 +46,12 @@ export class Scheduler {
       throw new Error('Invalid priority')
     }
 
-    if (this.#running < this.#concurrency) {
-      this.#running++
+    if (
+      this.#running > 1 &&
+      Atomics.load(this.#stateView, Scheduler.#RUNNING_INDEX) < this.#concurrency
+    ) {
+      Atomics.add(this.#stateView, Scheduler.#RUNNING_INDEX, 1)
+      this.#running += 1
       return fn(this.#next)
     }
 
@@ -42,21 +62,33 @@ export class Scheduler {
     } else {
       this.#normalQueue.push(fn)
     }
+    this.#pending += 1
   }
 
   #next = () => {
-    this.#counter++
-    this.#running--
-
-    const fn =
-      ((this.#counter & 63) === 0 && this.#lowQueue.shift()) ??
-      ((this.#counter & 15) === 0 && this.#normalQueue.shift()) ??
-      this.#highQueue.shift() ??
-      this.#normalQueue.shift() ??
-      this.#lowQueue.shift()
-
-    if (fn) {
-      this.#running++
+    Atomics.sub(this.#stateView, Scheduler.#RUNNING_INDEX, 1)
+    this.#running -= 1
+
+    while (
+      this.#pending > 0 &&
+      (this.#running < 1 ||
+        Atomics.load(this.#stateView, Scheduler.#RUNNING_INDEX) < this.#concurrency)
+    ) {
+      const counter = Atomics.add(this.#stateView, Scheduler.#COUNTER_INDEX, 1)
+      const fn =
+        ((counter & 63) === 0 && this.#lowQueue.shift()) ??
+        ((counter & 15) === 0 && this.#normalQueue.shift()) ??
+        this.#highQueue.shift() ??
+        this.#normalQueue.shift() ??
+        this.#lowQueue.shift()
+
+      if (!fn) {
+        break
+      }
+
+      Atomics.add(this.#stateView, Scheduler.#RUNNING_INDEX, 1)
+      this.#running += 1
+      this.#pending -= 1
       fn(this.#next)
     }
   }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nxtedition/scheduler",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "type": "module",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
@@ -20,5 +20,5 @@
     "rimraf": "^6.1.2",
     "typescript": "^5.9.3"
   },
-  "gitHead": "5e9be5511fdd2e10aecbab7cc007bdaf990f728a"
+  "gitHead": "a8531a2fd6f9e627ebb81eaaf9ea9e4bb8f85145"
 }