From 23ed840aa3cc6ba49f46334abd22aca82193b86b Mon Sep 17 00:00:00 2001
From: David Dworken <david@daviddworken.com>
Date: Tue, 14 Jan 2025 21:44:52 -0700
Subject: [PATCH] Add deep clean for users that have tons of pending deletion
 requests (#285)

---
 backend/server/internal/database/db.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/backend/server/internal/database/db.go b/backend/server/internal/database/db.go
index e7c3a9e..096b49d 100644
--- a/backend/server/internal/database/db.go
+++ b/backend/server/internal/database/db.go
@@ -544,6 +544,30 @@ func (db *DB) DeepClean(ctx context.Context) error {
 	if err != nil {
 		return err
 	}
+	err = db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
+		r := tx.Exec(`
+		CREATE TEMP TABLE high_del_req_users AS (
+			SELECT user_id
+			FROM deletion_requests
+			GROUP BY 1
+			HAVING COUNT(*) > 100000
+		)`)
+		if r.Error != nil {
+			return fmt.Errorf("failed to create list of users with too many deletion requests: %w", r.Error)
+		}
+		r = tx.Raw(`
+		DELETE FROM deletion_requests 
+			WHERE user_id IN (SELECT * FROM high_del_req_users) AND read_count > 25;
+		`)
+		if r.Error != nil {
+			return fmt.Errorf("failed to delete entries for overly active users: %w", r.Error)
+		}
+		fmt.Printf("Ran deep clean for overly active deletion requests and deleted %d rows\n", r.RowsAffected)
+		return nil
+	})
+	if err != nil {
+		return err
+	}
 	return db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
 		// Delete entries for integration test users
 		r := tx.Exec(`