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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
# frozen_string_literal: true
class Combustion::Database::Reset
# https://github.com/ruby/psych/pull/358/files#diff-fcdbfb11714f576f58ba9f866052bc79R322
RUBY_VERSION_WITH_NEW_SAFE_LOAD_METHOD_SIGNATURE = "2.6.0"
UnsupportedDatabase = Class.new StandardError
OPERATOR_PATTERNS = {
Combustion::Databases::MySQL => [/mysql/, /trilogy/],
Combustion::Databases::PostgreSQL => [/postgres/, /postgis/],
Combustion::Databases::SQLite => [/sqlite/],
Combustion::Databases::SQLServer => [/sqlserver/],
Combustion::Databases::Oracle => %w[ oci oracle ],
Combustion::Databases::Firebird => %w[ firebird ]
}.freeze
RAILS_DEFAULT_ENVIRONMENTS = %w[ development production test ].freeze
def self.call
new.call
end
def initialize
# TODO: remove when no longer support 2.5.8
if RUBY_VERSION >= RUBY_VERSION_WITH_NEW_SAFE_LOAD_METHOD_SIGNATURE
ActiveRecord::Base.configurations = YAML.safe_load(
ERB.new(database_yaml).result, :aliases => true
)
else
ActiveRecord::Base.configurations = YAML.safe_load(
ERB.new(database_yaml).result, [], [], true
)
end
end
def call
resettable_db_configs.each do |configuration|
adapter = configuration[:adapter] ||
configuration[:url].split("://").first
operator_class(adapter).new(configuration).reset
end
end
private
def database_yaml
File.read "#{Rails.root}/config/database.yml"
end
def operator_class(adapter)
klass = nil
OPERATOR_PATTERNS.each do |operator, keys|
klass = operator if keys.any? { |key| adapter[key] }
end
return klass if klass
raise UnsupportedDatabase, "Unsupported database type: #{adapter}"
end
# All database configs except Rails default environments
# that are not currently in use
def resettable_db_configs
if Combustion::VersionGate.call("activerecord", ">= 6.1")
return resettable_db_configs_for_6_1
end
all_configurations = ActiveRecord::Base.configurations.to_h
unused_environments = RAILS_DEFAULT_ENVIRONMENTS - [Rails.env.to_s]
resettable_environments = all_configurations.keys - unused_environments
all_configurations.
select { |name| resettable_environments.include?(name) }.
values.
collect(&:with_indifferent_access)
end
def resettable_db_configs_for_6_1
all_configurations = ActiveRecord::Base.configurations.configurations
unused_environments = RAILS_DEFAULT_ENVIRONMENTS - [Rails.env.to_s]
resettable_environments = all_configurations.collect(&:env_name).uniq -
unused_environments
all_configurations.
select { |config| resettable_environments.include?(config.env_name) }.
collect(&:configuration_hash)
end
end
|
