From a5f4bba440d7f9ea47046a0a561d49adf0a1e6d4 Mon Sep 17 00:00:00 2001 From: GitLab Bot Date: Wed, 16 Jun 2021 18:25:58 +0000 Subject: Add latest changes from gitlab-org/gitlab@14-0-stable-ee --- lib/gitlab/database/dynamic_model_helpers.rb | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) (limited to 'lib/gitlab/database/dynamic_model_helpers.rb') diff --git a/lib/gitlab/database/dynamic_model_helpers.rb b/lib/gitlab/database/dynamic_model_helpers.rb index 892f8291780..7439591be99 100644 --- a/lib/gitlab/database/dynamic_model_helpers.rb +++ b/lib/gitlab/database/dynamic_model_helpers.rb @@ -11,6 +11,25 @@ module Gitlab self.inheritance_column = :_type_disabled end end + + def each_batch(table_name, scope: ->(table) { table.all }, of: 1000) + if transaction_open? + raise <<~MSG.squish + each_batch should not run inside a transaction, you can disable + transactions by calling disable_ddl_transaction! in the body of + your migration class + MSG + end + + scope.call(define_batchable_model(table_name)) + .each_batch(of: of) { |batch| yield batch } + end + + def each_batch_range(table_name, scope: ->(table) { table.all }, of: 1000) + each_batch(table_name, scope: scope, of: of) do |batch| + yield batch.pluck('MIN(id), MAX(id)').first + end + end end end end -- cgit v1.2.1