summaryrefslogtreecommitdiff
path: root/lib/gitlab/background_migration/link_lfs_objects.rb
blob: 014bebc4258a1423143ebe0489b4366f5b708326 (plain)
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 Gitlab
  module BackgroundMigration
    # Create missing LfsObjectsProject records for forks
    class LinkLfsObjects
      # Model definition used for migration
      class ForkNetworkMember < ActiveRecord::Base
        self.table_name = 'fork_network_members'

        def self.with_non_existing_lfs_objects
          joins('JOIN lfs_objects_projects lop ON fork_network_members.forked_from_project_id = lop.project_id')
            .where(
              <<~SQL
                NOT EXISTS (
                  SELECT 1
                  FROM lfs_objects_projects
                  WHERE lfs_objects_projects.project_id = fork_network_members.project_id
                  AND lfs_objects_projects.lfs_object_id = lop.lfs_object_id
                )
              SQL
            )
        end
      end

      def perform(start_id, end_id)
        select_query =
          ForkNetworkMember
            .select('lop.lfs_object_id, fork_network_members.project_id')
            .with_non_existing_lfs_objects
            .where(project_id: start_id..end_id)

        return if select_query.empty?

        execute <<-SQL
          INSERT INTO lfs_objects_projects (lfs_object_id, project_id)
          #{select_query.to_sql}
        SQL
      end

      private

      def execute(sql)
        ::ActiveRecord::Base.connection.execute(sql)
      end
    end
  end
end