@klapeks/micro-db-utils 0.0.6 → 0.0.8

package/dist/connections/kafka/kafka.consumer.d.ts

@@ -1,4 +1,4 @@
-import type { Consumer, ConsumerConfig, EachMessagePayload } from "kafkajs";
+import type { Consumer, ConsumerConfig, ConsumerRunConfig, EachMessagePayload } from "kafkajs";
 import { KafkaConnection } from "./kafka";
 export type KafkaConnectionHandler = (message: any, payload: EachMessagePayload) => any;
 export declare class KafkaConsumer {
@@ -8,6 +8,6 @@ export declare class KafkaConsumer {
     constructor(connection: KafkaConnection, config: ConsumerConfig);
     private _needConnection;
     connect(): Promise<void>;
-    subscribe(topic: string, handler: KafkaConnectionHandler): Promise<void>;
+    subscribe(topic: string, handler: KafkaConnectionHandler, consumerOptions?: Omit<ConsumerRunConfig, "eachMessage">): Promise<void>;
     disconnect(): Promise<void>;
 }

package/dist/connections/kafka/kafka.consumer.js

@@ -1,4 +1,15 @@
 "use strict";
+var __assign = (this && this.__assign) || function () {
+    __assign = Object.assign || function(t) {
+        for (var s, i = 1, n = arguments.length; i < n; i++) {
+            s = arguments[i];
+            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
+                t[p] = s[p];
+        }
+        return t;
+    };
+    return __assign.apply(this, arguments);
+};
 var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
     function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
     return new (P || (P = Promise))(function (resolve, reject) {
@@ -62,7 +73,7 @@ var KafkaConsumer = /** @class */ (function () {
             });
         });
     };
-    KafkaConsumer.prototype.subscribe = function (topic, handler) {
+    KafkaConsumer.prototype.subscribe = function (topic, handler, consumerOptions) {
         return __awaiter(this, void 0, void 0, function () {
             var _this = this;
             return __generator(this, function (_a) {
@@ -79,9 +90,13 @@ var KafkaConsumer = /** @class */ (function () {
                         })];
                     case 3:
                         _a.sent();
+                        if (!consumerOptions)
+                            consumerOptions = {};
+                        if (!consumerOptions.partitionsConsumedConcurrently) {
+                            consumerOptions.partitionsConsumedConcurrently = 10;
+                        }
                         logger.log("Pre run:", topic);
-                        return [4 /*yield*/, this.consumer.run({
-                                eachMessage: function (payload) { return __awaiter(_this, void 0, void 0, function () {
+                        return [4 /*yield*/, this.consumer.run(__assign(__assign({}, consumerOptions), { eachMessage: function (payload) { return __awaiter(_this, void 0, void 0, function () {
                                     var msg;
                                     var _a;
                                     return __generator(this, function (_b) {
@@ -100,8 +115,7 @@ var KafkaConsumer = /** @class */ (function () {
                                                 return [2 /*return*/];
                                         }
                                     });
-                                }); },
-                            })];
+                                }); } }))];
                     case 4:
                         _a.sent();
                         logger.log("Listening:", topic);

package/dist/connections/nosql/mongo.d.ts

@@ -43,7 +43,7 @@ export declare class MongoDBConnection {
         readonly set: (model: T, value: number) => Promise<any>;
     };
     static getOptions(): MongoDBConnectOptions;
-    static init(databaseName: string): Promise<void>;
+    static init(databaseName: string, replicaSet?: string): Promise<void>;
 }
 export declare const MongoGeoPoint: MongooseSchema<Point, import("mongoose").Model<Point, any, any, any, import("mongoose").Document<unknown, any, Point, any> & Point & {
     _id: import("mongoose").Types.ObjectId;

package/dist/connections/nosql/mongo.js

@@ -139,7 +139,7 @@ var MongoDBConnection = /** @class */ (function () {
             database: ""
         };
     };
-    MongoDBConnection.init = function (databaseName) {
+    MongoDBConnection.init = function (databaseName, replicaSet) {
         return __awaiter(this, void 0, void 0, function () {
             var _a, host, port, password, username, mongoURI;
             return __generator(this, function (_b) {
@@ -147,8 +147,12 @@ var MongoDBConnection = /** @class */ (function () {
                     case 0:
                         if (_databaseName)
                             return [2 /*return*/];
+                        if (!replicaSet && process.env.MONGO_DATABASE_REPLICA_SET) {
+                            replicaSet = process.env.MONGO_DATABASE_REPLICA_SET;
+                        }
                         _a = this.getOptions(), host = _a.host, port = _a.port, password = _a.password, username = _a.username;
-                        mongoURI = 'mongodb://' + host + ':' + port + '/' + databaseName;
+                        mongoURI = 'mongodb://' + host + ':' + port + '/' + databaseName
+                            + (replicaSet ? '?replicaSet=' + replicaSet : undefined);
                         // logger.log(mongoURI);
                         if (!mongooseModule)
                             throw new Error("No mongoose module");

package/dist/connections/nosql/redis.d.ts

@@ -18,4 +18,10 @@ export declare class RedisConnection {
         /** in seconds */
         expiredInSeconds?: number;
     }): Promise<void>;
+    setLock(key: string, options?: {
+        expiredInSeconds?: number;
+    }): Promise<{
+        isAlreadyLocked: boolean;
+    }>;
+    unlock(key: string): Promise<boolean>;
 }

package/dist/connections/nosql/redis.js

@@ -129,6 +129,38 @@ var RedisConnection = /** @class */ (function () {
             });
         });
     };
+    RedisConnection.prototype.setLock = function (key, options) {
+        return __awaiter(this, void 0, void 0, function () {
+            var result, err_2;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        _a.trys.push([0, 2, , 3]);
+                        return [4 /*yield*/, this.redisClient.set(this.keyPrefix + key, "1", {
+                                NX: true,
+                                EX: (options === null || options === void 0 ? void 0 : options.expiredInSeconds) || 60
+                            })];
+                    case 1:
+                        result = _a.sent();
+                        return [2 /*return*/, {
+                                isAlreadyLocked: !result
+                            }];
+                    case 2:
+                        err_2 = _a.sent();
+                        logger.error("Error locking redis:", (0, utils_1.shortErrorParser)(err_2));
+                        throw err_2;
+                    case 3: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    RedisConnection.prototype.unlock = function (key) {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                return [2 /*return*/, this.remove(key)];
+            });
+        });
+    };
     return RedisConnection;
 }());
 exports.RedisConnection = RedisConnection;

package/dist/selectors/typeorm.select.handler.js

@@ -136,7 +136,7 @@ var SelectEntityHandler = /** @class */ (function (_super) {
                                         for (var _i = 0, _c = Object.keys(where); _i < _c.length; _i++) {
                                             var key = _c[_i];
                                             var placeholderKey = key + "_" + i;
-                                            if (typeof where[key] == undefined)
+                                            if (typeof where[key] === 'undefined')
                                                 continue;
                                             if (where[key] == null) {
                                                 qb2 = qb2.andWhere('key is NULL');

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@klapeks/micro-db-utils",
     "description": "klapeks utils for db and other storages",
-    "version": "0.0.6",
+    "version": "0.0.8",
     "type": "commonjs",
     "author": "Klapeks",
     "license": "ISC",