summaryrefslogtreecommitdiff
path: root/lib/gitlab/database/postgresql_adapter/schema_versions_copy_mixin.rb
blob: d8f96643dcb62aba8527d4d22815feda2ff0b29e (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
# frozen_string_literal: true

module Gitlab
  module Database
    module PostgresqlAdapter
      module SchemaVersionsCopyMixin
        extend ActiveSupport::Concern

        def dump_schema_information # :nodoc:
          versions = schema_migration.all_versions
          copy_versions_sql(versions) if versions.any?
        end

        private

        def copy_versions_sql(versions)
          sm_table = quote_table_name(schema_migration.table_name)

          sql = +"COPY #{sm_table} (version) FROM STDIN;\n"
          sql << versions.map { |v| Integer(v) }.sort.join("\n")
          sql << "\n\\.\n"

          sql
        end
      end
    end
  end
end