1 files changed, 31 insertions, 0 deletions
diff --git a/lib/gitlab/terraform/state_migration_helper.rb b/lib/gitlab/terraform/state_migration_helper.rb
new file mode 100644
index 00000000000..04c1cbd0373
--- /dev/null
+++ b/lib/gitlab/terraform/state_migration_helper.rb
@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Gitlab
+  module Terraform
+    class StateMigrationHelper
+      class << self
+        def migrate_to_remote_storage(&block)
+          migrate_in_batches(
+            ::Terraform::StateVersion.with_files_stored_locally.preload_state,
+            ::Terraform::StateUploader::Store::REMOTE,
+            &block
+          )
+        end
+
+        private
+
+        def batch_size
+          ENV.fetch('MIGRATION_BATCH_SIZE', 10).to_i
+        end
+
+        def migrate_in_batches(versions, store, &block)
+          versions.find_each(batch_size: batch_size) do |version| # rubocop:disable CodeReuse/ActiveRecord
+            version.file.migrate!(store)
+
+            yield version if block_given?
+          end
+        end
+      end
+    end
+  end
+end