mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
36 lines
1 KiB
Ruby
36 lines
1 KiB
Ruby
require 'connection_pool'
|
|
require 'redis'
|
|
require 'redis/namespace'
|
|
|
|
module Sidekiq
|
|
class RedisConnection
|
|
def self.create(options={})
|
|
url = options[:url] || determine_redis_provider || 'redis://localhost:6379/0'
|
|
driver = options[:driver] || 'ruby'
|
|
# need a connection for Fetcher and Retry
|
|
size = options[:size] || (Sidekiq.server? ? (Sidekiq.options[:concurrency] + 2) : 5)
|
|
namespace = options[:namespace] || Sidekiq.options[:namespace]
|
|
|
|
ConnectionPool.new(:timeout => 1, :size => size) do
|
|
build_client(url, namespace, driver)
|
|
end
|
|
end
|
|
|
|
def self.build_client(url, namespace, driver)
|
|
client = Redis.connect(:url => url, :driver => driver)
|
|
if namespace
|
|
Redis::Namespace.new(namespace, :redis => client)
|
|
else
|
|
client
|
|
end
|
|
end
|
|
private_class_method :build_client
|
|
|
|
# Not public
|
|
def self.determine_redis_provider
|
|
return ENV['REDISTOGO_URL'] if ENV['REDISTOGO_URL']
|
|
provider = ENV['REDIS_PROVIDER'] || 'REDIS_URL'
|
|
ENV[provider]
|
|
end
|
|
end
|
|
end
|