@klapeks/micro-db-utils 0.0.1

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 (49) hide show
  1. package/dist/backups/mysql.backup.d.ts +7 -0
  2. package/dist/backups/mysql.backup.js +156 -0
  3. package/dist/connections/cloud/s3.d.ts +23 -0
  4. package/dist/connections/cloud/s3.js +169 -0
  5. package/dist/connections/index.d.ts +5 -0
  6. package/dist/connections/index.js +25 -0
  7. package/dist/connections/kafka/index.d.ts +3 -0
  8. package/dist/connections/kafka/index.js +21 -0
  9. package/dist/connections/kafka/kafka.consumer.d.ts +13 -0
  10. package/dist/connections/kafka/kafka.consumer.js +128 -0
  11. package/dist/connections/kafka/kafka.d.ts +16 -0
  12. package/dist/connections/kafka/kafka.js +205 -0
  13. package/dist/connections/kafka/kafka.producer.d.ts +11 -0
  14. package/dist/connections/kafka/kafka.producer.js +101 -0
  15. package/dist/connections/mysql.connection.d.ts +15 -0
  16. package/dist/connections/mysql.connection.js +165 -0
  17. package/dist/connections/nosql/mongo.d.ts +56 -0
  18. package/dist/connections/nosql/mongo.js +192 -0
  19. package/dist/connections/nosql/redis.d.ts +310 -0
  20. package/dist/connections/nosql/redis.js +129 -0
  21. package/dist/connections/quiet.require.d.ts +1 -0
  22. package/dist/connections/quiet.require.js +12 -0
  23. package/dist/index.d.ts +4 -0
  24. package/dist/index.js +20 -0
  25. package/dist/migrations/index.d.ts +1 -0
  26. package/dist/migrations/index.js +17 -0
  27. package/dist/migrations/mysql.migrations.d.ts +6 -0
  28. package/dist/migrations/mysql.migrations.js +220 -0
  29. package/dist/selectors/abstract.select.handler.d.ts +26 -0
  30. package/dist/selectors/abstract.select.handler.js +202 -0
  31. package/dist/selectors/index.d.ts +3 -0
  32. package/dist/selectors/index.js +19 -0
  33. package/dist/selectors/raw-mysql.select.handler.d.ts +17 -0
  34. package/dist/selectors/raw-mysql.select.handler.js +149 -0
  35. package/dist/selectors/typeorm.select.handler.d.ts +21 -0
  36. package/dist/selectors/typeorm.select.handler.js +187 -0
  37. package/dist/utils/columns.types.d.ts +100 -0
  38. package/dist/utils/columns.types.js +29 -0
  39. package/dist/utils/index.d.ts +3 -0
  40. package/dist/utils/index.js +21 -0
  41. package/dist/utils/iso.date.time.d.ts +2 -0
  42. package/dist/utils/iso.date.time.js +27 -0
  43. package/dist/utils/sqlite3.utils.d.ts +4 -0
  44. package/dist/utils/sqlite3.utils.js +60 -0
  45. package/dist/utils/where.utils.d.ts +3 -0
  46. package/dist/utils/where.utils.js +22 -0
  47. package/index.d.ts +1 -0
  48. package/index.js +2 -0
  49. package/package.json +37 -0
