summaryrefslogtreecommitdiff
path: root/lib/gitlab/database/schema_cleaner.rb
blob: 8f93da2b66ca3fb4cfd7a4ecabd9b1aa6cc3ac78 (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
# frozen_string_literal: true

module Gitlab
  module Database
    class SchemaCleaner
      attr_reader :original_schema

      def initialize(original_schema)
        @original_schema = original_schema
      end

      def clean(io)
        structure = original_schema.dup

        # Remove noise
        structure.gsub!(/^COMMENT ON EXTENSION.*/, '')
        structure.gsub!(/^SET.+/, '')
        structure.gsub!(/^SELECT pg_catalog\.set_config\('search_path'.+/, '')
        structure.gsub!(/^--.*/, "\n")

        # We typically don't assume we're working with the public schema.
        # pg_dump uses fully qualified object names though, since we have multiple schemas
        # in the database.
        #
        # The intention here is to not introduce an assumption about the standard schema,
        # unless we have a good reason to do so.
        structure.gsub!(/public\.(\w+)/, '\1')
        structure.gsub!(/CREATE EXTENSION IF NOT EXISTS (\w+) WITH SCHEMA public;/, 'CREATE EXTENSION IF NOT EXISTS \1;')

        structure.gsub!(/\n{3,}/, "\n\n")

        io << structure.strip
        io << <<~MSG
          -- schema_migrations.version information is no longer stored in this file,
          -- but instead tracked in the db/schema_migrations directory
          -- see https://gitlab.com/gitlab-org/gitlab/-/issues/218590 for details
        MSG

        nil
      end
    end
  end
end