blob: c95b5b64a43cc05c42cb8b1240132c2311a09635 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
# frozen_string_literal: true
# Provides a way to execute nullify behaviour in batches
# to avoid query timeouts for really big tables
# Assumes that associations have `dependent: :nullify` statement
module BatchNullifyDependentAssociations
extend ActiveSupport::Concern
class_methods do
def dependent_associations_to_nullify
reflect_on_all_associations(:has_many).select { |assoc| assoc.options[:dependent] == :nullify }
end
end
def nullify_dependent_associations_in_batches(exclude: [], batch_size: 100)
self.class.dependent_associations_to_nullify.each do |association|
next if association.name.in?(exclude)
loop do
# rubocop:disable GitlabSecurity/PublicSend
update_count = public_send(association.name).limit(batch_size).update_all(association.foreign_key => nil)
# rubocop:enable GitlabSecurity/PublicSend
break if update_count < batch_size
end
end
end
end
|