nestworker 2.0.8 → 2.1.0

package/dist/health/worker.health.d.ts

@@ -0,0 +1,46 @@
+import { WorkerService } from '../core/worker.service';
+export interface WorkerHealthResult {
+    status: 'up' | 'down';
+    details: {
+        poolSize: number;
+        idle: number;
+        busy: number;
+        queued: number;
+        warmingUp: number;
+    };
+    error?: string;
+}
+/**
+ * WorkerHealthIndicator — plugs into NestJS Terminus.
+ *
+ * @example
+ *   // health.controller.ts
+ *   import { HealthCheck, HealthCheckService } from '@nestjs/terminus';
+ *   import { WorkerHealthIndicator } from 'nestworker';
+ *
+ *   @Controller('health')
+ *   export class HealthController {
+ *     constructor(
+ *       private health: HealthCheckService,
+ *       private workerHealth: WorkerHealthIndicator,
+ *     ) {}
+ *
+ *     @Get()
+ *     @HealthCheck()
+ *     check() {
+ *       return this.health.check([
+ *         () => this.workerHealth.check('workers'),
+ *       ]);
+ *     }
+ *   }
+ */
+export declare class WorkerHealthIndicator {
+    private readonly workerService;
+    constructor(workerService: WorkerService);
+    /**
+     * Returns a Terminus-compatible health indicator result.
+     * Reports 'down' when warmingUp > 0 (pool not fully ready) or when
+     * queued jobs exceed the pool size (backpressure signal).
+     */
+    check(key: string): Record<string, WorkerHealthResult>;
+}

package/dist/health/worker.health.js

@@ -0,0 +1,77 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkerHealthIndicator = void 0;
+const common_1 = require("@nestjs/common");
+const worker_service_1 = require("../core/worker.service");
+/**
+ * WorkerHealthIndicator — plugs into NestJS Terminus.
+ *
+ * @example
+ *   // health.controller.ts
+ *   import { HealthCheck, HealthCheckService } from '@nestjs/terminus';
+ *   import { WorkerHealthIndicator } from 'nestworker';
+ *
+ *   @Controller('health')
+ *   export class HealthController {
+ *     constructor(
+ *       private health: HealthCheckService,
+ *       private workerHealth: WorkerHealthIndicator,
+ *     ) {}
+ *
+ *     @Get()
+ *     @HealthCheck()
+ *     check() {
+ *       return this.health.check([
+ *         () => this.workerHealth.check('workers'),
+ *       ]);
+ *     }
+ *   }
+ */
+let WorkerHealthIndicator = class WorkerHealthIndicator {
+    workerService;
+    constructor(workerService) {
+        this.workerService = workerService;
+    }
+    /**
+     * Returns a Terminus-compatible health indicator result.
+     * Reports 'down' when warmingUp > 0 (pool not fully ready) or when
+     * queued jobs exceed the pool size (backpressure signal).
+     */
+    check(key) {
+        const stats = this.workerService.stats();
+        const isDown = stats.warmingUp > 0 ||
+            stats.queued > stats.poolSize;
+        const result = {
+            status: isDown ? 'down' : 'up',
+            details: stats,
+            ...(isDown && stats.warmingUp > 0
+                ? { error: `${stats.warmingUp} worker(s) still warming up` }
+                : {}),
+            ...(isDown && stats.queued > stats.poolSize
+                ? { error: `Queue depth (${stats.queued}) exceeds pool size (${stats.poolSize})` }
+                : {}),
+        };
+        if (isDown) {
+            throw Object.assign(new Error(`Worker pool unhealthy`), {
+                [key]: result,
+                causes: [result.error],
+            });
+        }
+        return { [key]: result };
+    }
+};
+exports.WorkerHealthIndicator = WorkerHealthIndicator;
+exports.WorkerHealthIndicator = WorkerHealthIndicator = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [worker_service_1.WorkerService])
+], WorkerHealthIndicator);
+//# sourceMappingURL=worker.health.js.map

