Module: Legion::Data::Connection

Defined in:
lib/legion/data/connection.rb

Defined Under Namespace

Classes: SlowQueryLogger

Constant Summary collapse

ADAPTERS = 
%i[sqlite mysql2 postgres].freeze

Class Attribute Summary collapse

Class Method Summary collapse

Class Attribute Details

.sequel ⇒ Object

Returns the value of attribute sequel.



36
37
38
 
# File 'lib/legion/data/connection.rb', line 36

def sequel
  @sequel
end

Class Method Details

.adapter ⇒ Object 



38
39
40
 
# File 'lib/legion/data/connection.rb', line 38

def adapter
  @adapter ||= Legion::Settings[:data][:adapter]&.to_sym || :sqlite
end

.connect_with_replicas ⇒ Object 



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
 
# File 'lib/legion/data/connection.rb', line 83

def connect_with_replicas
  return unless adapter == :postgres

  replica_url  = Legion::Settings[:data][:read_replica_url]
  replica_list = Array(Legion::Settings[:data][:replicas]).dup

  replica_list.prepend(replica_url) if replica_url && !replica_url.empty?
  replica_list.uniq!
  replica_list.compact!

  return if replica_list.empty?

  @sequel.extension(:server_block)

  replica_list.each_with_index do |url, idx|
    @sequel.add_servers("read_#{idx}": url)
  end

  @replica_servers = replica_list.each_with_index.map { |_, idx| :"read_#{idx}" }
  Legion::Logging.debug "Registered #{@replica_servers.size} read replica(s)" if defined?(Legion::Logging)
end

.creds_builder(final_creds = {}) ⇒ Object 



138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
 
# File 'lib/legion/data/connection.rb', line 138

def creds_builder(final_creds = {})
  final_creds.merge! Legion::Data::Settings.creds(adapter)
  final_creds.merge! Legion::Settings[:data][:creds] if Legion::Settings[:data][:creds].is_a? Hash

  port = final_creds[:port]
  merge_tls_creds(final_creds, adapter: adapter, port: port)

  return final_creds if Legion::Settings[:vault].nil?

  if Legion::Settings[:vault][:connected] && ::Vault.sys.mounts.key?(:database)
    temp_vault_creds = Legion::Crypt.read('database/creds/legion')
    final_creds[:user] = temp_vault_creds[:username]
    final_creds[:password] = temp_vault_creds[:password]
  end

  final_creds
end

.merge_tls_creds(creds, adapter:, port:) ⇒ Object 



115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
 
# File 'lib/legion/data/connection.rb', line 115

def merge_tls_creds(creds, adapter:, port:)
  return creds if adapter == :sqlite
  return creds unless defined?(Legion::Crypt::TLS)

  tls = Legion::Crypt::TLS.resolve(data_tls_settings, port: port)
  return creds unless tls[:enabled]

  case adapter
  when :postgres
    creds[:sslmode]     = tls[:verify] == :none ? 'require' : 'verify-full'
    creds[:sslrootcert] = tls[:ca] if tls[:ca]
    creds[:sslcert]     = tls[:cert] if tls[:cert]
    creds[:sslkey]      = tls[:key] if tls[:key]
  when :mysql2
    creds[:ssl_mode] = tls[:verify] == :none ? 'required' : 'verify_identity'
    creds[:sslca]    = tls[:ca] if tls[:ca]
    creds[:sslcert]  = tls[:cert] if tls[:cert]
    creds[:sslkey]   = tls[:key] if tls[:key]
  end

  creds
end

.read_server ⇒ Object 



105
106
107
108
109
 
# File 'lib/legion/data/connection.rb', line 105

def read_server
  return :default if @replica_servers.nil? || @replica_servers.empty?

  :read_0
end

.replica_servers ⇒ Object 



111
112
113
 
# File 'lib/legion/data/connection.rb', line 111

def replica_servers
  @replica_servers || []
end

.setup ⇒ Object 



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
 
# File 'lib/legion/data/connection.rb', line 42

def setup
  @sequel = if adapter == :sqlite
              ::Sequel.sqlite(sqlite_path)
            else
              begin
                ::Sequel.connect(adapter: adapter, **creds_builder)
              rescue StandardError => e
                raise unless dev_fallback?

                if defined?(Legion::Logging)
                  Legion::Logging.warn(
                    "Shared DB unreachable (#{e.message}), dev_mode fallback to SQLite"
                  )
                end
                @adapter = :sqlite
                ::Sequel.sqlite(sqlite_path)
              end
            end
  Legion::Settings[:data][:connected] = true
  if defined?(Legion::Logging)
    if adapter == :sqlite
      Legion::Logging.info "Connected to SQLite at #{sqlite_path}"
    else
      creds = Legion::Data::Settings.creds(adapter)
      user = creds[:user] || creds[:username] || 'unknown'
      host = creds[:host] || '127.0.0.1'
      port = creds[:port]
      db   = creds[:database] || creds[:db]
      Legion::Logging.info "Connected to #{adapter}://#{user}@#{host}:#{port}/#{db}"
    end
  end
  configure_logging
  connect_with_replicas
end

.shutdown ⇒ Object 



77
78
79
80
81
 
# File 'lib/legion/data/connection.rb', line 77

def shutdown
  @sequel&.disconnect
  Legion::Settings[:data][:connected] = false
  Legion::Logging.info 'Legion::Data connection closed' if defined?(Legion::Logging)
end