summaryrefslogtreecommitdiff
path: root/packages/backend/src/queue/processors/db/delete-drive-files.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/backend/src/queue/processors/db/delete-drive-files.ts')
-rw-r--r--packages/backend/src/queue/processors/db/delete-drive-files.ts56
1 files changed, 56 insertions, 0 deletions
diff --git a/packages/backend/src/queue/processors/db/delete-drive-files.ts b/packages/backend/src/queue/processors/db/delete-drive-files.ts
new file mode 100644
index 0000000000..8a28468b0d
--- /dev/null
+++ b/packages/backend/src/queue/processors/db/delete-drive-files.ts
@@ -0,0 +1,56 @@
+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<DbUserJobData>, done: any): Promise<void> {
+ 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();
+}