@rockster/core 0.0.1 → 0.0.4

package/tasks/functions/get-database-connection-name.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-database-connection-name.js","sourceRoot":"./","sources":["tasks/functions/get-database-connection-name.ts"],"names":[],"mappings":";;AAAA,8DAIC;AAJD,SAAgB,yBAAyB,CAAC,UAAmB,KAAK;IAC/D,OAAO,OAAO;QACX,CAAC,CAAC,gBAAgB,OAAO,CAAC,GAAG,EAAE;QAC/B,CAAC,CAAC,eAAe,OAAO,CAAC,GAAG,EAAE,CAAC;AACrC,CAAC"}

package/tasks/functions/get-database-uri.js

@@ -6,3 +6,4 @@ const getDatabaseUri = (applicationName) => {
     return `postgresql://${options.username}:${options.password}@${options.host}:${options.port}/${options.database}?connection_timeout=${options.connectTimeoutMS}&application_name=${applicationName}`;
 };
 exports.getDatabaseUri = getDatabaseUri;
+//# sourceMappingURL=get-database-uri.js.map

package/tasks/functions/get-database-uri.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-database-uri.js","sourceRoot":"./","sources":["tasks/functions/get-database-uri.ts"],"names":[],"mappings":";;;AAAO,MAAM,cAAc,GAAG,CAAC,eAAuB,EAAE,EAAE;IACvD,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;IACtC,OAAO,gBAAgB,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,QAAQ,uBAAuB,OAAO,CAAC,gBAAgB,qBAAqB,eAAe,EAAE,CAAC;AACxM,CAAC,CAAA;AAHY,QAAA,cAAc,kBAG1B"}

package/tasks/functions/get-partitions-of-table-script.js

@@ -8,3 +8,4 @@ function getPartitionsOfTableScript(tableName) {
       WHERE inhparent = '${tableName}'::regclass;
    `;
 }
+//# sourceMappingURL=get-partitions-of-table-script.js.map

package/tasks/functions/get-partitions-of-table-script.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-partitions-of-table-script.js","sourceRoot":"./","sources":["tasks/functions/get-partitions-of-table-script.ts"],"names":[],"mappings":";;AAAA,gEAMC;AAND,SAAgB,0BAA0B,CAAC,SAAiB;IACzD,OAAQ;;;2BAGgB,SAAS;IAChC,CAAA;AACJ,CAAC"}

package/tasks/functions/get-pending-task-script.d.ts

@@ -1,2 +1,2 @@
 import { ITaskNaming } from "../interfaces/task-naming";
-export declare function getPendingTaskScript(naming: ITaskNaming): string;
+export declare function getPendingTaskScript(naming: ITaskNaming, partitionId: number, limit?: number): string;

package/tasks/functions/get-pending-task-script.js

@@ -2,18 +2,22 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.getPendingTaskScript = getPendingTaskScript;
 const tasks_1 = require("@rockster/common/tasks");
-function getPendingTaskScript(naming) {
+function getPendingTaskScript(naming, partitionId, limit = 50) {
     return `
-      WITH ${naming.table} AS (
-         SELECT * FROM ${naming.table}
-         WHERE status = '${tasks_1.TaskMessageStatus.pending}'
-         ORDER BY "createdAt" ASC
-         LIMIT 1000
-         FOR UPDATE SKIP LOCKED
-      )
-      UPDATE ${naming.table}
-      SET status = '${tasks_1.TaskMessageStatus.processing}', "lastModifiedAt" = (now())
-      WHERE id IN (SELECT id FROM ${naming.table})
-      RETURNING *;
-   `;
+    WITH picked AS (
+      SELECT id
+      FROM ${naming.table}_${partitionId}
+      WHERE status = '${tasks_1.TaskMessageStatus.pending}'
+      ORDER BY created_at ASC
+      LIMIT ${limit}
+      FOR UPDATE SKIP LOCKED
+    )
+    UPDATE ${naming.table}_${partitionId}
+    SET
+      status = '${tasks_1.TaskMessageStatus.processing}',
+      last_modified_at = now()
+    WHERE id IN (SELECT id FROM picked)
+    RETURNING *;
+  `;
 }