package/dist/health/worker.health.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.health.js","sourceRoot":"","sources":["../../src/health/worker.health.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,2DAAuD;AAcvD;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AAEI,IAAM,qBAAqB,GAA3B,MAAM,qBAAqB;IACH;IAA7B,YAA6B,aAA4B;QAA5B,kBAAa,GAAb,aAAa,CAAe;IAAG,CAAC;IAE7D;;;;OAIG;IACH,KAAK,CAAC,GAAW;QACf,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QACzC,MAAM,MAAM,GACV,KAAK,CAAC,SAAS,GAAG,CAAC;YACnB,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,QAAQ,CAAC;QAEhC,MAAM,MAAM,GAAuB;YACjC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;YAC9B,OAAO,EAAE,KAAK;YACd,GAAG,CAAC,MAAM,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC;gBAC/B,CAAC,CAAC,EAAE,KAAK,EAAE,GAAG,KAAK,CAAC,SAAS,6BAA6B,EAAE;gBAC5D,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,QAAQ;gBACzC,CAAC,CAAC,EAAE,KAAK,EAAE,gBAAgB,KAAK,CAAC,MAAM,wBAAwB,KAAK,CAAC,QAAQ,GAAG,EAAE;gBAClF,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,CAAC,EAAE;gBACtD,CAAC,GAAG,CAAC,EAAE,MAAM;gBACb,MAAM,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC;aACvB,CAAC,CAAC;QACL,CAAC;QAED,OAAO,EAAE,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B,CAAC;CACF,CAAA;AAlCY,sDAAqB;gCAArB,qBAAqB;IADjC,IAAA,mBAAU,GAAE;qCAEiC,8BAAa;GAD9C,qBAAqB,CAkCjC"}

package/dist/index.d.ts

@@ -1,5 +1,10 @@
 export { WorkerModule } from './core/worker.module';
 export { WorkerService } from './core/worker.service';
+export type { RunOptions } from './core/worker.service';
 export { WorkerClass, WorkerTask } from './decorators/worker-task.decorator';
-export type { WorkerModuleOptions, TaskPriority } from './core/worker.interfaces';
 export type { WorkerTaskOptions } from './decorators/worker-task.decorator';
+export { WorkerHealthIndicator } from './health/worker.health';
+export type { WorkerHealthResult } from './health/worker.health';
+export { WorkerMetricsService } from './metrics/worker.metrics';
+export type { WorkerMetricsSnapshot } from './metrics/worker.metrics';
+export type { WorkerJob, WorkerModuleOptions, WorkerModuleAsyncOptions, TaskPriority, DeadLetterEvent, PoolStats, SerializedError, } from './core/worker.interfaces';

package/dist/index.js

@@ -1,11 +1,20 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.WorkerTask = exports.WorkerClass = exports.WorkerService = exports.WorkerModule = void 0;
+exports.WorkerMetricsService = exports.WorkerHealthIndicator = exports.WorkerTask = exports.WorkerClass = exports.WorkerService = exports.WorkerModule = void 0;
+// Module
 var worker_module_1 = require("./core/worker.module");
 Object.defineProperty(exports, "WorkerModule", { enumerable: true, get: function () { return worker_module_1.WorkerModule; } });
+// Services
 var worker_service_1 = require("./core/worker.service");
 Object.defineProperty(exports, "WorkerService", { enumerable: true, get: function () { return worker_service_1.WorkerService; } });
+// Decorators
 var worker_task_decorator_1 = require("./decorators/worker-task.decorator");
 Object.defineProperty(exports, "WorkerClass", { enumerable: true, get: function () { return worker_task_decorator_1.WorkerClass; } });
 Object.defineProperty(exports, "WorkerTask", { enumerable: true, get: function () { return worker_task_decorator_1.WorkerTask; } });
