"[feat] Emails retention periods/cleanup #158" (#286)

2025-11-29 02:43:41 +05:30
parent 05a02844b2
commit 357d561a8e
4 changed files with 94 additions and 4 deletions
@@ -0,0 +1,73 @@
+import {Queue, Worker} from "bullmq";
+import {db} from "~/server/db";
+import {getRedis} from "~/server/redis";
+import {logger} from "../logger/log";
+import {DEFAULT_QUEUE_OPTIONS} from "../queue/queue-constants";
+import {env} from "~/env";
+import {isSelfHosted, isEmailCleanupEnabled} from "~/utils/common";
+
+const CLEANUP_QUEUE_NAME = "cleanup-email-bodies";
+
+const CLEANUP_CRON = "0 0 * * *"; // default: midnight UTC
+
+// Only initialize if self hosted and cleanup enabled
+if (isSelfHosted() && isEmailCleanupEnabled()) {
+    const CLEANUP_DAYS = env.EMAIL_CLEANUP_DAYS!;
+
+    /**
+     * Initialize Queue
+     */
+    const cleanupQueue = new Queue(CLEANUP_QUEUE_NAME, {
+        connection: getRedis(),
+    });
+
+    const worker = new Worker(
+        CLEANUP_QUEUE_NAME,
+        async () => {
+            logger.info(`[Cleanup] Starting cleanup for emails older than ${CLEANUP_DAYS} days...`);
+
+            const cutoffDate = new Date();
+            cutoffDate.setDate(cutoffDate.getDate() - CLEANUP_DAYS);
+
+            const result = await db.email.updateMany({
+                where: {
+                    createdAt: {lt: cutoffDate},
+                    OR: [
+                        {text: {not: null}},
+                        {html: {not: null}},
+                    ],
+                },
+                data: {
+                    text: null,
+                    html: null,
+                },
+            });
+
+            logger.info(`[Cleanup] Emails cleaned: ${result.count}`);
+        },
+        {
+            connection: getRedis(),
+        }
+    );
+
+    await cleanupQueue.upsertJobScheduler(
+        "scheduled-email-cleanup",
+        {
+            pattern: CLEANUP_CRON,
+            tz: "UTC",
+        },
+        {
+            opts: {
+                ...DEFAULT_QUEUE_OPTIONS,
+            },
+        }
+    );
+
+    worker.on("completed", (job) => {
+        logger.info({jobId: job.id}, ` Email Body cleanup job completed`);
+    });
+
+    worker.on("failed", (job, err) => {
+        logger.error({err, jobId: job?.id}, `Email Body cleanup job failed`);
+    });
+}