lib/gitlab/database/custom_structure.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
43
44

# frozen_string_literal: true

module Gitlab
  module Database
    class CustomStructure
      CUSTOM_DUMP_FILE = 'db/gitlab_structure.sql'

      def dump
        File.open(self.class.custom_dump_filepath, 'wb') do |io|
          io << "-- this file tracks custom GitLab data, such as foreign keys referencing partitioned tables\n"
          io << "-- more details can be found in the issue: https://gitlab.com/gitlab-org/gitlab/-/issues/201872\n"
          io << "SET search_path=public;\n\n"

          dump_partitioned_foreign_keys(io) if partitioned_foreign_keys_exist?
        end
      end

      def self.custom_dump_filepath
        Rails.root.join(CUSTOM_DUMP_FILE)
      end

      private

      def dump_partitioned_foreign_keys(io)
        io << "COPY partitioned_foreign_keys (#{partitioned_fk_columns.join(", ")}) FROM STDIN;\n"

        PartitioningMigrationHelpers::PartitionedForeignKey.find_each do |fk|
          io << fk.attributes.values_at(*partitioned_fk_columns).join("\t") << "\n"
        end
        io << "\\.\n"
      end

      def partitioned_foreign_keys_exist?
        return false unless PartitioningMigrationHelpers::PartitionedForeignKey.table_exists?

        PartitioningMigrationHelpers::PartitionedForeignKey.exists?
      end

      def partitioned_fk_columns
        @partitioned_fk_columns ||= PartitioningMigrationHelpers::PartitionedForeignKey.column_names
      end
    end
  end
end