app/workers/concerns/update_repository_storage_worker.rb


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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

# frozen_string_literal: true

module UpdateRepositoryStorageWorker
  extend ActiveSupport::Concern
  include ApplicationWorker

  included do
    idempotent!
    feature_category :gitaly
    urgency :throttled
  end

  def perform(container_id, new_repository_storage_key, repository_storage_move_id = nil)
    repository_storage_move =
      if repository_storage_move_id
        find_repository_storage_move(repository_storage_move_id)
      else
        # maintain compatibility with workers queued before release
        container = find_container(container_id)
        container.repository_storage_moves.create!(
          source_storage_name: container.repository_storage,
          destination_storage_name: new_repository_storage_key
        )
      end

    update_repository_storage(repository_storage_move)
  end

  private

  def find_repository_storage_move(repository_storage_move_id)
    raise NotImplementedError
  end

  def find_container(container_id)
    raise NotImplementedError
  end

  def update_repository_storage(repository_storage_move)
    raise NotImplementedError
  end
end