Add changes (TEAMMATES#12965)

xenosf · Apr 14, 2024 · 1292dbe · 1292dbe
1 parent 0509982
commit 1292dbe
Showing 1 changed file with 26 additions and 1 deletion.
diff --git a/...ient/java/teammates/client/scripts/sql/DataMigrationForAccountAndReadNotificationSql.java b/...ient/java/teammates/client/scripts/sql/DataMigrationForAccountAndReadNotificationSql.java
@@ -20,6 +20,11 @@
 import com.google.cloud.datastore.QueryResults;
 import com.googlecode.objectify.cmd.Query;
 
+import jakarta.persistence.TypedQuery;
+import jakarta.persistence.criteria.CriteriaBuilder;
+import jakarta.persistence.criteria.CriteriaQuery;
+import jakarta.persistence.criteria.Root;
+
 // import jakarta.persistence.criteria.CriteriaDelete;
 
 import teammates.client.connector.DatastoreClient;
@@ -126,6 +131,27 @@ private void doMigration(teammates.storage.entity.Account entity) {
      * Migrates the entity.
      */
     protected void migrateEntity(teammates.storage.entity.Account oldAccount) {
+        HibernateUtil.beginTransaction();
+
+        CriteriaBuilder cb = HibernateUtil.getCriteriaBuilder();
+        CriteriaQuery<teammates.storage.sqlentity.Account> cr = cb.createQuery(
+                teammates.storage.sqlentity.Account.class);
+        Root<teammates.storage.sqlentity.Account> root = cr.from(teammates.storage.sqlentity.Account.class);
+        cr.select(root).where(cb.equal(root.get("googleId"), oldAccount.getGoogleId()));
+
+        TypedQuery<teammates.storage.sqlentity.Account> query = HibernateUtil.createQuery(cr);
+
+
+        boolean isEntityInDb = query.getResultList().size() != 0;
+        HibernateUtil.commitTransaction();
+
+        // In db, but somehow not set as migrated.
+        if (isEntityInDb) {
+            oldAccount.setMigrated(true);
+            entitiesOldAccountSavingBuffer.add(oldAccount);    
+            return;
+        };
+
         teammates.storage.sqlentity.Account newAccount = new teammates.storage.sqlentity.Account(
                 oldAccount.getGoogleId(),
                 oldAccount.getName(),
@@ -136,7 +162,6 @@ protected void migrateEntity(teammates.storage.entity.Account oldAccount) {
         oldAccount.setMigrated(true);
         entitiesOldAccountSavingBuffer.add(oldAccount);
         migrateReadNotification(oldAccount, newAccount);
-
     }
 
     private void migrateReadNotification(teammates.storage.entity.Account oldAccount,