@@ -0,0 +1,7 @@
1
+ import { DatabaseOptions } from "@klapeks/utils";
2
+ export declare function runRawBackup({ module, options, tables, outputFolder }: {
3
+ module: string;
4
+ options: DatabaseOptions;
5
+ tables: string[];
6
+ outputFolder: string;
7
+ }): Promise<void>;
@@ -0,0 +1,156 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __generator = (this && this.__generator) || function (thisArg, body) {
12
+ var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
13
+ return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
14
+ function verb(n) { return function (v) { return step([n, v]); }; }
15
+ function step(op) {
16
+ if (f) throw new TypeError("Generator is already executing.");
17
+ while (g && (g = 0, op[0] && (_ = 0)), _) try {
18
+ if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
19
+ if (y = 0, t) op = [op[0] & 2, t.value];
20
+ switch (op[0]) {
21
+ case 0: case 1: t = op; break;
22
+ case 4: _.label++; return { value: op[1], done: false };
23
+ case 5: _.label++; y = op[1]; op = [0]; continue;
24
+ case 7: op = _.ops.pop(); _.trys.pop(); continue;
25
+ default:
26
+ if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
27
+ if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
28
+ if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
29
+ if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
30
+ if (t[2]) _.ops.pop();
31
+ _.trys.pop(); continue;
32
+ }
33
+ op = body.call(thisArg, _);
34
+ } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
35
+ if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
36
+ }
37
+ };
38
+ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
39
+ if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
40
+ if (ar || !(i in from)) {
41
+ if (!ar) ar = Array.prototype.slice.call(from, 0, i);
42
+ ar[i] = from[i];
43
+ }
44
+ }
45
+ return to.concat(ar || Array.prototype.slice.call(from));
46
+ };
47
+ var __importDefault = (this && this.__importDefault) || function (mod) {
48
+ return (mod && mod.__esModule) ? mod : { "default": mod };
49
+ };
50
+ Object.defineProperty(exports, "__esModule", { value: true });
51
+ exports.runRawBackup = void 0;
52
+ var path_1 = __importDefault(require("path"));
53
+ var fs_1 = __importDefault(require("fs"));
54
+ var child_process_1 = require("child_process");
55
+ var os_1 = __importDefault(require("os"));
56
+ var utils_1 = require("@klapeks/utils");
57
+ var pad = function (s) { return s < 10 ? "0".concat(s) : s; };
58
+ function dateToStr(date) {
59
+ return date.getFullYear()
60
+ + '-' + pad(date.getMonth() + 1)
61
+ + '-' + pad(date.getDate());
62
+ }
63
+ function runRawBackup(_a) {
64
+ var _b;
65
+ var module = _a.module, options = _a.options, tables = _a.tables, outputFolder = _a.outputFolder;
66
+ return __awaiter(this, void 0, void 0, function () {
67
+ var date, filePath_1, file_1, dump_1, stat, err_1;
68
+ return __generator(this, function (_c) {
69
+ switch (_c.label) {
70
+ case 0:
71
+ if (!options.type)
72
+ return [2 /*return*/];
73
+ if (!(tables === null || tables === void 0 ? void 0 : tables.length))
74
+ return [2 /*return*/];
75
+ if (os_1.default.platform() == 'win32')
76
+ return [2 /*return*/];
77
+ if (!fs_1.default.existsSync(outputFolder)) {
78
+ fs_1.default.mkdirSync(outputFolder, { recursive: true });
79
+ }
80
+ utils_1.logger.log("Starting database backup:", options.type, '|', tables);
81
+ date = new Date();
82
+ if (!(options.type == 'mysql')) return [3 /*break*/, 4];
83
+ filePath_1 = path_1.default.join(outputFolder, 'backup-' + dateToStr(date) + '-' + module + '.sql');
84
+ if (fs_1.default.existsSync(filePath_1))
85
+ fs_1.default.rmSync(filePath_1, { force: true });
86
+ file_1 = fs_1.default.createWriteStream(filePath_1);
87
+ file_1.write('-- BSB v4 dump. Tables: ' + tables.join(', ') + '\n');
88
+ file_1.write('-- DateTime: ' + new Date().toISOString().replace('T', ' ').replace('Z', '') + '\n');
89
+ _c.label = 1;
90
+ case 1:
91
+ _c.trys.push([1, 3, , 4]);
92
+ dump_1 = (0, child_process_1.spawn)('mysqldump', __spreadArray([
93
+ '--skip-extended-insert',
94
+ '--host=' + options.host,
95
+ '--port=' + options.port,
96
+ '--user=' + options.username,
97
+ '--password=' + options.password,
98
+ '--databases', options.database,
99
+ '--tables'
100
+ ], tables, true));
101
+ return [4 /*yield*/, new Promise(function (resolve, reject) {
102
+ var error = '';
103
+ dump_1.stdout.on('data', function (data) { return file_1.write(data); });
104
+ dump_1.stderr.on('data', function (data) {
105
+ if (String(data).toLowerCase().includes('[warning]'))
106
+ return;
107
+ error += String(data);
108
+ });
109
+ dump_1.once('exit', function (code) {
110
+ dump_1.kill();
111
+ if (code) {
112
+ utils_1.logger.log('child process exited with code ' + code);
113
+ return reject(error);
114
+ }
115
+ resolve("Done");
116
+ });
117
+ })];
118
+ case 2:
119
+ _c.sent();
120
+ file_1.end();
121
+ utils_1.logger.log("└ Backup completed and saved at\n", filePath_1);
122
+ stat = fs_1.default.statSync(filePath_1);
123
+ utils_1.logger.log("└ Backup size:", +(stat.size / 1024).toFixed(2), 'KB', '/', +(stat.size / 1024 / 1024).toFixed(2), 'MB');
124
+ return [2 /*return*/];
125
+ case 3:
126
+ err_1 = _c.sent();
127
+ file_1.end(function () {
128
+ fs_1.default.rmSync(filePath_1, { force: true });
129
+ });
130
+ if ((_b = err_1 === null || err_1 === void 0 ? void 0 : err_1.sqlMessage) === null || _b === void 0 ? void 0 : _b.toLowerCase().includes("unknown database"))
131
+ return [2 /*return*/];
132
+ utils_1.logger.error("Error while backuping database:", err_1);
133
+ return [2 /*return*/];
134
+ case 4: throw "Unknown database type for backup: " + options.type;
135
+ }
136
+ });
137
+ });
138
+ }
139
+ exports.runRawBackup = runRawBackup;
140
+ // export function runMicroBackup(micro: MicroServer, dataSource: DataSource) {
141
+ // return runRawBackup({
142
+ // module: micro.options.microServer,
143
+ // options: dataSourceOptions(),
144
+ // tables: Object.values(dataSource.options.entities || []).map((m: any) => {
145
+ // if (m instanceof EntitySchema) return m.options.name;
146
+ // return undefined;
147
+ // }).filter(Boolean) as string[],
148
+ // outputFolder: minEnv.coreDataFile('backups')
149
+ // })
150
+ // }
151
+ // export async function initBackups(micro: MicroServer, dataSource: DataSource) {
152
+ // cron.schedule('50 23 * * *', async () => {
153
+ // await runMicroBackup(micro, dataSource);
154
+ // });
155
+ // await runMicroBackup(micro, dataSource);
156
+ // }
@@ -0,0 +1,23 @@
1
+ /// <reference types="node" />
2
+ /// <reference types="node" />
3
+ import type { S3Client } from "@aws-sdk/client-s3";
4
+ export interface S3ConnectionConfig {
5
+ bucket: string;
6
+ url: string;
7
+ accessKeyId: string;
8
+ secretAccessKey: string;
9
+ }
10
+ export declare class S3Connection {
11
+ readonly client: S3Client;
12
+ private _config;
13
+ constructor(config: S3ConnectionConfig);
14
+ get bucketName(): string;
15
+ presignedUploadObjectUrl(path: string, mimeType: string): Promise<{
16
+ url: string;
17
+ mimeType: string;
18
+ }>;
19
+ uploadObject(path: string, buffer: Buffer, mimeType: string): Promise<import("@aws-sdk/client-s3").PutObjectCommandOutput>;
20
+ listByPrefix(prefix: string): Promise<import("@aws-sdk/client-s3").ListObjectsCommandOutput>;
21
+ deleteFiles(keys: string[]): Promise<import("@aws-sdk/client-s3").DeleteObjectsCommandOutput | undefined>;
22
+ removeDirectory(directory: string): Promise<void>;
23
+ }
@@ -0,0 +1,169 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __generator = (this && this.__generator) || function (thisArg, body) {
12
+ var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
13
+ return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
14
+ function verb(n) { return function (v) { return step([n, v]); }; }
15
+ function step(op) {
16
+ if (f) throw new TypeError("Generator is already executing.");
17
+ while (g && (g = 0, op[0] && (_ = 0)), _) try {
18
+ if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
19
+ if (y = 0, t) op = [op[0] & 2, t.value];
20
+ switch (op[0]) {
21
+ case 0: case 1: t = op; break;
22
+ case 4: _.label++; return { value: op[1], done: false };
23
+ case 5: _.label++; y = op[1]; op = [0]; continue;
24
+ case 7: op = _.ops.pop(); _.trys.pop(); continue;
25
+ default:
26
+ if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
27
+ if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
28
+ if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
29
+ if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
30
+ if (t[2]) _.ops.pop();
31
+ _.trys.pop(); continue;
32
+ }
33
+ op = body.call(thisArg, _);
34
+ } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
35
+ if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
36
+ }
37
+ };
38
+ Object.defineProperty(exports, "__esModule", { value: true });
39
+ exports.S3Connection = void 0;
40
+ var utils_1 = require("@klapeks/utils");
41
+ var quiet_require_1 = require("../quiet.require");
42
+ var s3ClientModule = (0, quiet_require_1.quietRequire)('@aws-sdk/client-s3');
43
+ var s3PresignerModule = (0, quiet_require_1.quietRequire)('@aws-sdk/s3-request-presigner');
44
+ var logger = new utils_1.Logger("S3/R2");
45
+ var fixPath = function (path) {
46
+ while (path[0] == '/')
47
+ path = path.substring(1);
48
+ if (path.includes('?'))
49
+ path = path.split('?')[0];
50
+ return path;
51
+ };
52
+ var S3Connection = /** @class */ (function () {
53
+ function S3Connection(config) {
54
+ if (!s3ClientModule)
55
+ throw "No @aws-sdk/client-s3 module installed";
56
+ this._config = config;
57
+ this.client = new s3ClientModule.S3Client({
58
+ region: "auto",
59
+ endpoint: config.url,
60
+ credentials: {
61
+ accessKeyId: config.accessKeyId,
62
+ secretAccessKey: config.secretAccessKey
63
+ },
64
+ });
65
+ logger.log("S3 connection created:", config.bucket);
66
+ }
67
+ Object.defineProperty(S3Connection.prototype, "bucketName", {
68
+ get: function () {
69
+ return this._config.bucket;
70
+ },
71
+ enumerable: false,
72
+ configurable: true
73
+ });
74
+ S3Connection.prototype.presignedUploadObjectUrl = function (path, mimeType) {
75
+ return __awaiter(this, void 0, void 0, function () {
76
+ var url;
77
+ return __generator(this, function (_a) {
78
+ switch (_a.label) {
79
+ case 0:
80
+ if (!s3PresignerModule)
81
+ throw "No @aws-sdk/s3-request-presigner module";
82
+ return [4 /*yield*/, s3PresignerModule.getSignedUrl(this.client, new s3ClientModule.PutObjectCommand({
83
+ Bucket: this.bucketName,
84
+ Key: fixPath(path),
85
+ ContentType: mimeType,
86
+ }), { expiresIn: 3600 })];
87
+ case 1:
88
+ url = _a.sent();
89
+ return [2 /*return*/, { url: url, mimeType: mimeType }];
90
+ }
91
+ });
92
+ });
93
+ };
94
+ S3Connection.prototype.uploadObject = function (path, buffer, mimeType) {
95
+ return __awaiter(this, void 0, void 0, function () {
96
+ return __generator(this, function (_a) {
97
+ switch (_a.label) {
98
+ case 0: return [4 /*yield*/, this.client.send(new s3ClientModule.PutObjectCommand({
99
+ Bucket: this.bucketName,
100
+ Key: fixPath(path),
101
+ Body: buffer,
102
+ ContentType: mimeType,
103
+ }))];
104
+ case 1: return [2 /*return*/, _a.sent()];
105
+ }
106
+ });
107
+ });
108
+ };
109
+ S3Connection.prototype.listByPrefix = function (prefix) {
110
+ return __awaiter(this, void 0, void 0, function () {
111
+ return __generator(this, function (_a) {
112
+ switch (_a.label) {
113
+ case 0:
114
+ while (prefix[0] == '/')
115
+ prefix = prefix.substring(1);
116
+ return [4 /*yield*/, this.client.send(new s3ClientModule.ListObjectsCommand({
117
+ Bucket: this.bucketName, Prefix: prefix
118
+ }))];
119
+ case 1: return [2 /*return*/, _a.sent()];
120
+ }
121
+ });
122
+ });
123
+ };
124
+ S3Connection.prototype.deleteFiles = function (keys) {
125
+ return __awaiter(this, void 0, void 0, function () {
126
+ return __generator(this, function (_a) {
127
+ switch (_a.label) {
128
+ case 0:
129
+ if (!keys.length)
130
+ return [2 /*return*/];
131
+ return [4 /*yield*/, this.client.send(new s3ClientModule.DeleteObjectsCommand({
132
+ Bucket: this.bucketName,
133
+ Delete: {
134
+ Objects: keys.map(function (k) { return ({ Key: fixPath(k) }); })
135
+ }
136
+ }))];
137
+ case 1: return [2 /*return*/, _a.sent()];
138
+ }
139
+ });
140
+ });
141
+ };
142
+ S3Connection.prototype.removeDirectory = function (directory) {
143
+ var _a, _b;
144
+ return __awaiter(this, void 0, void 0, function () {
145
+ var listedObjects, result;
146
+ return __generator(this, function (_c) {
147
+ switch (_c.label) {
148
+ case 0:
149
+ while (directory[0] == '/')
150
+ directory = directory.substring(1);
151
+ return [4 /*yield*/, this.listByPrefix(directory)];
152
+ case 1:
153
+ listedObjects = _c.sent();
154
+ listedObjects.Contents = (_a = listedObjects.Contents) === null || _a === void 0 ? void 0 : _a.filter(function (k) { return k.Key; });
155
+ logger.log("content:", listedObjects.Contents);
156
+ if (!((_b = listedObjects.Contents) === null || _b === void 0 ? void 0 : _b.length))
157
+ return [2 /*return*/];
158
+ return [4 /*yield*/, this.deleteFiles(listedObjects.Contents.map(function (k) { return k.Key; }))];
159
+ case 2:
160
+ result = _c.sent();
161
+ logger.log("Deleted:", result);
162
+ return [2 /*return*/];
163
+ }
164
+ });
165
+ });
166
+ };
167
+ return S3Connection;
168
+ }());
169
+ exports.S3Connection = S3Connection;
@@ -0,0 +1,5 @@
1
+ export * from './mysql.connection';
2
+ export * from './nosql/redis';
3
+ export * from './nosql/mongo';
4
+ export * from './kafka';
5
+ export * from './cloud/s3';
@@ -0,0 +1,25 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
16
+ Object.defineProperty(exports, "__esModule", { value: true });
17
+ // RDBMS
18
+ __exportStar(require("./mysql.connection"), exports);
19
+ // NoSQL
20
+ __exportStar(require("./nosql/redis"), exports);
21
+ __exportStar(require("./nosql/mongo"), exports);
22
+ // Brokers
23
+ __exportStar(require("./kafka"), exports);
24
+ // Cloud: S3 + R2
25
+ __exportStar(require("./cloud/s3"), exports);
@@ -0,0 +1,3 @@
1
+ export * from './kafka.consumer';
2
+ export * from './kafka.producer';
3
+ export { KafkaConnection } from './kafka';
@@ -0,0 +1,21 @@
1
+ "use strict";
2
+ var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
3
+ if (k2 === undefined) k2 = k;
4
+ var desc = Object.getOwnPropertyDescriptor(m, k);
5
+ if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
6
+ desc = { enumerable: true, get: function() { return m[k]; } };
7
+ }
8
+ Object.defineProperty(o, k2, desc);
9
+ }) : (function(o, m, k, k2) {
10
+ if (k2 === undefined) k2 = k;
11
+ o[k2] = m[k];
12
+ }));
13
+ var __exportStar = (this && this.__exportStar) || function(m, exports) {
14
+ for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
15
+ };
16
+ Object.defineProperty(exports, "__esModule", { value: true });
17
+ exports.KafkaConnection = void 0;
18
+ __exportStar(require("./kafka.consumer"), exports);
19
+ __exportStar(require("./kafka.producer"), exports);
20
+ var kafka_1 = require("./kafka");
21
+ Object.defineProperty(exports, "KafkaConnection", { enumerable: true, get: function () { return kafka_1.KafkaConnection; } });
@@ -0,0 +1,13 @@
1
+ import type { Consumer, ConsumerConfig, EachMessagePayload } from "kafkajs";
2
+ import { KafkaConnection } from "./kafka";
3
+ export type KafkaConnectionHandler = (message: any, payload: EachMessagePayload) => any;
4
+ export declare class KafkaConsumer {
5
+ readonly connection: KafkaConnection;
6
+ private readonly config;
7
+ readonly consumer: Consumer;
8
+ constructor(connection: KafkaConnection, config: ConsumerConfig);
9
+ private _needConnection;
10
+ connect(): Promise<void>;
11
+ subscribe(topic: string, handler: KafkaConnectionHandler): Promise<void>;
12
+ disconnect(): Promise<void>;
13
+ }
@@ -0,0 +1,128 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ var __generator = (this && this.__generator) || function (thisArg, body) {
12
+ var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
13
+ return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
14
+ function verb(n) { return function (v) { return step([n, v]); }; }
15
+ function step(op) {
16
+ if (f) throw new TypeError("Generator is already executing.");
17
+ while (g && (g = 0, op[0] && (_ = 0)), _) try {
18
+ if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
19
+ if (y = 0, t) op = [op[0] & 2, t.value];
20
+ switch (op[0]) {
21
+ case 0: case 1: t = op; break;
22
+ case 4: _.label++; return { value: op[1], done: false };
23
+ case 5: _.label++; y = op[1]; op = [0]; continue;
24
+ case 7: op = _.ops.pop(); _.trys.pop(); continue;
25
+ default:
26
+ if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
27
+ if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
28
+ if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
29
+ if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
30
+ if (t[2]) _.ops.pop();
31
+ _.trys.pop(); continue;
32
+ }
33
+ op = body.call(thisArg, _);
34
+ } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
35
+ if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
36
+ }
37
+ };
38
+ Object.defineProperty(exports, "__esModule", { value: true });
39
+ exports.KafkaConsumer = void 0;
40
+ var utils_1 = require("@klapeks/utils");
41
+ var logger = new utils_1.Logger("Kafka Consumer");
42
+ var KafkaConsumer = /** @class */ (function () {
43
+ function KafkaConsumer(connection, config) {
44
+ this.connection = connection;
45
+ this.config = config;
46
+ this._needConnection = true;
47
+ this.consumer = connection.kafka.consumer(config);
48
+ }
49
+ KafkaConsumer.prototype.connect = function () {
50
+ return __awaiter(this, void 0, void 0, function () {
51
+ return __generator(this, function (_a) {
52
+ switch (_a.label) {
53
+ case 0:
54
+ if (!this._needConnection)
55
+ return [2 /*return*/];
56
+ return [4 /*yield*/, this.consumer.connect()];
57
+ case 1:
58
+ _a.sent();
59
+ this._needConnection = false;
60
+ return [2 /*return*/];
61
+ }
62
+ });
63
+ });
64
+ };
65
+ KafkaConsumer.prototype.subscribe = function (topic, handler) {
66
+ return __awaiter(this, void 0, void 0, function () {
67
+ var _this = this;
68
+ return __generator(this, function (_a) {
69
+ switch (_a.label) {
70
+ case 0:
71
+ if (!this._needConnection) return [3 /*break*/, 2];
72
+ return [4 /*yield*/, this.connect()];
73
+ case 1:
74
+ _a.sent();
75
+ _a.label = 2;
76
+ case 2: return [4 /*yield*/, this.consumer.subscribe({
77
+ topic: topic,
78
+ fromBeginning: true,
79
+ })];
80
+ case 3:
81
+ _a.sent();
82
+ logger.log("Pre run:", topic);
83
+ return [4 /*yield*/, this.consumer.run({
84
+ eachMessage: function (payload) { return __awaiter(_this, void 0, void 0, function () {
85
+ var msg;
86
+ var _a;
87
+ return __generator(this, function (_b) {
88
+ switch (_b.label) {
89
+ case 0:
90
+ msg = (_a = payload.message.value) === null || _a === void 0 ? void 0 : _a.toString();
91
+ if (!msg)
92
+ return [2 /*return*/];
93
+ try {
94
+ msg = JSON.parse(msg);
95
+ }
96
+ catch (err) { }
97
+ return [4 /*yield*/, handler(msg, payload)];
98
+ case 1:
99
+ _b.sent();
100
+ return [2 /*return*/];
101
+ }
102
+ });
103
+ }); },
104
+ })];
105
+ case 4:
106
+ _a.sent();
107
+ logger.log("Listening:", topic);
108
+ return [2 /*return*/];
109
+ }
110
+ });
111
+ });
112
+ };
113
+ KafkaConsumer.prototype.disconnect = function () {
114
+ return __awaiter(this, void 0, void 0, function () {
115
+ return __generator(this, function (_a) {
116
+ switch (_a.label) {
117
+ case 0: return [4 /*yield*/, this.consumer.disconnect()];
118
+ case 1:
119
+ _a.sent();
120
+ logger.log("Consumer disconnected:", this.config.groupId);
121
+ return [2 /*return*/];
122
+ }
123
+ });
124
+ });
125
+ };
126
+ return KafkaConsumer;
127
+ }());
128
+ exports.KafkaConsumer = KafkaConsumer;
@@ -0,0 +1,16 @@
1
+ import type { ConsumerConfig, Kafka, ProducerConfig } from "kafkajs";
2
+ import { KafkaProducer } from "./kafka.producer";
3
+ import { KafkaConsumer } from "./kafka.consumer";
4
+ export declare const instances: {
5
+ producers: KafkaProducer[];
6
+ consumers: KafkaConsumer[];
7
+ };
8
+ export declare class KafkaConnection {
9
+ readonly clientId: string;
10
+ readonly kafka: Kafka;
11
+ constructor(clientId: string);
12
+ createTopic(topic: string, partitions: number, replicas: number): Promise<void>;
13
+ createProducer(config?: ProducerConfig): KafkaProducer;
14
+ createConsumer(config: ConsumerConfig): KafkaConsumer;
15
+ static cleanExit(): Promise<void>;
16
+ }