Add support for variadic hset invocation #1217

Original pull request: #1220.

src/main/java/io/lettuce/core/AbstractRedisAsyncCommands.java

@@ -772,6 +772,11 @@ public RedisFuture<Boolean> hset(K key, K field, V value) {
         return dispatch(commandBuilder.hset(key, field, value));
     }
 
+    @Override
+    public RedisFuture<Long> hset(K key, Map<K, V> map) {
+        return dispatch(commandBuilder.hset(key, map));
+    }
+
     @Override
     public RedisFuture<Boolean> hsetnx(K key, K field, V value) {
         return dispatch(commandBuilder.hsetnx(key, field, value));

src/main/java/io/lettuce/core/AbstractRedisReactiveCommands.java

@@ -837,6 +837,11 @@ public Mono<Boolean> hset(K key, K field, V value) {
         return createMono(() -> commandBuilder.hset(key, field, value));
     }
 
+    @Override
+    public Mono<Long> hset(K key, Map<K, V> map) {
+        return createMono(() -> commandBuilder.hset(key, map));
+    }
+
     @Override
     public Mono<Boolean> hsetnx(K key, K field, V value) {
         return createMono(() -> commandBuilder.hsetnx(key, field, value));

src/main/java/io/lettuce/core/RedisCommandBuilder.java

@@ -1051,6 +1051,15 @@ Command<K, V, Boolean> hset(K key, K field, V value) {
         return createCommand(HSET, new BooleanOutput<>(codec), args);
     }
 
+    Command<K, V, Long> hset(K key, Map<K, V> map) {
+        notNullKey(key);
+        LettuceAssert.notNull(map, "Map " + MUST_NOT_BE_NULL);
+        LettuceAssert.isTrue(!map.isEmpty(), "Map " + MUST_NOT_BE_EMPTY);
+
+        CommandArgs<K, V> args = new CommandArgs<>(codec).addKey(key).add(map);
+        return createCommand(HSET, new IntegerOutput<>(codec), args);
+    }
+
     Command<K, V, Boolean> hsetnx(K key, K field, V value) {
         notNullKey(key);
         LettuceAssert.notNull(field, "Field " + MUST_NOT_BE_NULL);

src/main/java/io/lettuce/core/api/async/RedisHashAsyncCommands.java

@@ -249,6 +249,15 @@ public interface RedisHashAsyncCommands<K, V> {
      */
     RedisFuture<Boolean> hset(K key, K field, V value);
 
+    /**
+     * Set multiple hash fields to multiple values.
+     *
+     * @param key the key of the hash
+     * @param map the field/value pairs to update
+     * @return Long integer-reply: the number of fields that were added.
+     */
+    RedisFuture<Long> hset(K key, Map<K, V> map);
+
     /**
      * Set the value of a hash field, only if the field does not exist.
      *

src/main/java/io/lettuce/core/api/reactive/RedisHashReactiveCommands.java

@@ -250,6 +250,15 @@ public interface RedisHashReactiveCommands<K, V> {
      */
     Mono<Boolean> hset(K key, K field, V value);
 
+    /**
+     * Set multiple hash fields to multiple values.
+     *
+     * @param key the key of the hash
+     * @param map the field/value pairs to update
+     * @return Long integer-reply: the number of fields that were added.
+     */
+    Mono<Long> hset(K key, Map<K, V> map);
+
     /**
      * Set the value of a hash field, only if the field does not exist.
      *

src/main/java/io/lettuce/core/api/sync/RedisHashCommands.java

@@ -249,6 +249,15 @@ public interface RedisHashCommands<K, V> {
      */
     Boolean hset(K key, K field, V value);
 
+    /**
+     * Set multiple hash fields to multiple values.
+     *
+     * @param key the key of the hash
+     * @param map the field/value pairs to update
+     * @return Long integer-reply: the number of fields that were added.
+     */
+    Long hset(K key, Map<K, V> map);
+
     /**
      * Set the value of a hash field, only if the field does not exist.
      *

src/main/java/io/lettuce/core/cluster/api/async/NodeSelectionHashAsyncCommands.java

@@ -249,6 +249,15 @@ public interface NodeSelectionHashAsyncCommands<K, V> {
      */
     AsyncExecutions<Boolean> hset(K key, K field, V value);
 
+    /**
+     * Set multiple hash fields to multiple values.
+     *
+     * @param key the key of the hash
+     * @param map the field/value pairs to update
+     * @return Long integer-reply: the number of fields that were added.
+     */
+    AsyncExecutions<Long> hset(K key, Map<K, V> map);
+
     /**
      * Set the value of a hash field, only if the field does not exist.
      *

src/main/java/io/lettuce/core/cluster/api/sync/NodeSelectionHashCommands.java

@@ -249,6 +249,15 @@ public interface NodeSelectionHashCommands<K, V> {
      */
     Executions<Boolean> hset(K key, K field, V value);
 
+    /**
+     * Set multiple hash fields to multiple values.
+     *
+     * @param key the key of the hash
+     * @param map the field/value pairs to update
+     * @return Long integer-reply: the number of fields that were added.
+     */
+    Executions<Long> hset(K key, Map<K, V> map);
+
     /**
      * Set the value of a hash field, only if the field does not exist.
      *

src/main/templates/io/lettuce/core/api/RedisHashCommands.java

@@ -248,6 +248,15 @@ public interface RedisHashCommands<K, V> {
      */
     Boolean hset(K key, K field, V value);
 
+    /**
+     * Set multiple hash fields to multiple values.
+     *
+     * @param key the key of the hash
+     * @param map the field/value pairs to update
+     * @return Long integer-reply: the number of fields that were added.
+     */
+    Long hset(K key, Map<K, V> map);
+
     /**
      * Set the value of a hash field, only if the field does not exist.
      *

src/test/java/io/lettuce/core/commands/HashCommandIntegrationTests.java

@@ -223,6 +223,18 @@ void hset() {
         assertThat(redis.hset(key, "one", "1")).isFalse();
     }
 
+    @Test
+    void hsetMap() {
+        Map<String, String> hash = new LinkedHashMap<>();
+        hash.put("two", "2");
+        hash.put("three", "3");
+        assertThat(redis.hset(key, hash)).isEqualTo(2);
+
+        hash.put("two", "second");
+        assertThat(redis.hset(key, hash)).isEqualTo(0);
+        assertThat(redis.hget(key, "two")).isEqualTo("second");
+    }
+
     @Test
     void hsetnx() {
         redis.hset(key, "one", "1");