import * as Bull from 'bull'; import { queueLogger } from '../../logger'; import { deleteFileSync } from '@/services/drive/delete-file'; import { Users, DriveFiles } from '@/models/index'; import { MoreThan } from 'typeorm'; import { DbUserJobData } from '@/queue/types'; const logger = queueLogger.createSubLogger('delete-drive-files'); export async function deleteDriveFiles(job: Bull.Job, done: any): Promise { logger.info(`Deleting drive files of ${job.data.user.id} ...`); const user = await Users.findOne(job.data.user.id); if (user == null) { done(); return; } let deletedCount = 0; let cursor: any = null; while (true) { const files = await DriveFiles.find({ where: { userId: user.id, ...(cursor ? { id: MoreThan(cursor) } : {}), }, take: 100, order: { id: 1, }, }); if (files.length === 0) { job.progress(100); break; } cursor = files[files.length - 1].id; for (const file of files) { await deleteFileSync(file); deletedCount++; } const total = await DriveFiles.count({ userId: user.id, }); job.progress(deletedCount / total); } logger.succ(`All drive files (${deletedCount}) of ${user.id} has been deleted.`); done(); }