+//# sourceMappingURL=get-pending-task-script.js.map

package/tasks/functions/get-pending-task-script.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"get-pending-task-script.js","sourceRoot":"./","sources":["tasks/functions/get-pending-task-script.ts"],"names":[],"mappings":";;AAGA,oDAiBC;AApBD,kDAA2D;AAG3D,SAAgB,oBAAoB,CAAC,MAAmB,EAAE,WAAmB,EAAE,KAAK,GAAG,EAAE;IACtF,OAAO;;;aAGG,MAAM,CAAC,KAAK,IAAI,WAAW;wBAChB,yBAAiB,CAAC,OAAO;;cAEnC,KAAK;;;aAGN,MAAM,CAAC,KAAK,IAAI,WAAW;;kBAEtB,yBAAiB,CAAC,UAAU;;;;GAI3C,CAAC;AACJ,CAAC"}

package/tasks/functions/throw-task-error.js

@@ -5,3 +5,4 @@ const task_error_1 = require("../models/task-error");
 function throwTaskError(message, context) {
     throw new task_error_1.TaskError(message, context);
 }
+//# sourceMappingURL=throw-task-error.js.map

package/tasks/functions/throw-task-error.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"throw-task-error.js","sourceRoot":"./","sources":["tasks/functions/throw-task-error.ts"],"names":[],"mappings":";;AAEA,wCAKC;AAPD,qDAAiD;AAEjD,SAAgB,cAAc,CAC3B,OAAe,EACf,OAAgB;IAEhB,MAAM,IAAI,sBAAS,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;AACzC,CAAC"}

package/tasks/index.d.ts

@@ -1,3 +1,5 @@
 export * from './interfaces';
 export * from './decorators';
 export * from './functions/throw-task-error';
+export * from './tasks.module';
+export * from './models';

package/tasks/index.js

@@ -17,3 +17,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./interfaces"), exports);
 __exportStar(require("./decorators"), exports);
 __exportStar(require("./functions/throw-task-error"), exports);
+__exportStar(require("./tasks.module"), exports);
+__exportStar(require("./models"), exports);
+//# sourceMappingURL=index.js.map

package/tasks/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"./","sources":["tasks/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,+CAA6B;AAC7B,+CAA6B;AAC7B,+DAA6C;AAC7C,iDAA+B;AAC/B,2CAAyB"}

package/tasks/interfaces/index.js

@@ -20,3 +20,4 @@ __exportStar(require("./task-post-intercept"), exports);
 __exportStar(require("./task-runner"), exports);
 __exportStar(require("./task-search-intercept"), exports);
 __exportStar(require("./task-settings"), exports);
+//# sourceMappingURL=index.js.map

package/tasks/interfaces/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"./","sources":["tasks/interfaces/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gDAA8B;AAC9B,gDAA8B;AAC9B,wDAAsC;AACtC,gDAA8B;AAC9B,0DAAwC;AACxC,kDAAgC"}

package/tasks/interfaces/task-naming.d.ts

@@ -1,11 +1,11 @@
 export interface ITaskNaming {
     table: string;
-    trigger: string;
-    channel: string;
     postAction?: string;
     findAction?: string;
     retryAction?: string;
-    function: string;
+    queryAction?: string;
+    controller?: string;
     model?: string;
+    alias?: string;
     partition?: (partitionId: number) => ITaskNaming;
 }

package/tasks/interfaces/task-naming.js

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

package/tasks/interfaces/task-naming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-naming.js","sourceRoot":"./","sources":["tasks/interfaces/task-naming.ts"],"names":[],"mappings":""}

package/tasks/interfaces/task-object.d.ts

@@ -4,14 +4,16 @@ import { IObject } from "../../core";
 import { ITaskSettings } from "./task-settings";
 import { PgTableWithColumns } from "drizzle-orm/pg-core";
 import { ITaskNaming } from "./task-naming";