+// Health
+var worker_health_1 = require("./health/worker.health");
+Object.defineProperty(exports, "WorkerHealthIndicator", { enumerable: true, get: function () { return worker_health_1.WorkerHealthIndicator; } });
+// Metrics
+var worker_metrics_1 = require("./metrics/worker.metrics");
+Object.defineProperty(exports, "WorkerMetricsService", { enumerable: true, get: function () { return worker_metrics_1.WorkerMetricsService; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,sDAAkD;AAA1C,6GAAA,YAAY,OAAA;AACpB,wDAAoD;AAA5C,+GAAA,aAAa,OAAA;AACrB,4EAA2E;AAAnE,oHAAA,WAAW,OAAA;AAAE,mHAAA,UAAU,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,SAAS;AACT,sDAAsD;AAA7C,6GAAA,YAAY,OAAA;AAErB,WAAW;AACX,wDAAuD;AAA9C,+GAAA,aAAa,OAAA;AAGtB,aAAa;AACb,4EAA6E;AAApE,oHAAA,WAAW,OAAA;AAAE,mHAAA,UAAU,OAAA;AAGhC,SAAS;AACT,wDAA+D;AAAtD,sHAAA,qBAAqB,OAAA;AAG9B,UAAU;AACV,2DAAiE;AAAxD,sHAAA,oBAAoB,OAAA"}

package/dist/metrics/worker.metrics.d.ts

@@ -0,0 +1,65 @@
+import { OnModuleInit, OnModuleDestroy } from '@nestjs/common';
+import { WorkerService } from '../core/worker.service';
+export interface WorkerMetricsSnapshot {
+    /** Total jobs dispatched since startup */
+    jobsTotal: number;
+    /** Jobs that completed successfully */
+    jobsSuccess: number;
+    /** Jobs that failed (after all retries) */
+    jobsFailed: number;
+    /** Jobs that timed out */
+    jobsTimeout: number;
+    /** Jobs sent to dead letter queue */
+    jobsDead: number;
+    /** Current queue depth */
+    queueDepth: number;
+    /** Current idle worker count */
+    idleWorkers: number;
+    /** Current busy worker count */
+    busyWorkers: number;
+    /** Per-task duration histogram (p50, p95, p99) in ms */
+    durations: Record<string, {
+        p50: number;
+        p95: number;
+        p99: number;
+        count: number;
+    }>;
+}
+/**
+ * WorkerMetricsService — collects runtime metrics from the worker pool.
+ *
+ * Designed to be framework-agnostic: read the snapshot and push to
+ * Prometheus, Datadog, CloudWatch, or any other metrics provider.
+ *
+ * @example
+ *   // Prometheus integration
+ *   @Get('metrics')
+ *   async metrics() {
+ *     const snap = this.workerMetrics.snapshot();
+ *     return [
+ *       `nestworker_jobs_total ${snap.jobsTotal}`,
+ *       `nestworker_jobs_success ${snap.jobsSuccess}`,
+ *       `nestworker_queue_depth ${snap.queueDepth}`,
+ *     ].join('\n');
+ *   }
+ */
+export declare class WorkerMetricsService implements OnModuleInit, OnModuleDestroy {
+    private readonly workerService;
+    private jobsTotal;
+    private jobsSuccess;
+    private jobsFailed;
+    private jobsTimeout;
+    private jobsDead;
+    /** Raw duration samples per task key — capped at 1000 samples */
+    private readonly durationSamples;
+    /** Interval handle for periodic pool-stats polling */
+    private statsInterval?;
+    constructor(workerService: WorkerService);
+    onModuleInit(): void;
+    onModuleDestroy(): void;
+    /** Returns a point-in-time snapshot of all metrics */
+    snapshot(): WorkerMetricsSnapshot;
+    /** Reset all counters (useful in tests) */
+    reset(): void;
+    private pushDuration;
+}

package/dist/metrics/worker.metrics.js

@@ -0,0 +1,122 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkerMetricsService = void 0;
+const common_1 = require("@nestjs/common");
+const worker_service_1 = require("../core/worker.service");
+/**
+ * WorkerMetricsService — collects runtime metrics from the worker pool.
+ *
+ * Designed to be framework-agnostic: read the snapshot and push to
+ * Prometheus, Datadog, CloudWatch, or any other metrics provider.
+ *
+ * @example
+ *   // Prometheus integration
+ *   @Get('metrics')
+ *   async metrics() {
+ *     const snap = this.workerMetrics.snapshot();
+ *     return [
+ *       `nestworker_jobs_total ${snap.jobsTotal}`,
+ *       `nestworker_jobs_success ${snap.jobsSuccess}`,
+ *       `nestworker_queue_depth ${snap.queueDepth}`,
+ *     ].join('\n');
+ *   }
+ */
+let WorkerMetricsService = class WorkerMetricsService {
+    workerService;
+    jobsTotal = 0;
+    jobsSuccess = 0;
+    jobsFailed = 0;
+    jobsTimeout = 0;
+    jobsDead = 0;
+    /** Raw duration samples per task key — capped at 1000 samples */
+    durationSamples = new Map();
+    /** Interval handle for periodic pool-stats polling */
+    statsInterval;
+    constructor(workerService) {
+        this.workerService = workerService;
+    }
+    onModuleInit() {
+        // Task start
+        this.workerService.onTaskStart((job) => {
+            this.jobsTotal++;
+        });
+        // Task end: record duration + success
+        this.workerService.onTaskEnd((job, durationMs) => {
+            const key = `${job.serviceName}.${job.methodName}`;
+            this.jobsSuccess++;
+            this.pushDuration(key, durationMs);
+        });
+        // Task error: record failure
+        this.workerService.onTaskError((job, error) => {
+            this.jobsFailed++;
+            if (error.name === 'TimeoutError')
+                this.jobsTimeout++;
+        });
+        // Dead letter: separate counter
+        this.workerService.onDead(() => {
+            this.jobsDead++;
+        });
+    }
+    onModuleDestroy() {
+        if (this.statsInterval)
+            clearInterval(this.statsInterval);
+    }
+    /** Returns a point-in-time snapshot of all metrics */
+    snapshot() {
+        const stats = this.workerService.stats();
+        const durations = {};
+        for (const [key, samples] of this.durationSamples) {
+            durations[key] = computePercentiles(samples);
+        }
+        return {
+            jobsTotal: this.jobsTotal,
+            jobsSuccess: this.jobsSuccess,
+            jobsFailed: this.jobsFailed,
+            jobsTimeout: this.jobsTimeout,
+            jobsDead: this.jobsDead,
+            queueDepth: stats.queued,
+            idleWorkers: stats.idle,
+            busyWorkers: stats.busy,
+            durations,
+        };
+    }
+    /** Reset all counters (useful in tests) */
+    reset() {
+        this.jobsTotal = this.jobsSuccess = this.jobsFailed =
+            this.jobsTimeout = this.jobsDead = 0;
+        this.durationSamples.clear();
+    }
+    pushDuration(key, ms) {
+        if (!this.durationSamples.has(key))
+            this.durationSamples.set(key, []);
+        const samples = this.durationSamples.get(key);
+        samples.push(ms);
+        // Keep memory bounded — reservoir sampling after 1000 entries
+        if (samples.length > 1000) {
+            const idx = Math.floor(Math.random() * samples.length);
+            samples.splice(idx, 1);
+        }
+    }
+};
+exports.WorkerMetricsService = WorkerMetricsService;
+exports.WorkerMetricsService = WorkerMetricsService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [worker_service_1.WorkerService])
+], WorkerMetricsService);
+function computePercentiles(samples) {
+    if (samples.length === 0)
+        return { p50: 0, p95: 0, p99: 0, count: 0 };
+    const sorted = [...samples].sort((a, b) => a - b);
+    const p = (pct) => sorted[Math.ceil((pct / 100) * sorted.length) - 1] ?? 0;
+    return { p50: p(50), p95: p(95), p99: p(99), count: samples.length };
+}
+//# sourceMappingURL=worker.metrics.js.map

