Accept a hash as value in hset

redis · Jun 9, 2020 · 87df483 · 87df483
1 parent ad7191f
commit 87df483
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/lib/redis.rb b/lib/redis.rb
@@ -2109,11 +2109,16 @@ def hlen(key)
   #
   # @example
   #   redis.hset("hash", "f1", "v1", "f2", "v2") # => 2
+  #   redis.hset("hash", { "f1" => "v1", "f2" => "v2" }) # => 2
   #
   # @param [String] key
-  # @param [Array<String>] attrs array of fields and values
+  # @param [Array<String> | Hash<String, String>] attrs array or hash of fields and values
   # @return [Integer] The number of fields that were added to the hash
   def hset(key, *attrs)
+    if attrs.size == 1 && attrs.first.is_a?(Hash)
+      attrs = attrs.first.flatten
+    end
+
     synchronize do |client|
       client.call([:hset, key, *attrs])
     end

diff --git a/test/lint/hashes.rb b/test/lint/hashes.rb
@@ -15,6 +15,11 @@ def test_variadic_hset
 
         assert_equal "s1", r.hget("foo", "f1")
         assert_equal "s2", r.hget("foo", "f2")
+
+        assert_equal 2, r.hset("bar", { "f1" => "s1", "f2" => "s2" })
+
+        assert_equal "s1", r.hget("bar", "f1")
+        assert_equal "s2", r.hget("bar", "f2")
       end
     end