+import { TaskPostIntercept } from "./task-post-intercept";
 export interface ITaskObject extends IObject {
     type: ObjectType.task;
     settings: ITaskSettings<unknown>;
     payload: GetTypeCallback;
     MessageSchema: PgTableWithColumns<any>;
     MessageModel: Instantiable;
+    MessageController: Instantiable;
+    Query: MethodDecorator;
     naming: ITaskNaming;
     nextPartitionId: () => number;
-    postIntercept?: boolean;
-    searchIntercept?: boolean;
+    postInterceptors?: TaskPostIntercept[];
 }

package/tasks/interfaces/task-object.js

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

package/tasks/interfaces/task-object.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-object.js","sourceRoot":"./","sources":["tasks/interfaces/task-object.ts"],"names":[],"mappings":""}

package/tasks/interfaces/task-post-intercept.d.ts

@@ -1,8 +1 @@
-import { IRequestContext } from "../../core/interfaces/request-context";
-export type TaskOnPostArgs<T> = {
-    requestContext: IRequestContext;
-    messages: T[];
-};
-export interface ITaskPostIntercept<T> {
-    onPost(args: TaskOnPostArgs<T>): Promise<void>;
-}
+export type TaskPostIntercept = () => (target: any, propertyKey: string) => void;

package/tasks/interfaces/task-post-intercept.js

@@ -1,2 +1,3 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=task-post-intercept.js.map

package/tasks/interfaces/task-post-intercept.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-post-intercept.js","sourceRoot":"./","sources":["tasks/interfaces/task-post-intercept.ts"],"names":[],"mappings":""}

package/tasks/interfaces/task-runner.d.ts

