Merge pull request #396 from redis/dup2

Add ability to duplicate connections.

Author: pietern

lib/redis.rb

@@ -27,6 +27,7 @@ def self.current=(redis)
   include MonitorMixin
 
   def initialize(options = {})
+    @options = options.dup
     @original_client = @client = Client.new(options)
 
     super() # Monitor#initialize
@@ -2422,6 +2423,10 @@ def inspect
     "#<Redis client v#{Redis::VERSION} for #{id}>"
   end
 
+  def dup
+    self.class.new(@options)
+  end
+
   def method_missing(command, *args)
     synchronize do |client|
       client.call([command] + args)

lib/redis/distributed.rb

@@ -16,9 +16,10 @@ def message
     attr_reader :ring
 
     def initialize(node_configs, options = {})
-      @tag = options.delete(:tag) || /^\{(.+?)\}/
-      @ring = options.delete(:ring) || HashRing.new
-      @default_options = options
+      @tag = options[:tag] || /^\{(.+?)\}/
+      @ring = options[:ring] || HashRing.new
+      @node_configs = node_configs.dup
+      @default_options = options.dup
       node_configs.each { |node_config| add_node(node_config) }
       @subscribed_node = nil
     end
@@ -807,6 +808,10 @@ def inspect
       "#<Redis client v#{Redis::VERSION} for #{nodes.map(&:id).join(', ')}>"
     end
 
+    def dup
+      self.class.new(@node_configs, @default_options)
+    end
+
   protected
 
     def on_each_node(command, *args)

test/distributed_internals_test.rb

@@ -38,4 +38,33 @@ def test_override_id
     assert_equal redis.nodes.last.client.id,  "test1"
     assert_equal "#<Redis client v#{Redis::VERSION} for #{redis.nodes.map(&:id).join(', ')}>", redis.inspect
   end
+
+  def test_can_be_duped_to_create_a_new_connection
+    redis = Redis::Distributed.new(NODES)
+
+    clients = redis.info[0]["connected_clients"].to_i
+
+    r2 = redis.dup
+    r2.ping
+
+    assert_equal clients + 1, redis.info[0]["connected_clients"].to_i
+  end
+
+  def test_keeps_options_after_dup
+    r1 = Redis::Distributed.new(NODES, :tag => /^(\w+):/)
+
+    assert_raise(Redis::Distributed::CannotDistribute) do
+      r1.sinter("foo", "bar")
+    end
+
+    assert_equal [], r1.sinter("baz:foo", "baz:bar")
+
+    r2 = r1.dup
+
+    assert_raise(Redis::Distributed::CannotDistribute) do
+      r2.sinter("foo", "bar")
+    end
+
+    assert_equal [], r2.sinter("baz:foo", "baz:bar")
+  end
 end

test/helper.rb

@@ -141,6 +141,9 @@ module Generic
     def setup
       @log = StringIO.new
       @redis = init _new_client
+
+      # Run GC to make sure orphaned connections are closed.
+      GC.start
     end
 
     def teardown

test/internals_test.rb

@@ -342,4 +342,13 @@ def test_does_not_change_self_client_options
     assert_equal 1, redis.client.options[:db]
     assert_equal "foo", redis.client.options[:scheme]
   end
+
+  def test_can_be_duped_to_create_a_new_connection
+    clients = r.info["connected_clients"].to_i
+
+    r2 = r.dup
+    r2.ping
+
+    assert_equal clients + 1, r.info["connected_clients"].to_i
+  end
 end