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();