@betterinternship/core 2.5.18 → 2.5.20

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,4 +1,4 @@
1
- import { Task } from '@betterinternship/db';
1
+ import { Task, Process } from '@betterinternship/db';
2
2
  import { JsonValue } from '@prisma/client/runtime/client';
3
3
  export interface BaseJson {
4
4
  [key: string]: any;
@@ -23,7 +23,7 @@ export declare class BaseTaskInitiator<TaskInputs extends JsonValue, TaskResult
23
23
  private responseQueueId;
24
24
  constructor(queueId: string, handler: TaskResponseHandler<TaskResult>);
25
25
  init(handler: TaskResponseHandler<TaskResult>): Promise<void>;
26
- sendTaskRequest(processId: string, inputs: TaskInputs, options: {
26
+ sendTaskRequest(process: Process, inputs: TaskInputs, options: {
27
27
  internalId?: string;
28
28
  parentId?: string;
29
29
  maxRetries?: number;
@@ -35,6 +35,6 @@ export declare class BaseTaskRunner<TaskInputs extends JsonValue, TaskResult ext
35
35
  private responseQueueId;
36
36
  private handler;
37
37
  constructor(queueId: string, handler: TaskRequestHandler<TaskInputs, TaskResult>);
38
- listen(processId?: string): Promise<void>;
38
+ listen(processName?: string): Promise<void>;
39
39
  }
40
40
  export {};
@@ -26,7 +26,7 @@ export class BaseTaskInitiator {
26
26
  }
27
27
  async init(handler) {
28
28
  const brokerAPI = await BrokerAPI.instance();
29
- return await brokerAPI.addQueueHandler(this.responseQueueId, async (taskId) => {
29
+ return await brokerAPI.registerQueueHandler(this.responseQueueId, async (taskId) => {
30
30
  const task = await tasks.findFirst({ where: { id: taskId } });
31
31
  await handler({
32
32
  result: task?.result,
@@ -35,14 +35,15 @@ export class BaseTaskInitiator {
35
35
  });
36
36
  });
37
37
  }
38
- async sendTaskRequest(processId, inputs, options) {
38
+ async sendTaskRequest(process, inputs, options) {
39
39
  const brokerAPI = await BrokerAPI.instance();
40
40
  const task = await tasks.create({
41
41
  data: {
42
- internal_id: options.internalId ?? `${processId}.${v4()}`,
42
+ internal_id: options.internalId ?? `${process.id}.${v4()}`,
43
43
  parent_id: options.parentId,
44
44
  handler_id: this.handlerId,
45
- process_id: processId,
45
+ process_id: process.id,
46
+ process_name: process.name,
46
47
  inputs: inputs,
47
48
  },
48
49
  });
@@ -58,11 +59,13 @@ export class BaseTaskRunner {
58
59
  this.responseQueueId = `response.${queueId}`;
59
60
  this.handler = handler;
60
61
  }
61
- async listen(processId) {
62
- const processTask = async (taskId) => {
62
+ async listen(processName) {
63
+ const taskRunner = async (taskId, actions) => {
63
64
  const task = await tasks.findFirst({ where: { id: taskId } });
64
- if (processId && processId !== task?.process_id)
65
+ if (processName && processName !== task?.process_name)
65
66
  return;
67
+ if (task?.parent_id && !task?.inputs)
68
+ return actions.hold();
66
69
  const response = await this.handler(task);
67
70
  await tasks.update({
68
71
  where: { id: taskId },
@@ -72,10 +75,14 @@ export class BaseTaskRunner {
72
75
  status: response.status,
73
76
  },
74
77
  });
78
+ if (response.error)
79
+ actions.reject();
80
+ else
81
+ actions.resolve();
75
82
  await brokerAPI.addToQueue(this.responseQueueId, taskId);
76
83
  };
77
84
  const brokerAPI = await BrokerAPI.instance();
78
- await brokerAPI.addQueueHandler(this.requestQueueId, processTask);
85
+ await brokerAPI.registerQueueHandler(this.requestQueueId, taskRunner);
79
86
  }
80
87
  }
81
88
  //# sourceMappingURL=base.task.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"base.task.js","sourceRoot":"","sources":["../../../lib/broker/base.task.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,KAAK,EAAQ,MAAM,sBAAsB,CAAC;AACnD,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAO1B,MAAM,OAAO,WAAW;IAYtB,MAAM,CAAa;IACnB,WAAW,CAAS;IACpB,UAAU,CAAS;IACnB,SAAS,CAAgB;IACzB,UAAU,CAAS;IACnB,SAAS,CAAO;CACjB;AAED,MAAM,OAAO,YAAY;IAGvB,MAAM,CAAa;IACnB,KAAK,CAAS;IACd,MAAM,CAAS;CAChB;AAUD,MAAM,OAAO,iBAAiB;IAIpB,SAAS,CAAS;IAClB,cAAc,CAAS;IACvB,eAAe,CAAS;IAQhC,YAAY,OAAe,EAAE,OAAwC;QACnE,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,WAAW,OAAO,EAAE,CAAC;QAC3C,IAAI,CAAC,eAAe,GAAG,YAAY,OAAO,EAAE,CAAC;QAC7C,KAAK,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;IAQD,KAAK,CAAC,IAAI,CAAC,OAAwC;QACjD,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7C,OAAO,MAAM,SAAU,CAAC,eAAe,CACrC,IAAI,CAAC,eAAe,EACpB,KAAK,EAAE,MAAc,EAAE,EAAE;YACvB,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAC9D,MAAM,OAAO,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,MAAoB;gBAClC,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE;gBACxB,MAAM,EAAE,IAAI,EAAE,MAAM,IAAI,QAAQ;aACjC,CAAC,CAAC;QACL,CAAC,CACF,CAAC;IACJ,CAAC;IAWD,KAAK,CAAC,eAAe,CACnB,SAAiB,EACjB,MAAkB,EAClB,OAIC;QAED,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC;YAC9B,IAAI,EAAE;gBACJ,WAAW,EAAE,OAAO,CAAC,UAAU,IAAI,GAAG,SAAS,IAAI,EAAE,EAAE,EAAE;gBACzD,SAAS,EAAE,OAAO,CAAC,QAAQ;gBAC3B,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,UAAU,EAAE,SAAS;gBACrB,MAAM,EAAE,MAAwB;aACjC;SACF,CAAC,CAAC;QAEH,OAAO,SAAU,CAAC,UAAU,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACtE,CAAC;CACF;AAUD,MAAM,OAAO,cAAc;IAIjB,cAAc,CAAS;IACvB,eAAe,CAAS;IACxB,OAAO,CAA6C;IAQ5D,YACE,OAAe,EACf,OAAmD;QAEnD,IAAI,CAAC,cAAc,GAAG,WAAW,OAAO,EAAE,CAAC;QAC3C,IAAI,CAAC,eAAe,GAAG,YAAY,OAAO,EAAE,CAAC;QAC7C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAOD,KAAK,CAAC,MAAM,CAAC,SAAkB;QAE7B,MAAM,WAAW,GAAG,KAAK,EAAE,MAAc,EAAE,EAAE;YAC3C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAG9D,IAAI,SAAS,IAAI,SAAS,KAAK,IAAI,EAAE,UAAU;gBAAE,OAAO;YAGxD,MAAM,QAAQ,GAA6B,MAAM,IAAI,CAAC,OAAO,CAC3D,IAA0C,CAC3C,CAAC;YACF,MAAM,KAAK,CAAC,MAAM,CAAC;gBACjB,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,EAAE;oBAC7B,KAAK,EAAE,QAAQ,CAAC,KAAK,IAAI,EAAE;oBAC3B,MAAM,EAAE,QAAQ,CAAC,MAAM;iBACxB;aACF,CAAC,CAAC;YACH,MAAM,SAAU,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QAC5D,CAAC,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7C,MAAM,SAAU,CAAC,eAAe,CAAC,IAAI,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;IACrE,CAAC;CACF"}
1
+ {"version":3,"file":"base.task.js","sourceRoot":"","sources":["../../../lib/broker/base.task.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,SAAS,EAA0C,MAAM,aAAa,CAAC;AAChF,OAAO,EAAE,KAAK,EAAiB,MAAM,sBAAsB,CAAC;AAC5D,OAAO,EAAE,EAAE,EAAE,MAAM,MAAM,CAAC;AAO1B,MAAM,OAAO,WAAW;IAYtB,MAAM,CAAa;IACnB,WAAW,CAAS;IACpB,UAAU,CAAS;IACnB,SAAS,CAAgB;IACzB,UAAU,CAAS;IACnB,SAAS,CAAO;CACjB;AAED,MAAM,OAAO,YAAY;IAGvB,MAAM,CAAa;IACnB,KAAK,CAAS;IACd,MAAM,CAAS;CAChB;AAUD,MAAM,OAAO,iBAAiB;IAIpB,SAAS,CAAS;IAClB,cAAc,CAAS;IACvB,eAAe,CAAS;IAQhC,YAAY,OAAe,EAAE,OAAwC;QACnE,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,WAAW,OAAO,EAAE,CAAC;QAC3C,IAAI,CAAC,eAAe,GAAG,YAAY,OAAO,EAAE,CAAC;QAC7C,KAAK,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1B,CAAC;IAQD,KAAK,CAAC,IAAI,CAAC,OAAwC;QACjD,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7C,OAAO,MAAM,SAAU,CAAC,oBAAoB,CAC1C,IAAI,CAAC,eAAe,EACpB,KAAK,EAAE,MAAc,EAAE,EAAE;YACvB,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAC9D,MAAM,OAAO,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,MAAoB;gBAClC,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,EAAE;gBACxB,MAAM,EAAE,IAAI,EAAE,MAAM,IAAI,QAAQ;aACjC,CAAC,CAAC;QACL,CAAC,CACF,CAAC;IACJ,CAAC;IAWD,KAAK,CAAC,eAAe,CACnB,OAAgB,EAChB,MAAkB,EAClB,OAIC;QAED,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7C,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC;YAC9B,IAAI,EAAE;gBACJ,WAAW,EAAE,OAAO,CAAC,UAAU,IAAI,GAAG,OAAO,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE;gBAC1D,SAAS,EAAE,OAAO,CAAC,QAAQ;gBAC3B,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,UAAU,EAAE,OAAO,CAAC,EAAE;gBACtB,YAAY,EAAE,OAAO,CAAC,IAAI;gBAC1B,MAAM,EAAE,MAAwB;aACjC;SACF,CAAC,CAAC;QAEH,OAAO,SAAU,CAAC,UAAU,CAAC,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACtE,CAAC;CACF;AAUD,MAAM,OAAO,cAAc;IAIjB,cAAc,CAAS;IACvB,eAAe,CAAS;IACxB,OAAO,CAA6C;IAQ5D,YACE,OAAe,EACf,OAAmD;QAEnD,IAAI,CAAC,cAAc,GAAG,WAAW,OAAO,EAAE,CAAC;QAC3C,IAAI,CAAC,eAAe,GAAG,YAAY,OAAO,EAAE,CAAC;QAC7C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAOD,KAAK,CAAC,MAAM,CAAC,WAAoB;QAE/B,MAAM,UAAU,GAAG,KAAK,EACtB,MAAc,EACd,OAIC,EACD,EAAE;YACF,MAAM,IAAI,GAAG,MAAM,KAAK,CAAC,SAAS,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,CAAC,CAAC;YAG9D,IAAI,WAAW,IAAI,WAAW,KAAK,IAAI,EAAE,YAAY;gBAAE,OAAO;YAC9D,IAAI,IAAI,EAAE,SAAS,IAAI,CAAC,IAAI,EAAE,MAAM;gBAAE,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;YAG5D,MAAM,QAAQ,GAA6B,MAAM,IAAI,CAAC,OAAO,CAC3D,IAA0C,CAC3C,CAAC;YACF,MAAM,KAAK,CAAC,MAAM,CAAC;gBACjB,KAAK,EAAE,EAAE,EAAE,EAAE,MAAM,EAAE;gBACrB,IAAI,EAAE;oBACJ,MAAM,EAAE,QAAQ,CAAC,MAAM,IAAI,EAAE;oBAC7B,KAAK,EAAE,QAAQ,CAAC,KAAK,IAAI,EAAE;oBAC3B,MAAM,EAAE,QAAQ,CAAC,MAAM;iBACxB;aACF,CAAC,CAAC;YAGH,IAAI,QAAQ,CAAC,KAAK;gBAAE,OAAO,CAAC,MAAM,EAAE,CAAC;;gBAChC,OAAO,CAAC,OAAO,EAAE,CAAC;YAGvB,MAAM,SAAU,CAAC,UAAU,CAAC,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QAC5D,CAAC,CAAC;QAEF,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7C,MAAM,SAAU,CAAC,oBAAoB,CAAC,IAAI,CAAC,cAAc,EAAE,UAAU,CAAC,CAAC;IACzE,CAAC;CACF"}
@@ -1,19 +1,28 @@
1
- import * as amqp from 'amqplib';
2
- export declare const Q: (...args: string[]) => string;
3
- export interface RequestBody {
4
- taskId: string;
5
- }
6
- export declare class BrokerAPI {
7
- connection: amqp.ChannelModel;
8
- channels: Record<string, amqp.Channel>;
9
- private static __instance;
10
- private constructor();
11
- static instance(): Promise<BrokerAPI | null>;
12
- private assertQueue;
13
- private addToQueueWithHeaders;
14
- addToQueue(queueId: string, taskId: string, options?: {
15
- retries?: number;
16
- maxRetries?: number;
17
- }): Promise<boolean>;
18
- addQueueHandler<T>(queueId: string, handler: (taskId: string, resolve: () => void, reject: () => void) => Promise<T>): Promise<void>;
19
- }
1
+ import * as amqp from 'amqplib';
2
+ export declare const Q: (...args: string[]) => string;
3
+ export interface RequestBody {
4
+ taskId: string;
5
+ }
6
+ export type TaskResolver = () => void;
7
+ export type TaskRejecter = () => void;
8
+ export type TaskHolder = () => void;
9
+ export type TaskHandler<T> = (taskId: string, actions: {
10
+ resolve: TaskResolver;
11
+ reject: TaskRejecter;
12
+ hold: TaskHolder;
13
+ }) => Promise<T>;
14
+ export declare class BrokerAPI {
15
+ connection: amqp.ChannelModel;
16
+ channels: Record<string, amqp.Channel>;
17
+ handlers: Record<string, TaskHandler<any>[]>;
18
+ private static __instance;
19
+ private constructor();
20
+ static instance(): Promise<BrokerAPI | null>;
21
+ private assertQueue;
22
+ private addToQueueWithHeaders;
23
+ addToQueue(queueId: string, taskId: string, options?: {
24
+ retries?: number;
25
+ maxRetries?: number;
26
+ }): Promise<boolean>;
27
+ registerQueueHandler<T>(queueId: string, handler: TaskHandler<T>): Promise<void>;
28
+ }
@@ -1,95 +1,112 @@
1
- import * as amqp from 'amqplib';
2
- import { ENV } from '../env.js';
3
- const RABBITMQ_ENABLED = ENV.RABBITMQ_ENABLED;
4
- const RABBITMQ_URL = ENV.RABBITMQ_URL;
5
- if (!RABBITMQ_URL && RABBITMQ_ENABLED)
6
- throw new Error('[ERROR:ENV] Missing RabbitMQ configuration.');
7
- export const Q = (...args) => {
8
- if (ENV.ENVIRONMENT === 'production')
9
- return args.join('.');
10
- else
11
- return ['dev', ...args].join('.');
12
- };
13
- export class BrokerAPI {
14
- connection;
15
- channels;
16
- static __instance = null;
17
- constructor() { }
18
- static async instance() {
19
- if (!RABBITMQ_ENABLED)
20
- return Promise.resolve(null);
21
- if (BrokerAPI.__instance)
22
- return Promise.resolve(BrokerAPI.__instance);
23
- const instance = new BrokerAPI();
24
- instance.connection = await amqp.connect(RABBITMQ_URL);
25
- instance.channels = {
26
- asserter: await instance.connection.createChannel(),
27
- reader: await instance.connection.createChannel(),
28
- writer: await instance.connection.createChannel(),
29
- };
30
- BrokerAPI.__instance = instance;
31
- return instance;
32
- }
33
- async assertQueue(queueId, options) {
34
- const result = await this.channels.asserter.assertQueue(queueId, {
35
- durable: true,
36
- ...options,
37
- });
38
- console.log(`[BROKER] Queue "${queueId}" has been asserted.`);
39
- return result;
40
- }
41
- async addToQueueWithHeaders(queueId, taskId, headers, options) {
42
- await this.assertQueue(queueId);
43
- const bodyStr = JSON.stringify({ taskId });
44
- const result = this.channels.writer.sendToQueue(queueId, Buffer.from(bodyStr), {
45
- persistent: true,
46
- contentType: 'application/json',
47
- headers: {
48
- ...headers,
49
- 'x-retries': options?.retries ?? 0,
50
- 'x-max-retries': options?.maxRetries ?? 3,
51
- 'x-first-published': Date.now(),
52
- },
53
- messageId: taskId,
54
- });
55
- if (result) {
56
- console.log(`[BROKER] Message sent to queue "${queueId}": ${bodyStr}.`);
57
- }
58
- else {
59
- console.log(`[BROKER] Message could not sent to queue "${queueId}": ${bodyStr}.`);
60
- }
61
- return Promise.resolve(result);
62
- }
63
- async addToQueue(queueId, taskId, options) {
64
- await this.assertQueue(queueId);
65
- return this.addToQueueWithHeaders(queueId, taskId, {}, options);
66
- }
67
- async addQueueHandler(queueId, handler) {
68
- await this.assertQueue(queueId);
69
- const resolve = (message) => () => {
70
- this.channels.writer.ack(message);
71
- };
72
- const reject = (message) => () => {
73
- const content = message?.content?.toString?.();
74
- const request = JSON.parse(content);
75
- const headers = message.properties.headers ?? {};
76
- const retries = headers['x-retries'] ?? 0;
77
- const maxRetries = headers['x-max-retries'] ?? 3;
78
- if (retries >= maxRetries) {
79
- this.channels.writer.ack(message);
80
- }
81
- else {
82
- void this.addToQueueWithHeaders(queueId, request.taskId, headers, {
83
- retries,
84
- maxRetries,
85
- }).then(() => this.channels.writer.ack(message));
86
- }
87
- };
88
- await this.channels.reader.consume(queueId, (message) => {
89
- const content = message?.content?.toString?.();
90
- const request = JSON.parse(content);
91
- void handler(request.taskId, resolve(message), reject(message));
92
- });
93
- }
94
- }
1
+ import * as amqp from 'amqplib';
2
+ import { ENV } from '../env.js';
3
+ const RABBITMQ_ENABLED = ENV.RABBITMQ_ENABLED;
4
+ const RABBITMQ_URL = ENV.RABBITMQ_URL;
5
+ if (!RABBITMQ_URL && RABBITMQ_ENABLED)
6
+ throw new Error('[ERROR:ENV] Missing RabbitMQ configuration.');
7
+ export const Q = (...args) => {
8
+ if (ENV.ENVIRONMENT === 'production')
9
+ return args.join('.');
10
+ else
11
+ return ['dev', ...args].join('.');
12
+ };
13
+ export class BrokerAPI {
14
+ connection;
15
+ channels;
16
+ handlers;
17
+ static __instance = null;
18
+ constructor() { }
19
+ static async instance() {
20
+ if (!RABBITMQ_ENABLED)
21
+ return Promise.resolve(null);
22
+ if (BrokerAPI.__instance)
23
+ return Promise.resolve(BrokerAPI.__instance);
24
+ const instance = new BrokerAPI();
25
+ instance.connection = await amqp.connect(RABBITMQ_URL);
26
+ instance.channels = {
27
+ asserter: await instance.connection.createChannel(),
28
+ reader: await instance.connection.createChannel(),
29
+ writer: await instance.connection.createChannel(),
30
+ };
31
+ BrokerAPI.__instance = instance;
32
+ return instance;
33
+ }
34
+ async assertQueue(queueId, options) {
35
+ const result = await this.channels.asserter.assertQueue(queueId, {
36
+ durable: true,
37
+ ...options,
38
+ });
39
+ console.log(`[BROKER] Queue "${queueId}" has been asserted.`);
40
+ return result;
41
+ }
42
+ async addToQueueWithHeaders(queueId, taskId, headers, options) {
43
+ await this.assertQueue(queueId);
44
+ const bodyStr = JSON.stringify({ taskId });
45
+ const result = this.channels.writer.sendToQueue(queueId, Buffer.from(bodyStr), {
46
+ persistent: true,
47
+ contentType: 'application/json',
48
+ headers: {
49
+ ...headers,
50
+ 'x-retries': options?.retries ?? 0,
51
+ 'x-max-retries': options?.maxRetries ?? 3,
52
+ 'x-first-published': Date.now(),
53
+ },
54
+ messageId: taskId,
55
+ });
56
+ if (result) {
57
+ console.log(`[BROKER] Message sent to queue "${queueId}": ${bodyStr}.`);
58
+ }
59
+ else {
60
+ console.log(`[BROKER] Message could not sent to queue "${queueId}": ${bodyStr}.`);
61
+ }
62
+ return Promise.resolve(result);
63
+ }
64
+ async addToQueue(queueId, taskId, options) {
65
+ await this.assertQueue(queueId);
66
+ return this.addToQueueWithHeaders(queueId, taskId, {}, options);
67
+ }
68
+ async registerQueueHandler(queueId, handler) {
69
+ await this.assertQueue(queueId);
70
+ if (Object.keys(this.handlers).includes(queueId)) {
71
+ this.handlers[queueId].push(handler);
72
+ return;
73
+ }
74
+ const brokerAPI = this;
75
+ this.handlers[queueId] = [handler];
76
+ const resolve = (message) => () => {
77
+ this.channels.writer.ack(message);
78
+ };
79
+ const hold = (message) => () => {
80
+ this.channels.writer.nack(message, false, true);
81
+ };
82
+ const reject = (queueId, message) => () => {
83
+ const content = message?.content?.toString?.();
84
+ const request = JSON.parse(content);
85
+ const headers = message.properties.headers ?? {};
86
+ const retries = headers['x-retries'] ?? 0;
87
+ const maxRetries = headers['x-max-retries'] ?? 3;
88
+ if (retries >= maxRetries) {
89
+ this.channels.writer.ack(message);
90
+ }
91
+ else {
92
+ void this.addToQueueWithHeaders(queueId, request.taskId, headers, {
93
+ retries: retries + 1,
94
+ maxRetries,
95
+ }).then(() => this.channels.writer.ack(message));
96
+ }
97
+ };
98
+ await this.channels.reader.consume(queueId, (message) => {
99
+ const handlers = brokerAPI.handlers;
100
+ const content = message?.content?.toString?.();
101
+ const request = JSON.parse(content);
102
+ for (const handler of handlers[queueId]) {
103
+ void handler(request.taskId, {
104
+ resolve: resolve(message),
105
+ reject: reject(queueId, message),
106
+ hold: hold(message),
107
+ });
108
+ }
109
+ });
110
+ }
111
+ }
95
112
  //# sourceMappingURL=broker.js.map
@@ -1 +1 @@
1
- {"version":3,"file":"broker.js","sourceRoot":"","sources":["../../../lib/broker/broker.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAEhC,MAAM,gBAAgB,GAAG,GAAG,CAAC,gBAAgB,CAAC;AAC9C,MAAM,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC;AACtC,IAAI,CAAC,YAAY,IAAI,gBAAgB;IACnC,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;AAMjE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAc,EAAU,EAAE;IAC7C,IAAI,GAAG,CAAC,WAAW,KAAK,YAAY;QAAE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;;QACvD,OAAO,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzC,CAAC,CAAC;AAaF,MAAM,OAAO,SAAS;IAEpB,UAAU,CAAoB;IAC9B,QAAQ,CAA+B;IAG/B,MAAM,CAAC,UAAU,GAAqB,IAAI,CAAC;IAGnD,gBAAuB,CAAC;IAOxB,MAAM,CAAC,KAAK,CAAC,QAAQ;QACnB,IAAI,CAAC,gBAAgB;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,SAAS,CAAC,UAAU;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAIvE,MAAM,QAAQ,GAAG,IAAI,SAAS,EAAE,CAAC;QACjC,QAAQ,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAa,CAAC,CAAC;QAIxD,QAAQ,CAAC,QAAQ,GAAG;YAClB,QAAQ,EAAE,MAAM,QAAQ,CAAC,UAAU,CAAC,aAAa,EAAE;YACnD,MAAM,EAAE,MAAM,QAAQ,CAAC,UAAU,CAAC,aAAa,EAAE;YACjD,MAAM,EAAE,MAAM,QAAQ,CAAC,UAAU,CAAC,aAAa,EAAE;SAClD,CAAC;QAEF,SAAS,CAAC,UAAU,GAAG,QAAQ,CAAC;QAChC,OAAO,QAAQ,CAAC;IAClB,CAAC;IASO,KAAK,CAAC,WAAW,CACvB,OAAe,EACf,OAAkC;QAElC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,OAAO,EAAE;YAC/D,OAAO,EAAE,IAAI;YACb,GAAG,OAAO;SACX,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,mBAAmB,OAAO,sBAAsB,CAAC,CAAC;QAC9D,OAAO,MAAM,CAAC;IAChB,CAAC;IASO,KAAK,CAAC,qBAAqB,CACjC,OAAe,EACf,MAAc,EACd,OAA4B,EAC5B,OAAmD;QAEnD,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,WAAW,CAC7C,OAAO,EACP,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EACpB;YACE,UAAU,EAAE,IAAI;YAChB,WAAW,EAAE,kBAAkB;YAC/B,OAAO,EAAE;gBACP,GAAG,OAAO;gBACV,WAAW,EAAE,OAAO,EAAE,OAAO,IAAI,CAAC;gBAClC,eAAe,EAAE,OAAO,EAAE,UAAU,IAAI,CAAC;gBACzC,mBAAmB,EAAE,IAAI,CAAC,GAAG,EAAE;aAChC;YACD,SAAS,EAAE,MAAM;SAClB,CACF,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,mCAAmC,OAAO,MAAM,OAAO,GAAG,CAAC,CAAC;QAC1E,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CACT,6CAA6C,OAAO,MAAM,OAAO,GAAG,CACrE,CAAC;QACJ,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAAe,EACf,MAAc,EACd,OAAmD;QAEnD,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAChC,OAAO,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IAClE,CAAC;IASD,KAAK,CAAC,eAAe,CACnB,OAAe,EACf,OAIe;QAEf,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAGhC,MAAM,OAAO,GAAG,CAAC,OAA4B,EAAE,EAAE,CAAC,GAAG,EAAE;YACrD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC,CAAC;QAGF,MAAM,MAAM,GAAG,CAAC,OAA4B,EAAE,EAAE,CAAC,GAAG,EAAE;YACpD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAgB,CAAC;YACnD,MAAM,OAAO,GAAwB,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,EAAE,CAAC;YACtE,MAAM,OAAO,GAAI,OAAO,CAAC,WAAW,CAAY,IAAI,CAAC,CAAC;YACtD,MAAM,UAAU,GAAI,OAAO,CAAC,eAAe,CAAY,IAAI,CAAC,CAAC;YAG7D,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;gBAC1B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,KAAK,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE;oBAChE,OAAO;oBACP,UAAU;iBACX,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACnD,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAChC,OAAO,EACP,CAAC,OAA4B,EAAE,EAAE;YAC/B,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAgB,CAAC;YACnD,KAAK,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QAClE,CAAC,CACF,CAAC;IACJ,CAAC"}
1
+ {"version":3,"file":"broker.js","sourceRoot":"","sources":["../../../lib/broker/broker.ts"],"names":[],"mappings":"AASA,OAAO,KAAK,IAAI,MAAM,SAAS,CAAC;AAChC,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAEhC,MAAM,gBAAgB,GAAG,GAAG,CAAC,gBAAgB,CAAC;AAC9C,MAAM,YAAY,GAAG,GAAG,CAAC,YAAY,CAAC;AACtC,IAAI,CAAC,YAAY,IAAI,gBAAgB;IACnC,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;AAMjE,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAc,EAAU,EAAE;IAC7C,IAAI,GAAG,CAAC,WAAW,KAAK,YAAY;QAAE,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;;QACvD,OAAO,CAAC,KAAK,EAAE,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AACzC,CAAC,CAAC;AAyBF,MAAM,OAAO,SAAS;IAEpB,UAAU,CAAoB;IAC9B,QAAQ,CAA+B;IACvC,QAAQ,CAAqC;IAGrC,MAAM,CAAC,UAAU,GAAqB,IAAI,CAAC;IAGnD,gBAAuB,CAAC;IAOxB,MAAM,CAAC,KAAK,CAAC,QAAQ;QACnB,IAAI,CAAC,gBAAgB;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,SAAS,CAAC,UAAU;YAAE,OAAO,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAIvE,MAAM,QAAQ,GAAG,IAAI,SAAS,EAAE,CAAC;QACjC,QAAQ,CAAC,UAAU,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAa,CAAC,CAAC;QAIxD,QAAQ,CAAC,QAAQ,GAAG;YAClB,QAAQ,EAAE,MAAM,QAAQ,CAAC,UAAU,CAAC,aAAa,EAAE;YACnD,MAAM,EAAE,MAAM,QAAQ,CAAC,UAAU,CAAC,aAAa,EAAE;YACjD,MAAM,EAAE,MAAM,QAAQ,CAAC,UAAU,CAAC,aAAa,EAAE;SAClD,CAAC;QAEF,SAAS,CAAC,UAAU,GAAG,QAAQ,CAAC;QAChC,OAAO,QAAQ,CAAC;IAClB,CAAC;IASO,KAAK,CAAC,WAAW,CACvB,OAAe,EACf,OAAkC;QAElC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC,OAAO,EAAE;YAC/D,OAAO,EAAE,IAAI;YACb,GAAG,OAAO;SACX,CAAC,CAAC;QACH,OAAO,CAAC,GAAG,CAAC,mBAAmB,OAAO,sBAAsB,CAAC,CAAC;QAC9D,OAAO,MAAM,CAAC;IAChB,CAAC;IASO,KAAK,CAAC,qBAAqB,CACjC,OAAe,EACf,MAAc,EACd,OAA4B,EAC5B,OAAmD;QAEnD,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,WAAW,CAC7C,OAAO,EACP,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,EACpB;YACE,UAAU,EAAE,IAAI;YAChB,WAAW,EAAE,kBAAkB;YAC/B,OAAO,EAAE;gBACP,GAAG,OAAO;gBACV,WAAW,EAAE,OAAO,EAAE,OAAO,IAAI,CAAC;gBAClC,eAAe,EAAE,OAAO,EAAE,UAAU,IAAI,CAAC;gBACzC,mBAAmB,EAAE,IAAI,CAAC,GAAG,EAAE;aAChC;YACD,SAAS,EAAE,MAAM;SAClB,CACF,CAAC;QAEF,IAAI,MAAM,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,mCAAmC,OAAO,MAAM,OAAO,GAAG,CAAC,CAAC;QAC1E,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,GAAG,CACT,6CAA6C,OAAO,MAAM,OAAO,GAAG,CACrE,CAAC;QACJ,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,UAAU,CACd,OAAe,EACf,MAAc,EACd,OAAmD;QAEnD,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAChC,OAAO,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,OAAO,CAAC,CAAC;IAClE,CAAC;IASD,KAAK,CAAC,oBAAoB,CACxB,OAAe,EACf,OAAuB;QAEvB,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAGhC,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACjD,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,OAAO;QACT,CAAC;QAGD,MAAM,SAAS,GAAG,IAAI,CAAC;QAGvB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAGnC,MAAM,OAAO,GAAG,CAAC,OAA4B,EAAE,EAAE,CAAC,GAAG,EAAE;YACrD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpC,CAAC,CAAC;QAGF,MAAM,IAAI,GAAG,CAAC,OAA4B,EAAE,EAAE,CAAC,GAAG,EAAE;YAClD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC;QAGF,MAAM,MAAM,GAAG,CAAC,OAAe,EAAE,OAA4B,EAAE,EAAE,CAAC,GAAG,EAAE;YACrE,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAgB,CAAC;YACnD,MAAM,OAAO,GAAwB,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,EAAE,CAAC;YACtE,MAAM,OAAO,GAAI,OAAO,CAAC,WAAW,CAAY,IAAI,CAAC,CAAC;YACtD,MAAM,UAAU,GAAI,OAAO,CAAC,eAAe,CAAY,IAAI,CAAC,CAAC;YAG7D,IAAI,OAAO,IAAI,UAAU,EAAE,CAAC;gBAC1B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,KAAK,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE;oBAChE,OAAO,EAAE,OAAO,GAAG,CAAC;oBACpB,UAAU;iBACX,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;YACnD,CAAC;QACH,CAAC,CAAC;QAGF,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,CAChC,OAAO,EACP,CAAC,OAA4B,EAAE,EAAE;YAC/B,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,CAAC;YACpC,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAgB,CAAC;YAEnD,KAAK,MAAM,OAAO,IAAI,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBACxC,KAAK,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE;oBAC3B,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC;oBACzB,MAAM,EAAE,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC;oBAChC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC;iBACpB,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CACF,CAAC;IACJ,CAAC"}