Update migrations for backfilling (#1188)

db/migrate/20241122053814_add_locatable_to_locations.rb

@@ -1,9 +1,7 @@
 class AddLocatableToLocations < ActiveRecord::Migration[7.2]
-  def change
-    safety_assured do
-      remove_reference :locations, :organization, foreign_key: true
+  disable_ddl_transaction!
 
-      add_reference :locations, :locatable, polymorphic: true, null: false, index: true
-    end
+  def change
+    add_reference :locations, :locatable, polymorphic: true, null: true, index: {algorithm: :concurrently}
   end
 end

db/migrate/20241127052013_backfill_locatable_for_locations.rb

@@ -0,0 +1,22 @@
+class BackfillLocatableForLocations < ActiveRecord::Migration[7.2]
+  def up
+    # All locatables were previously organizations
+    Location.find_each do |location|
+      location.update!(
+        locatable_type: "Organization",
+        locatable_id: location.organization_id
+      )
+    end
+
+    safety_assured do
+      change_column_null :locations, :locatable_id, false
+      change_column_null :locations, :locatable_type, false
+    end
+  end
+
+  def down
+    Location.find_each do |location|
+      location.update!(organization_id: location.locatable_id)
+    end
+  end
+end

db/migrate/20241127052542_remove_organization_from_locations.rb

@@ -0,0 +1,7 @@
+class RemoveOrganizationFromLocations < ActiveRecord::Migration[7.2]
+  def change
+    safety_assured do
+      remove_reference :locations, :organization, foreign_key: true
+    end
+  end
+end