@nxtedition/scheduler 1.0.4 → 1.0.6

package/lib/index.d.ts

@@ -4,10 +4,9 @@ export declare class Scheduler {
     static NORMAL: 1;
     static HIGH: 2;
     get concurrency(): number;
-    static makeState(concurrency?: number): SharedArrayBuffer;
-    constructor({ concurrency, stateBuffer }?: {
-        concurrency?: number | undefined;
-        stateBuffer?: SharedArrayBuffer | undefined;
+    static makeSharedState(concurrency?: number): SharedArrayBuffer;
+    constructor(opts?: SharedArrayBuffer | {
+        concurrency?: number;
     });
-    schedule(fn: (next: Function) => any, priority?: 1): any;
+    schedule(fn: (next: Function) => any, priority?: 0 | 1 | 2 | "low" | "normal" | "high"): any;
 }

package/lib/index.js

@@ -14,6 +14,7 @@ export class Scheduler {
 
   #running = 0
   #pending = 0
+  #counter = 0
 
   #lowQueue = new FixedQueue()
   #normalQueue = new FixedQueue()
@@ -23,7 +24,7 @@ export class Scheduler {
     return this.#concurrency
   }
 
-  static makeState(concurrency         = 0) {
+  static makeSharedState(concurrency         = 0) {
     if (concurrency < 0 || !Number.isInteger(concurrency)) {
       throw new Error('Invalid concurrency')
     }
@@ -33,31 +34,44 @@ export class Scheduler {
     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)
+  constructor(opts                                               = {}) {
+    if (opts instanceof SharedArrayBuffer) {
+      this.#stateView = new Uint32Array(opts)
+      this.#concurrency = Atomics.load(this.#stateView, Scheduler.#CONCURRENCY_INDEX)
+    } else {
+      this.#concurrency = opts?.concurrency ?? 0
+    }
   }
 
-  schedule(fn                         , priority = Scheduler.NORMAL) {
+  schedule(fn                         , priority                                        = Scheduler.NORMAL) {
     if (typeof fn !== 'function') {
       throw new TypeError('First argument must be a function')
     }
 
     if (priority == null) {
       priority = Scheduler.NORMAL
-    }
-
-    if (!Number.isInteger(priority)) {
+    } else if (typeof priority === 'string') {
+      if (priority === 'low') {
+        priority = Scheduler.LOW
+      } else if (priority === 'normal') {
+        priority = Scheduler.NORMAL
+      } else if (priority === 'high') {
+        priority = Scheduler.HIGH
+      } else {
+        throw new Error('Invalid priority')
+      }
+    } else if (!Number.isInteger(priority)) {
       throw new Error('Invalid priority')
     }
 
-    if (
-      this.#running < 1 ||
-      Atomics.load(this.#stateView, Scheduler.#RUNNING_INDEX) < this.#concurrency
-    ) {
-      Atomics.add(this.#stateView, Scheduler.#RUNNING_INDEX, 1)
+    const running = this.#stateView
+      ? Atomics.load(this.#stateView, Scheduler.#RUNNING_INDEX)
+      : this.#running
+
+    if (this.#running < 1 || running < this.#concurrency) {
+      if (this.#stateView) {
+        Atomics.add(this.#stateView, Scheduler.#RUNNING_INDEX, 1)
+      }
       this.#running += 1
       return fn(this.#next)
     }
@@ -73,11 +87,15 @@ export class Scheduler {
   }
 
   #next = () => {
-    let running = Atomics.sub(this.#stateView, Scheduler.#RUNNING_INDEX, 1) - 1
+    let running = this.#stateView
+      ? Atomics.sub(this.#stateView, Scheduler.#RUNNING_INDEX, 1) - 1
+      : this.#running - 1
     this.#running -= 1
 
     while (this.#pending > 0 && (this.#running < 1 || running < this.#concurrency)) {
-      const counter = Atomics.add(this.#stateView, Scheduler.#COUNTER_INDEX, 1)
+      const counter = this.#stateView
+        ? Atomics.add(this.#stateView, Scheduler.#COUNTER_INDEX, 1)
+        : this.#counter++
       const fn =
         ((counter & 63) === 0 && this.#lowQueue.shift()) ??
         ((counter & 15) === 0 && this.#normalQueue.shift()) ??
@@ -89,7 +107,9 @@ export class Scheduler {
         break
       }
 
-      running = Atomics.add(this.#stateView, Scheduler.#RUNNING_INDEX, 1) + 1
+      running = this.#stateView
+        ? Atomics.add(this.#stateView, Scheduler.#RUNNING_INDEX, 1) + 1
+        : this.#running + 1
       this.#running += 1
       this.#pending -= 1
       fn(this.#next)

package/package.json

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