~cytrogen/masto-fe

0ad2413b35287958f59073a5b63aecc659a64d98 — Eugen Rochko 2 years ago f1c1fa1
Fix n+1 query in settings migration (#24783)

1 files changed, 16 insertions(+), 13 deletions(-)

M db/migrate/20230215074423_move_user_settings.rb
M db/migrate/20230215074423_move_user_settings.rb => db/migrate/20230215074423_move_user_settings.rb +16 -13
@@ 59,26 59,29 @@ class MoveUserSettings < ActiveRecord::Migration[6.1]
  end

  def up
    User.find_each do |user|
      previous_settings = LegacySetting.where(thing_type: 'User', thing_id: user.id).index_by(&:var)
    User.find_in_batches do |users|
      previous_settings_for_batch = LegacySetting.where(thing_type: 'User', thing_id: users.map(&:id)).group_by(&:thing_id)

      user_settings = {}
      users.each do |user|
        previous_settings = previous_settings_for_batch[user.id]&.index_by(&:var) || {}
        user_settings     = {}

      MAPPING.each do |legacy_key, new_key|
        value = previous_settings[legacy_key]&.value
        MAPPING.each do |legacy_key, new_key|
          value = previous_settings[legacy_key]&.value

        next if value.blank?
          next if value.blank?

        if value.is_a?(Hash)
          value.each do |nested_key, nested_value|
            user_settings[MAPPING[legacy_key][nested_key.to_sym]] = nested_value
          if value.is_a?(Hash)
            value.each do |nested_key, nested_value|
              user_settings[MAPPING[legacy_key][nested_key.to_sym]] = nested_value
            end
          else
            user_settings[new_key] = value
          end
        else
          user_settings[new_key] = value
        end
      end

      user.update_column('settings', Oj.dump(user_settings)) # rubocop:disable Rails/SkipsModelValidations
        user.update_column('settings', Oj.dump(user_settings)) # rubocop:disable Rails/SkipsModelValidations
      end
    end
  end