@@ -1,5 +1,6 @@
+import { ITaskMessage } from "@rockster/common/tasks";
 export type TaskOnRun<T> = {
-    message: T;
+    message: ITaskMessage<T>;
 };
 export interface ITaskRunner<T> {
     onRun(args: TaskOnRun<T>): Promise<void>;

package/tasks/interfaces/task-runner.js

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

package/tasks/interfaces/task-runner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-runner.js","sourceRoot":"./","sources":["tasks/interfaces/task-runner.ts"],"names":[],"mappings":""}

package/tasks/interfaces/task-search-intercept.js

@@ -1,2 +1,3 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=task-search-intercept.js.map

package/tasks/interfaces/task-search-intercept.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-search-intercept.js","sourceRoot":"./","sources":["tasks/interfaces/task-search-intercept.ts"],"names":[],"mappings":""}

package/tasks/interfaces/task-settings.d.ts

@@ -1,7 +1,8 @@
 import { GetTypeCallback } from "@rockster/common";
+import { TaskPostIntercept } from "./task-post-intercept";
 export interface ITaskSettings<T> {
     /**Executor for any message */
-    payload: GetTypeCallback;
+    input: GetTypeCallback;
     /**Public action: false */
     public?: boolean;
     /**Default: 1 */
@@ -12,4 +13,5 @@ export interface ITaskSettings<T> {
     retentionDays?: number;
     /**limit: 100. Indicates how many messages the runner will recieve to proccess*/
     messagesLimit?: number;
+    postInterceptors?: TaskPostIntercept[];
 }

package/tasks/interfaces/task-settings.js

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

package/tasks/interfaces/task-settings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-settings.js","sourceRoot":"./","sources":["tasks/interfaces/task-settings.ts"],"names":[],"mappings":""}

package/tasks/models/index.d.ts

@@ -0,0 +1,3 @@
+export * from './task-error';
+export * from './task-message-find';
+export * from './task-message';

package/tasks/models/index.js

@@ -0,0 +1,20 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./task-error"), exports);
+__exportStar(require("./task-message-find"), exports);
+__exportStar(require("./task-message"), exports);
+//# sourceMappingURL=index.js.map

package/tasks/models/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"./","sources":["tasks/models/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,+CAA6B;AAC7B,sDAAoC;AACpC,iDAA+B"}

package/tasks/models/task-error.js

@@ -14,7 +14,7 @@ const class_pipe_1 = require("@rockster/class-pipe");
 class TaskError extends Error {
     constructor(message, context) {
         super(message);
-        this.context = context;
+        this.context = context ?? '';
     }
 }
 exports.TaskError = TaskError;
@@ -38,3 +38,4 @@ __decorate([
     (0, class_pipe_1.IsString)(),
     __metadata("design:type", String)
 ], TaskError.prototype, "infos", void 0);
+//# sourceMappingURL=task-error.js.map

package/tasks/models/task-error.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-error.js","sourceRoot":"./","sources":["tasks/models/task-error.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,qDAG8B;AAG9B,MAAa,SACV,SAAQ,KAAK;IAGb,YACG,OAAe,EACf,OAAgB;QAEhB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;IAChC,CAAC;CAiBH;AA3BD,8BA2BC;AAbE;IAFC,IAAA,mBAAM,GAAE;IACR,IAAA,qBAAQ,GAAE;;0CACK;AAIhB;IAFC,IAAA,mBAAM,GAAE;IACR,IAAA,qBAAQ,GAAE;;0CACM;AAIjB;IAFC,IAAA,mBAAM,GAAE;IACR,IAAA,qBAAQ,GAAE;;wCACI;AAIf;IAFC,IAAA,mBAAM,GAAE;IACR,IAAA,qBAAQ,GAAE;;wCACI"}

package/tasks/models/task-message-find.js

@@ -19,3 +19,4 @@ __decorate([
     (0, class_pipe_1.IsOptional)(),
     __metadata("design:type", String)
 ], TaskMessageFind.prototype, "search", void 0);
+//# sourceMappingURL=task-message-find.js.map

package/tasks/models/task-message-find.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-message-find.js","sourceRoot":"./","sources":["tasks/models/task-message-find.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,qDAA4D;AAE5D,MAAa,eAAe;CAK3B;AALD,0CAKC;AADE;IAFC,IAAA,qBAAQ,GAAE;IACV,IAAA,uBAAU,GAAE;;+CACG"}

package/tasks/models/task-message.d.ts

@@ -1,13 +1,13 @@
 import { ITaskMessage, TaskMessageStatus, ITaskError } from '@rockster/common/tasks';
-import { TaskError } from './task-error';
 export declare abstract class TaskMessage<T> implements ITaskMessage<T> {
     id?: string;
     status?: TaskMessageStatus;
     createdAt?: Date;
     lastModifiedAt?: Date;
     attempts?: number;
-    internalError?: TaskError;
+    internalError?: ITaskError;
     error?: ITaskError;
-    payload: T;
     partitionId: number;
+    runtimeMs?: number;
+    payload: T;
 }

package/tasks/models/task-message.js

@@ -13,45 +13,72 @@ exports.TaskMessage = void 0;
 const tasks_1 = require("@rockster/common/tasks");
 const class_pipe_1 = require("@rockster/class-pipe");
 const task_error_1 = require("./task-error");
+const column_decorator_1 = require("../../database/decorators/column.decorator");
+const id_decorator_1 = require("../../database/decorators/id.decorator");
 class TaskMessage {
 }
 exports.TaskMessage = TaskMessage;
 __decorate([
     (0, class_pipe_1.Expose)(),
     (0, class_pipe_1.IsRequired)(),
+    (0, class_pipe_1.IsString)(),
+    (0, id_decorator_1.Id)({ type: 'varchar' }),
     __metadata("design:type", String)
 ], TaskMessage.prototype, "id", void 0);
 __decorate([
     (0, class_pipe_1.Expose)(),
+    (0, class_pipe_1.IsIn)(tasks_1.TaskMessageStatus),
+    (0, column_decorator_1.Column)({
+        type: 'varchar',
+        default: tasks_1.TaskMessageStatus.pending
+    }),
     __metadata("design:type", String)
 ], TaskMessage.prototype, "status", void 0);
 __decorate([
     (0, class_pipe_1.Expose)(),
+    (0, class_pipe_1.IsDate)(),
+    (0, column_decorator_1.Column)({
+        type: 'timestamptz',
+        default: "NOW()"
+    }),
     __metadata("design:type", Date)
 ], TaskMessage.prototype, "createdAt", void 0);
 __decorate([
     (0, class_pipe_1.Expose)(),
+    (0, class_pipe_1.IsDate)(),
+    (0, column_decorator_1.Column)({
+        type: 'timestamptz',
+        default: "NOW()"
+    }),
     __metadata("design:type", Date)
 ], TaskMessage.prototype, "lastModifiedAt", void 0);
 __decorate([
     (0, class_pipe_1.Expose)(),
+    (0, class_pipe_1.IsInt)(),
+    (0, column_decorator_1.Column)({ nullable: true }),
     __metadata("design:type", Number)
 ], TaskMessage.prototype, "attempts", void 0);
 __decorate([
     (0, class_pipe_1.IsType)(() => task_error_1.TaskError),
-    __metadata("design:type", task_error_1.TaskError)
+    (0, column_decorator_1.Column)({ type: 'jsonb', nullable: true }),
+    __metadata("design:type", Object)
 ], TaskMessage.prototype, "internalError", void 0);
 __decorate([
     (0, class_pipe_1.Expose)(),
     (0, class_pipe_1.IsType)(() => task_error_1.TaskError),
+    (0, column_decorator_1.Column)({ type: 'jsonb', nullable: true }),
     __metadata("design:type", Object)
 ], TaskMessage.prototype, "error", void 0);
 __decorate([
     (0, class_pipe_1.Expose)(),
-    (0, class_pipe_1.IsRequired)(),
-    __metadata("design:type", Object)
-], TaskMessage.prototype, "payload", void 0);
+    (0, class_pipe_1.IsInt)(),
+    (0, id_decorator_1.Id)({ type: 'int' }),
+    __metadata("design:type", Number)
+], TaskMessage.prototype, "partitionId", void 0);
 __decorate([
     (0, class_pipe_1.Expose)(),
+    (0, class_pipe_1.IsInt)(),
+    (0, column_decorator_1.Column)({ nullable: true }),
     __metadata("design:type", Number)
-], TaskMessage.prototype, "partitionId", void 0);
+], TaskMessage.prototype, "runtimeMs", void 0);
+//# sourceMappingURL=task-message.js.map

