• Index
• Search by APIs
• Search by Words
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• nestjs-throttler-storage-redis-master
  • .github
    • dependabot.yml
    • workflows
      • tests.yml
  • .eslintignore
  • src
    • throttler-storage-redis.interface.ts
    • type.ts
    • throttler-storage-redis.service.ts
    • index.ts
  • .prettierrc
  • LICENSE
  • test
    • controller.e2e-spec.ts
    • utility
      • httpromise.ts
      • redis.ts
    • jest-e2e.json
    • app
      • app.module.ts
      • app.service.ts
      • controllers
        • limit.controller.ts
        • controller.module.ts
        • app.controller.ts
        • default.controller.ts
      • main.ts
  • commitlint.config.js
  • .npmrc
  • CONTRIBUTING.md
  • CHANGELOG.md
  • tsconfig.build.json
  • README.md
  • package.json
  • CODE_OF_CONDUCT.md
  • .eslintrc.js
  • tsconfig.json
  • .gitignore
  • .vscode
    • settings.json
  • docker-compose.yml
  • nest-cli.json

import { Injectable, OnModuleDestroy } from '@nestjs/common';
import Redis, { RedisOptions } from 'ioredis';
import { ThrottlerStorageRedis } from './throttler-storage-redis.interface';

@Injectable()
export class ThrottlerStorageRedisService implements ThrottlerStorageRedis, OnModuleDestroy {
  redis: Redis;
  disconnectRequired?: boolean;
  scanCount: number;

  constructor(redis?: Redis, scanCount?: number);
  constructor(options?: RedisOptions, scanCount?: number);
  constructor(url?: string, scanCount?: number);
  constructor(redisOrOptions?: Redis | RedisOptions | string, scanCount?: number) {
    this.scanCount = typeof scanCount === 'undefined' ? 1000 : scanCount;

    if (redisOrOptions instanceof Redis) {
      this.redis = redisOrOptions;
    } else if (typeof redisOrOptions === 'string') {
      this.redis = new Redis(redisOrOptions as string);
      this.disconnectRequired = true;
    } else {
      this.redis = new Redis(redisOrOptions);
      this.disconnectRequired = true;
    }
  }

  async getRecord(key: string): Promise<number[]> {
    const ttls = (
      await this.redis.scan(
        0,
        'MATCH',
        `${this.redis?.options?.keyPrefix}${key}:*`,
        'COUNT',
        this.scanCount,
      )
    ).pop();
    return (ttls as string[]).map((k) => parseInt(k.split(':').pop())).sort();
  }

  async addRecord(key: string, ttl: number): Promise<void> {
    await this.redis.set(`${key}:${Date.now() + ttl * 1000}`, ttl, 'EX', ttl);
  }

  onModuleDestroy() {
    if (this.disconnectRequired) {
      this.redis?.disconnect(false);
    }
  }
}