@wavezync/nestjs-pgboss 1.0.0 → 2.0.0

package/dist/decorators/job.decorator.d.ts

@@ -3,5 +3,6 @@ export declare const JOB_NAME = "JOB_NAME";
 export declare const JOB_OPTIONS = "JOB_OPTIONS";
 export declare const CRON_EXPRESSION = "CRON_EXPRESSION";
 export declare const CRON_OPTIONS = "CRON_OPTIONS";
+export declare const PG_BOSS_JOB_METADATA = "PG_BOSS_JOB_METADATA";
 export declare function Job<_TData extends object = any>(name: string, options?: JobOptions): (target: any, key: string, descriptor: PropertyDescriptor) => void;
 export declare function CronJob<_TData extends object = any>(name: string, cron: string, options?: JobOptions): (target: any, key: string, descriptor: PropertyDescriptor) => void;

package/dist/decorators/job.decorator.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CRON_OPTIONS = exports.CRON_EXPRESSION = exports.JOB_OPTIONS = exports.JOB_NAME = void 0;
+exports.PG_BOSS_JOB_METADATA = exports.CRON_OPTIONS = exports.CRON_EXPRESSION = exports.JOB_OPTIONS = exports.JOB_NAME = void 0;
 exports.Job = Job;
 exports.CronJob = CronJob;
 const common_1 = require("@nestjs/common");
@@ -8,10 +8,12 @@ exports.JOB_NAME = "JOB_NAME";
 exports.JOB_OPTIONS = "JOB_OPTIONS";
 exports.CRON_EXPRESSION = "CRON_EXPRESSION";
 exports.CRON_OPTIONS = "CRON_OPTIONS";
+exports.PG_BOSS_JOB_METADATA = "PG_BOSS_JOB_METADATA";
 function Job(name, options = {}) {
     return (target, key, descriptor) => {
         (0, common_1.SetMetadata)(exports.JOB_NAME, name)(target, key, descriptor);
         (0, common_1.SetMetadata)(exports.JOB_OPTIONS, options)(target, key, descriptor);
+        (0, common_1.SetMetadata)(exports.PG_BOSS_JOB_METADATA, { jobName: name, workOptions: options })(target, key, descriptor);
     };
 }
 function CronJob(name, cron, options = {}) {
@@ -19,6 +21,7 @@ function CronJob(name, cron, options = {}) {
         (0, common_1.SetMetadata)(exports.JOB_NAME, name)(target, key, descriptor);
         (0, common_1.SetMetadata)(exports.CRON_EXPRESSION, cron)(target, key, descriptor);
         (0, common_1.SetMetadata)(exports.CRON_OPTIONS, options)(target, key, descriptor);
+        (0, common_1.SetMetadata)(exports.PG_BOSS_JOB_METADATA, { jobName: name, workOptions: options })(target, key, descriptor);
     };
 }
 //# sourceMappingURL=job.decorator.js.map

package/dist/decorators/job.decorator.js.map

@@ -1 +1 @@
-{"version":3,"file":"job.decorator.js","sourceRoot":"","sources":["../../lib/decorators/job.decorator.ts"],"names":[],"mappings":";;;AAQA,kBAQC;AAED,0BAUC;AA5BD,2CAA6C;AAGhC,QAAA,QAAQ,GAAG,UAAU,CAAC;AACtB,QAAA,WAAW,GAAG,aAAa,CAAC;AAC5B,QAAA,eAAe,GAAG,iBAAiB,CAAC;AACpC,QAAA,YAAY,GAAG,cAAc,CAAC;AAE3C,SAAgB,GAAG,CACjB,IAAY,EACZ,UAAsB,EAAE;IAExB,OAAO,CAAC,MAAW,EAAE,GAAW,EAAE,UAA8B,EAAE,EAAE;QAClE,IAAA,oBAAW,EAAC,gBAAQ,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QACrD,IAAA,oBAAW,EAAC,mBAAW,EAAE,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;IAC7D,CAAC,CAAC;AACJ,CAAC;AAED,SAAgB,OAAO,CACrB,IAAY,EACZ,IAAY,EACZ,UAAsB,EAAE;IAExB,OAAO,CAAC,MAAW,EAAE,GAAW,EAAE,UAA8B,EAAE,EAAE;QAClE,IAAA,oBAAW,EAAC,gBAAQ,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QACrD,IAAA,oBAAW,EAAC,uBAAe,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QAC5D,IAAA,oBAAW,EAAC,oBAAY,EAAE,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;IAC9D,CAAC,CAAC;AACJ,CAAC"}
+{"version":3,"file":"job.decorator.js","sourceRoot":"","sources":["../../lib/decorators/job.decorator.ts"],"names":[],"mappings":";;;AASA,kBAaC;AAED,0BAeC;AAvCD,2CAA6C;AAGhC,QAAA,QAAQ,GAAG,UAAU,CAAC;AACtB,QAAA,WAAW,GAAG,aAAa,CAAC;AAC5B,QAAA,eAAe,GAAG,iBAAiB,CAAC;AACpC,QAAA,YAAY,GAAG,cAAc,CAAC;AAC9B,QAAA,oBAAoB,GAAG,sBAAsB,CAAC;AAE3D,SAAgB,GAAG,CACjB,IAAY,EACZ,UAAsB,EAAE;IAExB,OAAO,CAAC,MAAW,EAAE,GAAW,EAAE,UAA8B,EAAE,EAAE;QAClE,IAAA,oBAAW,EAAC,gBAAQ,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QACrD,IAAA,oBAAW,EAAC,mBAAW,EAAE,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QAC3D,IAAA,oBAAW,EAAC,4BAAoB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CACxE,MAAM,EACN,GAAG,EACH,UAAU,CACX,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AAED,SAAgB,OAAO,CACrB,IAAY,EACZ,IAAY,EACZ,UAAsB,EAAE;IAExB,OAAO,CAAC,MAAW,EAAE,GAAW,EAAE,UAA8B,EAAE,EAAE;QAClE,IAAA,oBAAW,EAAC,gBAAQ,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QACrD,IAAA,oBAAW,EAAC,uBAAe,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QAC5D,IAAA,oBAAW,EAAC,oBAAY,EAAE,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,UAAU,CAAC,CAAC;QAC5D,IAAA,oBAAW,EAAC,4BAAoB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CACxE,MAAM,EACN,GAAG,EACH,UAAU,CACX,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC"}

