Move redis connection logic into class. Add support for namespaces via redis-namespace gem

Gemfile.lock

@@ -0,0 +1,34 @@
+PATH
+  remote: .
+  specs:
+    sidekiq (0.5.1)
+      celluloid
+      connection_pool
+      multi_json
+      redis
+      redis-namespace
+
+GEM
+  remote: http://rubygems.org/
+  specs:
+    celluloid (0.8.0)
+    connection_pool (0.1.0)
+    minitest (2.11.1)
+    multi_json (1.0.4)
+    rake (0.9.2.2)
+    redis (2.2.2)
+    redis-namespace (1.1.0)
+      redis (< 3.0.0)
+    simplecov (0.5.4)
+      multi_json (~> 1.0.3)
+      simplecov-html (~> 0.5.3)
+    simplecov-html (0.5.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  minitest
+  rake
+  sidekiq!
+  simplecov

lib/sidekiq/cli.rb

@@ -1,9 +1,9 @@
 require 'optparse'
 require 'sidekiq/version'
 require 'sidekiq/util'
+require 'sidekiq/redis_connection'
 require 'sidekiq/client'
 require 'sidekiq/manager'
-require 'connection_pool'
 
 module Sidekiq
   class CLI
@@ -18,8 +18,9 @@ module Sidekiq
     FOREVER = 2_000_000_000
 
     def run
-      ::Sidekiq::Client.redis = ConnectionPool.new { Redis.connect(:url => @options[:server]) }
-      manager = Sidekiq::Manager.new(@options[:server], @options)
+      ::Sidekiq::Client.redis = ::Sidekiq::RedisConnection.create(@options[:server], @options[:namespace])
+      manager_redis = ::Sidekiq::RedisConnection.create(@options[:server], @options[:namespace], false)
+      manager = Sidekiq::Manager.new(manager_redis, @options)
       begin
         log 'Starting processing, hit Ctrl-C to stop'
         manager.start!
@@ -77,6 +78,10 @@ module Sidekiq
           @options[:verbose] = true
         end
 
+        o.on "-n", "--namespace NAMESPACE", "namespace worker queues are under" do |arg|
+          @options[:namespace] = arg
+        end
+
         o.on "-s", "--server LOCATION", "Where to find Redis" do |arg|
           @options[:server] = arg
         end

lib/sidekiq/client.rb

@@ -1,6 +1,7 @@
 require 'multi_json'
 require 'redis'
 
+require 'sidekiq/redis_connection'
 require 'sidekiq/middleware/chain'
 require 'sidekiq/middleware/client/unique_jobs'
 
@@ -12,10 +13,7 @@ module Sidekiq
 
     def self.redis
       @redis ||= begin
-        # autoconfig for Heroku
-        hash = {}
-        hash[:url] = ENV['REDISTOGO_URL'] if ENV['REDISTOGO_URL']
-        Redis.connect(hash)
+        RedisConnection.create
       end
     end
 

lib/sidekiq/manager.rb

@@ -18,14 +18,14 @@ module Sidekiq
 
     trap_exit :processor_died
 
-    def initialize(location, options={})
-      log "Booting sidekiq #{Sidekiq::VERSION} with Redis at #{location}"
+    def initialize(redis, options={})
+      log "Booting sidekiq #{Sidekiq::VERSION} with Redis at #{redis.client.location}"
       verbose options.inspect
       @count = options[:processor_count] || 25
       @queues = options[:queues]
       @queue_idx = 0
       @queues_size = @queues.size
-      @redis = Redis.connect(:url => location)
+      @redis = redis
       @done_callback = nil
 
       @done = false

lib/sidekiq/redis_connection.rb

@@ -0,0 +1,37 @@
+require 'connection_pool'
+require 'redis/namespace'
+
+module Sidekiq
+  class RedisConnection
+    def self.create(url = nil, namespace = nil, pool = true)
+      @namespace = namespace if namespace
+      @url = url if url
+      if pool
+        ConnectionPool.new { connect }
+      else
+        connect
+      end
+    end
+
+    def self.connect
+      r = Redis.connect(:url => url)
+      if namespace
+        Redis::Namespace.new(namespace, :redis => r)
+      else
+        r
+      end
+    end
+
+    def self.namespace
+      @namespace ||= nil
+    end
+
+    def self.url
+      ENV['REDISTOGO_URL'] || @url
+    end
+
+    def self.namespace=(namespace)
+      @namespace = namespace
+    end
+  end
+end

sidekiq.gemspec

@@ -14,6 +14,7 @@ Gem::Specification.new do |gem|
   gem.require_paths = ["lib"]
   gem.version       = Sidekiq::VERSION
   gem.add_dependency                  'redis'
+  gem.add_dependency                  'redis-namespace'
   gem.add_dependency                  'connection_pool'
   gem.add_dependency                  'celluloid'
   gem.add_dependency                  'multi_json'

test/test_manager.rb

@@ -6,7 +6,7 @@ require 'connection_pool'
 class TestManager < MiniTest::Unit::TestCase
   describe 'with redis' do
     before do
-      Sidekiq::Client.redis = @redis = Redis.connect(:url => 'redis://localhost/sidekiq_test')
+      Sidekiq::Client.redis = @redis = Sidekiq::RedisConnection.create('redis://localhost/sidekiq_test', nil, false)
       @redis.flushdb
       $processed = 0
     end
@@ -25,7 +25,8 @@ class TestManager < MiniTest::Unit::TestCase
       Sidekiq::Client.push(:foo, 'class' => IntegrationWorker, 'args' => [1, 2])
 
       q = TimedQueue.new
-      mgr = Sidekiq::Manager.new("redis://localhost/sidekiq_test", :queues => [:foo])
+      redis = Sidekiq::RedisConnection.create('redis://localhost/sidekiq_test', nil, false)
+      mgr = Sidekiq::Manager.new(redis, :queues => [:foo])
       mgr.when_done do |_|
         q << 'done' if $processed == 2
       end