package/tasks/models/task-message.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-message.js","sourceRoot":"./","sources":["tasks/models/task-message.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,kDAIgC;AAChC,qDAQ8B;AAC9B,6CAAyC;AACzC,iFAAoE;AACpE,yEAA4D;AAE5D,MAAsB,WAAW;CAyDhC;AAzDD,kCAyDC;AAnDE;IAJC,IAAA,mBAAM,GAAE;IACR,IAAA,uBAAU,GAAE;IACZ,IAAA,qBAAQ,GAAE;IACV,IAAA,iBAAE,EAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;;uCACZ;AAQZ;IANC,IAAA,mBAAM,GAAE;IACR,IAAA,iBAAI,EAAC,yBAAiB,CAAC;IACvB,IAAA,yBAAM,EAAC;QACL,IAAI,EAAE,SAAS;QACf,OAAO,EAAE,yBAAiB,CAAC,OAAO;KACpC,CAAC;;2CACyB;AAQ3B;IANC,IAAA,mBAAM,GAAE;IACR,IAAA,mBAAM,GAAE;IACR,IAAA,yBAAM,EAAC;QACL,IAAI,EAAE,aAAa;QACnB,OAAO,EAAE,OAAO;KAClB,CAAC;8BACU,IAAI;8CAAC;AAQjB;IANC,IAAA,mBAAM,GAAE;IACR,IAAA,mBAAM,GAAE;IACR,IAAA,yBAAM,EAAC;QACL,IAAI,EAAE,aAAa;QACnB,OAAO,EAAE,OAAO;KAClB,CAAC;8BACe,IAAI;mDAAC;AAKtB;IAHC,IAAA,mBAAM,GAAE;IACR,IAAA,kBAAK,GAAE;IACP,IAAA,yBAAM,EAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;6CACT;AAIlB;IAFC,IAAA,mBAAM,EAAC,GAAG,EAAE,CAAC,sBAAS,CAAC;IACvB,IAAA,yBAAM,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;kDACf;AAK3B;IAHC,IAAA,mBAAM,GAAE;IACR,IAAA,mBAAM,EAAC,GAAG,EAAE,CAAC,sBAAS,CAAC;IACvB,IAAA,yBAAM,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;0CACvB;AAKnB;IAHC,IAAA,mBAAM,GAAE;IACR,IAAA,kBAAK,GAAE;IACP,IAAA,iBAAE,EAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;;gDACA;AAKpB;IAHC,IAAA,mBAAM,GAAE;IACR,IAAA,kBAAK,GAAE;IACP,IAAA,yBAAM,EAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;;8CACR"}

