Sha256: fac6bb64eeabac9bfb7a30ffd9cbbdf85256f1fee704fe180d464f7b254d80c9
Contents?: true
Size: 1.26 KB
Versions: 8
Compression:
Stored size: 1.26 KB
Contents
require 'sequel'
require 'erb'
Inferno::Application.register_provider(:db) do
prepare do
target_container.start :logging
require 'yaml'
Sequel::Model.plugin :json_serializer
config_path = File.expand_path('database.yml', File.join(Dir.pwd, 'config'))
config_contents = ERB.new(File.read(config_path)).result
config = YAML.safe_load(config_contents)[ENV.fetch('APP_ENV', nil)]
.merge(logger: Inferno::Application['logger'])
connection_attempts_remaining = ENV.fetch('MAX_DB_CONNECTION_ATTEMPTS', '10').to_i
connection_retry_delay = ENV.fetch('DB_CONNECTION_RETRY_DELAY', '5').to_i
connection = nil
loop do
connection = Sequel.connect(config)
break
rescue StandardError => e
connection_attempts_remaining -= 1
if connection_attempts_remaining.positive?
Inferno::Application['logger'].error("Unable to connect to database: #{e.message}")
Inferno::Application['logger'].error("#{connection_attempts_remaining} connection attempts remaining.")
sleep connection_retry_delay
next
end
raise
end
connection.sql_log_level = :debug
register('db.config', config)
register('db.connection', connection)
end
start do
Sequel.extension :migration
end
end
Version data entries
8 entries across 8 versions & 1 rubygems