package/dist/handler-scanner.service.d.ts

@@ -0,0 +1,11 @@
+import { Reflector, ModulesContainer } from "@nestjs/core";
+import { PgBossService } from "./pgboss.service";
+export declare class HandlerScannerService {
+    private readonly pgBossService;
+    private readonly reflector;
+    private readonly modulesContainer;
+    private readonly logger;
+    constructor(pgBossService: PgBossService, reflector: Reflector, modulesContainer: ModulesContainer);
+    scanAndRegisterHandlers(): Promise<void>;
+    private scanProvider;
+}

package/dist/handler-scanner.service.js

@@ -0,0 +1,66 @@
+"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.HandlerScannerService = void 0;
+const common_1 = require("@nestjs/common");
+const core_1 = require("@nestjs/core");
+const pgboss_service_1 = require("./pgboss.service");
+const job_decorator_1 = require("./decorators/job.decorator");
+const consts_1 = require("./utils/consts");
+let HandlerScannerService = class HandlerScannerService {
+    constructor(pgBossService, reflector, modulesContainer) {
+        this.pgBossService = pgBossService;
+        this.reflector = reflector;
+        this.modulesContainer = modulesContainer;
+        this.logger = new common_1.Logger(consts_1.LOGGER);
+    }
+    async scanAndRegisterHandlers() {
+        for (const module of this.modulesContainer.values()) {
+            const providers = [...module.providers.values()];
+            for (const provider of providers) {
+                this.scanProvider(provider);
+            }
+        }
+    }
+    async scanProvider(provider) {
+        const { instance } = provider;
+        if (!instance || typeof instance !== "object")
+            return;
+        const prototype = Object.getPrototypeOf(instance);
+        const methodNames = Object.getOwnPropertyNames(prototype).filter((method) => method !== "constructor" && typeof instance[method] === "function");
+        for (const methodName of methodNames) {
+            const methodRef = instance[methodName];
+            const jobName = this.reflector.get(job_decorator_1.JOB_NAME, methodRef);
+            const jobOptions = this.reflector.get(job_decorator_1.JOB_OPTIONS, methodRef);
+            const cronExpression = this.reflector.get(job_decorator_1.CRON_EXPRESSION, methodRef);
+            const cronOptions = this.reflector.get(job_decorator_1.CRON_OPTIONS, methodRef);
+            if (jobName) {
+                const boundHandler = methodRef.bind(instance);
+                if (cronExpression) {
+                    await this.pgBossService.registerCronJob(jobName, cronExpression, boundHandler, {}, cronOptions);
+                    this.logger.log(`Registered cron job: ${jobName}`);
+                }
+                else {
+                    await this.pgBossService.registerJob(jobName, boundHandler, jobOptions);
+                    this.logger.log(`Registered job: ${jobName}`);
+                }
+            }
+        }
+    }
+};
+exports.HandlerScannerService = HandlerScannerService;
+exports.HandlerScannerService = HandlerScannerService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [pgboss_service_1.PgBossService,
+        core_1.Reflector,
+        core_1.ModulesContainer])
+], HandlerScannerService);
+//# sourceMappingURL=handler-scanner.service.js.map

package/dist/handler-scanner.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler-scanner.service.js","sourceRoot":"","sources":["../lib/handler-scanner.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAAoD;AACpD,uCAA2D;AAC3D,qDAAiD;AACjD,8DAKoC;AAGpC,2CAAsC;AAG/B,IAAM,qBAAqB,GAA3B,MAAM,qBAAqB;IAGhC,YACmB,aAA4B,EAC5B,SAAoB,EACpB,gBAAkC;QAFlC,kBAAa,GAAb,aAAa,CAAe;QAC5B,cAAS,GAAT,SAAS,CAAW;QACpB,qBAAgB,GAAhB,gBAAgB,CAAkB;QALpC,WAAM,GAAG,IAAI,eAAM,CAAC,eAAM,CAAC,CAAC;IAM1C,CAAC;IAEJ,KAAK,CAAC,uBAAuB;QAC3B,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,gBAAgB,CAAC,MAAM,EAAE,EAAE,CAAC;YACpD,MAAM,SAAS,GAAG,CAAC,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;YAEjD,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;gBACjC,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,QAA8B;QACvD,MAAM,EAAE,QAAQ,EAAE,GAAG,QAAQ,CAAC;QAC9B,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;YAAE,OAAO;QAEtD,MAAM,SAAS,GAAG,MAAM,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QAClD,MAAM,WAAW,GAAG,MAAM,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,MAAM,CAC9D,CAAC,MAAM,EAAE,EAAE,CACT,MAAM,KAAK,aAAa,IAAI,OAAO,QAAQ,CAAC,MAAM,CAAC,KAAK,UAAU,CACrE,CAAC;QAEF,KAAK,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;YACrC,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;YACvC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAS,wBAAQ,EAAE,SAAS,CAAC,CAAC;YAChE,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CACnC,2BAAW,EACX,SAAS,CACV,CAAC;YACF,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CACvC,+BAAe,EACf,SAAS,CACV,CAAC;YACF,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAM,4BAAY,EAAE,SAAS,CAAC,CAAC;YAErE,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAE9C,IAAI,cAAc,EAAE,CAAC;oBACnB,MAAM,IAAI,CAAC,aAAa,CAAC,eAAe,CACtC,OAAO,EACP,cAAc,EACd,YAAY,EACZ,EAAE,EACF,WAAW,CACZ,CAAC;oBACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,wBAAwB,OAAO,EAAE,CAAC,CAAC;gBACrD,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAClC,OAAO,EACP,YAAY,EACZ,UAAU,CACX,CAAC;oBACF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,mBAAmB,OAAO,EAAE,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;CACF,CAAA;AAjEY,sDAAqB;gCAArB,qBAAqB;IADjC,IAAA,mBAAU,GAAE;qCAKuB,8BAAa;QACjB,gBAAS;QACF,uBAAgB;GAN1C,qBAAqB,CAiEjC"}

package/dist/index.d.ts

@@ -2,3 +2,4 @@ export * from "./pgboss.module";
 export * from "./pgboss.service";
 export * from "./decorators/job.decorator";
 export * from "./interfaces/pgboss-module-options.interface";
+export * from "./handler-scanner.service";

package/dist/index.js

