Sha256: 604e732d6cb7db83e8e651b69275b1177cc4c199ce66aa2d8a785ce7ecabb9ae
Contents?: true
Size: 1.64 KB
Versions: 23
Compression:
Stored size: 1.64 KB
Contents
require 'redis-namespace'
require 'redis/distributed'
require 'uri'
require 'singleton'
module Routemaster
class RedisBroker
include Singleton
DEFAULT_NAMESPACE = 'rm'.freeze
def initialize
@_connections = {}
_cleanup
end
def get(name, urls: [])
_check_for_fork
@_connections[name] ||= begin
parsed_url = URI.parse(urls.first)
namespace = parsed_url.path.split('/')[2] || DEFAULT_NAMESPACE
Redis::Namespace.new(namespace, redis: Redis::Distributed.new(urls))
end
end
def cleanup
_cleanup
end
# Allow to inject pre-built Redis clients
#
# Before storing a new connection, ensures that any previously
# set client is properly closed.
#
def inject(clients={})
@_injected_clients = true
clients.each_pair do |name, client|
_close_if_present(@_connections[name])
@_connections[name] = Redis::Namespace.new(DEFAULT_NAMESPACE, redis: client)
end
end
private
# Do not clean up if the clients are injected by the host application.
# In that case connections should be managed the server or worker processes.
#
def _check_for_fork
return if @_injected_clients
return if Process.pid == @_pid
_cleanup
end
def _cleanup
@_pid = Process.pid
@_connections.each_value { |conn| _close_if_present(conn) }
@_connections = {}
end
def _close_if_present(connection)
if connection.respond_to?(:redis)
connection.redis.quit
end
end
end
end
Version data entries
23 entries across 23 versions & 1 rubygems