12345678910111213141516171819202122232425262728293031323334 |
- class StatsStorePolymorphicAssociation < ActiveRecord::Migration[5.2]
- def change
- return if !Setting.exists?(name: 'system_init_done')
- # create ObjectLookup ID -> Model map
- object_lookup_map = ObjectLookup.all.pluck(:id, :name)
- # create empty, indexed polymorphic association columns
- add_reference :stats_stores, :stats_storable, polymorphic: true, index: true
- # set last run 20 min in future to avoid scheduler errors until restart
- Scheduler.find_by(method: 'Stats.generate').update(last_run: 20.minutes.from_now)
- # migrate column data in the most performance way
- object_lookup_map.each do |id, model|
- StatsStore.where(stats_store_object_id: id)
- .update_all("stats_storable_id = o_id, stats_storable_type = '#{model}'") # rubocop:disable Rails/SkipsModelValidations
- end
- remove_unneeded_columns
- end
- private
- def remove_unneeded_columns
- # remove home made "polymorphic association" columns
- remove_column :stats_stores, :o_id
- remove_column :stats_stores, :stats_store_object_id
- # remove unused/obsolete columns
- remove_column :stats_stores, :related_stats_store_object_id
- remove_column(:stats_stores, :related_o_id) if StatsStore.column_names.include?('related_o_id')
- end
- end
|