generated from tailored/router-template
56 lines
1.5 KiB
TypeScript
56 lines
1.5 KiB
TypeScript
import { redis } from '@/modules/redis.ts';
|
|
import { Queue, Worker } from 'bullmq';
|
|
import { clamp } from 'lodash-es';
|
|
import { nanoid } from 'nanoid';
|
|
|
|
export const queue = new Queue('TEST_QUEUE');
|
|
export const sleep = (ms: number) => {
|
|
return new Promise((resolve) => setTimeout(resolve, ms));
|
|
};
|
|
export const worker = new Worker(
|
|
'TEST_QUEUE',
|
|
async (job) => {
|
|
const startTime = Date.now();
|
|
console.log('job', job.name, job.data);
|
|
await sleep(1000);
|
|
const endTime = Date.now();
|
|
const duration = endTime - startTime;
|
|
job.updateProgress(50);
|
|
const isRetry = job.attemptsMade > 0;
|
|
console.log('job attemptsMade', job.attemptsMade, 'isRetry', isRetry);
|
|
if (job.data.index === 1 && !isRetry) {
|
|
// set fail
|
|
throw new Error('test error');
|
|
}
|
|
return {
|
|
startTime: startTime,
|
|
endTime: endTime,
|
|
duration: duration,
|
|
};
|
|
},
|
|
{
|
|
connection: redis,
|
|
limiter: {
|
|
max: 1,
|
|
duration: 10000,
|
|
},
|
|
},
|
|
);
|
|
worker.on('completed', async (job) => {
|
|
console.log('job completed', job.name, job.id, job.returnvalue);
|
|
});
|
|
worker.on('failed', async (job) => {
|
|
console.log('job failed', job.name, job.id, job.failedReason, await job.getState());
|
|
});
|
|
|
|
queue.pause();
|
|
await queue.drain();
|
|
const arr = Array.from({ length: 2 }, (_, i) => i + 1);
|
|
for (const i of arr) {
|
|
queue.add('job' + i, { name: 'test' + i, index: i }, { jobId: 'job' + i, removeOnFail: true, removeOnComplete: true, attempts: 3 });
|
|
}
|
|
const job = await queue.getJob('job1');
|
|
console.log('job', job);
|
|
await sleep(4000)
|
|
queue.resume();
|