@@ -18,4 +18,5 @@ __exportStar(require("./pgboss.module"), exports);
 __exportStar(require("./pgboss.service"), exports);
 __exportStar(require("./decorators/job.decorator"), exports);
 __exportStar(require("./interfaces/pgboss-module-options.interface"), exports);
+__exportStar(require("./handler-scanner.service"), exports);
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../lib/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAAgC;AAChC,mDAAiC;AACjC,6DAA2C;AAC3C,+EAA6D"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../lib/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,kDAAgC;AAChC,mDAAiC;AACjC,6DAA2C;AAC3C,+EAA6D;AAC7D,4DAA0C"}

package/dist/interfaces/handler-metadata.interface.d.ts

@@ -0,0 +1,7 @@
+import { JobOptions } from "pg-boss";
+export interface HandlerMetadata {
+    jobName: string;
+    workOptions?: JobOptions;
+    cronExpression?: string;
+    cronOptions?: JobOptions;
+}

package/dist/interfaces/handler-metadata.interface.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=handler-metadata.interface.js.map

package/dist/interfaces/handler-metadata.interface.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handler-metadata.interface.js","sourceRoot":"","sources":["../../lib/interfaces/handler-metadata.interface.ts"],"names":[],"mappings":""}

package/dist/pgboss.module.d.ts

@@ -1,7 +1,17 @@
-import { DynamicModule } from "@nestjs/common";
+import { DynamicModule, OnModuleInit, OnApplicationBootstrap, OnModuleDestroy } from "@nestjs/common";
+import PgBoss from "pg-boss";
 import { PgBossModuleAsyncOptions } from "./interfaces/pgboss-module-options.interface";
-export declare class PgBossModule {
+import { HandlerScannerService } from "./handler-scanner.service";
+export declare class PgBossModule implements OnModuleInit, OnApplicationBootstrap, OnModuleDestroy {
+    private readonly boss;
+    private readonly handlerScannerService;
+    private readonly logger;
+    constructor(boss: PgBoss, handlerScannerService: HandlerScannerService);
     static forRootAsync(options: PgBossModuleAsyncOptions): DynamicModule;
     private static createAsyncProviders;
     private static createAsyncOptionsProvider;
+    onModuleInit(): void;
+    onApplicationBootstrap(): Promise<void>;
+    onModuleDestroy(): Promise<void>;
+    private setupWorkers;
 }

package/dist/pgboss.module.js

@@ -5,6 +5,12 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
     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);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
@@ -12,17 +18,27 @@ var PgBossModule_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PgBossModule = void 0;
 const common_1 = require("@nestjs/common");
+const core_1 = require("@nestjs/core");
 const pg_boss_1 = __importDefault(require("pg-boss"));
+const rxjs_1 = require("rxjs");
 const pgboss_service_1 = require("./pgboss.service");
 const consts_1 = require("./utils/consts");