package/dist/metrics/worker.metrics.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.metrics.js","sourceRoot":"","sources":["../../src/metrics/worker.metrics.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA2E;AAC3E,2DAAuD;AAwBvD;;;;;;;;;;;;;;;;;GAiBG;AAEI,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAaF;IAZrB,SAAS,GAAM,CAAC,CAAC;IACjB,WAAW,GAAI,CAAC,CAAC;IACjB,UAAU,GAAK,CAAC,CAAC;IACjB,WAAW,GAAI,CAAC,CAAC;IACjB,QAAQ,GAAO,CAAC,CAAC;IAEzB,iEAAiE;IAChD,eAAe,GAAG,IAAI,GAAG,EAAoB,CAAC;IAE/D,sDAAsD;IAC9C,aAAa,CAAkB;IAEvC,YAA6B,aAA4B;QAA5B,kBAAa,GAAb,aAAa,CAAe;IAAG,CAAC;IAE7D,YAAY;QACV,aAAa;QACb,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,GAAc,EAAE,EAAE;YAChD,IAAI,CAAC,SAAS,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,sCAAsC;QACtC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC,GAAc,EAAE,UAAkB,EAAE,EAAE;YAClE,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC;YACnD,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,6BAA6B;QAC7B,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,GAAc,EAAE,KAAsB,EAAE,EAAE;YACxE,IAAI,CAAC,UAAU,EAAE,CAAC;YAClB,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc;gBAAE,IAAI,CAAC,WAAW,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,gCAAgC;QAChC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,GAAG,EAAE;YAC7B,IAAI,CAAC,QAAQ,EAAE,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,eAAe;QACb,IAAI,IAAI,CAAC,aAAa;YAAE,aAAa,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAC5D,CAAC;IAED,sDAAsD;IACtD,QAAQ;QACN,MAAM,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QACzC,MAAM,SAAS,GAAuC,EAAE,CAAC;QAEzD,KAAK,MAAM,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,SAAS,CAAC,GAAG,CAAC,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;QAED,OAAO;YACL,SAAS,EAAI,IAAI,CAAC,SAAS;YAC3B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,UAAU,EAAG,IAAI,CAAC,UAAU;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,QAAQ,EAAK,IAAI,CAAC,QAAQ;YAC1B,UAAU,EAAG,KAAK,CAAC,MAAM;YACzB,WAAW,EAAE,KAAK,CAAC,IAAI;YACvB,WAAW,EAAE,KAAK,CAAC,IAAI;YACvB,SAAS;SACV,CAAC;IACJ,CAAC;IAED,2CAA2C;IAC3C,KAAK;QACH,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU;YACjD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAC;QACvC,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC/B,CAAC;IAEO,YAAY,CAAC,GAAW,EAAE,EAAU;QAC1C,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;QACtE,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC;QAC/C,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,8DAA8D;QAC9D,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,EAAE,CAAC;YAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;YACvD,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;CACF,CAAA;AAnFY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;qCAciC,8BAAa;GAb9C,oBAAoB,CAmFhC;AAED,SAAS,kBAAkB,CACzB,OAAiB;IAEjB,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;IACtE,MAAM,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,MAAM,CAAC,GAAG,CAAC,GAAW,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC;IACnF,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC;AACvE,CAAC"}

