redis-smq-benchmarks 9.0.9-next.0

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 (57) hide show
  1. package/CHANGELOG.md +19 -0
  2. package/LICENSE +22 -0
  3. package/README.md +192 -0
  4. package/README.template.md +186 -0
  5. package/dist/esm/bin/cli.d.ts +2 -0
  6. package/dist/esm/bin/cli.d.ts.map +1 -0
  7. package/dist/esm/bin/cli.js +121 -0
  8. package/dist/esm/bin/cli.js.map +1 -0
  9. package/dist/esm/index.d.ts +2 -0
  10. package/dist/esm/index.d.ts.map +1 -0
  11. package/dist/esm/index.js +2 -0
  12. package/dist/esm/index.js.map +1 -0
  13. package/dist/esm/src/common/base-benchmark.d.ts +39 -0
  14. package/dist/esm/src/common/base-benchmark.d.ts.map +1 -0
  15. package/dist/esm/src/common/base-benchmark.js +83 -0
  16. package/dist/esm/src/common/base-benchmark.js.map +1 -0
  17. package/dist/esm/src/common/redis-server.d.ts +13 -0
  18. package/dist/esm/src/common/redis-server.d.ts.map +1 -0
  19. package/dist/esm/src/common/redis-server.js +22 -0
  20. package/dist/esm/src/common/redis-server.js.map +1 -0
  21. package/dist/esm/src/consumer-benchmark.d.ts +9 -0
  22. package/dist/esm/src/consumer-benchmark.d.ts.map +1 -0
  23. package/dist/esm/src/consumer-benchmark.js +61 -0
  24. package/dist/esm/src/consumer-benchmark.js.map +1 -0
  25. package/dist/esm/src/e2e-benchmark.d.ts +4 -0
  26. package/dist/esm/src/e2e-benchmark.d.ts.map +1 -0
  27. package/dist/esm/src/e2e-benchmark.js +159 -0
  28. package/dist/esm/src/e2e-benchmark.js.map +1 -0
  29. package/dist/esm/src/helpers/create-worker.d.ts +13 -0
  30. package/dist/esm/src/helpers/create-worker.d.ts.map +1 -0
  31. package/dist/esm/src/helpers/create-worker.js +24 -0
  32. package/dist/esm/src/helpers/create-worker.js.map +1 -0
  33. package/dist/esm/src/helpers/ensure-queue.d.ts +4 -0
  34. package/dist/esm/src/helpers/ensure-queue.d.ts.map +1 -0
  35. package/dist/esm/src/helpers/ensure-queue.js +10 -0
  36. package/dist/esm/src/helpers/ensure-queue.js.map +1 -0
  37. package/dist/esm/src/helpers/prefill-queue.d.ts +4 -0
  38. package/dist/esm/src/helpers/prefill-queue.d.ts.map +1 -0
  39. package/dist/esm/src/helpers/prefill-queue.js +23 -0
  40. package/dist/esm/src/helpers/prefill-queue.js.map +1 -0
  41. package/dist/esm/src/producer-benchmark.d.ts +8 -0
  42. package/dist/esm/src/producer-benchmark.d.ts.map +1 -0
  43. package/dist/esm/src/producer-benchmark.js +47 -0
  44. package/dist/esm/src/producer-benchmark.js.map +1 -0
  45. package/dist/esm/src/threads/consumer-worker-thread.d.ts +2 -0
  46. package/dist/esm/src/threads/consumer-worker-thread.d.ts.map +1 -0
  47. package/dist/esm/src/threads/consumer-worker-thread.js +59 -0
  48. package/dist/esm/src/threads/consumer-worker-thread.js.map +1 -0
  49. package/dist/esm/src/threads/producer-worker-thread.d.ts +2 -0
  50. package/dist/esm/src/threads/producer-worker-thread.d.ts.map +1 -0
  51. package/dist/esm/src/threads/producer-worker-thread.js +54 -0
  52. package/dist/esm/src/threads/producer-worker-thread.js.map +1 -0
  53. package/dist/esm/src/types/index.d.ts +57 -0
  54. package/dist/esm/src/types/index.d.ts.map +1 -0
  55. package/dist/esm/src/types/index.js +6 -0
  56. package/dist/esm/src/types/index.js.map +1 -0
  57. package/package.json +69 -0
