wangdl/api-server
2
0
Fork 0
Code Issues 45 Pull Requests Actions Packages Projects Releases Wiki Activity

fix: ThrottlerStorage v6 interface again
All checks were successful
Deploy API Server / build-and-deploy (push) Successful in 37s
Details

Browse Source
This commit is contained in:
WangDL 2026-05-22 23:16:06 +08:00
parent 178225f212
commit c7052ee48e
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/common/throttle/redis-throttler.storage.ts
View File

@ -1,19 +1,20 @@
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): Promise<{ totalHits: number; timeToExpire: number }> {
const redisKey = `throttle:${key}`;
async increment(key: string, ttl: number, limit: number, blockDuration: number, throttlerName: string): Promise<ThrottlerStorageRecord> {
const redisKey = `throttle:${throttlerName}:${key}`;
try {
const result = await this.redis.incr(redisKey);
const hits = await this.redis.incr(redisKey);
await this.redis.expire(redisKey, Math.ceil(ttl / 1000));
return { totalHits: result, timeToExpire: ttl };
return { totalHits: hits, timeToExpire: ttl, isBlocked: hits > limit, timeToBlockExpire: hits > limit ? blockDuration : 0 };
} catch {
return { totalHits: 1, timeToExpire: ttl };
return { totalHits: 1, timeToExpire: ttl, isBlocked: false, timeToBlockExpire: 0 };
}
}
}