@lostgradient/weft 0.2.0 → 0.3.0

package/dist/worker/registry.js

@@ -1,5 +1,10 @@
-import { compareScores, scoreWorker } from "./registry/fair-share.js";
-import { matchesWorkerCapabilities, pickLeastLoaded } from "./registry/routing.js";
+import { FairShareCounters } from "./registry/fair-share.js";
+import {
+  matchesWorkerCapabilities,
+  pickFairShare,
+  pickLeastLoaded,
+  pickRoundRobin
+} from "./registry/routing.js";
 import {
   deploymentHealth,
   projectDeploymentSummaries,
@@ -19,7 +24,7 @@ export class WorkerRegistry {
     this.#deploymentDrainStates = new Map;
     this.#policy = options?.policy ?? "least-loaded";
     this.#roundRobinCursor = new Map;
-    this.#fairShareCounts = new Map;
+    this.#fairShareCounts = new FairShareCounters;
   }
   get policy() {
     return this.#policy;
@@ -51,7 +56,7 @@ export class WorkerRegistry {
     if (info === void 0)
       return;
     this.#workers.delete(workerId);
-    this.#fairShareCounts.delete(workerId);
+    this.#fairShareCounts.purge(workerId);
     for (const [operationId, task] of this.#inFlightTasks)
       if (task.workerId === workerId)
         this.#inFlightTasks.delete(operationId);
@@ -99,39 +104,23 @@ export class WorkerRegistry {
   }
   #selectByPolicy(eligible, queue, activityName, fairShareKey) {
     if (this.#policy === "round-robin")
-      return this.#pickRoundRobin(eligible, queue, activityName);
+      return pickRoundRobin(eligible, this.#roundRobinCursor, queue, activityName);
     if (this.#policy === "fair-share" && fairShareKey !== void 0)
-      return this.#pickFairShare(eligible, fairShareKey);
+      return pickFairShare(eligible, this.#fairShareCounts, fairShareKey);
     return pickLeastLoaded(eligible);
   }
-  #pickRoundRobin(eligible, queue, activityName) {
-    const key = `${queue ?? "__default__"}::${activityName}`, cursor = this.#roundRobinCursor.get(key) ?? 0, pick = eligible[cursor % eligible.length];
-    this.#roundRobinCursor.set(key, cursor + 1);
-    return pick;
-  }
-  #pickFairShare(eligible, fairShareKey) {
-    const winner = eligible.map((worker) => scoreWorker({
-      id: worker.id,
-      inFlight: worker.inFlight,
-      keyLoad: this.#fairShareCounts.get(worker.id)?.get(fairShareKey) ?? 0
-    })).reduce((best, candidate) => compareScores(candidate, best) < 0 ? candidate : best);
-    return this.#workers.get(winner.id);
-  }
-  assignTask(workerId, operationId, visibilityTimeout, fairShareKey) {
+  assignTask(workerId, operationId, visibilityTimeout, fairShareKey, attemptToken) {
     const deadline = Date.now() + visibilityTimeout, task = {
       operationId,
       workerId,
       deadline,
       visibilityTimeout
     };
+    if (attemptToken !== void 0)
+      task.attemptToken = attemptToken;
     if (fairShareKey !== void 0) {
       task.fairShareKey = fairShareKey;
-      let workerCounts = this.#fairShareCounts.get(workerId);
-      if (workerCounts === void 0) {
-        workerCounts = new Map;
-        this.#fairShareCounts.set(workerId, workerCounts);
-      }
-      workerCounts.set(fairShareKey, (workerCounts.get(fairShareKey) ?? 0) + 1);
+      this.#fairShareCounts.increment(workerId, fairShareKey);
     }
     this.#inFlightTasks.set(operationId, task);
     this.taskAssigned(workerId);
@@ -164,6 +153,14 @@ export class WorkerRegistry {
   isAssignedToWorker(operationId, workerId) {
     return this.#inFlightTasks.get(operationId)?.workerId === workerId;
   }
+  isAssignedToAttempt(operationId, workerId, attemptToken) {
+    const task = this.#inFlightTasks.get(operationId);
+    if (task === void 0 || task.workerId !== workerId)
+      return !1;
+    if (task.attemptToken === void 0 || attemptToken === void 0)
+      return !0;
+    return task.attemptToken === attemptToken;
+  }
   isAssigned(operationId) {
     return this.#inFlightTasks.has(operationId);
   }
@@ -285,16 +282,7 @@ export class WorkerRegistry {
   #releaseFairShare(task) {
     if (task.fairShareKey === void 0)
       return;
-    const workerCounts = this.#fairShareCounts.get(task.workerId);
-    if (workerCounts === void 0)
-      return;
-    const current = workerCounts.get(task.fairShareKey) ?? 0, next = Math.max(0, current - 1);
-    if (next === 0) {
-      workerCounts.delete(task.fairShareKey);
-      if (workerCounts.size === 0)
-        this.#fairShareCounts.delete(task.workerId);
-    } else
-      workerCounts.set(task.fairShareKey, next);
+    this.#fairShareCounts.release(task.workerId, task.fairShareKey);
   }
 }
 function createDrainRecord(options) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lostgradient/weft",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "A Bun-native durable execution engine with pluggable key-value storage.",
   "keywords": [
     "bun",
@@ -107,6 +107,12 @@
       "import": "./dist/storage/turso.js",
       "default": "./dist/storage/turso.js"
     },
+    "./storage/neon": {
+      "types": "./dist/storage/neon.d.ts",
+      "bun": "./dist/storage/neon.js",
+      "import": "./dist/storage/neon.js",
+      "default": "./dist/storage/neon.js"
+    },
     "./storage/indexeddb": {
       "types": "./dist/storage/indexeddb.d.ts",
       "import": "./dist/storage/indexeddb.js",
@@ -202,18 +208,19 @@
     "lint": "oxlint --type-aware --tsconfig ./tsconfig.json --report-unused-disable-directives-severity=error && bun run scripts/check-lint-disables.ts && bun run scripts/check-import-cycles.ts && bun run scripts/check-internal-imports.ts && bun run scripts/check-engine-internals-field-access.ts && bun run scripts/check-definition-vocabulary.ts && bun run scripts/check-catalog-completeness.ts && bun run scripts/check-catalog-drift.ts && bun run scripts/check-type-ergonomics.ts",
     "lint:fix": "oxlint --fix --type-aware --tsconfig ./tsconfig.json",
     "markdown-doctests:inventory": "bun run scripts/extract-markdown-doctests.ts inventory",
-    "prepack": "bun run build && bun run verify:exports && bun run verify:portability && bun run verify:markdown-doctests && bun run verify:jsdoc && bun run verify:jsdoc:doctests && bun run verify:jsdoc:declarations && bun run check:package-contents && bun run validate:package-consumers",
+    "prepack": "bun run scripts/run-gates.ts prepack",
     "pr:title": "bun run scripts/pr-title.ts",
     "prepare": "husky",
     "prepublishOnly": "bun run validate",
     "start": "bun run src/index.ts",
-    "test": "bun test --timeout 15000 --path-ignore-patterns='src/benchmarks/**' --path-ignore-patterns='examples/checkout/**' --path-ignore-patterns='src/service-worker/service-worker-browser.test.ts' --path-ignore-patterns='src/storage/indexeddb-browser.test.ts' --path-ignore-patterns='src/storage/web-extension-browser.test.ts' && bun test src/benchmarks --timeout 15000 --parallel=1",
+    "test": "bun test --timeout 15000 --path-ignore-patterns='src/benchmarks/**' --path-ignore-patterns='examples/checkout/**' --path-ignore-patterns='src/service-worker/service-worker-browser.test.ts' --path-ignore-patterns='src/storage/indexeddb-browser.test.ts' --path-ignore-patterns='src/storage/web-extension-browser.test.ts'",
+    "test:benchmarks": "bun test src/benchmarks --timeout 15000 --parallel=1",
     "test:browser-smoke": "WEFT_BROWSER_SMOKE=1 bun test --timeout 20000 src/service-worker/service-worker-browser.test.ts src/storage/indexeddb-browser.test.ts src/storage/web-extension-browser.test.ts",
     "test:coverage": "bun run scripts/check-coverage.ts",
     "typecheck": "tsc --noEmit",
     "typecheck:package-root": "bun run build && tsc --noEmit -p tsconfig.package-test-d.json",
     "typecheck:tests": "tsc --noEmit -p tsconfig.test-d.json && bun run typecheck:package-root",
-    "validate": "bun run lint && bun run typecheck && bun run typecheck:tests && bun run verify:documentation && bun run verify:no-test-sleeps && bun run verify:public-api-jsdoc && bun run test",
+    "validate": "bun run scripts/run-gates.ts validate",
     "validate:package-consumers": "bun run scripts/validate-package-consumers.ts",
     "verify:documentation": "bun run scripts/verify-documentation.ts",
     "verify:exports": "bun run scripts/verify-tree-shaking.ts",
@@ -248,7 +255,9 @@
     "zod": "^4.3.6"
   },
   "devDependencies": {
+    "@electric-sql/pglite": "^0.5.1",
     "@libsql/client": "^0.17.2",
+    "@neondatabase/serverless": "^1.1.0",
     "@types/bun": "^1.3.13",
     "better-sqlite3": "^12.8.0",
     "chalk": "^5.6.2",
@@ -267,6 +276,7 @@
   },
   "peerDependencies": {
     "@libsql/client": "^0.17.2",
+    "@neondatabase/serverless": "^1.1.0",
     "@opentelemetry/api": "^1.0.0",
     "better-sqlite3": "^12.8.0",
     "lmdb": "^3.5.2"
@@ -275,6 +285,9 @@
     "@libsql/client": {
       "optional": true
     },
+    "@neondatabase/serverless": {
+      "optional": true
+    },
     "@opentelemetry/api": {
       "optional": true
     },