package/tasks/services/task-manager.service.js

@@ -8,7 +8,6 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TaskManagerService = void 0;
 const class_injector_1 = require("@rockster/class-injector");
-const constants_1 = require("../constants");
 const get_check_table_exists_script_1 = require("../functions/get-check-table-exists-script");
 const get_create_table_script_1 = require("../functions/get-create-table-script");
 const get_partitions_of_table_script_1 = require("../functions/get-partitions-of-table-script");
@@ -23,8 +22,9 @@ let TaskManagerService = class TaskManagerService {
         const tableExists = await this
             .checkTableExists(tableName);
         if (!tableExists) {
+            await core.database.dataSource.query(`DROP TABLE ${tableName}`);
             const createTableScript = (0, get_create_table_script_1.getCreateTablescript)(tableName);
-            await constants_1.constants.database.execute(createTableScript);
+            await core.database.dataSource.query(createTableScript);
         }
     }
     async createPartitionsIfNotExists(object) {
@@ -35,26 +35,24 @@ let TaskManagerService = class TaskManagerService {
             if (exists)
                 continue;
             const createScript = (0, get_create_table_partition_script_1.getCreateTablePartitionScript)(object.naming, partitionId);
-            await constants_1.constants.database.execute(createScript);
+            await core.database.dataSource.query(createScript);
         }
     }
     async checkPartitionExists(naming, partitionId) {
         const tableName = naming.partition(partitionId).table;
         const script = (0, get_partitions_of_table_script_1.getPartitionsOfTableScript)(naming.table);
-        const result = await constants_1.constants.database.execute(script);
-        const partitions = result.rows;
+        const partitions = await core.database.dataSource.query(script);
         return partitions.find((partition) => {
             return partition.tableName === tableName;
         });
     }
     async checkTableExists(name) {
-        const result = await constants_1.constants
-            .database
-            .execute((0, get_check_table_exists_script_1.getCheckTableExistsScript)(name));
-        return result.rows[0].exists;
+        const result = await core.database.dataSource.query((0, get_check_table_exists_script_1.getCheckTableExistsScript)(name));
+        return result[0].exists;
     }
 };
 exports.TaskManagerService = TaskManagerService;
 exports.TaskManagerService = TaskManagerService = __decorate([
     (0, class_injector_1.Injectable)()
 ], TaskManagerService);
+//# sourceMappingURL=task-manager.service.js.map