package/dist/worker/worker-runtime.js

@@ -1,28 +1,73 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 const node_worker_threads_1 = require("node:worker_threads");
+const node_async_hooks_1 = require("node:async_hooks");
+const node_crypto_1 = __importDefault(require("node:crypto"));
 const worker_container_1 = require("../di/worker-container");
-/**
- * Worker Runtime – entry point for every worker thread.
- *
- * Receives SerializedService[] via workerData and boots a WorkerContainer
- * which eval()s the stripped class sources and reconstructs the full
- * service graph (deps injected via constructor) without any require() calls
- * into NestJS-annotated files.
- *
- * Jobs are processed serially via a queue — one postMessage() reply per job,
- * which is exactly what the pool's worker.once('message') expects.
- */
 const services = node_worker_threads_1.workerData?.services ?? [];
 const container = new worker_container_1.WorkerContainer();
 container.load(services);
-// Build flat dispatch map: "ServiceName.methodName" → bound callable
-const tasks = {};
+const instances = new Map();
 for (const svc of services) {
-    const inst = container.get(svc.name);
-    for (const { methodName } of svc.methods) {
-        tasks[`${svc.name}.${methodName}`] = (...args) => inst[methodName](...args);
+    instances.set(svc.name, container.get(svc.name));
+}
+node_worker_threads_1.parentPort?.postMessage({ type: 'worker:ready' });
+// ── Pending IPC calls ─────────────────────────────────────────────────────
+const pendingIpc = new Map();
+// ── Pending AbortControllers (keyed by abortSignalId) ────────────────────
+const pendingAborts = new Map();
+// ── Internal ALS for context propagation ─────────────────────────────────
+// One ALS instance per context key — we use a single generic ALS that holds
+// the full context map, then the user's ALS instances read from it via
+// the context restoration path in WorkerService.
+const workerAls = new node_async_hooks_1.AsyncLocalStorage();
+node_worker_threads_1.parentPort?.on('message', (msg) => {
+    const message = msg;
+    // IPC result from main thread
+    if (message.type === 'ipc:result') {
+        const res = message;
+        const pending = pendingIpc.get(res.callId);
+        if (!pending)
+            return;
+        pendingIpc.delete(res.callId);
+        res.ok ? pending.resolve(res.data) : pending.reject(new Error(res.error ?? 'IPC failed'));
+        return;
+    }
+    // Abort signal from main thread
+    if (message.type === 'abort') {
+        const abort = message;
+        pendingAborts.get(abort.abortSignalId)?.abort();
+        return;
+    }
+    // Normal job
+    queue.push(message);
+    runNext();
+});
+function buildProxy(descriptor) {
+    const proxy = {};
+    for (const methodName of descriptor.methodNames) {
+        proxy[methodName] = (...args) => new Promise((resolve, reject) => {
+            const callId = node_crypto_1.default.randomUUID();
+            pendingIpc.set(callId, { resolve, reject });
+            const request = {
+                type: 'ipc:invoke', callId,
+                propertyKey: descriptor.propertyKey,
+                methodName, args,
+            };
+            try {
+                node_worker_threads_1.parentPort?.postMessage(request);
+            }
+            catch (err) {
+                pendingIpc.delete(callId);
+                reject(new Error(`IPC proxy "${descriptor.propertyKey}.${methodName}" could not serialise args: ` +
+                    `${err.message}`));
+            }
+        });
     }
+    return proxy;
 }
 const queue = [];
 let busy = false;
@@ -30,26 +75,78 @@ async function runNext() {
     if (busy || queue.length === 0)
         return;
     busy = true;
-    const payload = queue.shift();
-    const key = `${payload.serviceName}.${payload.methodName}`;
+    const job = queue.shift();
+    // Set up AbortController for this job
+    const abortController = new AbortController();
+    if (job.abortSignalId) {
+        pendingAborts.set(job.abortSignalId, abortController);
+    }
+    // Inject proxy stubs
+    if (job.proxyServices?.length) {
+        const inst = instances.get(job.serviceName);
+        if (inst) {
+            for (const descriptor of job.proxyServices) {
+                inst[descriptor.propertyKey] = buildProxy(descriptor);
+            }
+        }
+    }
+    // Restore ALS context and run the task inside it
+    const alsContext = job.alsContext ?? {};
+    const run = async () => {
+        const inst = instances.get(job.serviceName);
+        if (!inst)
+            throw new Error(`Service "${job.serviceName}" is not registered`);
+        const fn = inst[job.methodName];
+        if (typeof fn !== 'function') {
+            throw new Error(`Task "${job.serviceName}.${job.methodName}" is not registered`);
+        }
+        // Inject AbortSignal as a last argument if the job has an abortSignalId.
+        // Call via inst.method() — not as a detached fn() — so `this` is preserved.
+        const args = job.abortSignalId
+            ? [...job.args, abortController.signal]
+            : job.args;
+        return inst[job.methodName](...args);
+    };
     try {
-        const fn = tasks[key];
-        if (!fn)
-            throw new Error(`Task "${key}" is not registered`);
-        const data = await fn(...payload.args);
+        // Run the task inside the ALS context so any nested ALS.getStore() calls
+        // inside the task body see the propagated main-thread context.
+        const data = await workerAls.run(alsContext, run);
         node_worker_threads_1.parentPort?.postMessage({ ok: true, data });
     }
     catch (error) {
         const e = error;
-        node_worker_threads_1.parentPort?.postMessage({ ok: false, error: { message: e.message, stack: e.stack } });
+        const serialized = {
+            name: e.name ?? 'Error',
+            message: e.message,
+            stack: e.stack,
+            code: e.code,
+            // Capture any extra own enumerable properties (e.g. HttpException.status)
+            extra: serializeExtraProps(e),
+        };
+        node_worker_threads_1.parentPort?.postMessage({ ok: false, error: serialized });
     }
     finally {
+        if (job.abortSignalId) {
+            pendingAborts.delete(job.abortSignalId);
+        }
         busy = false;
         runNext();
     }
 }
-node_worker_threads_1.parentPort?.on('message', (payload) => {
-    queue.push(payload);
-    runNext();
-});
+function serializeExtraProps(err) {
+    const skip = new Set(['name', 'message', 'stack', 'code']);
+    const extra = {};
+    let hasExtra = false;
+    for (const key of Object.keys(err)) {
+        if (skip.has(key))
+            continue;
+        try {
+            structuredClone(err[key]);
+            extra[key] = err[key];
+            hasExtra = true;
+        }
+        catch { /* skip non-cloneable */ }
+    }
+    return hasExtra ? extra : undefined;
+}
 //# sourceMappingURL=worker-runtime.js.map

package/dist/worker/worker-runtime.js.map

@@ -1 +1 @@
-{"version":3,"file":"worker-runtime.js","sourceRoot":"","sources":["../../src/worker/worker-runtime.ts"],"names":[],"mappings":";;AAAA,6DAA6D;AAC7D,6DAAyD;AAGzD;;;;;;;;;;GAUG;AAEH,MAAM,QAAQ,GAAwB,gCAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;AACjE,MAAM,SAAS,GAAG,IAAI,kCAAe,EAAE,CAAC;AACxC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAEzB,qEAAqE;AACrE,MAAM,KAAK,GAAoD,EAAE,CAAC;AAClE,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC3B,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAkD,GAAG,CAAC,IAAI,CAAC,CAAC;IACtF,KAAK,MAAM,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;QACzC,KAAK,CAAC,GAAG,GAAG,CAAC,IAAI,IAAI,UAAU,EAAE,CAAC,GAAG,CAAC,GAAG,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IAC9E,CAAC;AACH,CAAC;AAID,MAAM,KAAK,GAAU,EAAE,CAAC;AACxB,IAAI,IAAI,GAAG,KAAK,CAAC;AAEjB,KAAK,UAAU,OAAO;IACpB,IAAI,IAAI,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IACvC,IAAI,GAAG,IAAI,CAAC;IACZ,MAAM,OAAO,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;IAC/B,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;IAC3D,IAAI,CAAC;QACH,MAAM,EAAE,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;QACtB,IAAI,CAAC,EAAE;YAAE,MAAM,IAAI,KAAK,CAAC,SAAS,GAAG,qBAAqB,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;QACvC,gCAAU,EAAE,WAAW,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,MAAM,CAAC,GAAG,KAAc,CAAC;QACzB,gCAAU,EAAE,WAAW,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;IACxF,CAAC;YAAS,CAAC;QACT,IAAI,GAAG,KAAK,CAAC;QACb,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,gCAAU,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC,OAAY,EAAE,EAAE;IACzC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACpB,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC,CAAC"}
+{"version":3,"file":"worker-runtime.js","sourceRoot":"","sources":["../../src/worker/worker-runtime.ts"],"names":[],"mappings":";;;;;AAAA,6DAA6D;AAC7D,uDAAqD;AACrD,8DAAiC;AACjC,6DAAyD;AAYzD,MAAM,QAAQ,GAAwB,gCAAU,EAAE,QAAQ,IAAI,EAAE,CAAC;AACjE,MAAM,SAAS,GAAG,IAAI,kCAAe,EAAE,CAAC;AACxC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;AAIzB,MAAM,SAAS,GAAG,IAAI,GAAG,EAA2B,CAAC;AACrD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC3B,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,GAAG,CAAkB,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,gCAAU,EAAE,WAAW,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC,CAAC;AAElD,6EAA6E;AAC7E,MAAM,UAAU,GAAG,IAAI,GAAG,EAGvB,CAAC;AAEJ,4EAA4E;AAC5E,MAAM,aAAa,GAAG,IAAI,GAAG,EAA2B,CAAC;AAEzD,4EAA4E;AAC5E,4EAA4E;AAC5E,uEAAuE;AACvE,iDAAiD;AACjD,MAAM,SAAS,GAAG,IAAI,oCAAiB,EAA2B,CAAC;AAEnE,gCAAU,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC,GAAY,EAAE,EAAE;IACzC,MAAM,OAAO,GAAG,GAA2B,CAAC;IAE5C,8BAA8B;IAC9B,IAAK,OAA6B,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACzD,MAAM,GAAG,GAAG,OAA4B,CAAC;QACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC3C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,UAAU,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9B,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,YAAY,CAAC,CAAC,CAAC;QAC1F,OAAO;IACT,CAAC;IAED,gCAAgC;IAChC,IAAK,OAA6B,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;QACpD,MAAM,KAAK,GAAG,OAA6B,CAAC;QAC5C,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,KAAK,EAAE,CAAC;QAChD,OAAO;IACT,CAAC;IAED,aAAa;IACb,KAAK,CAAC,IAAI,CAAC,OAAoB,CAAC,CAAC;IACjC,OAAO,EAAE,CAAC;AACZ,CAAC,CAAC,CAAC;AAEH,SAAS,UAAU,CAAC,UAAkC;IACpD,MAAM,KAAK,GAAoB,EAAE,CAAC;IAClC,KAAK,MAAM,UAAU,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;QAChD,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,IAAe,EAAoB,EAAE,CAC3D,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YAC9B,MAAM,MAAM,GAAG,qBAAM,CAAC,UAAU,EAAE,CAAC;YACnC,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5C,MAAM,OAAO,GAAqB;gBAChC,IAAI,EAAE,YAAY,EAAE,MAAM;gBAC1B,WAAW,EAAE,UAAU,CAAC,WAAW;gBACnC,UAAU,EAAE,IAAI;aACjB,CAAC;YACF,IAAI,CAAC;gBACH,gCAAU,EAAE,WAAW,CAAC,OAAO,CAAC,CAAC;YACnC,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC1B,MAAM,CAAC,IAAI,KAAK,CACd,cAAc,UAAU,CAAC,WAAW,IAAI,UAAU,8BAA8B;oBAChF,GAAI,GAAa,CAAC,OAAO,EAAE,CAC5B,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;IACP,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,KAAK,GAAgB,EAAE,CAAC;AAC9B,IAAI,IAAI,GAAG,KAAK,CAAC;AAEjB,KAAK,UAAU,OAAO;IACpB,IAAI,IAAI,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IACvC,IAAI,GAAG,IAAI,CAAC;IACZ,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,EAAG,CAAC;IAE3B,sCAAsC;IACtC,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;IAC9C,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;QACtB,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,aAAa,EAAE,eAAe,CAAC,CAAC;IACxD,CAAC;IAED,qBAAqB;IACrB,IAAI,GAAG,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC5C,IAAI,IAAI,EAAE,CAAC;YACT,KAAK,MAAM,UAAU,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;gBAC3C,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC,UAAU,CAA+C,CAAC;YACtG,CAAC;QACH,CAAC;IACH,CAAC;IAED,iDAAiD;IACjD,MAAM,UAAU,GAAG,GAAG,CAAC,UAAU,IAAI,EAAE,CAAC;IAExC,MAAM,GAAG,GAAG,KAAK,IAAI,EAAE;QACrB,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,YAAY,GAAG,CAAC,WAAW,qBAAqB,CAAC,CAAC;QAE7E,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAChC,IAAI,OAAO,EAAE,KAAK,UAAU,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,UAAU,qBAAqB,CAAC,CAAC;QACnF,CAAC;QAED,yEAAyE;QACzE,4EAA4E;QAC5E,MAAM,IAAI,GAAG,GAAG,CAAC,aAAa;YAC5B,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,IAAI,EAAE,eAAe,CAAC,MAAM,CAAC;YACvC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC;QAEb,OAAO,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC;IACvC,CAAC,CAAC;IAEF,IAAI,CAAC;QACH,yEAAyE;QACzE,+DAA+D;QAC/D,MAAM,IAAI,GAAG,MAAM,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAClD,gCAAU,EAAE,WAAW,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC9C,CAAC;IAAC,OAAO,KAAc,EAAE,CAAC;QACxB,MAAM,CAAC,GAAG,KAAmE,CAAC;QAC9E,MAAM,UAAU,GAAoB;YAClC,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,OAAO;YACvB,OAAO,EAAE,CAAC,CAAC,OAAO;YAClB,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,0EAA0E;YAC1E,KAAK,EAAE,mBAAmB,CAAC,CAAC,CAAC;SAC9B,CAAC;QACF,gCAAU,EAAE,WAAW,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;IAC5D,CAAC;YAAS,CAAC;QACT,IAAI,GAAG,CAAC,aAAa,EAAE,CAAC;YACtB,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAC1C,CAAC;QACD,IAAI,GAAG,KAAK,CAAC;QACb,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAC1B,GAAoC;IAEpC,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC;IAC3D,MAAM,KAAK,GAA4B,EAAE,CAAC;IAC1C,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YAAE,SAAS;QAC5B,IAAI,CAAC;YACH,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1B,KAAK,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;YACtB,QAAQ,GAAG,IAAI,CAAC;QAClB,CAAC;QAAC,MAAM,CAAC,CAAC,wBAAwB,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC;AACtC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nestworker",
-  "version": "2.0.8",
+  "version": "2.1.0",
   "description": "Enterprise-grade worker thread module for NestJS — priority pool, auto-discovery, DI in workers",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",