mirror of
https://github.com/mperham/sidekiq.git
synced 2022-11-09 13:52:34 -05:00
76 lines
2.5 KiB
Ruby
76 lines
2.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "connection_pool"
|
|
require "uri"
|
|
require "sidekiq/redis_client_adapter"
|
|
|
|
module Sidekiq
|
|
module RedisConnection
|
|
class << self
|
|
def create(options = {})
|
|
symbolized_options = options.transform_keys(&:to_sym)
|
|
symbolized_options[:url] ||= determine_redis_provider
|
|
|
|
if symbolized_options[:logger]
|
|
log_info(symbolized_options)
|
|
symbolized_options.delete(:logger)
|
|
end
|
|
|
|
size = symbolized_options.delete(:size) || 5
|
|
pool_timeout = symbolized_options.delete(:pool_timeout) || 1
|
|
|
|
redis_config = Sidekiq::RedisClientAdapter.new(symbolized_options)
|
|
ConnectionPool.new(timeout: pool_timeout, size: size) do
|
|
redis_config.new_client
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def log_info(options)
|
|
redacted = "REDACTED"
|
|
|
|
# Deep clone so we can muck with these options all we want and exclude
|
|
# params from dump-and-load that may contain objects that Marshal is
|
|
# unable to safely dump.
|
|
keys = options.keys - [:logger, :ssl_params]
|
|
scrubbed_options = Marshal.load(Marshal.dump(options.slice(*keys)))
|
|
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
|
|
scrubbed_options[:sentinels]&.each do |sentinel|
|
|
sentinel[:password] = redacted if sentinel[:password]
|
|
end
|
|
options[:logger].info("Sidekiq #{Sidekiq::VERSION} connecting to Redis with options #{scrubbed_options}")
|
|
end
|
|
|
|
def determine_redis_provider
|
|
# If you have this in your environment:
|
|
# MY_REDIS_URL=redis://hostname.example.com:1238/4
|
|
# then set:
|
|
# REDIS_PROVIDER=MY_REDIS_URL
|
|
# and Sidekiq will find your custom URL variable with no custom
|
|
# initialization code at all.
|
|
#
|
|
p = ENV["REDIS_PROVIDER"]
|
|
if p && p =~ /:/
|
|
raise <<~EOM
|
|
REDIS_PROVIDER should be set to the name of the variable which contains the Redis URL, not a URL itself.
|
|
Platforms like Heroku will sell addons that publish a *_URL variable. You need to tell Sidekiq with REDIS_PROVIDER, e.g.:
|
|
|
|
REDISTOGO_URL=redis://somehost.example.com:6379/4
|
|
REDIS_PROVIDER=REDISTOGO_URL
|
|
EOM
|
|
end
|
|
|
|
ENV[
|
|
p || "REDIS_URL"
|
|
]
|
|
end
|
|
end
|
|
end
|
|
end
|