shard-crystal-db/src/db/database.cr

101 lines
2.5 KiB
Crystal
Raw Normal View History

require "http/params"
require "weak_ref"
module DB
# Acts as an entry point for database access.
# Connections are managed by a pool.
# The connection pool can be configured from URI parameters:
#
# - initial_pool_size (default 1)
2016-10-22 00:49:16 +00:00
# - max_pool_size (default 0 = unlimited)
# - max_idle_pool_size (default 1)
# - checkout_timeout (default 5.0)
# - retry_attempts (default 1)
# - retry_delay (in seconds, default 1.0)
#
# It should be created from DB module. See `DB#open`.
#
# Refer to `QueryMethods` for documentation about querying the database.
class Database
# :nodoc:
getter driver
# :nodoc:
getter pool
# Returns the uri with the connection settings to the database
2016-02-03 21:29:09 +00:00
getter uri
getter? prepared_statements : Bool
@pool : Pool(Connection)
2016-08-18 03:55:43 +00:00
@setup_connection : Connection -> Nil
@statements_cache = StringKeyCache(PoolStatement).new
2016-06-16 15:14:57 +00:00
# :nodoc:
2016-06-16 15:14:57 +00:00
def initialize(@driver : Driver, @uri : URI)
2016-08-30 18:22:27 +00:00
params = HTTP::Params.parse(uri.query || "")
@prepared_statements = DB.fetch_bool(params, "prepared_statements", true)
pool_options = @driver.connection_pool_options(params)
2016-08-18 03:55:43 +00:00
@setup_connection = ->(conn : Connection) {}
@pool = uninitialized Pool(Connection) # in order to use self in the factory proc
2016-08-18 03:55:43 +00:00
@pool = Pool.new(**pool_options) {
conn = @driver.build_connection(self).as(Connection)
@setup_connection.call conn
conn
}
end
def setup_connection(&proc : Connection -> Nil)
@setup_connection = proc
@pool.each_resource do |conn|
@setup_connection.call conn
end
end
# Closes all connection to the database.
def close
@statements_cache.each_value &.close
@statements_cache.clear
@pool.close
end
# :nodoc:
def build(query)
@statements_cache.fetch(query) { PoolStatement.new(self, query) }
end
2016-02-03 19:57:54 +00:00
# :nodoc:
def checkout_some(candidates : Enumerable(WeakRef(Connection))) : {Connection, Bool}
@pool.checkout_some candidates
end
# :nodoc:
def return_to_pool(connection)
@pool.release connection
end
# yields a connection from the pool
# the connection is returned to the pool after
# when the block ends
def using_connection
connection = @pool.checkout
2016-07-11 15:56:40 +00:00
begin
yield connection
ensure
return_to_pool connection
end
end
# :nodoc:
def retry
@pool.retry do
yield
end
end
include QueryMethods
end
end