1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
require 'connection_pool'
require 'redis'
require 'uri'
module Sidekiq
class RedisConnection
class << self
def create(options={})
url = options[:url] || determine_redis_provider
if url
options[:url] = url
end
# need a connection for Fetcher and Retry
size = options[:size] || (Sidekiq.server? ? (Sidekiq.options[:concurrency] + 2) : 5)
pool_timeout = options[:pool_timeout] || 1
log_info(options)
ConnectionPool.new(:timeout => pool_timeout, :size => size) do
build_client(options)
end
end
private
def build_client(options)
namespace = options[:namespace]
client = Redis.new client_opts(options)
if namespace
require 'redis/namespace'
Redis::Namespace.new(namespace, :redis => client)
else
client
end
end
def client_opts(options)
opts = options.dup
if opts[:namespace]
opts.delete(:namespace)
end
if opts[:network_timeout]
opts[:timeout] = opts[:network_timeout]
opts.delete(:network_timeout)
end
opts[:driver] = opts[:driver] || 'ruby'
opts
end
def log_info(options)
# Don't log Redis AUTH password
redacted = "REDACTED"
scrubbed_options = options.dup
if scrubbed_options[:url] && (uri = URI.parse(scrubbed_options[:url])) && uri.password
uri.password = redacted
scrubbed_options[:url] = uri.to_s
end
if scrubbed_options[:password]
scrubbed_options[:password] = redacted
end
if Sidekiq.server?
Sidekiq.logger.info("Booting Sidekiq #{Sidekiq::VERSION} with redis options #{scrubbed_options}")
else
Sidekiq.logger.debug("#{Sidekiq::NAME} client with redis options #{scrubbed_options}")
end
end
def determine_redis_provider
ENV[ENV['REDIS_PROVIDER'] || 'REDIS_URL']
end
end
end
end
|