package/tasks/services/task-manager.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"task-manager.service.js","sourceRoot":"./","sources":["tasks/services/task-manager.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,6DAAsD;AAGtD,8FAAuF;AACvF,kFAA4E;AAC5E,gGAAyF;AAEzF,sGAA+F;AAGxF,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IAE5B,KAAK,CAAC,SAAS,CAAC,MAA+B;QAC5C,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC7B,MAAM,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,CAAC;IAClD,CAAC;IAES,KAAK,CAAC,sBAAsB,CAAC,SAAiB;QACrD,MAAM,WAAW,GAAG,MAAM,IAAI;aAC1B,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAEhC,IAAI,CAAC,WAAW,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,cAAc,SAAS,EAAE,CAAC,CAAC;YAChE,MAAM,iBAAiB,GAAG,IAAA,8CAAoB,EAAC,SAAS,CAAC,CAAC;YAC1D,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;QAC3D,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,2BAA2B,CAAC,MAA+B;QACxE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;QACjC,KAAI,IAAI,WAAW,GAAG,CAAC,EAAE,WAAW,IAAI,QAAQ,CAAC,UAAU,EAAE,WAAW,EAAE,EAAE,CAAC;YAC1E,MAAM,MAAM,GAAG,MAAM,IAAI;iBACrB,oBAAoB,CAClB,MAAM,CAAC,MAAM,EACb,WAAW,CACb,CAAC;YACL,IAAI,MAAM;gBAAE,SAAS;YACrB,MAAM,YAAY,GAAG,IAAA,iEAA6B,EAC/C,MAAM,CAAC,MAAM,EACb,WAAW,CACb,CAAC;YACF,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QACtD,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,oBAAoB,CACjC,MAAmB,EACnB,WAAmB;QAEnB,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC;QACtD,MAAM,MAAM,GAAG,IAAA,2DAA0B,EAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACxD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAChE,OAAO,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE;YAClC,OAAO,SAAS,CAAC,SAAS,KAAK,SAAS,CAAC;QAC5C,CAAC,CAAC,CAAC;IACN,CAAC;IAES,KAAK,CAAC,gBAAgB,CAAC,IAAY;QAC1C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,IAAA,yDAAyB,EAAC,IAAI,CAAC,CAAC,CAAA;QACpF,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;IAC3B,CAAC;CACH,CAAA;AApDY,gDAAkB;6BAAlB,kBAAkB;IAD9B,IAAA,2BAAU,GAAE;GACA,kBAAkB,CAoD9B"}

package/tasks/services/task-runner.service.d.ts

@@ -2,15 +2,22 @@ import { ObjectNote } from "@rockster/class-memory";
 import { ITaskObject } from "../interfaces/task-object";
 import { ITaskRunner } from "../interfaces/task-runner";
 import { TaskMessage } from "../models/task-message";
+import { ITaskMessage } from "@rockster/common/tasks";
+import { Repository } from "typeorm";
 type TaskRunner = {
     object: ObjectNote<ITaskObject>;
     runner: ITaskRunner<any>;
+    repository: Repository<ITaskMessage>;
 };
 export declare class TaskRunnerService {
     private static running;
+    private static done;
     tasks: TaskRunner[];
     activeTimer(): void;
+    constructor();
     startRunner(object: ObjectNote<ITaskObject>): Promise<void>;
+    protected processMessages(): Promise<void>;
     protected eachMessage(task: TaskRunner, message: TaskMessage<any>): Promise<void>;
+    protected mapMessage(row: any): TaskMessage<any>;
 }
 export {};

package/tasks/services/task-runner.service.js

@@ -5,84 +5,106 @@ 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 TaskRunnerService_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TaskRunnerService = void 0;
 const get_pending_task_script_1 = require("../functions/get-pending-task-script");
-const constants_1 = require("../constants");
+const use_repository_1 = require("../../database/functions/use-repository");
 const tasks_1 = require("@rockster/common/tasks");
 const class_injector_1 = require("@rockster/class-injector");
