WangDL
c7052ee48e
All checks were successful
Deploy API Server / build-and-deploy (push) Successful in 37s
21 lines
978 B
TypeScript
21 lines
978 B
TypeScript
import { Injectable } from '@nestjs/common';
|
|
import { ThrottlerStorage } from '@nestjs/throttler';
|
|
import { ThrottlerStorageRecord } from '@nestjs/throttler/dist/throttler-storage-record.interface';
|
|
import { RedisService } from '../../infrastructure/redis/redis.service';
|
|
|
|
@Injectable()
|
|
export class RedisThrottlerStorage implements ThrottlerStorage {
|
|
constructor(private readonly redis: RedisService) {}
|
|
|
|
async increment(key: string, ttl: number, limit: number, blockDuration: number, throttlerName: string): Promise<ThrottlerStorageRecord> {
|
|
const redisKey = `throttle:${throttlerName}:${key}`;
|
|
try {
|
|
const hits = await this.redis.incr(redisKey);
|
|
await this.redis.expire(redisKey, Math.ceil(ttl / 1000));
|
|
return { totalHits: hits, timeToExpire: ttl, isBlocked: hits > limit, timeToBlockExpire: hits > limit ? blockDuration : 0 };
|
|
} catch {
|
|
return { totalHits: 1, timeToExpire: ttl, isBlocked: false, timeToBlockExpire: 0 };
|
|
}
|
|
}
|
|
}
|