Add Redis::Distributed#mset_nonatomic & mapped_mset_nonatomic

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Unreleased
 
+# 5.0.8
+
+- Add `Redis::Distributed#mset_nonatomic` & `mapped_mset_nonatomic`
+
 # 5.0.7
 
 - Fix compatibility with `redis-client 0.15.0` when using Redis Sentinel. Fix #1209.

lib/redis/distributed.rb

@@ -302,10 +302,22 @@ def mset(*)
       raise CannotDistribute, :mset
     end
 
+    def mset_nonatomic(*args)
+      group_args_by_node(args.each_slice(2)) do |node, subargs|
+        node.mset(*subargs)
+      end
+    end
+
     def mapped_mset(_hash)
       raise CannotDistribute, :mapped_mset
     end
 
+    def mapped_mset_nonatomic(hash)
+      group_hash_by_node(hash) do |node, subhash|
+        node.mapped_mset(subhash)
+      end
+    end
+
     # Set multiple keys to multiple values, only if none of the keys exist.
     def msetnx(*)
       raise CannotDistribute, :msetnx
@@ -1089,5 +1101,17 @@ def ensure_same_node(command, keys)
 
       yield(node_for(keys.first))
     end
+
+    def group_args_by_node(args)
+      nodes = args.each_slice(2).group_by { |k, _v| node_for(k) }
+      nodes.each do |node, grouped_args|
+        yield node, grouped_args.flatten
+      end
+    end
+
+    def group_hash_by_node(hash, &block)
+      nodes = hash.group_by { |k, _v| node_for(k) }
+      nodes.each(&block)
+    end
   end
 end

test/distributed/commands_on_strings_test.rb

@@ -37,12 +37,26 @@ def test_mset
     end
   end
 
+  def test_mset_nonatomic
+    r.mset_nonatomic(:foo, "s1", :bar, "s2")
+
+    assert_equal 's1', r.get('foo')
+    assert_equal 's2', r.get('bar')
+  end
+
   def test_mset_mapped
     assert_raises Redis::Distributed::CannotDistribute do
       r.mapped_mset(foo: "s1", bar: "s2")
     end
   end
 
+  def test_mset_mapped_nonatomic
+    r.mapped_mset_nonatomic(foo: "s1", bar: "s2")
+
+    assert_equal 's1', r.get('foo')
+    assert_equal 's2', r.get('bar')
+  end
+
   def test_msetnx
     assert_raises Redis::Distributed::CannotDistribute do
       r.set("foo", "s1")