@betterinternship/core 2.5.11 → 2.5.13

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.
Files changed (7) hide show
  1. package/dist/lib/broker/base.task.d.ts +40 -36
  2. package/dist/lib/broker/base.task.js +78 -76
  3. package/dist/lib/broker/base.task.js.map +1 -1
  4. package/dist/lib/broker/broker.d.ts +19 -19
  5. package/dist/lib/broker/broker.js +94 -94
  6. package/dist/tsconfig.tsbuildinfo +1 -1
  7. package/package.json +5 -2
package/dist/lib/broker/base.task.d.ts CHANGED
@@ -1,36 +1,40 @@
1
- import { Task } from '@betterinternship/db';
2
- import { JsonValue } from '@prisma/client/runtime/client';
3
- export declare class TaskRequest<TaskInputs extends JsonValue> implements Pick<Task, 'inputs' | 'internal_id' | 'handler_id' | 'parent_id' | 'process_id' | 'timestamp'> {
4
- inputs: TaskInputs;
5
- internal_id: string;
6
- handler_id: string;
7
- parent_id: string | null;
8
- process_id: string;
9
- timestamp: Date;
10
- }
11
- export declare class TaskResponse<TaskResult extends JsonValue> implements Pick<Task, 'result' | 'status' | 'error'> {
12
- result: TaskResult;
13
- error: string;
14
- status: string;
15
- }
16
- type TaskResponseHandler<TaskResult extends JsonValue> = (response: TaskResponse<TaskResult>) => Promise<void>;
17
- export declare class BaseTaskInitiator<TaskInputs extends JsonValue, TaskResult extends JsonValue> {
18
- private handlerId;
19
- private requestQueueId;
20
- private responseQueueId;
21
- constructor(queueId: string, handler: TaskResponseHandler<TaskResult>);
22
- init(handler: TaskResponseHandler<TaskResult>): Promise<void>;
23
- sendTaskRequest(processId: string, inputs: TaskInputs, options: {
24
- internalId?: string;
25
- parentId?: string;
26
- maxRetries?: number;
27
- }): Promise<boolean>;
28
- }
29
- type TaskRequestHandler<TaskInputs extends JsonValue, TaskResult extends JsonValue> = (task: TaskRequest<TaskInputs>) => Promise<TaskResponse<TaskResult>>;
30
- export declare class BaseTaskRunner<TaskInputs extends JsonValue, TaskResult extends JsonValue> {
31
- private requestQueueId;
32
- private responseQueueId;
33
- constructor(queueId: string, handler: TaskRequestHandler<TaskInputs, TaskResult>);
34
- private init;
35
- }
36
- export {};
1
+ import { Task } from '@betterinternship/db';
2
+ import { JsonValue } from '@prisma/client/runtime/client';
3
+ export interface BaseJson {
4
+ [key: string]: any;
5
+ }
6
+ export declare class TaskRequest<TaskInputs extends JsonValue> implements Pick<Task, 'inputs' | 'internal_id' | 'handler_id' | 'parent_id' | 'process_id' | 'timestamp'> {
7
+ inputs: TaskInputs;
8
+ internal_id: string;
9
+ handler_id: string;
10
+ parent_id: string | null;
11
+ process_id: string;
12
+ timestamp: Date;
13
+ }
14
+ export declare class TaskResponse<TaskResult extends JsonValue> implements Pick<Task, 'result' | 'status' | 'error'> {
15
+ result: TaskResult;
16
+ error: string;
17
+ status: string;
18
+ }
19
+ type TaskResponseHandler<TaskResult extends JsonValue> = (response: TaskResponse<TaskResult>) => Promise<void>;
20
+ export declare class BaseTaskInitiator<TaskInputs extends JsonValue, TaskResult extends JsonValue> {
21
+ private handlerId;
22
+ private requestQueueId;
23
+ private responseQueueId;
24
+ constructor(queueId: string, handler: TaskResponseHandler<TaskResult>);
25
+ init(handler: TaskResponseHandler<TaskResult>): Promise<void>;
26
+ sendTaskRequest(processId: string, inputs: TaskInputs, options: {
27
+ internalId?: string;
28
+ parentId?: string;
29
+ maxRetries?: number;
30
+ }): Promise<boolean>;
31
+ }
32
+ type TaskRequestHandler<TaskInputs extends JsonValue, TaskResult extends JsonValue> = (task: TaskRequest<TaskInputs>) => Promise<TaskResponse<TaskResult>>;
33
+ export declare class BaseTaskRunner<TaskInputs extends JsonValue, TaskResult extends JsonValue> {
34
+ private requestQueueId;
35
+ private responseQueueId;
36
+ private handler;
37
+ constructor(queueId: string, handler: TaskRequestHandler<TaskInputs, TaskResult>);
38
+ listen(): Promise<void>;
39
+ }
40
+ export {};
package/dist/lib/broker/base.task.js CHANGED
@@ -1,77 +1,79 @@
1
- import { BrokerAPI } from './broker.js';
2
- import { tasks } from '@betterinternship/db';
3
- import { v4 } from 'uuid';
4
- export class TaskRequest {
5
- inputs;
6
- internal_id;
7
- handler_id;
8
- parent_id;
9
- process_id;
10
- timestamp;
11
- }
12
- export class TaskResponse {
13
- result;
14
- error;
15
- status;
16
- }
17
- export class BaseTaskInitiator {
18
- handlerId;
19
- requestQueueId;
20
- responseQueueId;
21
- constructor(queueId, handler) {
22
- this.handlerId = queueId;
23
- this.requestQueueId = `request.${queueId}`;
24
- this.responseQueueId = `response.${queueId}`;
25
- void this.init(handler);
26
- }
27
- async init(handler) {
28
- const brokerAPI = await BrokerAPI.instance();
29
- return await brokerAPI.addQueueHandler(this.responseQueueId, async (taskId) => {
30
- const task = await tasks.findFirst({ where: { id: taskId } });
31
- await handler({
32
- result: task?.result,
33
- error: task?.error ?? '',
34
- status: task?.status ?? 'failed',
35
- });
36
- });
37
- }
38
- async sendTaskRequest(processId, inputs, options) {
39
- const brokerAPI = await BrokerAPI.instance();
40
- const task = await tasks.create({
41
- data: {
42
- internal_id: options.internalId ?? `${processId}.${v4()}`,
43
- parent_id: options.parentId,
44
- handler_id: this.handlerId,
45
- process_id: processId,
46
- inputs: inputs,
47
- },
48
- });
49
- return brokerAPI.addToQueue(this.requestQueueId, task.id, options);
50
- }
51
- }
52
- export class BaseTaskRunner {
53
- requestQueueId;
54
- responseQueueId;
55
- constructor(queueId, handler) {
56
- this.requestQueueId = `request.${queueId}`;
57
- this.responseQueueId = `response.${queueId}`;
58
- void this.init(handler);
59
- }
60
- async init(handler) {
61
- const brokerAPI = await BrokerAPI.instance();
62
- await brokerAPI.addQueueHandler(this.requestQueueId, async (taskId) => {
63
- const task = await tasks.findFirst({ where: { id: taskId } });
64
- const response = await handler(task);
65
- await tasks.update({
66
- where: { id: taskId },
67
- data: {
68
- result: response.result ?? {},
69
- error: response.error ?? '',
70
- status: response.status,
71
- },
72
- });
73
- await brokerAPI.addToQueue(this.responseQueueId, taskId);
74
- });
75
- }
76
- }
1
+ import { BrokerAPI } from './broker.js';
2
+ import { tasks } from '@betterinternship/db';
3
+ import { v4 } from 'uuid';
4
+ export class TaskRequest {
5
+ inputs;
6
+ internal_id;
7
+ handler_id;
8
+ parent_id;
9
+ process_id;
10
+ timestamp;
11
+ }
12
+ export class TaskResponse {
13
+ result;
14
+ error;
15
+ status;
16
+ }
17
+ export class BaseTaskInitiator {
18
+ handlerId;
19
+ requestQueueId;
20
+ responseQueueId;
21
+ constructor(queueId, handler) {
22
+ this.handlerId = queueId;
23
+ this.requestQueueId = `request.${queueId}`;
24
+ this.responseQueueId = `response.${queueId}`;
25
+ void this.init(handler);
26
+ }
27
+ async init(handler) {
28
+ const brokerAPI = await BrokerAPI.instance();
29
+ return await brokerAPI.addQueueHandler(this.responseQueueId, async (taskId) => {
30
+ const task = await tasks.findFirst({ where: { id: taskId } });
31
+ await handler({
32
+ result: task?.result,
33
+ error: task?.error ?? '',
34
+ status: task?.status ?? 'failed',
35
+ });
36
+ });
37
+ }
38
+ async sendTaskRequest(processId, inputs, options) {
39
+ const brokerAPI = await BrokerAPI.instance();
40
+ const task = await tasks.create({
41
+ data: {
42
+ internal_id: options.internalId ?? `${processId}.${v4()}`,
43
+ parent_id: options.parentId,
44
+ handler_id: this.handlerId,
45
+ process_id: processId,
46
+ inputs: inputs,
47
+ },
48
+ });
49
+ return brokerAPI.addToQueue(this.requestQueueId, task.id, options);
50
+ }
51
+ }
52
+ export class BaseTaskRunner {
53
+ requestQueueId;
54
+ responseQueueId;
55
+ handler;
56
+ constructor(queueId, handler) {
57
+ this.requestQueueId = `request.${queueId}`;
58
+ this.responseQueueId = `response.${queueId}`;
59
+ this.handler = handler;
60
+ }
61
+ async listen() {
62
+ const processTask = async (taskId) => {
63
+ const task = await tasks.findFirst({ where: { id: taskId } });
64
+ const response = await this.handler(task);
65
+ await tasks.update({
66
+ where: { id: taskId },
67
+ data: {
68
+ result: response.result ?? {},
69
+ error: response.error ?? '',
70
+ status: response.status,
71
+ },
72
+ });
73
+ await brokerAPI.addToQueue(this.responseQueueId, taskId);
74
+ };
75
+ const brokerAPI = await BrokerAPI.instance();
76
+ await brokerAPI.addQueueHandler(this.requestQueueId, processTask);
77
+ }
78
+ }
77
79
  //# sourceMappingURL=base.task.js.map
package/dist/lib/broker/base.task.js.map CHANGED
@@ -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;AAG1B,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;IAQhC,YACE,OAAe,EACf,OAAmD;QAEnD,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;IASO,KAAK,CAAC,IAAI,CAAC,OAAmD;QACpE,MAAM,SAAS,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE,CAAC;QAC7C,MAAM,SAAU,CAAC,eAAe,CAC9B,IAAI,CAAC,cAAc,EACnB,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,QAAQ,GAA6B,MAAM,OAAO,CACtD,IAA0C,CAC3C,CAAC;YAEF,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,CACF,CAAC;IACJ,CAAC;CACF"}
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;QAEV,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;YAC9D,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"}
package/dist/lib/broker/broker.d.ts CHANGED
@@ -1,19 +1,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 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 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
+ }
package/dist/lib/broker/broker.js CHANGED
@@ -1,95 +1,95 @@
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
+ 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
+ }
95
95
  //# sourceMappingURL=broker.js.map