add: cleanup scripts

This commit is contained in:
Carl-Gerhard Lindesvärd
2025-11-03 13:25:17 +01:00
parent 730c953bbc
commit 9bae0fb2db
2 changed files with 489 additions and 0 deletions

View File

@@ -0,0 +1,206 @@
/**
* Migration Script: Migrate Delayed Jobs to New Queue Names
*
* This script migrates delayed jobs from old queue names (e.g., "sessions")
* to new queue names with hash tags (e.g., "{sessions}").
*
* Active/waiting jobs are ignored - only delayed jobs are migrated.
*
* Usage:
* npx tsx apps/worker/scripts/migrate-delayed-jobs.ts
*
* Options:
* --dry-run Show what would be migrated without actually doing it
* --queue Migrate specific queue only (sessions, cron, notification, misc)
*
* # Dry run (recommended first)
* npx tsx apps/worker/scripts/migrate-delayed-jobs.ts --dry-run
*
* Migrate all queues
* npx tsx apps/worker/scripts/migrate-delayed-jobs.ts
*
* Migrate specific queue only
* npx tsx apps/worker/scripts/migrate-delayed-jobs.ts --queue=sessions
* npx tsx apps/worker/scripts/migrate-delayed-jobs.ts --queue=misc
*
*/
import type {
CronQueuePayload,
MiscQueuePayload,
NotificationQueuePayload,
SessionsQueuePayload,
} from '@openpanel/queue';
import { getRedisQueue } from '@openpanel/redis';
import { Queue } from 'bullmq';
interface MigrationStats {
queue: string;
total: number;
migrated: number;
failed: number;
skipped: number;
}
const isDryRun = process.argv.includes('--dry-run');
const specificQueue = process.argv
.find((arg) => arg.startsWith('--queue='))
?.split('=')[1];
console.log('🚀 Starting delayed jobs migration');
console.log(
`Mode: ${isDryRun ? 'DRY RUN (no changes will be made)' : 'LIVE MIGRATION'}`,
);
console.log(`Queue filter: ${specificQueue || 'all queues'}`);
console.log('---\n');
async function migrateDelayedJobs<T>(
oldQueueName: string,
newQueueName: string,
): Promise<MigrationStats> {
const stats: MigrationStats = {
queue: oldQueueName,
total: 0,
migrated: 0,
failed: 0,
skipped: 0,
};
const connection = getRedisQueue();
const oldQueue = new Queue<T>(oldQueueName, { connection });
const newQueue = new Queue<T>(newQueueName, { connection });
try {
console.log(`\n📦 Processing queue: ${oldQueueName}${newQueueName}`);
// Get all delayed jobs from old queue
const delayedJobs = await oldQueue.getDelayed();
stats.total = delayedJobs.length;
console.log(` Found ${stats.total} delayed jobs`);
if (stats.total === 0) {
console.log(' ✓ No delayed jobs to migrate');
return stats;
}
for (const job of delayedJobs) {
try {
const delay = job.opts.delay || 0;
const remainingDelay = Math.max(0, job.timestamp + delay - Date.now());
console.log(
` - Job ${job.id}: ${job.name}, delay: ${Math.round(remainingDelay / 1000)}s remaining`,
);
if (!isDryRun) {
// Add to new queue with remaining delay
await newQueue.add(job.name || 'migrated-job', job.data, {
...job.opts,
delay: remainingDelay,
jobId: job.id, // Preserve job ID if possible
attempts: job.opts.attempts,
backoff: job.opts.backoff,
});
// Remove from old queue
await job.remove();
stats.migrated++;
console.log(' ✓ Migrated');
} else {
stats.migrated++;
console.log(' ✓ Would migrate (dry run)');
}
} catch (error) {
stats.failed++;
console.error(
` ✗ Failed to migrate job ${job.id}:`,
error instanceof Error ? error.message : error,
);
}
}
console.log(`\n Summary for ${oldQueueName}:`);
console.log(` - Total: ${stats.total}`);
console.log(` - Migrated: ${stats.migrated}`);
console.log(` - Failed: ${stats.failed}`);
console.log(` - Skipped: ${stats.skipped}`);
} catch (error) {
console.error(` ✗ Error processing queue ${oldQueueName}:`, error);
} finally {
await oldQueue.close();
await newQueue.close();
}
return stats;
}
async function main() {
const queuesToMigrate: Array<{ old: string; new: string }> = [
{ old: 'sessions', new: '{sessions}' },
{ old: 'misc', new: '{misc}' },
];
// Filter to specific queue if requested
const filtered = specificQueue
? queuesToMigrate.filter((q) => q.old === specificQueue)
: queuesToMigrate;
if (filtered.length === 0) {
console.error(
`❌ Queue "${specificQueue}" not found. Valid queues: sessions, cron, notification, misc`,
);
process.exit(1);
}
const allStats: MigrationStats[] = [];
for (const { old: oldName, new: newName } of filtered) {
const stats = await migrateDelayedJobs(oldName, newName);
allStats.push(stats);
}
// Print summary
console.log(`\n${'='.repeat(50)}`);
console.log('📊 MIGRATION SUMMARY');
console.log(`${'='.repeat(50)}\n`);
let totalJobs = 0;
let totalMigrated = 0;
let totalFailed = 0;
for (const stats of allStats) {
totalJobs += stats.total;
totalMigrated += stats.migrated;
totalFailed += stats.failed;
}
console.log(`Total jobs found: ${totalJobs}`);
console.log(`Successfully migrated: ${totalMigrated}`);
console.log(`Failed: ${totalFailed}`);
console.log(
`\n${isDryRun ? '⚠️ This was a DRY RUN - no changes were made' : '✅ Migration complete!'}`,
);
if (totalFailed > 0) {
console.log(
'\n⚠ Some jobs failed to migrate. Check the logs above for details.',
);
process.exit(1);
}
if (isDryRun && totalMigrated > 0) {
console.log('\n💡 Run without --dry-run to perform the actual migration');
}
}
main()
.then(() => {
console.log('\n✨ Done!');
process.exit(0);
})
.catch((error) => {
console.error('\n❌ Migration failed:', error);
process.exit(1);
});