Make account req data migration script rerunnable (#12932)

TEAMMATES · Mar 26, 2024 · 716fdc4 · 716fdc4
1 parent 1be2adb
commit 716fdc4
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/src/client/java/teammates/client/scripts/sql/DataMigrationForAccountRequestSql.java b/src/client/java/teammates/client/scripts/sql/DataMigrationForAccountRequestSql.java
@@ -1,9 +1,15 @@
 package teammates.client.scripts.sql;
 
+// CHECKSTYLE.OFF:ImportOrder
 import com.googlecode.objectify.cmd.Query;
 
+import jakarta.persistence.criteria.CriteriaDelete;
+
+import teammates.common.util.HibernateUtil;
 import teammates.storage.sqlentity.AccountRequest;
 
+// CHECKSTYLE.ON:ImportOrder
+
 /**
  * Data migration class for account request entity.
  */
@@ -33,7 +39,8 @@ protected boolean isPreview() {
      */
     @Override
     protected void setMigrationCriteria() {
-        // No migration criteria currently needed.
+        // Prepare clean db before migration
+        cleanAccountRequestInSql();
     }
 
     /**
@@ -73,4 +80,15 @@ protected void migrateEntity(teammates.storage.entity.AccountRequest oldEntity)
 
         saveEntityDeferred(newEntity);
     }
+
+    private void cleanAccountRequestInSql() {
+        HibernateUtil.beginTransaction();
+
+        CriteriaDelete<AccountRequest> cdAccountReq = HibernateUtil.getCriteriaBuilder()
+                .createCriteriaDelete(AccountRequest.class);
+        cdAccountReq.from(AccountRequest.class);
+        HibernateUtil.executeDelete(cdAccountReq);
+
+        HibernateUtil.commitTransaction();
+    }
 }