[AC-1682] Setting FlexibleCollections = 1 only for Orgs that had data migrated in previous steps

util/MySqlMigrations/HelperScripts/2024-03-22_00_EnableOrgsCollectionEnhancements.sql

@@ -131,9 +131,23 @@
     SET `U`.`AccountRevisionDate` = UTC_TIMESTAMP();
 
 -- Step 5: Set `FlexibleCollections` = 1 for all organizations that have not yet been migrated.
-    UPDATE `Organization`
+    UPDATE `Organization` AS `O`
-    SET `FlexibleCollections` = 1
+    JOIN (
-    WHERE `FlexibleCollections` = 0;
+        SELECT DISTINCT `TG`.`OrganizationId`
+        FROM `TempGroupsAccessAll` AS `TG`
+
+        UNION
+
+        SELECT DISTINCT `TU`.`OrganizationId`
+        FROM `TempUsersAccessAll` AS `TU`
+
+        UNION
+
+        SELECT DISTINCT `OU`.`OrganizationId`
+        FROM `TempUserManagers` AS `OU`
+    ) AS `TempOrgIds` ON `O`.`Id` = `TempOrgIds`.`OrganizationId`
+    SET `O`.`FlexibleCollections` = 1
+    WHERE `O`.`FlexibleCollections` = 0;
 
 -- Step 6: Drop the temporary tables
     DROP TEMPORARY TABLE IF EXISTS `TempGroupsAccessAll`;

util/PostgresMigrations/HelperScripts/2024-03-22_00_EnableOrgsCollectionEnhancements.psql

@@ -124,9 +124,24 @@
     );
 
 -- Step 5: Set "FlexibleCollections" = true for all organizations that have not yet been migrated.
-    UPDATE "Organization"
+    UPDATE "Organization" AS "O"
     SET "FlexibleCollections" = true
-    WHERE "FlexibleCollections" = false;
+    FROM (
+        SELECT DISTINCT "TG"."OrganizationId"
+        FROM "TempGroupsAccessAll" AS "TG"
+
+        UNION
+
+        SELECT DISTINCT "TU"."OrganizationId"
+        FROM "TempUsersAccessAll" AS "TU"
+
+        UNION
+
+        SELECT DISTINCT "OU"."OrganizationId"
+        FROM "TempUserManagers" AS "OU"
+    ) AS "TempOrgIds"
+    WHERE "O"."Id" = "TempOrgIds"."OrganizationId"
+    AND "O"."FlexibleCollections" = false;
 
 -- Step 6: Drop the temporary tables
     DROP TABLE IF EXISTS "TempGroupsAccessAll";

util/SqliteMigrations/HelperScripts/2024-03-22_00_EnableOrgsCollectionEnhancements.sql

@@ -144,8 +144,20 @@
 -- Step 5: Set "FlexibleCollections" = 1 for all organizations that have not yet been migrated.
     UPDATE "Organization"
     SET "FlexibleCollections" = 1
-    WHERE "FlexibleCollections" = 0;
+    WHERE "Id" IN (
+        SELECT DISTINCT "TG"."OrganizationId"
+        FROM "TempGroupsAccessAll" AS "TG"
 
+        UNION
+
+        SELECT DISTINCT "TU"."OrganizationId"
+        FROM "TempUsersAccessAll" AS "TU"
+
+        UNION
+
+        SELECT DISTINCT "OU"."OrganizationId"
+        FROM "TempUserManagers" AS "OU"
+    ) AND "FlexibleCollections" = 0;
 
 -- Step 6: Drop the temporary tables
     DROP TABLE IF EXISTS "TempGroupsAccessAll";