Module: Gitlab::Patch::DatabaseConfig

Extended by:
ActiveSupport::Concern
Defined in:
lib/gitlab/patch/database_config.rb

Instance Method Summary collapse

Instance Method Details

#database_configurationObject


68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
# File 'lib/gitlab/patch/database_config.rb', line 68

def database_configuration
  super.to_h do |env, configs|
    if Gitlab.ee?
      if !configs.key?("geo") && File.exist?(Rails.root.join("config/database_geo.yml"))
        configs["geo"] = Rails.application.config_for(:database_geo).stringify_keys
      end

      if configs.key?("geo")
        migrations_paths = Array(configs["geo"]["migrations_paths"])
        migrations_paths << "ee/db/geo/migrate" if migrations_paths.empty?
        migrations_paths << "ee/db/geo/post_migrate" unless ENV['SKIP_POST_DEPLOYMENT_MIGRATIONS']

        configs["geo"]["migrations_paths"] = migrations_paths.uniq
        configs["geo"]["schema_migrations_path"] = "ee/db/geo/schema_migrations" if configs["geo"]["schema_migrations_path"].blank?
      end
    end

    [env, configs]
  end
end

#load_database_yamlObject


34
35
36
37
38
# File 'lib/gitlab/patch/database_config.rb', line 34

def load_database_yaml
  return super unless Gitlab.ee?

  super.deep_merge(load_geo_database_yaml)
end

#load_geo_database_yamlObject

This method is taken from Rails to load a database YAML file without evaluating ERB. This allows us to create the rake tasks for the Geo tracking database without filling in the configuration values or loading the environment. To be removed when we start configure Geo tracking database in database.yml instead of custom database_geo.yml

github.com/rails/rails/blob/v6.1.4/railties/lib/rails/application/configuration.rb#L255


47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# File 'lib/gitlab/patch/database_config.rb', line 47

def load_geo_database_yaml
  path = Rails.root.join("config/database_geo.yml")
  return {} unless File.exist?(path)

  require "rails/application/dummy_erb_compiler"

  yaml = DummyERB.new(Pathname.new(path).read).result
  config = YAML.load(yaml) || {} # rubocop:disable Security/YAMLLoad

  config.to_h do |env, configs|
    # This check is taken from Rails where the transformation
    # of a flat database.yml is done into `primary:`
    # https://github.com/rails/rails/blob/v6.1.4/activerecord/lib/active_record/database_configurations.rb#L169
    if configs.is_a?(Hash) && !configs.all? { |_, v| v.is_a?(Hash) }
      configs = { "geo" => configs }
    end

    [env, configs]
  end
end