@mondart/nestjs-common-module 2.3.5 → 2.4.1

package/dist/lib/bottleneck/services/bottleneck.service.js

@@ -0,0 +1,93 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+var BottleneckService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BottleneckService = void 0;
+const common_1 = require("@nestjs/common");
+const bottleneck_1 = require("bottleneck");
+const bottleneck_options_interface_1 = require("../interfaces/bottleneck-options.interface");
+const ioredis_1 = require("ioredis");
+let BottleneckService = BottleneckService_1 = class BottleneckService {
+    constructor(options) {
+        this.options = options;
+        this.logger = new common_1.Logger(BottleneckService_1.name, {
+            timestamp: true,
+        });
+        this.cache = new Map();
+        const { useRedis, redis, appName, ...rest } = this.options;
+        const redisOptions = useRedis && redis
+            ? {
+                datastore: 'ioredis',
+                Redis: ioredis_1.default,
+                clientOptions: {
+                    host: redis.host,
+                    port: redis.port,
+                    username: redis.username,
+                    password: redis.password,
+                    db: redis.database,
+                    tls: redis.tls ? {} : undefined,
+                },
+                timeout: redis.ttl,
+            }
+            : undefined;
+        this.group = new bottleneck_1.default.Group({
+            id: `bottleneck-${appName ?? 'app'}`,
+            maxConcurrent: 1,
+            minTime: 0,
+            clearDatastore: true,
+            ...redisOptions,
+            ...rest,
+        });
+        this.group.on('idle', (key) => {
+            setTimeout(() => this.group.deleteKey(key).catch(() => { }), 30_000);
+        });
+    }
+    getLimiter(key, opts) {
+        const limiter = this.group.key(key);
+        if (opts)
+            limiter.updateSettings(opts);
+        return limiter;
+    }
+    scheduleForKey(key, task, opts) {
+        const limiter = this.getLimiter(key, opts);
+        this.logger.debug(`Scheduling task for key ${key}`);
+        return limiter.schedule(task);
+    }
+    async runQueuedTask(key, fn) {
+        if (this.cache.has(key)) {
+            this.logger.debug(`Cache hit for key ${key}`);
+            return this.cache.get(key);
+        }
+        const limiter = this.getLimiter(key);
+        const taskPromise = limiter.schedule(async () => {
+            try {
+                return await fn();
+            }
+            finally {
+                this.cache.delete(key);
+            }
+        });
+        this.cache.set(key, taskPromise);
+        return taskPromise;
+    }
+    async onModuleDestroy() {
+        await this.group.disconnect();
+    }
+};
+exports.BottleneckService = BottleneckService;
+exports.BottleneckService = BottleneckService = BottleneckService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, common_1.Inject)(bottleneck_options_interface_1.BOTTLENECK_MODULE_OPTIONS)),
+    __metadata("design:paramtypes", [Object])
+], BottleneckService);

package/dist/lib/index.d.ts

@@ -4,3 +4,4 @@ export * from './caching';
 export * from './keycloak';
 export * from './core-crud';
 export * from './health-check';
+export * from './bottleneck';

package/dist/lib/index.js

@@ -20,3 +20,4 @@ __exportStar(require("./caching"), exports);
 __exportStar(require("./keycloak"), exports);
 __exportStar(require("./core-crud"), exports);
 __exportStar(require("./health-check"), exports);
+__exportStar(require("./bottleneck"), exports);