1 files changed, 58 insertions, 0 deletions
diff --git a/app/models/project_repository_storage_move.rb b/app/models/project_repository_storage_move.rb
new file mode 100644
index 00000000000..e88cc5cfca6
--- /dev/null
+++ b/app/models/project_repository_storage_move.rb
@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+# ProjectRepositoryStorageMove are details of repository storage moves for a
+# project. For example, moving a project to another gitaly node to help
+# balance storage capacity.
+class ProjectRepositoryStorageMove < ApplicationRecord
+  include AfterCommitQueue
+
+  belongs_to :project, inverse_of: :repository_storage_moves
+
+  validates :project, presence: true
+  validates :state, presence: true
+  validates :source_storage_name,
+    on: :create,
+    presence: true,
+    inclusion: { in: ->(_) { Gitlab.config.repositories.storages.keys } }
+  validates :destination_storage_name,
+    on: :create,
+    presence: true,
+    inclusion: { in: ->(_) { Gitlab.config.repositories.storages.keys } }
+
+  state_machine initial: :initial do
+    event :schedule do
+      transition initial: :scheduled
+    end
+
+    event :start do
+      transition scheduled: :started
+    end
+
+    event :finish do
+      transition started: :finished
+    end
+
+    event :do_fail do
+      transition [:initial, :scheduled, :started] => :failed
+    end
+
+    after_transition initial: :scheduled do |storage_move, _|
+      storage_move.run_after_commit do
+        ProjectUpdateRepositoryStorageWorker.perform_async(
+          storage_move.project_id,
+          storage_move.destination_storage_name,
+          storage_move.id
+        )
+      end
+    end
+
+    state :initial, value: 1
+    state :scheduled, value: 2
+    state :started, value: 3
+    state :finished, value: 4
+    state :failed, value: 5
+  end
+
+  scope :order_created_at_desc, -> { order(created_at: :desc) }
+  scope :with_projects, -> { includes(project: :route) }
+end