08af99ed52
- Added new models for premium tier configurations and assignments, enabling feature limits for free and premium users. - Introduced GDPR deletion logging and commands for user data management. - Enhanced logging configuration with retention days for audit logs. - Updated bot commands and job processing to include new premium and GDPR functionalities. - Improved localization with new keys for premium features and GDPR commands in both English and German. - Added UI components for managing premium settings and logging retention in the WebUI.
78 lines
1.8 KiB
TypeScript
78 lines
1.8 KiB
TypeScript
import { Queue } from 'bullmq';
|
|
import { redis } from './redis';
|
|
|
|
export const OWNER_QUEUE_NAMES = [
|
|
'moderation',
|
|
'backups',
|
|
'automod',
|
|
'verification',
|
|
'reminders',
|
|
'giveaways',
|
|
'tickets',
|
|
'birthdays',
|
|
'stats',
|
|
'feeds',
|
|
'schedules',
|
|
'guild-backups',
|
|
'suggestions',
|
|
'presence',
|
|
'retention'
|
|
] as const;
|
|
|
|
export type OwnerQueueName = (typeof OWNER_QUEUE_NAMES)[number];
|
|
|
|
const globalForOwnerQueues = globalThis as unknown as {
|
|
__nexumiOwnerQueues?: Map<string, Queue>;
|
|
};
|
|
|
|
function getQueue(name: string): Queue {
|
|
const cache = globalForOwnerQueues.__nexumiOwnerQueues ?? new Map<string, Queue>();
|
|
globalForOwnerQueues.__nexumiOwnerQueues = cache;
|
|
let queue = cache.get(name);
|
|
if (!queue) {
|
|
queue = new Queue(name, { connection: redis });
|
|
cache.set(name, queue);
|
|
}
|
|
return queue;
|
|
}
|
|
|
|
export interface QueueStats {
|
|
name: string;
|
|
waiting: number;
|
|
active: number;
|
|
completed: number;
|
|
failed: number;
|
|
delayed: number;
|
|
}
|
|
|
|
export async function getAllQueueStats(): Promise<QueueStats[]> {
|
|
return Promise.all(
|
|
OWNER_QUEUE_NAMES.map(async (name) => {
|
|
const queue = getQueue(name);
|
|
const counts = await queue.getJobCounts('wait', 'active', 'completed', 'failed', 'delayed');
|
|
return {
|
|
name,
|
|
waiting: counts.wait,
|
|
active: counts.active,
|
|
completed: counts.completed,
|
|
failed: counts.failed,
|
|
delayed: counts.delayed
|
|
};
|
|
})
|
|
);
|
|
}
|
|
|
|
export async function retryFailedJobs(queueName: string, limit = 25): Promise<number> {
|
|
if (!(OWNER_QUEUE_NAMES as readonly string[]).includes(queueName)) {
|
|
throw new Error('Unknown queue');
|
|
}
|
|
const queue = getQueue(queueName);
|
|
const failed = await queue.getFailed(0, limit - 1);
|
|
let retried = 0;
|
|
for (const job of failed) {
|
|
await job.retry();
|
|
retried += 1;
|
|
}
|
|
return retried;
|
|
}
|