boki/stock-bot
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

fixed cache keys

Browse source
This commit is contained in:
Boki 2025-06-22 20:34:35 -04:00
parent db3aa9c330
commit 19dfda2392
13 changed files with 286 additions and 221 deletions
Download patch file Download diff file Expand all files Collapse all files

27
libs/services/queue/src/rate-limiter.ts
View file

@ -1,9 +1,6 @@
import { RateLimiterRedis, RateLimiterRes } from 'rate-limiter-flexible';
import { getLogger } from '@stock-bot/logger';
import type { RateLimitConfig as BaseRateLimitConfig, RateLimitRule } from './types';
const logger = getLogger('rate-limiter');
// Extend the base config to add rate-limiter specific fields
export interface RateLimitConfig extends BaseRateLimitConfig {
keyPrefix?: string;
@ -12,8 +9,14 @@ export interface RateLimitConfig extends BaseRateLimitConfig {
export class QueueRateLimiter {
private limiters = new Map<string, RateLimiterRedis>();
private rules: RateLimitRule[] = [];
private readonly logger: any;
constructor(private redisClient: ReturnType<typeof import('./utils').getRedisConnection>) {}
constructor(
private redisClient: ReturnType<typeof import('./utils').getRedisConnection>,
logger?: any
) {
this.logger = logger || console;
}
/**
* Add a rate limit rule
@ -32,7 +35,7 @@ export class QueueRateLimiter {
this.limiters.set(key, limiter);
logger.info('Rate limit rule added', {
this.logger.info('Rate limit rule added', {
level: rule.level,
queueName: rule.queueName,
handler: rule.handler,
@ -72,7 +75,7 @@ export class QueueRateLimiter {
const limiter = this.limiters.get(key);
if (!limiter) {
logger.warn('Rate limiter not found for rule', { key, rule: applicableRule });
this.logger.warn('Rate limiter not found for rule', { key, rule: applicableRule });
return { allowed: true };
}
@ -87,7 +90,7 @@ export class QueueRateLimiter {
appliedRule: applicableRule,
};
} catch (error) {
logger.error('Rate limit check failed', { queueName, handler, operation, error });
this.logger.error('Rate limit check failed', { queueName, handler, operation, error });
// On error, allow the request to proceed
return { allowed: true };
}
@ -148,7 +151,7 @@ export class QueueRateLimiter {
};
} catch (rejRes) {
if (rejRes instanceof RateLimiterRes) {
logger.warn('Rate limit exceeded', {
this.logger.warn('Rate limit exceeded', {
key,
retryAfter: rejRes.msBeforeNext,
});
@ -260,7 +263,7 @@ export class QueueRateLimiter {
limit,
};
} catch (error) {
logger.error('Failed to get rate limit status', { queueName, handler, operation, error });
this.logger.error('Failed to get rate limit status', { queueName, handler, operation, error });
return {
queueName,
handler,
@ -288,10 +291,10 @@ export class QueueRateLimiter {
}
} else {
// Reset broader scope - this is more complex with the new hierarchy
logger.warn('Broad reset not implemented yet', { queueName, handler, operation });
this.logger.warn('Broad reset not implemented yet', { queueName, handler, operation });
}
logger.info('Rate limits reset', { queueName, handler, operation });
this.logger.info('Rate limits reset', { queueName, handler, operation });
}
/**
@ -318,7 +321,7 @@ export class QueueRateLimiter {
this.rules.splice(ruleIndex, 1);
this.limiters.delete(key);
logger.info('Rate limit rule removed', { level, queueName, handler, operation });
this.logger.info('Rate limit rule removed', { level, queueName, handler, operation });
return true;
}