module SlavePools

# File lib/slave_pools.rb, line 16
def config
  @config ||= SlavePools::Config.new
end

# File lib/slave_pools.rb, line 35
def current
  proxy.current
end

# File lib/slave_pools.rb, line 55
def log(level, message)
  logger.send(level, "[SlavePools] #{message}")
end

# File lib/slave_pools.rb, line 59
def logger
  ActiveRecord::Base.logger
end

# File lib/slave_pools.rb, line 39
def next_slave!
  proxy.next_slave!
end

# File lib/slave_pools.rb, line 51
def pools
  Thread.current[:slave_pools] ||= SlavePools::Pools.new
end

# File lib/slave_pools.rb, line 28
def proxy
  Thread.current[:slave_pools_proxy] ||= SlavePools::ConnectionProxy.new(
    ActiveRecord::Base,
    SlavePools.pools
  )
end

# File lib/slave_pools.rb, line 20
def setup!
  ConnectionProxy.generate_safe_delegations

  ActiveRecord::Base.send(:extend, SlavePools::Hijack)

  log :info, "Proxy loaded with: #{pools.keys.join(', ')}"
end

# File lib/slave_pools.rb, line 47
def with_master
  proxy.with_master{ yield }
end

# File lib/slave_pools.rb, line 43
def with_pool(*a)
  proxy.with_pool(*a){ yield }
end