File: consistency_fix_service.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (48 lines) | stat: -rw-r--r-- 1,334 bytes parent folder | download
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
43
44
45
46
47
48
# frozen_string_literal: true

module Database
  class ConsistencyFixService
    def initialize(source_model:, target_model:, sync_event_class:, source_sort_key:, target_sort_key:)
      @source_model = source_model
      @target_model = target_model
      @sync_event_class = sync_event_class
      @source_sort_key = source_sort_key
      @target_sort_key = target_sort_key
    end

    attr_accessor :source_model, :target_model, :sync_event_class, :source_sort_key, :target_sort_key

    def execute(ids:)
      ids.each do |id|
        if source_object(id) && target_object(id)
          create_sync_event_for(id)
        elsif target_object(id)
          target_object(id).destroy!
        end
      end
      sync_event_class.enqueue_worker
    end

    private

    # rubocop: disable CodeReuse/ActiveRecord
    def source_object(id)
      source_model.find_by(source_sort_key => id)
    end

    def target_object(id)
      target_model.find_by(target_sort_key => id)
    end
    # rubocop: enable CodeReuse/ActiveRecord

    def create_sync_event_for(id)
      if source_model == Namespace
        sync_event_class.create!(namespace_id: id)
      elsif source_model == Project
        sync_event_class.create!(project_id: id)
      else
        raise("Unknown Source Model #{source_model.name}")
      end
    end
  end
end