app/services/snippets/schedule_bulk_repository_shard_moves_service.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

# frozen_string_literal: true

module Snippets
  # Tries to schedule a move for every snippet with repositories on the source shard
  class ScheduleBulkRepositoryShardMovesService
    include ScheduleBulkRepositoryShardMovesMethods
    extend ::Gitlab::Utils::Override

    private

    override :repository_klass
    def repository_klass
      SnippetRepository
    end

    override :container_klass
    def container_klass
      Snippet
    end

    override :container_column
    def container_column
      :snippet_id
    end

    override :schedule_bulk_worker_klass
    def self.schedule_bulk_worker_klass
      ::Snippets::ScheduleBulkRepositoryShardMovesWorker
    end
  end
end