-const drizzle_orm_1 = require("drizzle-orm");
 let TaskRunnerService = TaskRunnerService_1 = class TaskRunnerService {
+    activeTimer() {
+        TaskRunnerService_1.done = true;
+    }
     constructor() {
         this.tasks = [];
+        setInterval(() => this.processMessages(), 500);
     }
-    activeTimer() {
-        setInterval(async () => {
+    async startRunner(object) {
+        const runner = await (0, class_injector_1.createInstance)(object.objectClass);
+        const repository = (0, use_repository_1.useRepository)(object.MessageModel);
+        this.tasks.push({
+            object: object,
+            runner: runner,
+            repository: repository
+        });
+    }
+    async processMessages() {
+        try {
             if (TaskRunnerService_1.running)
                 return;
             TaskRunnerService_1.running = true;
-            const promises = [];
             for (const task of this.tasks) {
-                const query = (0, get_pending_task_script_1.getPendingTaskScript)(task.object.naming);
-                const result = await constants_1.constants.database.execute(query);
-                const messages = result.rows;
-                messages.forEach((message) => {
-                    promises.push(this.eachMessage(task, message));
-                });
+                const query = (0, get_pending_task_script_1.getPendingTaskScript)(task.object.naming, core.workerId);
+                const messages = (await task.repository.query(query))[0];
+                await Promise.all(messages.map((message) => this.eachMessage(task, this.mapMessage(message))));
             }
-            await Promise.all(promises);
             TaskRunnerService_1.running = false;
-        }, 500);
-    }
-    async startRunner(object) {
-        const runner = await (0, class_injector_1.createInstance)(object.objectClass);
-        this.tasks.push({
-            object: object,
-            runner: runner
-        });
+        }
+        catch (error) {
+            TaskRunnerService_1.running = false;
+        }
     }
     async eachMessage(task, message) {
-        const table = task
-            .object
-            .MessageSchema;
         const stopwatch = getTimer();
         try {
             stopwatch.start();
             await task
                 .runner
                 .onRun({
-                message: message.payload
+                message: message
             });
-            await constants_1.constants
-                .database
-                .update(table)
-                .set({
+            await task.repository.update({
+                id: message.id,
+                partitionId: message.partitionId
+            }, {
                 status: tasks_1.TaskMessageStatus.completed,
                 lastModifiedAt: new Date(),
-                runtimeMS: stopwatch.stop()
-            }).where((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(table.id, message.id), (0, drizzle_orm_1.eq)(table.partitionId, message.partitionId)));
+                runtimeMs: stopwatch.stop(),
+                payload: message.payload
+            });
         }
         catch (error) {
             const taskError = error;
             taskError.infos = getErrorInfos(error).message;
-            const isExpose = error.constructor.name === 'TaskError';
-            await constants_1.constants
-                .database
-                .update(table)
-                .set({
+            const isExpose = error.constructor.name !== 'Error';
+            await task.repository.update({
+                id: message.id,
+                partitionId: message.partitionId
+            }, {
                 status: tasks_1.TaskMessageStatus.failed,
                 lastModifiedAt: new Date(),
                 attempts: message.attempts + 1,
-                runtineMS: stopwatch.stop(),
+                runtimeMs: stopwatch.stop(),
                 internalError: taskError,
-                error: isExpose ? taskError : null
-            }).where((0, drizzle_orm_1.and)((0, drizzle_orm_1.eq)(table.id, message.id), (0, drizzle_orm_1.eq)(table.partitionId, message.partitionId)));
+                error: isExpose ? taskError : null,
+                payload: message.payload
+            });
         }
     }
+    mapMessage(row) {
+        return {
+            id: row.id,
+            partitionId: row.partition_id,
+            status: row.status,
+            createdAt: row.created_at,
+            lastModifiedAt: row.last_modified_at,
+            attempts: row.attempts,
+            error: row.error,
+            internalError: row.internal_error,
+            payload: row.payload,
+            runtimeMs: row.runtime_ms
+        };
+    }
 };
 exports.TaskRunnerService = TaskRunnerService;
 TaskRunnerService.running = false;
+TaskRunnerService.done = false;
 exports.TaskRunnerService = TaskRunnerService = TaskRunnerService_1 = __decorate([
-    (0, class_injector_1.Injectable)()
+    (0, class_injector_1.Injectable)(),
+    __metadata("design:paramtypes", [])
 ], TaskRunnerService);
+//# sourceMappingURL=task-runner.service.js.map