@@ -0,0 +1,83 @@
1
+ import { EWorkerMessageType, } from '../types/index.js';
2
+ import { createWorker } from '../helpers/create-worker.js';
3
+ export class BaseBenchmark {
4
+ showProgress = false;
5
+ redisConfig;
6
+ queue;
7
+ totalMessages;
8
+ workerCount;
9
+ workerPath;
10
+ workerLabel;
11
+ workers = [];
12
+ completedWorkers = 0;
13
+ totalProcessed = 0;
14
+ totalProcessingTime = 0;
15
+ startTime = 0;
16
+ constructor(config) {
17
+ this.redisConfig = config.redisConfig;
18
+ this.queue = config.queue;
19
+ this.totalMessages = config.totalMessages;
20
+ this.workerCount = config.workerCount;
21
+ this.workerPath = config.workerPath;
22
+ this.workerLabel = config.workerLabel;
23
+ this.showProgress = config.showProgress;
24
+ }
25
+ calculateMessageDistribution() {
26
+ const messagesPerWorker = Math.floor(this.totalMessages / this.workerCount);
27
+ const remainingMessages = this.totalMessages % this.workerCount;
28
+ return { count: messagesPerWorker, remainder: remainingMessages };
29
+ }
30
+ createMessageHandler(onComplete) {
31
+ return (msg) => {
32
+ if (msg.type === EWorkerMessageType.COMPLETED) {
33
+ this.completedWorkers++;
34
+ const { workerId, processed, timeTaken } = msg.data;
35
+ this.totalProcessed += Number(processed);
36
+ this.totalProcessingTime += Number(timeTaken);
37
+ console.log(`${this.workerLabel} ${workerId} completed: ${processed} messages in ${(Number(timeTaken) / 1000).toFixed(2)}s (${(Number(processed) / (Number(timeTaken) / 1000)).toFixed(0)} msg/s)`);
38
+ if (this.completedWorkers === this.workerCount) {
39
+ const totalTime = Date.now() - this.startTime;
40
+ onComplete({
41
+ totalTime,
42
+ totalWorkerTime: this.totalProcessingTime,
43
+ total: this.totalProcessed,
44
+ workerCount: this.workerCount,
45
+ });
46
+ }
47
+ }
48
+ else if (msg.type === EWorkerMessageType.PROGRESS &&
49
+ this.showProgress) {
50
+ const { workerId, processed } = msg.data;
51
+ console.log(`${this.workerLabel} ${workerId} progress: ${processed} messages`);
52
+ }
53
+ };
54
+ }
55
+ async createWorkers(onMessage) {
56
+ const { count: messagesPerWorker, remainder: remainingMessages } = this.calculateMessageDistribution();
57
+ console.log(`Messages per ${this.workerLabel} (approx): ${messagesPerWorker}`);
58
+ this.startTime = Date.now();
59
+ for (let i = 0; i < this.workerCount; i++) {
60
+ let workerMessageCount = messagesPerWorker;
61
+ if (i < remainingMessages) {
62
+ workerMessageCount += 1;
63
+ }
64
+ console.log(`${this.workerLabel} ${i + 1} will handle ${workerMessageCount} messages`);
65
+ const worker = createWorker({
66
+ redisConfig: this.redisConfig,
67
+ workerPath: this.workerPath,
68
+ workerId: i,
69
+ expectedMessages: workerMessageCount,
70
+ queue: this.queue,
71
+ onMessage,
72
+ });
73
+ this.workers.push(worker);
74
+ }
75
+ }
76
+ async shutdownWorkers() {
77
+ await Promise.all(this.workers.map((worker) => {
78
+ worker.removeAllListeners();
79
+ return worker.terminate();
80
+ })).catch(() => void 0);
81
+ }
82
+ }
83
+ //# sourceMappingURL=base-benchmark.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"base-benchmark.js","sourceRoot":"","sources":["../../../../src/common/base-benchmark.ts"],"names":[],"mappings":"AAWA,OAAO,EACL,kBAAkB,GAGnB,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAc3D,MAAM,OAAgB,aAAa;IACvB,YAAY,GAAG,KAAK,CAAC;IACrB,WAAW,CAAe;IAC1B,KAAK,CAAe;IACpB,aAAa,CAAS;IACtB,WAAW,CAAS;IACpB,UAAU,CAAS;IACnB,WAAW,CAAS;IACpB,OAAO,GAAa,EAAE,CAAC;IACvB,gBAAgB,GAAG,CAAC,CAAC;IACrB,cAAc,GAAG,CAAC,CAAC;IACnB,mBAAmB,GAAG,CAAC,CAAC;IACxB,SAAS,GAAG,CAAC,CAAC;IAExB,YAAY,MAAwB;QAClC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC1B,IAAI,CAAC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC;QACpC,IAAI,CAAC,WAAW,GAAG,MAAM,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;IAC1C,CAAC;IAIS,4BAA4B;QAIpC,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5E,MAAM,iBAAiB,GAAG,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,WAAW,CAAC;QAChE,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,SAAS,EAAE,iBAAiB,EAAE,CAAC;IACpE,CAAC;IAES,oBAAoB,CAC5B,UAA8C;QAE9C,OAAO,CAAC,GAAmB,EAAE,EAAE;YAC7B,IAAI,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,SAAS,EAAE,CAAC;gBAC9C,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACxB,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;gBACpD,IAAI,CAAC,cAAc,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC;gBACzC,IAAI,CAAC,mBAAmB,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC;gBAE9C,OAAO,CAAC,GAAG,CACT,GAAG,IAAI,CAAC,WAAW,IAAI,QAAQ,eAAe,SAAS,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CACvL,CAAC;gBAEF,IAAI,IAAI,CAAC,gBAAgB,KAAK,IAAI,CAAC,WAAW,EAAE,CAAC;oBAC/C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;oBAC9C,UAAU,CAAC;wBACT,SAAS;wBACT,eAAe,EAAE,IAAI,CAAC,mBAAmB;wBACzC,KAAK,EAAE,IAAI,CAAC,cAAc;wBAC1B,WAAW,EAAE,IAAI,CAAC,WAAW;qBAC9B,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;iBAAM,IACL,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,QAAQ;gBACxC,IAAI,CAAC,YAAY,EACjB,CAAC;gBACD,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;gBACzC,OAAO,CAAC,GAAG,CACT,GAAG,IAAI,CAAC,WAAW,IAAI,QAAQ,cAAc,SAAS,WAAW,CAClE,CAAC;YACJ,CAAC;QACH,CAAC,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,aAAa,CAAC,SAAyB;QACrD,MAAM,EAAE,KAAK,EAAE,iBAAiB,EAAE,SAAS,EAAE,iBAAiB,EAAE,GAC9D,IAAI,CAAC,4BAA4B,EAAE,CAAC;QAEtC,OAAO,CAAC,GAAG,CACT,gBAAgB,IAAI,CAAC,WAAW,cAAc,iBAAiB,EAAE,CAClE,CAAC;QAEF,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;YAC1C,IAAI,kBAAkB,GAAG,iBAAiB,CAAC;YAC3C,IAAI,CAAC,GAAG,iBAAiB,EAAE,CAAC;gBAC1B,kBAAkB,IAAI,CAAC,CAAC;YAC1B,CAAC;YAED,OAAO,CAAC,GAAG,CACT,GAAG,IAAI,CAAC,WAAW,IAAI,CAAC,GAAG,CAAC,gBAAgB,kBAAkB,WAAW,CAC1E,CAAC;YAEF,MAAM,MAAM,GAAG,YAAY,CAAC;gBAC1B,WAAW,EAAE,IAAI,CAAC,WAAW;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,QAAQ,EAAE,CAAC;gBACX,gBAAgB,EAAE,kBAAkB;gBACpC,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS;aACV,CAAC,CAAC;YACH,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAES,KAAK,CAAC,eAAe;QAC7B,MAAM,OAAO,CAAC,GAAG,CACf,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;YAC1B,MAAM,CAAC,kBAAkB,EAAE,CAAC;YAC5B,OAAO,MAAM,CAAC,SAAS,EAAE,CAAC;QAC5B,CAAC,CAAC,CACH,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IACxB,CAAC;CAGF"}
@@ -0,0 +1,13 @@
1
+ import { ERedisConfigClient } from 'redis-smq-common';
2
+ export declare const redisConfig: {
3
+ client: ERedisConfigClient;
4
+ options: {
5
+ host: string;
6
+ port: number;
7
+ db: number;
8
+ showFriendlyErrorStack: boolean;
9
+ };
10
+ };
11
+ export declare function startRedisServer(): Promise<void>;
12
+ export declare function shutdownRedisServer(): Promise<void>;
13
+ //# sourceMappingURL=redis-server.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"redis-server.d.ts","sourceRoot":"","sources":["../../../../src/common/redis-server.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,kBAAkB,EAAe,MAAM,kBAAkB,CAAC;AAEnE,eAAO,MAAM,WAAW;;;;;;;;CAQvB,CAAC;AAIF,wBAAsB,gBAAgB,kBAKrC;AAED,wBAAsB,mBAAmB,kBAGxC"}
@@ -0,0 +1,22 @@
1
+ import { ERedisConfigClient, RedisServer } from 'redis-smq-common';
2
+ export const redisConfig = {
3
+ client: ERedisConfigClient.IOREDIS,
4
+ options: {
5
+ host: '127.0.0.1',
6
+ port: 0,
7
+ db: 0,
8
+ showFriendlyErrorStack: true,
9
+ },
10
+ };
11
+ const redisServer = new RedisServer();
12
+ export async function startRedisServer() {
13
+ console.log('Starting Redis server...');
14
+ const redisPort = await redisServer.start();
15
+ redisConfig.options.port = redisPort;
16
+ console.log('Redis Server started on port: ' + redisPort);
17
+ }
18
+ export async function shutdownRedisServer() {
19
+ console.log('Shutting down Redis Server...');
20
+ await redisServer.shutdown();
21
+ }
22
+ //# sourceMappingURL=redis-server.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"redis-server.js","sourceRoot":"","sources":["../../../../src/common/redis-server.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,kBAAkB,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAEnE,MAAM,CAAC,MAAM,WAAW,GAAG;IACzB,MAAM,EAAE,kBAAkB,CAAC,OAAO;IAClC,OAAO,EAAE;QACP,IAAI,EAAE,WAAW;QACjB,IAAI,EAAE,CAAC;QACP,EAAE,EAAE,CAAC;QACL,sBAAsB,EAAE,IAAI;KAC7B;CACF,CAAC;AAEF,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AAEtC,MAAM,CAAC,KAAK,UAAU,gBAAgB;IACpC,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,CAAC;IACxC,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;IAC5C,WAAW,CAAC,OAAO,CAAC,IAAI,GAAG,SAAS,CAAC;IACrC,OAAO,CAAC,GAAG,CAAC,gCAAgC,GAAG,SAAS,CAAC,CAAC;AAC5D,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,OAAO,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IAC7C,MAAM,WAAW,CAAC,QAAQ,EAAE,CAAC;AAC/B,CAAC"}
@@ -0,0 +1,9 @@
1
+ import { ICallback } from 'redis-smq-common';
2
+ import { IBenchmarkResult } from './types/index.js';
3
+ import { BaseBenchmark } from './common/base-benchmark.js';
4
+ export declare class ConsumerBenchmark extends BaseBenchmark {
5
+ protected ensureQueue(): Promise<void>;
6
+ protected prefillQueue(): Promise<void>;
7
+ run(cb: ICallback<IBenchmarkResult>): void;
8
+ }
9
+ //# sourceMappingURL=consumer-benchmark.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"consumer-benchmark.d.ts","sourceRoot":"","sources":["../../../src/consumer-benchmark.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAGpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAE3D,qBAAa,iBAAkB,SAAQ,aAAa;cAClC,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;cAS5B,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAStC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,gBAAgB,CAAC,GAAG,IAAI;CA6ClD"}
@@ -0,0 +1,61 @@
1
+ import { async } from 'redis-smq-common';
2
+ import { ensureQueue } from './helpers/ensure-queue.js';
3
+ import { prefillQueue } from './helpers/prefill-queue.js';
4
+ import { BaseBenchmark } from './common/base-benchmark.js';
5
+ export class ConsumerBenchmark extends BaseBenchmark {
6
+ async ensureQueue() {
7
+ return new Promise((resolve, reject) => {
8
+ ensureQueue(this.queue, (err) => {
9
+ if (err)
10
+ reject(err);
11
+ else
12
+ resolve();
13
+ });
14
+ });
15
+ }
16
+ async prefillQueue() {
17
+ return new Promise((resolve, reject) => {
18
+ prefillQueue(this.queue, this.totalMessages, (err) => {
19
+ if (err)
20
+ reject(err);
21
+ else
22
+ resolve();
23
+ });
24
+ });
25
+ }
26
+ run(cb) {
27
+ console.log('Starting consumer throughput benchmark...');
28
+ console.log(`Queue: ${this.queue.ns}/${this.queue.name} | Messages: ${this.totalMessages} | Consumers: ${this.workerCount}`);
29
+ const onComplete = (result) => {
30
+ this.shutdownWorkers()
31
+ .then(() => {
32
+ const totalTime = Date.now() - this.startTime;
33
+ console.log('\n========== BENCHMARK COMPLETE ==========');
34
+ console.log(`Total messages consumed: ${result.total}`);
35
+ console.log(`Total time: ${(totalTime / 1000).toFixed(2)}s`);
36
+ console.log(`Overall throughput: ${(result.total / (totalTime / 1000)).toFixed(0)} messages/second`);
37
+ console.log('========================================\n');
38
+ cb(null, result);
39
+ })
40
+ .catch((err) => cb(err));
41
+ };
42
+ const messageHandler = this.createMessageHandler(onComplete);
43
+ async.series([
44
+ (cb) => this.ensureQueue()
45
+ .then(() => cb())
46
+ .catch(cb),
47
+ (cb) => this.prefillQueue()
48
+ .then(() => cb())
49
+ .catch(cb),
50
+ (cb) => {
51
+ this.createWorkers(messageHandler)
52
+ .then(() => cb())
53
+ .catch(cb);
54
+ },
55
+ ], (err) => {
56
+ if (err)
57
+ cb(err);
58
+ });
59
+ }
60
+ }
61
+ //# sourceMappingURL=consumer-benchmark.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"consumer-benchmark.js","sourceRoot":"","sources":["../../../src/consumer-benchmark.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAa,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAE1D,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAE3D,MAAM,OAAO,iBAAkB,SAAQ,aAAa;IACxC,KAAK,CAAC,WAAW;QACzB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC9B,IAAI,GAAG;oBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;oBAChB,OAAO,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAES,KAAK,CAAC,YAAY;QAC1B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,aAAa,EAAE,CAAC,GAAG,EAAE,EAAE;gBACnD,IAAI,GAAG;oBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;oBAChB,OAAO,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,GAAG,CAAC,EAA+B;QACxC,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;QACzD,OAAO,CAAC,GAAG,CACT,UAAU,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,gBAAgB,IAAI,CAAC,aAAa,iBAAiB,IAAI,CAAC,WAAW,EAAE,CAChH,CAAC;QAEF,MAAM,UAAU,GAAG,CAAC,MAAwB,EAAE,EAAE;YAC9C,IAAI,CAAC,eAAe,EAAE;iBACnB,IAAI,CAAC,GAAG,EAAE;gBACT,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC9C,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;gBAC1D,OAAO,CAAC,GAAG,CAAC,4BAA4B,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;gBACxD,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBAC7D,OAAO,CAAC,GAAG,CACT,uBAAuB,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAkB,CACxF,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;gBAC1D,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnB,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QAE7D,KAAK,CAAC,MAAM,CACV;YACE,CAAC,EAAE,EAAE,EAAE,CACL,IAAI,CAAC,WAAW,EAAE;iBACf,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;iBAChB,KAAK,CAAC,EAAE,CAAC;YACd,CAAC,EAAE,EAAE,EAAE,CACL,IAAI,CAAC,YAAY,EAAE;iBAChB,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;iBAChB,KAAK,CAAC,EAAE,CAAC;YACd,CAAC,EAAE,EAAE,EAAE;gBACL,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC;qBAC/B,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;qBAChB,KAAK,CAAC,EAAE,CAAC,CAAC;YACf,CAAC;SACF,EACD,CAAC,GAAG,EAAE,EAAE;YACN,IAAI,GAAG;gBAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,4 @@
1
+ import { ICallback } from 'redis-smq-common';
2
+ import { IE2EBenchmarkConfig, IE2EBenchmarkResult } from './types/index.js';
3
+ export declare function runE2EBenchmark(config: IE2EBenchmarkConfig, cb: ICallback<IE2EBenchmarkResult>): void;
4
+ //# sourceMappingURL=e2e-benchmark.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"e2e-benchmark.d.ts","sourceRoot":"","sources":["../../../src/e2e-benchmark.ts"],"names":[],"mappings":"AAUA,OAAO,EAAS,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAEL,mBAAmB,EACnB,mBAAmB,EACpB,MAAM,kBAAkB,CAAC;AAiB1B,wBAAgB,eAAe,CAC7B,MAAM,EAAE,mBAAmB,EAC3B,EAAE,EAAE,SAAS,CAAC,mBAAmB,CAAC,QAuOnC"}
@@ -0,0 +1,159 @@
1
+ import { ensureQueue } from './helpers/ensure-queue.js';
2
+ import { async } from 'redis-smq-common';
3
+ import { EWorkerMessageType, } from './types/index.js';
4
+ import { createWorker } from './helpers/create-worker.js';
5
+ export function runE2EBenchmark(config, cb) {
6
+ const { redisConfig, queue, totalMessages, producerWorkerPath, consumerWorkerPath, producerCount, consumerCount, showProgress, } = config;
7
+ console.log('Starting end-to-end throughput benchmark...');
8
+ console.log(`Queue: ${queue.ns}/${queue.name} | Messages: ${totalMessages} | Producers: ${producerCount} | Consumers: ${consumerCount}`);
9
+ const allWorkers = [];
10
+ let completedProducers = 0;
11
+ let completedConsumers = 0;
12
+ let totalProduced = 0;
13
+ let totalConsumed = 0;
14
+ let totalProductionTime = 0;
15
+ let totalConsumptionTime = 0;
16
+ let productionStartTime = 0;
17
+ let consumptionStartTime = 0;
18
+ let productionEndTime = 0;
19
+ let consumptionEndTime = 0;
20
+ let benchmarkStartTime = 0;
21
+ let benchmarkEndTime = 0;
22
+ const onProducerMessage = (msg) => {
23
+ if (msg.type === EWorkerMessageType.COMPLETED) {
24
+ completedProducers++;
25
+ const { workerId, processed, timeTaken } = msg.data;
26
+ totalProduced += Number(processed);
27
+ totalProductionTime += Number(timeTaken);
28
+ productionEndTime = Date.now();
29
+ console.log(`Producer ${workerId} completed: ${processed} messages in ${(Number(timeTaken) / 1000).toFixed(2)}s (${(Number(processed) / (Number(timeTaken) / 1000)).toFixed(0)} msg/s)`);
30
+ checkBenchmarkCompletion();
31
+ }
32
+ else if (msg.type === EWorkerMessageType.PROGRESS && showProgress) {
33
+ const { workerId, processed } = msg.data;
34
+ console.log(`Producer ${workerId} progress: ${processed} messages`);
35
+ }
36
+ };
37
+ const onConsumerMessage = (msg) => {
38
+ if (msg.type === EWorkerMessageType.COMPLETED) {
39
+ completedConsumers++;
40
+ const { workerId, processed, timeTaken } = msg.data;
41
+ totalConsumed += Number(processed);
42
+ totalConsumptionTime += Number(timeTaken);
43
+ consumptionEndTime = Date.now();
44
+ console.log(`Consumer ${workerId} completed: ${processed} messages in ${(Number(timeTaken) / 1000).toFixed(2)}s (${(Number(processed) / (Number(timeTaken) / 1000)).toFixed(0)} msg/s)`);
45
+ checkBenchmarkCompletion();
46
+ }
47
+ else if (msg.type === EWorkerMessageType.PROGRESS && showProgress) {
48
+ const { workerId, processed } = msg.data;
49
+ console.log(`Consumer ${workerId} progress: ${processed} messages`);
50
+ }
51
+ };
52
+ const checkBenchmarkCompletion = () => {
53
+ if (completedProducers === producerCount &&
54
+ completedConsumers === consumerCount) {
55
+ benchmarkEndTime = Date.now();
56
+ const totalTime = benchmarkEndTime - benchmarkStartTime;
57
+ const productionTime = productionEndTime - productionStartTime;
58
+ const consumptionTime = consumptionEndTime - consumptionStartTime;
59
+ const productionThroughput = totalProduced / (productionTime / 1000);
60
+ const consumptionThroughput = totalConsumed / (consumptionTime / 1000);
61
+ console.log('\n========== E2E BENCHMARK COMPLETE ==========');
62
+ console.log(`Production Phase:`);
63
+ console.log(` Total produced: ${totalProduced}`);
64
+ console.log(` Production time: ${(productionTime / 1000).toFixed(2)}s`);
65
+ console.log(` Production throughput: ${productionThroughput.toFixed(0)} msg/s`);
66
+ console.log(`\nConsumption Phase:`);
67
+ console.log(` Total consumed: ${totalConsumed}`);
68
+ console.log(` Consumption time: ${(consumptionTime / 1000).toFixed(2)}s`);
69
+ console.log(` Consumption throughput: ${consumptionThroughput.toFixed(0)} msg/s`);
70
+ console.log(`\nEnd-to-End:`);
71
+ console.log(` Total time: ${(totalTime / 1000).toFixed(2)}s`);
72
+ console.log(` Overall throughput: ${(totalProduced / (totalTime / 1000)).toFixed(0)} msg/s`);
73
+ console.log(` System backlog: ${totalProduced - totalConsumed} messages`);
74
+ if (totalProduced === totalConsumed) {
75
+ console.log(` Status: All messages processed successfully ✓`);
76
+ }
77
+ else {
78
+ console.log(` Status: ${totalProduced - totalConsumed} messages not consumed ⚠️`);
79
+ }
80
+ console.log('============================================\n');
81
+ const result = {
82
+ totalTime,
83
+ totalWorkerTime: totalProductionTime + totalConsumptionTime,
84
+ total: totalProduced + totalConsumed,
85
+ workerCount: producerCount + consumerCount,
86
+ productionTime,
87
+ consumptionTime,
88
+ messagesProduced: totalProduced,
89
+ messagesConsumed: totalConsumed,
90
+ productionThroughput,
91
+ consumptionThroughput,
92
+ };
93
+ const shutdownPromises = Promise.all(allWorkers.map((worker) => {
94
+ worker.removeAllListeners();
95
+ return worker.terminate();
96
+ }));
97
+ shutdownPromises.catch(() => void 0).finally(() => cb(null, result));
98
+ }
99
+ };
100
+ async.series([
101
+ (cb) => ensureQueue(queue, cb),
102
+ (cb) => {
103
+ console.log('Setting up producers and consumers...');
104
+ const messagesPerProducer = Math.floor(totalMessages / producerCount);
105
+ const remainingProducers = totalMessages % producerCount;
106
+ const messagesPerConsumer = Math.floor(totalMessages / consumerCount);
107
+ const remainingConsumers = totalMessages % consumerCount;
108
+ console.log(`Messages per producer (approx): ${messagesPerProducer}`);
109
+ console.log(`Messages per consumer (approx): ${messagesPerConsumer}`);
110
+ benchmarkStartTime = Date.now();
111
+ productionStartTime = Date.now();
112
+ consumptionStartTime = Date.now();
113
+ try {
114
+ for (let i = 0; i < producerCount; i++) {
115
+ let producerMessageCount = messagesPerProducer;
116
+ if (i < remainingProducers) {
117
+ producerMessageCount += 1;
118
+ }
119
+ console.log(`Producer ${i + 1} will produce ${producerMessageCount} messages`);
120
+ const producer = createWorker({
121
+ workerPath: producerWorkerPath,
122
+ workerId: i,
123
+ expectedMessages: producerMessageCount,
124
+ redisConfig,
125
+ queue,
126
+ onMessage: onProducerMessage,
127
+ });
128
+ allWorkers.push(producer);
129
+ }
130
+ for (let i = 0; i < consumerCount; i++) {
131
+ let consumerMessageCount = messagesPerConsumer;
132
+ if (i < remainingConsumers) {
133
+ consumerMessageCount += 1;
134
+ }
135
+ console.log(`Consumer ${i + 1} will consume ${consumerMessageCount} messages`);
136
+ const consumer = createWorker({
137
+ workerPath: consumerWorkerPath,
138
+ workerId: i,
139
+ expectedMessages: consumerMessageCount,
140
+ redisConfig,
141
+ queue,
142
+ onMessage: onConsumerMessage,
143
+ });
144
+ allWorkers.push(consumer);
145
+ }
146
+ cb();
147
+ }
148
+ catch (err) {
149
+ cb(err instanceof Error ? err : new Error(String(err)));
150
+ }
151
+ },
152
+ ], (err) => {
153
+ if (err) {
154
+ console.error('Benchmark setup failed:', err);
155
+ Promise.all(allWorkers.map((worker) => worker.terminate())).finally(() => cb(err));
156
+ }
157
+ });
158
+ }
159
+ //# sourceMappingURL=e2e-benchmark.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"e2e-benchmark.js","sourceRoot":"","sources":["../../../src/e2e-benchmark.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACxD,OAAO,EAAE,KAAK,EAAa,MAAM,kBAAkB,CAAC;AACpD,OAAO,EACL,kBAAkB,GAGnB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAgB1D,MAAM,UAAU,eAAe,CAC7B,MAA2B,EAC3B,EAAkC;IAElC,MAAM,EACJ,WAAW,EACX,KAAK,EACL,aAAa,EACb,kBAAkB,EAClB,kBAAkB,EAClB,aAAa,EACb,aAAa,EACb,YAAY,GACb,GAAG,MAAM,CAAC;IAEX,OAAO,CAAC,GAAG,CAAC,6CAA6C,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CACT,UAAU,KAAK,CAAC,EAAE,IAAI,KAAK,CAAC,IAAI,gBAAgB,aAAa,iBAAiB,aAAa,iBAAiB,aAAa,EAAE,CAC5H,CAAC;IAEF,MAAM,UAAU,GAAa,EAAE,CAAC;IAEhC,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAC5B,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAE7B,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAC5B,IAAI,oBAAoB,GAAG,CAAC,CAAC;IAC7B,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAE3B,IAAI,kBAAkB,GAAG,CAAC,CAAC;IAC3B,IAAI,gBAAgB,GAAG,CAAC,CAAC;IAEzB,MAAM,iBAAiB,GAAG,CAAC,GAAmB,EAAE,EAAE;QAChD,IAAI,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,SAAS,EAAE,CAAC;YAC9C,kBAAkB,EAAE,CAAC;YACrB,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YACpD,aAAa,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC;YACnC,mBAAmB,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC;YACzC,iBAAiB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE/B,OAAO,CAAC,GAAG,CACT,YAAY,QAAQ,eAAe,SAAS,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAC5K,CAAC;YAEF,wBAAwB,EAAE,CAAC;QAC7B,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,QAAQ,IAAI,YAAY,EAAE,CAAC;YACpE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YACzC,OAAO,CAAC,GAAG,CAAC,YAAY,QAAQ,cAAc,SAAS,WAAW,CAAC,CAAC;QACtE,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,iBAAiB,GAAG,CAAC,GAAmB,EAAE,EAAE;QAChD,IAAI,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,SAAS,EAAE,CAAC;YAC9C,kBAAkB,EAAE,CAAC;YACrB,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YACpD,aAAa,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC;YACnC,oBAAoB,IAAI,MAAM,CAAC,SAAS,CAAC,CAAC;YAC1C,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEhC,OAAO,CAAC,GAAG,CACT,YAAY,QAAQ,eAAe,SAAS,gBAAgB,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAC5K,CAAC;YAEF,wBAAwB,EAAE,CAAC;QAC7B,CAAC;aAAM,IAAI,GAAG,CAAC,IAAI,KAAK,kBAAkB,CAAC,QAAQ,IAAI,YAAY,EAAE,CAAC;YACpE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC;YACzC,OAAO,CAAC,GAAG,CAAC,YAAY,QAAQ,cAAc,SAAS,WAAW,CAAC,CAAC;QACtE,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,wBAAwB,GAAG,GAAG,EAAE;QAEpC,IACE,kBAAkB,KAAK,aAAa;YACpC,kBAAkB,KAAK,aAAa,EACpC,CAAC;YACD,gBAAgB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC9B,MAAM,SAAS,GAAG,gBAAgB,GAAG,kBAAkB,CAAC;YAExD,MAAM,cAAc,GAAG,iBAAiB,GAAG,mBAAmB,CAAC;YAC/D,MAAM,eAAe,GAAG,kBAAkB,GAAG,oBAAoB,CAAC;YAElE,MAAM,oBAAoB,GAAG,aAAa,GAAG,CAAC,cAAc,GAAG,IAAI,CAAC,CAAC;YACrE,MAAM,qBAAqB,GAAG,aAAa,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC,CAAC;YAEvE,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;YAC9D,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;YACjC,OAAO,CAAC,GAAG,CAAC,qBAAqB,aAAa,EAAE,CAAC,CAAC;YAClD,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,cAAc,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACzE,OAAO,CAAC,GAAG,CACT,4BAA4B,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CACpE,CAAC;YAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;YACpC,OAAO,CAAC,GAAG,CAAC,qBAAqB,aAAa,EAAE,CAAC,CAAC;YAClD,OAAO,CAAC,GAAG,CACT,uBAAuB,CAAC,eAAe,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAC9D,CAAC;YACF,OAAO,CAAC,GAAG,CACT,6BAA6B,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CACtE,CAAC;YAEF,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YAC7B,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YAC/D,OAAO,CAAC,GAAG,CACT,yBAAyB,CAAC,aAAa,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CACjF,CAAC;YACF,OAAO,CAAC,GAAG,CACT,qBAAqB,aAAa,GAAG,aAAa,WAAW,CAC9D,CAAC;YAEF,IAAI,aAAa,KAAK,aAAa,EAAE,CAAC;gBACpC,OAAO,CAAC,GAAG,CAAC,iDAAiD,CAAC,CAAC;YACjE,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CACT,aAAa,aAAa,GAAG,aAAa,2BAA2B,CACtE,CAAC;YACJ,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;YAG9D,MAAM,MAAM,GAAwB;gBAClC,SAAS;gBACT,eAAe,EAAE,mBAAmB,GAAG,oBAAoB;gBAC3D,KAAK,EAAE,aAAa,GAAG,aAAa;gBACpC,WAAW,EAAE,aAAa,GAAG,aAAa;gBAC1C,cAAc;gBACd,eAAe;gBACf,gBAAgB,EAAE,aAAa;gBAC/B,gBAAgB,EAAE,aAAa;gBAC/B,oBAAoB;gBACpB,qBAAqB;aACtB,CAAC;YAGF,MAAM,gBAAgB,GAAG,OAAO,CAAC,GAAG,CAClC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE;gBACxB,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC5B,OAAO,MAAM,CAAC,SAAS,EAAE,CAAC;YAC5B,CAAC,CAAC,CACH,CAAC;YACF,gBAAgB,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;QACvE,CAAC;IACH,CAAC,CAAC;IAEF,KAAK,CAAC,MAAM,CACV;QACE,CAAC,EAAE,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,EAAE,CAAC;QAC9B,CAAC,EAAE,EAAE,EAAE;YACL,OAAO,CAAC,GAAG,CAAC,uCAAuC,CAAC,CAAC;YAGrD,MAAM,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,aAAa,CAAC,CAAC;YACtE,MAAM,kBAAkB,GAAG,aAAa,GAAG,aAAa,CAAC;YAEzD,MAAM,mBAAmB,GAAG,IAAI,CAAC,KAAK,CAAC,aAAa,GAAG,aAAa,CAAC,CAAC;YACtE,MAAM,kBAAkB,GAAG,aAAa,GAAG,aAAa,CAAC;YAEzD,OAAO,CAAC,GAAG,CAAC,mCAAmC,mBAAmB,EAAE,CAAC,CAAC;YACtE,OAAO,CAAC,GAAG,CAAC,mCAAmC,mBAAmB,EAAE,CAAC,CAAC;YAEtE,kBAAkB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAChC,mBAAmB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACjC,oBAAoB,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAElC,IAAI,CAAC;gBAEH,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;oBACvC,IAAI,oBAAoB,GAAG,mBAAmB,CAAC;oBAC/C,IAAI,CAAC,GAAG,kBAAkB,EAAE,CAAC;wBAC3B,oBAAoB,IAAI,CAAC,CAAC;oBAC5B,CAAC;oBAED,OAAO,CAAC,GAAG,CACT,YAAY,CAAC,GAAG,CAAC,iBAAiB,oBAAoB,WAAW,CAClE,CAAC;oBAEF,MAAM,QAAQ,GAAG,YAAY,CAAC;wBAC5B,UAAU,EAAE,kBAAkB;wBAC9B,QAAQ,EAAE,CAAC;wBACX,gBAAgB,EAAE,oBAAoB;wBACtC,WAAW;wBACX,KAAK;wBACL,SAAS,EAAE,iBAAiB;qBAC7B,CAAC,CAAC;oBAEH,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC5B,CAAC;gBAGD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;oBACvC,IAAI,oBAAoB,GAAG,mBAAmB,CAAC;oBAC/C,IAAI,CAAC,GAAG,kBAAkB,EAAE,CAAC;wBAC3B,oBAAoB,IAAI,CAAC,CAAC;oBAC5B,CAAC;oBAED,OAAO,CAAC,GAAG,CACT,YAAY,CAAC,GAAG,CAAC,iBAAiB,oBAAoB,WAAW,CAClE,CAAC;oBAEF,MAAM,QAAQ,GAAG,YAAY,CAAC;wBAC5B,UAAU,EAAE,kBAAkB;wBAC9B,QAAQ,EAAE,CAAC;wBACX,gBAAgB,EAAE,oBAAoB;wBACtC,WAAW;wBACX,KAAK;wBACL,SAAS,EAAE,iBAAiB;qBAC7B,CAAC,CAAC;oBAEH,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC5B,CAAC;gBAED,EAAE,EAAE,CAAC;YACP,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,EAAE,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAC1D,CAAC;QACH,CAAC;KACF,EACD,CAAC,GAAG,EAAE,EAAE;QACN,IAAI,GAAG,EAAE,CAAC;YACR,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YAE9C,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,OAAO,CACjE,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,CACd,CAAC;QACJ,CAAC;IACH,CAAC,CACF,CAAC;AACJ,CAAC"}
@@ -0,0 +1,13 @@
1
+ import { Worker } from 'worker_threads';
2
+ import { IWorkerMessage } from '../types/index.js';
3
+ import { IQueueParams } from 'redis-smq';
4
+ import { IRedisConfig } from 'redis-smq-common';
5
+ export declare function createWorker(params: {
6
+ workerId: number;
7
+ workerPath: string;
8
+ queue: IQueueParams;
9
+ expectedMessages: number;
10
+ redisConfig: IRedisConfig;
11
+ onMessage: (message: IWorkerMessage) => void;
12
+ }): Worker;
13
+ //# sourceMappingURL=create-worker.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"create-worker.d.ts","sourceRoot":"","sources":["../../../../src/helpers/create-worker.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AACxC,OAAO,EAAe,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAEhD,wBAAgB,YAAY,CAAC,MAAM,EAAE;IACnC,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE,YAAY,CAAC;IACpB,gBAAgB,EAAE,MAAM,CAAC;IACzB,WAAW,EAAE,YAAY,CAAC;IAC1B,SAAS,EAAE,CAAC,OAAO,EAAE,cAAc,KAAK,IAAI,CAAC;CAC9C,UAgCA"}
@@ -0,0 +1,24 @@
1
+ import { Worker } from 'worker_threads';
2
+ export function createWorker(params) {
3
+ const { queue, workerId, expectedMessages, redisConfig } = params;
4
+ const workerData = {
5
+ redisConfig,
6
+ queue,
7
+ workerId,
8
+ expectedMessages,
9
+ };
10
+ const w = new Worker(params.workerPath, {
11
+ workerData,
12
+ stdout: true,
13
+ stderr: true,
14
+ });
15
+ w.on('message', params.onMessage);
16
+ w.on('error', (e) => {
17
+ console.error('Worker thread error:', e);
18
+ });
19
+ w.on('exit', (code) => {
20
+ console.warn(`Worker exited with code ${code}`);
21
+ });
22
+ return w;
23
+ }
24
+ //# sourceMappingURL=create-worker.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"create-worker.js","sourceRoot":"","sources":["../../../../src/helpers/create-worker.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAKxC,MAAM,UAAU,YAAY,CAAC,MAO5B;IAEC,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,gBAAgB,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC;IAClE,MAAM,UAAU,GAAgB;QAC9B,WAAW;QACX,KAAK;QACL,QAAQ;QACR,gBAAgB;KACjB,CAAC;IAGF,MAAM,CAAC,GAAG,IAAI,MAAM,CAAC,MAAM,CAAC,UAAU,EAAE;QACtC,UAAU;QACV,MAAM,EAAE,IAAI;QACZ,MAAM,EAAE,IAAI;KACb,CAAC,CAAC;IAMH,CAAC,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IAElC,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;QAClB,OAAO,CAAC,KAAK,CAAC,sBAAsB,EAAE,CAAC,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;QACpB,OAAO,CAAC,IAAI,CAAC,2BAA2B,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,CAAC;AACX,CAAC"}
@@ -0,0 +1,4 @@
1
+ import { ICallback } from 'redis-smq-common';
2
+ import { IQueueParams } from 'redis-smq';
3
+ export declare function ensureQueue(queue: IQueueParams, cb: ICallback<void>): void;
4
+ //# sourceMappingURL=ensure-queue.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ensure-queue.d.ts","sourceRoot":"","sources":["../../../../src/helpers/ensure-queue.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAGL,YAAY,EAEb,MAAM,WAAW,CAAC;AAEnB,wBAAgB,WAAW,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE,EAAE,SAAS,CAAC,IAAI,CAAC,QAWnE"}
@@ -0,0 +1,10 @@
1
+ import { EQueueDeliveryModel, EQueueType, RedisSMQ, } from 'redis-smq';
2
+ export function ensureQueue(queue, cb) {
3
+ const qm = RedisSMQ.createQueueManager();
4
+ qm.save(queue, EQueueType.FIFO_QUEUE, EQueueDeliveryModel.POINT_TO_POINT, (err) => {
5
+ if (err && err.name !== 'QueueAlreadyExistsError')
6
+ return cb(err);
7
+ cb();
8
+ });
9
+ }
10
+ //# sourceMappingURL=ensure-queue.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ensure-queue.js","sourceRoot":"","sources":["../../../../src/helpers/ensure-queue.ts"],"names":[],"mappings":"AAUA,OAAO,EACL,mBAAmB,EACnB,UAAU,EAEV,QAAQ,GACT,MAAM,WAAW,CAAC;AAEnB,MAAM,UAAU,WAAW,CAAC,KAAmB,EAAE,EAAmB;IAClE,MAAM,EAAE,GAAG,QAAQ,CAAC,kBAAkB,EAAE,CAAC;IACzC,EAAE,CAAC,IAAI,CACL,KAAK,EACL,UAAU,CAAC,UAAU,EACrB,mBAAmB,CAAC,cAAc,EAClC,CAAC,GAAG,EAAE,EAAE;QACN,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,yBAAyB;YAAE,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;QAClE,EAAE,EAAE,CAAC;IACP,CAAC,CACF,CAAC;AACJ,CAAC"}
@@ -0,0 +1,4 @@
1
+ import { ICallback } from 'redis-smq-common';
2
+ import { IQueueParams } from 'redis-smq';
3
+ export declare function prefillQueue(queue: IQueueParams, count: number, cb: ICallback): void;
4
+ //# sourceMappingURL=prefill-queue.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"prefill-queue.d.ts","sourceRoot":"","sources":["../../../../src/helpers/prefill-queue.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,YAAY,EAA+B,MAAM,WAAW,CAAC;AAEtE,wBAAgB,YAAY,CAC1B,KAAK,EAAE,YAAY,EACnB,KAAK,EAAE,MAAM,EACb,EAAE,EAAE,SAAS,QAiBd"}
@@ -0,0 +1,23 @@
1
+ import { ProducibleMessage, RedisSMQ } from 'redis-smq';
2
+ export function prefillQueue(queue, count, cb) {
3
+ const producer = RedisSMQ.startProducer((err) => {
4
+ if (err)
5
+ return cb(err);
6
+ let sent = 0;
7
+ const pump = () => {
8
+ if (sent < count) {
9
+ sent++;
10
+ const msg = new ProducibleMessage().setQueue(queue).setBody(`m${sent}`);
11
+ producer.produce(msg, (perr) => {
12
+ if (perr)
13
+ return cb(perr);
14
+ pump();
15
+ });
16
+ }
17
+ else
18
+ cb();
19
+ };
20
+ pump();
21
+ });
22
+ }
23
+ //# sourceMappingURL=prefill-queue.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"prefill-queue.js","sourceRoot":"","sources":["../../../../src/helpers/prefill-queue.ts"],"names":[],"mappings":"AAUA,OAAO,EAAgB,iBAAiB,EAAE,QAAQ,EAAE,MAAM,WAAW,CAAC;AAEtE,MAAM,UAAU,YAAY,CAC1B,KAAmB,EACnB,KAAa,EACb,EAAa;IAEb,MAAM,QAAQ,GAAG,QAAQ,CAAC,aAAa,CAAC,CAAC,GAAG,EAAE,EAAE;QAC9C,IAAI,GAAG;YAAE,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,MAAM,IAAI,GAAG,GAAG,EAAE;YAChB,IAAI,IAAI,GAAG,KAAK,EAAE,CAAC;gBACjB,IAAI,EAAE,CAAC;gBACP,MAAM,GAAG,GAAG,IAAI,iBAAiB,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBACxE,QAAQ,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,EAAE,EAAE;oBAC7B,IAAI,IAAI;wBAAE,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;oBAC1B,IAAI,EAAE,CAAC;gBACT,CAAC,CAAC,CAAC;YACL,CAAC;;gBAAM,EAAE,EAAE,CAAC;QACd,CAAC,CAAC;QACF,IAAI,EAAE,CAAC;IACT,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,8 @@
1
+ import { ICallback } from 'redis-smq-common';
2
+ import { IBenchmarkResult } from './types/index.js';
3
+ import { BaseBenchmark } from './common/base-benchmark.js';
4
+ export declare class ProducerBenchmark extends BaseBenchmark {
5
+ protected ensureQueue(): Promise<void>;
6
+ run(cb: ICallback<IBenchmarkResult>): void;
7
+ }
8
+ //# sourceMappingURL=producer-benchmark.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"producer-benchmark.d.ts","sourceRoot":"","sources":["../../../src/producer-benchmark.ts"],"names":[],"mappings":"AAAA,OAAO,EAAS,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAEpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAE3D,qBAAa,iBAAkB,SAAQ,aAAa;cAClC,WAAW,IAAI,OAAO,CAAC,IAAI,CAAC;IASrC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,gBAAgB,CAAC,GAAG,IAAI;CAyClD"}
@@ -0,0 +1,47 @@
1
+ import { async } from 'redis-smq-common';
2
+ import { ensureQueue } from './helpers/ensure-queue.js';
3
+ import { BaseBenchmark } from './common/base-benchmark.js';
4
+ export class ProducerBenchmark extends BaseBenchmark {
5
+ async ensureQueue() {
6
+ return new Promise((resolve, reject) => {
7
+ ensureQueue(this.queue, (err) => {
8
+ if (err)
9
+ reject(err);
10
+ else
11
+ resolve();
12
+ });
13
+ });
14
+ }
15
+ run(cb) {
16
+ console.log('Starting producer throughput benchmark...');
17
+ console.log(`Queue: ${this.queue.ns}/${this.queue.name} | Messages: ${this.totalMessages} | Producers: ${this.workerCount}`);
18
+ const onComplete = (result) => {
19
+ this.shutdownWorkers()
20
+ .then(() => {
21
+ const totalTime = Date.now() - this.startTime;
22
+ console.log('\n========== BENCHMARK COMPLETE ==========');
23
+ console.log(`Total messages produced: ${result.total}`);
24
+ console.log(`Total time: ${(totalTime / 1000).toFixed(2)}s`);
25
+ console.log(`Overall throughput: ${(result.total / (totalTime / 1000)).toFixed(0)} messages/second`);
26
+ console.log('========================================\n');
27
+ cb(null, result);
28
+ })
29
+ .catch((err) => cb(err));
30
+ };
31
+ const messageHandler = this.createMessageHandler(onComplete);
32
+ async.series([
33
+ (cb) => this.ensureQueue()
34
+ .then(() => cb())
35
+ .catch(cb),
36
+ (cb) => {
37
+ this.createWorkers(messageHandler)
38
+ .then(() => cb())
39
+ .catch(cb);
40
+ },
41
+ ], (err) => {
42
+ if (err)
43
+ cb(err);
44
+ });
45
+ }
46
+ }
47
+ //# sourceMappingURL=producer-benchmark.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"producer-benchmark.js","sourceRoot":"","sources":["../../../src/producer-benchmark.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAa,MAAM,kBAAkB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AAExD,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAE3D,MAAM,OAAO,iBAAkB,SAAQ,aAAa;IACxC,KAAK,CAAC,WAAW;QACzB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,WAAW,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE;gBAC9B,IAAI,GAAG;oBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;oBAChB,OAAO,EAAE,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,GAAG,CAAC,EAA+B;QACxC,OAAO,CAAC,GAAG,CAAC,2CAA2C,CAAC,CAAC;QACzD,OAAO,CAAC,GAAG,CACT,UAAU,IAAI,CAAC,KAAK,CAAC,EAAE,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,gBAAgB,IAAI,CAAC,aAAa,iBAAiB,IAAI,CAAC,WAAW,EAAE,CAChH,CAAC;QAEF,MAAM,UAAU,GAAG,CAAC,MAAwB,EAAE,EAAE;YAC9C,IAAI,CAAC,eAAe,EAAE;iBACnB,IAAI,CAAC,GAAG,EAAE;gBACT,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC;gBAC9C,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;gBAC1D,OAAO,CAAC,GAAG,CAAC,4BAA4B,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC;gBACxD,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBAC7D,OAAO,CAAC,GAAG,CACT,uBAAuB,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,kBAAkB,CACxF,CAAC;gBACF,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;gBAC1D,EAAE,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnB,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC;QAE7D,KAAK,CAAC,MAAM,CACV;YACE,CAAC,EAAE,EAAE,EAAE,CACL,IAAI,CAAC,WAAW,EAAE;iBACf,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;iBAChB,KAAK,CAAC,EAAE,CAAC;YACd,CAAC,EAAE,EAAE,EAAE;gBACL,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC;qBAC/B,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;qBAChB,KAAK,CAAC,EAAE,CAAC,CAAC;YACf,CAAC;SACF,EACD,CAAC,GAAG,EAAE,EAAE;YACN,IAAI,GAAG;gBAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC,CACF,CAAC;IACJ,CAAC;CACF"}
@@ -0,0 +1,2 @@
1
+ export {};
2
+ //# sourceMappingURL=consumer-worker-thread.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"consumer-worker-thread.d.ts","sourceRoot":"","sources":["../../../../src/threads/consumer-worker-thread.ts"],"names":[],"mappings":""}