Module: Legion::Data::Connection

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

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.



11
12
13
 
# File 'lib/legion/data/connection.rb', line 11

def sequel
  @sequel
end

Class Method Details

.adapter ⇒ Object 



13
14
15
 
# File 'lib/legion/data/connection.rb', line 13

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

.connect_with_replicas ⇒ Object 



58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/legion/data/connection.rb', line 58

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 



113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
 
# File 'lib/legion/data/connection.rb', line 113

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 



90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
 
# File 'lib/legion/data/connection.rb', line 90

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 



80
81
82
83
84
 
# File 'lib/legion/data/connection.rb', line 80

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

  :read_0
end

.replica_servers ⇒ Object 



86
87
88
 
# File 'lib/legion/data/connection.rb', line 86

def replica_servers
  @replica_servers || []
end

.setup ⇒ Object 



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

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 



52
53
54
55
56
 
# File 'lib/legion/data/connection.rb', line 52

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