+const handler_scanner_service_1 = require("./handler-scanner.service");
+const handleRetry_1 = require("./utils/handleRetry");
 let PgBossModule = PgBossModule_1 = class PgBossModule {
+    constructor(boss, handlerScannerService) {
+        this.boss = boss;
+        this.handlerScannerService = handlerScannerService;
+        this.logger = new common_1.Logger(consts_1.LOGGER);
+    }
     static forRootAsync(options) {
-        const logger = new common_1.Logger(PgBossModule_1.name);
+        const logger = new common_1.Logger(consts_1.LOGGER);
         const pgBossProvider = {
             provide: consts_1.PGBOSS_TOKEN,
             useFactory: async (pgBossOptions) => {
-                const boss = new pg_boss_1.default(pgBossOptions.connectionString);
-                await boss.start();
+                const boss = await (0, rxjs_1.lastValueFrom)((0, rxjs_1.defer)(() => new pg_boss_1.default({
+                    connectionString: pgBossOptions.connectionString,
+                }).start()).pipe((0, handleRetry_1.handleRetry)(pgBossOptions.retryAttempts, pgBossOptions.retryDelay, pgBossOptions.verboseRetryLog)));
                 logger.log("PgBoss started successfully");
                 return boss;
             },
@@ -32,8 +48,14 @@ let PgBossModule = PgBossModule_1 = class PgBossModule {
         return {
             module: PgBossModule_1,
             imports: options.imports || [],
-            providers: [...asyncProviders, pgBossProvider, pgboss_service_1.PgBossService],
-            exports: [pgboss_service_1.PgBossService],
+            providers: [
+                ...asyncProviders,
+                pgBossProvider,
+                pgboss_service_1.PgBossService,
+                handler_scanner_service_1.HandlerScannerService,
+                core_1.MetadataScanner,
+            ],
+            exports: [pgboss_service_1.PgBossService, consts_1.PGBOSS_TOKEN],
         };
     }
     static createAsyncProviders(options) {
@@ -43,10 +65,7 @@ let PgBossModule = PgBossModule_1 = class PgBossModule {
         const useClass = options.useClass;
         return [
             this.createAsyncOptionsProvider(options),
-            {
-                provide: useClass,
-                useClass,
-            },
+            { provide: useClass, useClass },
         ];
     }
     static createAsyncOptionsProvider(options) {
@@ -64,10 +83,32 @@ let PgBossModule = PgBossModule_1 = class PgBossModule {
             inject,
         };
     }
+    onModuleInit() { }
+    async onApplicationBootstrap() {
+        await this.setupWorkers();
+    }
+    async onModuleDestroy() {
+        try {
+            if (this.boss) {
+                await this.boss.stop();
+            }
+        }
+        catch (e) {
+            this.logger.error(e);
+        }
+    }
+    async setupWorkers() {
+        await this.handlerScannerService.scanAndRegisterHandlers();
+    }
 };
 exports.PgBossModule = PgBossModule;
 exports.PgBossModule = PgBossModule = PgBossModule_1 = __decorate([
     (0, common_1.Global)(),
-    (0, common_1.Module)({})
+    (0, common_1.Module)({
+        providers: [core_1.MetadataScanner, handler_scanner_service_1.HandlerScannerService],
+    }),
+    __param(0, (0, common_1.Inject)(consts_1.PGBOSS_TOKEN)),
+    __metadata("design:paramtypes", [pg_boss_1.default,
+        handler_scanner_service_1.HandlerScannerService])
 ], PgBossModule);
 //# sourceMappingURL=pgboss.module.js.map

package/dist/pgboss.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"pgboss.module.js","sourceRoot":"","sources":["../lib/pgboss.module.ts"],"names":[],"mappings":";;;;;;;;;;;;;AACA,2CAMwB;AACxB,sDAA6B;AAC7B,qDAAiD;AACjD,2CAA8D;AAQvD,IAAM,YAAY,oBAAlB,MAAM,YAAY;IACvB,MAAM,CAAC,YAAY,CAAC,OAAiC;QACnD,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC,cAAY,CAAC,IAAI,CAAC,CAAC;QAE7C,MAAM,cAAc,GAAa;YAC/B,OAAO,EAAE,qBAAY;YACrB,UAAU,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;gBAClC,MAAM,IAAI,GAAG,IAAI,iBAAM,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;gBACxD,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;gBACnB,MAAM,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;gBAC1C,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,EAAE,CAAC,uBAAc,CAAC;SACzB,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAE1D,OAAO;YACL,MAAM,EAAE,cAAY;YACpB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;YAC9B,SAAS,EAAE,CAAC,GAAG,cAAc,EAAE,cAAc,EAAE,8BAAa,CAAC;YAC7D,OAAO,EAAE,CAAC,8BAAa,CAAC;SACzB,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,oBAAoB,CACjC,OAAiC;QAEjC,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAC9C,OAAO,CAAC,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,QAAQ,GAAG,OAAO,CAAC,QAEA,CAAC;QAE1B,OAAO;YACL,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC;YACxC;gBACE,OAAO,EAAE,QAAQ;gBACjB,QAAQ;aACT;SACF,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,0BAA0B,CACvC,OAAiC;QAEjC,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,OAAO;gBACL,OAAO,EAAE,uBAAc;gBACvB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE;aAC7B,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAQ,CAAC,CAAC;QAElE,OAAO;YACL,OAAO,EAAE,uBAAc;YACvB,UAAU,EAAE,KAAK,EAAE,cAAoC,EAAE,EAAE,CACzD,cAAc,CAAC,mBAAmB,EAAE;YACtC,MAAM;SACP,CAAC;IACJ,CAAC;CACF,CAAA;AAjEY,oCAAY;uBAAZ,YAAY;IAFxB,IAAA,eAAM,GAAE;IACR,IAAA,eAAM,EAAC,EAAE,CAAC;GACE,YAAY,CAiExB"}
+{"version":3,"file":"pgboss.module.js","sourceRoot":"","sources":["../lib/pgboss.module.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;AAAA,2CASwB;AACxB,uCAA+C;AAC/C,sDAA6B;AAC7B,+BAA4C;AAC5C,qDAAiD;AACjD,2CAAsE;AAKtE,uEAAkE;AAClE,qDAAgD;AAMzC,IAAM,YAAY,oBAAlB,MAAM,YAAY;IAKvB,YACwB,IAA6B,EAClC,qBAA4C;QADtB,SAAI,GAAJ,IAAI,CAAQ;QAClC,0BAAqB,GAArB,qBAAqB,CAAuB;QAJ9C,WAAM,GAAG,IAAI,eAAM,CAAC,eAAM,CAAC,CAAC;IAK1C,CAAC;IAEJ,MAAM,CAAC,YAAY,CAAC,OAAiC;QACnD,MAAM,MAAM,GAAG,IAAI,eAAM,CAAC,eAAM,CAAC,CAAC;QAElC,MAAM,cAAc,GAAG;YACrB,OAAO,EAAE,qBAAY;YACrB,UAAU,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE;gBAClC,MAAM,IAAI,GAAG,MAAM,IAAA,oBAAa,EAC9B,IAAA,YAAK,EAAC,GAAG,EAAE,CACT,IAAI,iBAAM,CAAC;oBACT,gBAAgB,EAAE,aAAa,CAAC,gBAAgB;iBACjD,CAAC,CAAC,KAAK,EAAE,CACX,CAAC,IAAI,CACJ,IAAA,yBAAW,EACT,aAAa,CAAC,aAAa,EAC3B,aAAa,CAAC,UAAU,EACxB,aAAa,CAAC,eAAe,CAC9B,CACF,CACF,CAAC;gBACF,MAAM,CAAC,GAAG,CAAC,6BAA6B,CAAC,CAAC;gBAC1C,OAAO,IAAI,CAAC;YACd,CAAC;YACD,MAAM,EAAE,CAAC,uBAAc,CAAC;SACzB,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAE1D,OAAO;YACL,MAAM,EAAE,cAAY;YACpB,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,EAAE;YAC9B,SAAS,EAAE;gBACT,GAAG,cAAc;gBACjB,cAAc;gBACd,8BAAa;gBACb,+CAAqB;gBACrB,sBAAe;aAChB;YACD,OAAO,EAAE,CAAC,8BAAa,EAAE,qBAAY,CAAC;SACvC,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,oBAAoB,CAAC,OAAiC;QACnE,IAAI,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YAC9C,OAAO,CAAC,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC,CAAC,CAAC;QACpD,CAAC;QAED,MAAM,QAAQ,GAAG,OAAO,CAAC,QAEA,CAAC;QAE1B,OAAO;YACL,IAAI,CAAC,0BAA0B,CAAC,OAAO,CAAC;YACxC,EAAE,OAAO,EAAE,QAAQ,EAAE,QAAQ,EAAE;SAChC,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,0BAA0B,CAAC,OAAiC;QACzE,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,OAAO;gBACL,OAAO,EAAE,uBAAc;gBACvB,UAAU,EAAE,OAAO,CAAC,UAAU;gBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,EAAE;aAC7B,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,WAAW,CAAQ,CAAC,CAAC;QAElE,OAAO;YACL,OAAO,EAAE,uBAAc;YACvB,UAAU,EAAE,KAAK,EAAE,cAAoC,EAAE,EAAE,CACzD,cAAc,CAAC,mBAAmB,EAAE;YACtC,MAAM;SACP,CAAC;IACJ,CAAC;IAED,YAAY,KAAI,CAAC;IAEjB,KAAK,CAAC,sBAAsB;QAC1B,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,eAAe;QACnB,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;gBACd,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACzB,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,YAAY;QACxB,MAAM,IAAI,CAAC,qBAAqB,CAAC,uBAAuB,EAAE,CAAC;IAC7D,CAAC;CACF,CAAA;AAxGY,oCAAY;uBAAZ,YAAY;IAJxB,IAAA,eAAM,GAAE;IACR,IAAA,eAAM,EAAC;QACN,SAAS,EAAE,CAAC,sBAAe,EAAE,+CAAqB,CAAC;KACpD,CAAC;IAOG,WAAA,IAAA,eAAM,EAAC,qBAAY,CAAC,CAAA;qCAAwB,iBAAM;QACX,+CAAqB;GAPpD,YAAY,CAwGxB"}

package/dist/pgboss.service.d.ts

@@ -1,8 +1,9 @@
-import PgBoss, { BatchWorkOptions, JobOptions, WorkHandler } from "pg-boss";
+import PgBoss, { BatchWorkOptions, WorkHandler } from "pg-boss";
 export declare class PgBossService {
     private readonly boss;
     constructor(boss: PgBoss);
-    registerJob<TData extends object>(name: string, handler: WorkHandler<TData>, options?: BatchWorkOptions): Promise<void>;
-    scheduleJob<TData extends object>(name: string, data: TData, options?: JobOptions): Promise<void>;
+    scheduleJob<TData extends object>(name: string, data: TData, options?: PgBoss.SendOptions): Promise<void>;
+    scheduleCronJob<TData extends object>(name: string, cron: string, data?: TData, options?: PgBoss.ScheduleOptions): Promise<void>;
     registerCronJob<TData extends object>(name: string, cron: string, handler: WorkHandler<TData>, data?: TData, options?: PgBoss.ScheduleOptions): Promise<void>;
+    registerJob<TData extends object>(name: string, handler: WorkHandler<TData>, options?: BatchWorkOptions): Promise<void>;
 }

package/dist/pgboss.service.js

@@ -18,26 +18,30 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.PgBossService = void 0;
 const common_1 = require("@nestjs/common");
 const pg_boss_1 = __importDefault(require("pg-boss"));
+const common_2 = require("@nestjs/common");
 const consts_1 = require("./utils/consts");
 let PgBossService = class PgBossService {
     constructor(boss) {
         this.boss = boss;
     }
-    async registerJob(name, handler, options) {
-        await this.boss.work(name, options, handler);
-    }
     async scheduleJob(name, data, options) {
         await this.boss.send(name, data, options);
     }
+    async scheduleCronJob(name, cron, data, options) {
+        await this.boss.schedule(name, cron, data !== null && data !== void 0 ? data : {}, options !== null && options !== void 0 ? options : {});
+    }
     async registerCronJob(name, cron, handler, data, options) {
-        await this.boss.schedule(name, cron, data ? data : {}, options ? options : {});
+        await this.boss.schedule(name, cron, data !== null && data !== void 0 ? data : {}, options !== null && options !== void 0 ? options : {});
         await this.boss.work(name, handler);
     }
+    async registerJob(name, handler, options) {
+        await this.boss.work(name, options, handler);
+    }
 };
 exports.PgBossService = PgBossService;
 exports.PgBossService = PgBossService = __decorate([
     (0, common_1.Injectable)(),
-    __param(0, (0, common_1.Inject)(consts_1.PGBOSS_TOKEN)),
+    __param(0, (0, common_2.Inject)(consts_1.PGBOSS_TOKEN)),
     __metadata("design:paramtypes", [pg_boss_1.default])
 ], PgBossService);
 //# sourceMappingURL=pgboss.service.js.map

package/dist/pgboss.service.js.map

@@ -1 +1 @@
-{"version":3,"file":"pgboss.service.js","sourceRoot":"","sources":["../lib/pgboss.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AAAA,2CAAoD;AACpD,sDAA4E;AAC5E,2CAA8C;AAGvC,IAAM,aAAa,GAAnB,MAAM,aAAa;IACxB,YAAmD,IAAY;QAAZ,SAAI,GAAJ,IAAI,CAAQ;IAAG,CAAC;IAEnE,KAAK,CAAC,WAAW,CACf,IAAY,EACZ,OAA2B,EAC3B,OAA0B;QAE1B,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;IAED,KAAK,CAAC,WAAW,CACf,IAAY,EACZ,IAAW,EACX,OAAoB;QAEpB,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,IAAY,EACZ,IAAY,EACZ,OAA2B,EAC3B,IAAY,EACZ,OAAgC;QAEhC,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CACtB,IAAI,EACJ,IAAI,EACJ,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAChB,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CACvB,CAAC;QACF,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC;CACF,CAAA;AAlCY,sCAAa;wBAAb,aAAa;IADzB,IAAA,mBAAU,GAAE;IAEE,WAAA,IAAA,eAAM,EAAC,qBAAY,CAAC,CAAA;qCAAwB,iBAAM;GADpD,aAAa,CAkCzB"}
+{"version":3,"file":"pgboss.service.js","sourceRoot":"","sources":["../lib/pgboss.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;AAAA,2CAA4C;AAC5C,sDAAgE;AAChE,2CAAwC;AACxC,2CAA8C;AAGvC,IAAM,aAAa,GAAnB,MAAM,aAAa;IACxB,YAAmD,IAAY;QAAZ,SAAI,GAAJ,IAAI,CAAQ;IAAG,CAAC;IAEnE,KAAK,CAAC,WAAW,CACf,IAAY,EACZ,IAAW,EACX,OAA4B;QAE5B,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,IAAY,EACZ,IAAY,EACZ,IAAY,EACZ,OAAgC;QAEhC,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,EAAE,EAAE,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,EAAE,CAAC,CAAC;IAClE,CAAC;IAED,KAAK,CAAC,eAAe,CACnB,IAAY,EACZ,IAAY,EACZ,OAA2B,EAC3B,IAAY,EACZ,OAAgC;QAEhC,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,EAAE,EAAE,OAAO,aAAP,OAAO,cAAP,OAAO,GAAI,EAAE,CAAC,CAAC;QAChE,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,WAAW,CACf,IAAY,EACZ,OAA2B,EAC3B,OAA0B;QAE1B,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;CACF,CAAA;AAtCY,sCAAa;wBAAb,aAAa;IADzB,IAAA,mBAAU,GAAE;IAEE,WAAA,IAAA,eAAM,EAAC,qBAAY,CAAC,CAAA;qCAAwB,iBAAM;GADpD,aAAa,CAsCzB"}

package/dist/utils/consts.d.ts

@@ -1,2 +1,5 @@
+export declare const PG_BOSS_JOB_METADATA = "PG_BOSS_JOB_METADATA";
+export declare const PGBOSS_TOKEN: unique symbol;
+export declare const MODULE_OPTIONS_TOKEN: unique symbol;
 export declare const PGBOSS_OPTIONS = "PGBOSS_OPTIONS";
-export declare const PGBOSS_TOKEN = "PGBOSS_TOKEN";
+export declare const LOGGER = "nestjs-pgboss";

package/dist/utils/consts.js

@@ -1,6 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.PGBOSS_TOKEN = exports.PGBOSS_OPTIONS = void 0;
+exports.LOGGER = exports.PGBOSS_OPTIONS = exports.MODULE_OPTIONS_TOKEN = exports.PGBOSS_TOKEN = exports.PG_BOSS_JOB_METADATA = void 0;
+exports.PG_BOSS_JOB_METADATA = "PG_BOSS_JOB_METADATA";
+exports.PGBOSS_TOKEN = Symbol("PGBOSS_TOKEN");
+exports.MODULE_OPTIONS_TOKEN = Symbol("MODULE_OPTIONS_TOKEN");
 exports.PGBOSS_OPTIONS = "PGBOSS_OPTIONS";
-exports.PGBOSS_TOKEN = "PGBOSS_TOKEN";
+exports.LOGGER = "nestjs-pgboss";
 //# sourceMappingURL=consts.js.map

package/dist/utils/consts.js.map

@@ -1 +1 @@
-{"version":3,"file":"consts.js","sourceRoot":"","sources":["../../lib/utils/consts.ts"],"names":[],"mappings":";;;AAAa,QAAA,cAAc,GAAG,gBAAgB,CAAC;AAClC,QAAA,YAAY,GAAG,cAAc,CAAC"}
+{"version":3,"file":"consts.js","sourceRoot":"","sources":["../../lib/utils/consts.ts"],"names":[],"mappings":";;;AAAa,QAAA,oBAAoB,GAAG,sBAAsB,CAAC;AAC9C,QAAA,YAAY,GAAG,MAAM,CAAC,cAAc,CAAC,CAAC;AACtC,QAAA,oBAAoB,GAAG,MAAM,CAAC,sBAAsB,CAAC,CAAC;AACtD,QAAA,cAAc,GAAG,gBAAgB,CAAC;AAClC,QAAA,MAAM,GAAG,eAAe,CAAC"}

package/dist/utils/conversion-helper.d.ts

@@ -0,0 +1,2 @@
+import { JobOptions, BatchWorkOptions } from "pg-boss";
+export declare function convertToBatchWorkOptions(jobOptions?: JobOptions): BatchWorkOptions;

package/dist/utils/conversion-helper.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.convertToBatchWorkOptions = convertToBatchWorkOptions;
+function convertToBatchWorkOptions(jobOptions = {}) {
+    return Object.assign(Object.assign({}, jobOptions), { batchSize: 1 });
+}
+//# sourceMappingURL=conversion-helper.js.map

package/dist/utils/conversion-helper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"conversion-helper.js","sourceRoot":"","sources":["../../lib/utils/conversion-helper.ts"],"names":[],"mappings":";;AAEA,8DAOC;AAPD,SAAgB,yBAAyB,CACvC,aAAyB,EAAE;IAE3B,uCACK,UAAU,KACb,SAAS,EAAE,CAAC,IACZ;AACJ,CAAC"}

package/dist/utils/handleRetry.d.ts

@@ -0,0 +1 @@
+export declare function handleRetry(retryAttempts?: number, retryDelay?: number, verbose?: boolean, toRetry?: (err: any) => boolean): <T>(source: import("rxjs").Observable<T>) => import("rxjs").Observable<T>;

package/dist/utils/handleRetry.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.handleRetry = handleRetry;
+const rxjs_1 = require("rxjs");
+const operators_1 = require("rxjs/operators");
+function handleRetry(retryAttempts = 9, retryDelay = 3000, verbose = false, toRetry = (_err) => true) {
+    return (source) => source.pipe((0, operators_1.retryWhen)((attempts) => attempts.pipe((0, operators_1.take)(retryAttempts), (0, operators_1.mergeMap)((error, index) => {
+        const includeError = toRetry(error);
+        if (verbose && includeError) {
+            console.warn(`Attempt ${index + 1}: Retrying in ${retryDelay / 1000} seconds...`);
+        }
+        return includeError ? (0, rxjs_1.of)(error) : (0, rxjs_1.throwError)(error);
+    }), (0, operators_1.delay)(retryDelay))));
+}
+//# sourceMappingURL=handleRetry.js.map

package/dist/utils/handleRetry.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"handleRetry.js","sourceRoot":"","sources":["../../lib/utils/handleRetry.ts"],"names":[],"mappings":";;AAGA,kCA0BC;AA7BD,+BAAsC;AACtC,8CAAkE;AAElE,SAAgB,WAAW,CACzB,aAAa,GAAG,CAAC,EACjB,UAAU,GAAG,IAAI,EACjB,OAAO,GAAG,KAAK,EACf,UAAiC,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI;IAEpD,OAAO,CAAI,MAAoC,EAAE,EAAE,CACjD,MAAM,CAAC,IAAI,CACT,IAAA,qBAAS,EAAC,CAAC,QAAQ,EAAE,EAAE,CACrB,QAAQ,CAAC,IAAI,CACX,IAAA,gBAAI,EAAC,aAAa,CAAC,EACnB,IAAA,oBAAQ,EAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QACxB,MAAM,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,OAAO,IAAI,YAAY,EAAE,CAAC;YAC5B,OAAO,CAAC,IAAI,CACV,WAAW,KAAK,GAAG,CAAC,iBAClB,UAAU,GAAG,IACf,aAAa,CACd,CAAC;QACJ,CAAC;QACD,OAAO,YAAY,CAAC,CAAC,CAAC,IAAA,SAAE,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAA,iBAAU,EAAC,KAAK,CAAC,CAAC;IACtD,CAAC,CAAC,EACF,IAAA,iBAAK,EAAC,UAAU,CAAC,CAClB,CACF,CACF,CAAC;AACN,CAAC"}

package/lib/decorators/job.decorator.ts

@@ -5,6 +5,7 @@ export const JOB_NAME = "JOB_NAME";
 export const JOB_OPTIONS = "JOB_OPTIONS";
 export const CRON_EXPRESSION = "CRON_EXPRESSION";
 export const CRON_OPTIONS = "CRON_OPTIONS";
+export const PG_BOSS_JOB_METADATA = "PG_BOSS_JOB_METADATA";
 
 export function Job<_TData extends object = any>(
   name: string,
@@ -13,6 +14,11 @@ export function Job<_TData extends object = any>(
   return (target: any, key: string, descriptor: PropertyDescriptor) => {
     SetMetadata(JOB_NAME, name)(target, key, descriptor);
     SetMetadata(JOB_OPTIONS, options)(target, key, descriptor);
+    SetMetadata(PG_BOSS_JOB_METADATA, { jobName: name, workOptions: options })(
+      target,
+      key,
+      descriptor,
+    );
   };
 }
 
@@ -25,5 +31,10 @@ export function CronJob<_TData extends object = any>(
     SetMetadata(JOB_NAME, name)(target, key, descriptor);
     SetMetadata(CRON_EXPRESSION, cron)(target, key, descriptor);
     SetMetadata(CRON_OPTIONS, options)(target, key, descriptor);
+    SetMetadata(PG_BOSS_JOB_METADATA, { jobName: name, workOptions: options })(
+      target,
+      key,
+      descriptor,
+    );
   };
 }

package/lib/handler-scanner.service.ts

@@ -0,0 +1,80 @@
+import { Injectable, Logger } from "@nestjs/common";
+import { Reflector, ModulesContainer } from "@nestjs/core";
+import { PgBossService } from "./pgboss.service";
+import {
+  JOB_NAME,
+  JOB_OPTIONS,
+  CRON_EXPRESSION,
+  CRON_OPTIONS,
+} from "./decorators/job.decorator";
+import { InstanceWrapper } from "@nestjs/core/injector/instance-wrapper";
+import { BatchWorkOptions } from "pg-boss";
+import { LOGGER } from "utils/consts";
+
+@Injectable()
+export class HandlerScannerService {
+  private readonly logger = new Logger(LOGGER);
+
+  constructor(
+    private readonly pgBossService: PgBossService,
+    private readonly reflector: Reflector,
+    private readonly modulesContainer: ModulesContainer,
+  ) {}
+
+  async scanAndRegisterHandlers() {
+    for (const module of this.modulesContainer.values()) {
+      const providers = [...module.providers.values()];
+
+      for (const provider of providers) {
+        this.scanProvider(provider);
+      }
+    }
+  }
+
+  private async scanProvider(provider: InstanceWrapper<any>) {
+    const { instance } = provider;
+    if (!instance || typeof instance !== "object") return;
+
+    const prototype = Object.getPrototypeOf(instance);
+    const methodNames = Object.getOwnPropertyNames(prototype).filter(
+      (method) =>
+        method !== "constructor" && typeof instance[method] === "function",
+    );
+
+    for (const methodName of methodNames) {
+      const methodRef = instance[methodName];
+      const jobName = this.reflector.get<string>(JOB_NAME, methodRef);
+      const jobOptions = this.reflector.get<BatchWorkOptions>(
+        JOB_OPTIONS,
+        methodRef,
+      );
+      const cronExpression = this.reflector.get<string>(
+        CRON_EXPRESSION,
+        methodRef,
+      );
+      const cronOptions = this.reflector.get<any>(CRON_OPTIONS, methodRef);
+
+      if (jobName) {
+        const boundHandler = methodRef.bind(instance);
+
+        if (cronExpression) {
+          await this.pgBossService.registerCronJob(
+            jobName,
+            cronExpression,
+            boundHandler,
+            {},
+            cronOptions,
+          );
+          this.logger.log(`Registered cron job: ${jobName}`);
+        } else {
+          await this.pgBossService.registerJob(
+            jobName,
+            boundHandler,
+            jobOptions,
+          );
+          this.logger.log(`Registered job: ${jobName}`);
+        }
+      }
+    }
+  }
+}

package/lib/index.ts

@@ -2,3 +2,4 @@ export * from "./pgboss.module";
 export * from "./pgboss.service";
 export * from "./decorators/job.decorator";
 export * from "./interfaces/pgboss-module-options.interface";
+export * from "./handler-scanner.service";

package/lib/interfaces/handler-metadata.interface.ts

@@ -0,0 +1,8 @@
+import { JobOptions } from "pg-boss";
+
+export interface HandlerMetadata {
+  jobName: string;
+  workOptions?: JobOptions;
+  cronExpression?: string;
+  cronOptions?: JobOptions;
+}

package/lib/pgboss.module.ts

@@ -1,30 +1,58 @@
-// src/pg-boss.module.ts
 import {
-  Module,
   DynamicModule,
   Global,
-  Provider,
   Logger,
+  Module,
+  OnModuleInit,
+  OnApplicationBootstrap,
+  OnModuleDestroy,
+  Inject,
 } from "@nestjs/common";
+import { MetadataScanner } from "@nestjs/core";
 import PgBoss from "pg-boss";
+import { defer, lastValueFrom } from "rxjs";
 import { PgBossService } from "./pgboss.service";
-import { PGBOSS_OPTIONS, PGBOSS_TOKEN } from "./utils/consts";
+import { LOGGER, PGBOSS_OPTIONS, PGBOSS_TOKEN } from "./utils/consts";
 import {
   PgBossModuleAsyncOptions,
   PgBossOptionsFactory,
 } from "./interfaces/pgboss-module-options.interface";
+import { HandlerScannerService } from "./handler-scanner.service";
+import { handleRetry } from "utils/handleRetry";
 
 @Global()
-@Module({})
-export class PgBossModule {
+@Module({
+  providers: [MetadataScanner, HandlerScannerService],
+})
+export class PgBossModule
+  implements OnModuleInit, OnApplicationBootstrap, OnModuleDestroy
+{
+  private readonly logger = new Logger(LOGGER);
+
+  constructor(
+    @Inject(PGBOSS_TOKEN) private readonly boss: PgBoss,
+    private readonly handlerScannerService: HandlerScannerService,
+  ) {}
+
   static forRootAsync(options: PgBossModuleAsyncOptions): DynamicModule {
-    const logger = new Logger(PgBossModule.name);
+    const logger = new Logger(LOGGER);
 
-    const pgBossProvider: Provider = {
+    const pgBossProvider = {
       provide: PGBOSS_TOKEN,
       useFactory: async (pgBossOptions) => {
-        const boss = new PgBoss(pgBossOptions.connectionString);
-        await boss.start();
+        const boss = await lastValueFrom(
+          defer(() =>
+            new PgBoss({
+              connectionString: pgBossOptions.connectionString,
+            }).start(),
+          ).pipe(
+            handleRetry(
+              pgBossOptions.retryAttempts,
+              pgBossOptions.retryDelay,
+              pgBossOptions.verboseRetryLog,
+            ),
+          ),
+        );
         logger.log("PgBoss started successfully");
         return boss;
       },
@@ -36,14 +64,18 @@ export class PgBossModule {
     return {
       module: PgBossModule,
       imports: options.imports || [],
-      providers: [...asyncProviders, pgBossProvider, PgBossService],
-      exports: [PgBossService],
+      providers: [
+        ...asyncProviders,
+        pgBossProvider,
+        PgBossService,
+        HandlerScannerService,
+        MetadataScanner,
+      ],
+      exports: [PgBossService, PGBOSS_TOKEN],
     };
   }
 
-  private static createAsyncProviders(
-    options: PgBossModuleAsyncOptions,
-  ): Provider[] {
+  private static createAsyncProviders(options: PgBossModuleAsyncOptions) {
     if (options.useExisting || options.useFactory) {
       return [this.createAsyncOptionsProvider(options)];
     }
@@ -54,16 +86,11 @@ export class PgBossModule {
 
     return [
       this.createAsyncOptionsProvider(options),
-      {
-        provide: useClass,
-        useClass,
-      },
+      { provide: useClass, useClass },
     ];
   }
 
-  private static createAsyncOptionsProvider(
-    options: PgBossModuleAsyncOptions,
-  ): Provider {
+  private static createAsyncOptionsProvider(options: PgBossModuleAsyncOptions) {
     if (options.useFactory) {
       return {
         provide: PGBOSS_OPTIONS,
@@ -81,4 +108,24 @@ export class PgBossModule {
       inject,
     };
   }
+
+  onModuleInit() {}
+
+  async onApplicationBootstrap(): Promise<void> {
+    await this.setupWorkers();
+  }
+
+  async onModuleDestroy(): Promise<void> {
+    try {
+      if (this.boss) {
+        await this.boss.stop();
+      }
+    } catch (e) {
+      this.logger.error(e);
+    }
+  }
+
+  private async setupWorkers() {
+    await this.handlerScannerService.scanAndRegisterHandlers();
+  }
 }

package/lib/pgboss.service.ts

@@ -1,25 +1,27 @@
-import { Inject, Injectable } from "@nestjs/common";
-import PgBoss, { BatchWorkOptions, JobOptions, WorkHandler } from "pg-boss";
+import { Injectable } from "@nestjs/common";
+import PgBoss, { BatchWorkOptions, WorkHandler } from "pg-boss";
+import { Inject } from "@nestjs/common";
 import { PGBOSS_TOKEN } from "./utils/consts";
 
 @Injectable()
 export class PgBossService {
   constructor(@Inject(PGBOSS_TOKEN) private readonly boss: PgBoss) {}
 
-  async registerJob<TData extends object>(
+  async scheduleJob<TData extends object>(
     name: string,
-    handler: WorkHandler<TData>,
-    options?: BatchWorkOptions,
+    data: TData,
+    options?: PgBoss.SendOptions,
   ) {
-    await this.boss.work(name, options, handler);
+    await this.boss.send(name, data, options);
   }
 
-  async scheduleJob<TData extends object>(
+  async scheduleCronJob<TData extends object>(
     name: string,
-    data: TData,
-    options?: JobOptions,
+    cron: string,
+    data?: TData,
+    options?: PgBoss.ScheduleOptions,
   ) {
-    await this.boss.send(name, data, options);
+    await this.boss.schedule(name, cron, data ?? {}, options ?? {});
   }
 
   async registerCronJob<TData extends object>(
@@ -29,12 +31,15 @@ export class PgBossService {
     data?: TData,
     options?: PgBoss.ScheduleOptions,
   ) {
-    await this.boss.schedule(
-      name,
-      cron,
-      data ? data : {},
-      options ? options : {},
-    );
+    await this.boss.schedule(name, cron, data ?? {}, options ?? {});
     await this.boss.work(name, handler);
   }
+
+  async registerJob<TData extends object>(
+    name: string,
+    handler: WorkHandler<TData>,
+    options?: BatchWorkOptions,
+  ) {
+    await this.boss.work(name, options, handler);
+  }
 }

package/lib/utils/consts.ts

@@ -1,2 +1,5 @@
+export const PG_BOSS_JOB_METADATA = "PG_BOSS_JOB_METADATA";
+export const PGBOSS_TOKEN = Symbol("PGBOSS_TOKEN");
+export const MODULE_OPTIONS_TOKEN = Symbol("MODULE_OPTIONS_TOKEN");
 export const PGBOSS_OPTIONS = "PGBOSS_OPTIONS";
-export const PGBOSS_TOKEN = "PGBOSS_TOKEN";
+export const LOGGER = "nestjs-pgboss";

package/lib/utils/conversion-helper.ts

@@ -0,0 +1,10 @@
+import { JobOptions, BatchWorkOptions } from "pg-boss";
+
+export function convertToBatchWorkOptions(
+  jobOptions: JobOptions = {},
+): BatchWorkOptions {
+  return {
+    ...jobOptions,
+    batchSize: 1,
+  };
+}

package/lib/utils/handleRetry.ts

@@ -0,0 +1,30 @@
+import { of, throwError } from "rxjs";
+import { mergeMap, retryWhen, take, delay } from "rxjs/operators";
+
+export function handleRetry(
+  retryAttempts = 9,
+  retryDelay = 3000,
+  verbose = false,
+  toRetry: (err: any) => boolean = (_err: any) => true,
+) {
+  return <T>(source: import("rxjs").Observable<T>) =>
+    source.pipe(
+      retryWhen((attempts) =>
+        attempts.pipe(
+          take(retryAttempts),
+          mergeMap((error, index) => {
+            const includeError = toRetry(error);
+            if (verbose && includeError) {
+              console.warn(
+                `Attempt ${index + 1}: Retrying in ${
+                  retryDelay / 1000
+                } seconds...`,
+              );
+            }
+            return includeError ? of(error) : throwError(error);
+          }),
+          delay(retryDelay),
+        ),
+      ),
+    );
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wavezync/nestjs-pgboss",
-  "version": "1.0.0",
+  "version": "2.0.0",
   "description": "A NestJS module that integrates pg-boss for job scheduling and handling.",
   "license": "MIT",
   "author": "samaratungajs@wavezync.com",