Contents

require 'sequel'
require 'sequel/connection_pool/threaded'
require 'yaml'
require 'pact_broker/logging'
require 'erb'

module DB
  include PactBroker::Logging
  ##
  # Sequel by default does not test connections in its' connection pool before
  # handing them to a client. To enable connection testing you need to load the
  # "connection_validator" extension like below. The connection validator
  # extension is configurable, by default it only checks connections once per
  # hour:
  #
  # http://sequel.rubyforge.org/rdoc-plugins/files/lib/sequel/extensions/connection_validator_rb.html
  #
  # Because most of our applications so far are accessed infrequently, there is
  # very little overhead in checking each connection when it is requested. This
  # takes care of stale connections.
  #
  # A gotcha here is that it is not enough to enable the "connection_validator"
  # extension, we also need to specify that we want to use the threaded connection
  # pool, as noted in the documentation for the extension.
  #
  def self.connect db_credentials
    con = Sequel.connect(db_credentials.merge(:logger => logger, :pool_class => Sequel::ThreadedConnectionPool))
    con.extension(:connection_validator)
    con.pool.connection_validation_timeout = -1 #Check the connection on every request
    con
  end

  PACT_BROKER_DB ||= begin
    config = YAML.load(::ERB.new(File.read(File.join('./config', 'database.yml'))).result)
    rack_env = ENV['RACK_ENV'] || 'development'
    puts "Connecting using config #{rack_env}"
    connect config[rack_env]
  end

  def self.health_check
    PACT_BROKER_DB.synchronize do |c| c
      PACT_BROKER_DB.valid_connection? c
    end
  end
end

Version data entries

10 entries across 10 versions & 1 rubygems

Version	Path
pact_broker-0.0.10	lib/pact_broker/db.rb
pact_broker-0.0.9	lib/pact_broker/db.rb
pact_broker-0.0.8	lib/pact_broker/db.rb
pact_broker-0.0.7	lib/pact_broker/db.rb
pact_broker-0.0.6	lib/pact_broker/db.rb
pact_broker-0.0.5	lib/pact_broker/db.rb
pact_broker-0.0.4	lib/pact_broker/db.rb
pact_broker-0.0.3	lib/pact_broker/db.rb
pact_broker-0.0.2	lib/pact_broker/db.rb
pact_broker-0.0.1	lib/pact_broker/db.rb