Java: Add SETRANGE command.

glide-core/src/protobuf/redis_request.proto

@@ -135,6 +135,7 @@ enum RequestType {
     Rename = 91;
     DBSize = 92;
     Brpop = 93;
+    SetRange = 107;
 }
 
 message Command {

glide-core/src/socket_listener.rs

@@ -362,6 +362,7 @@ fn get_command(request: &Command) -> Option<Cmd> {
         RequestType::Rename => Some(cmd("RENAME")),
         RequestType::DBSize => Some(cmd("DBSIZE")),
         RequestType::Brpop => Some(cmd("BRPOP")),
+        RequestType::SetRange => Some(cmd("SETRANGE")),
     }
 }
 

java/client/src/main/java/glide/api/BaseClient.java

@@ -42,6 +42,7 @@
 import static redis_request.RedisRequestOuterClass.RequestType.SCard;
 import static redis_request.RedisRequestOuterClass.RequestType.SMembers;
 import static redis_request.RedisRequestOuterClass.RequestType.SRem;
+import static redis_request.RedisRequestOuterClass.RequestType.SetRange;
 import static redis_request.RedisRequestOuterClass.RequestType.SetString;
 import static redis_request.RedisRequestOuterClass.RequestType.Strlen;
 import static redis_request.RedisRequestOuterClass.RequestType.TTL;
@@ -319,6 +320,12 @@ public CompletableFuture<Long> strlen(@NonNull String key) {
         return commandManager.submitNewCommand(Strlen, new String[] {key}, this::handleLongResponse);
     }
 
+    @Override
+    public CompletableFuture<Long> setrange(@NonNull String key, int offset, @NonNull String value) {
+        return commandManager.submitNewCommand(
+                SetRange, new String[] {key, Integer.toString(offset), value}, this::handleLongResponse);
+    }
+
     @Override
     public CompletableFuture<String> hget(@NonNull String key, @NonNull String field) {
         return commandManager.submitNewCommand(

java/client/src/main/java/glide/api/commands/StringCommands.java

@@ -203,4 +203,25 @@ public interface StringCommands {
      * }</pre>
      */
     CompletableFuture<Long> strlen(String key);
+
+    /**
+     * Overwrites part of the string stored at <code>key</code>, starting at the specified <code>
+     * offset</code>, for the entire length of <code>value</code>.<br>
+     * If the <code>offset</code> is larger than the current length of the string at <code>key</code>,
+     * the string is padded with zero-bytes to make <code>offset</code> fit. Creates the key if it
+     * doesn't exist.
+     *
+     * @see <a href="https://redis.io/commands/setrange/">redis.io</a> for details.
+     * @param key The key of the string to update.
+     * @param offset The position in the string where <code>value</code> should be written.
+     * @param value The string which should be written at the position specified by <code>offset
+     *     </code>.
+     * @return The length of the string stored at <code>key</code> after it was modified.
+     * @example
+     *     <pre>{@code
+     * long len = client.setrange("key", 6, "Redis").get();
+     * assert len == 11L;
+     * }</pre>
+     */
+    CompletableFuture<Long> setrange(String key, int offset, String value);
 }

java/client/src/main/java/glide/api/models/BaseTransaction.java

@@ -50,6 +50,7 @@
 import static redis_request.RedisRequestOuterClass.RequestType.SCard;
 import static redis_request.RedisRequestOuterClass.RequestType.SMembers;
 import static redis_request.RedisRequestOuterClass.RequestType.SRem;
+import static redis_request.RedisRequestOuterClass.RequestType.SetRange;
 import static redis_request.RedisRequestOuterClass.RequestType.SetString;
 import static redis_request.RedisRequestOuterClass.RequestType.Strlen;
 import static redis_request.RedisRequestOuterClass.RequestType.TTL;
@@ -379,6 +380,27 @@ public T strlen(@NonNull String key) {
         return getThis();
     }
 
+    /**
+     * Overwrites part of the string stored at <code>key</code>, starting at the specified <code>
+     * offset</code>, for the entire length of <code>value</code>.<br>
+     * If the <code>offset</code> is larger than the current length of the string at <code>key</code>,
+     * the string is padded with zero-bytes to make <code>offset</code> fit. Creates the key if it
+     * doesn't exist.
+     *
+     * @see <a href="https://redis.io/commands/setrange/">redis.io</a> for details.
+     * @param key The key of the string to update.
+     * @param offset The position in the string where <code>value</code> should be written.
+     * @param value The string which should be written at the position specified by <code>offset
+     *     </code>.
+     * @return Command Response - The length of the string stored at <code>key</code> after it was
+     *     modified.
+     */
+    public T setrange(@NonNull String key, int offset, @NonNull String value) {
+        ArgsArray commandArgs = buildArgs(key, Integer.toString(offset), value);
+        protobufTransaction.addCommands(buildCommand(SetRange, commandArgs));
+        return getThis();
+    }
+
     /**
      * Retrieve the value associated with <code>field</code> in the hash stored at <code>key</code>.
      *

java/client/src/test/java/glide/api/RedisClientTest.java

@@ -62,6 +62,7 @@
 import static redis_request.RedisRequestOuterClass.RequestType.SMembers;
 import static redis_request.RedisRequestOuterClass.RequestType.SRem;
 import static redis_request.RedisRequestOuterClass.RequestType.Select;
+import static redis_request.RedisRequestOuterClass.RequestType.SetRange;
 import static redis_request.RedisRequestOuterClass.RequestType.SetString;
 import static redis_request.RedisRequestOuterClass.RequestType.Strlen;
 import static redis_request.RedisRequestOuterClass.RequestType.TTL;
@@ -938,6 +939,32 @@ public void strlen_returns_success() {
         assertEquals(value, payload);
     }
 
+    @SneakyThrows
+    @Test
+    public void setrange_returns_success() {
+        // setup
+        String key = "testKey";
+        int offset = 42;
+        String str = "pewpew";
+        String[] arguments = new String[] {key, Integer.toString(offset), str};
+        Long value = 10L;
+
+        CompletableFuture<Long> testResponse = new CompletableFuture<>();
+        testResponse.complete(value);
+
+        // match on protobuf request
+        when(commandManager.<Long>submitNewCommand(eq(SetRange), eq(arguments), any()))
+                .thenReturn(testResponse);
+
+        // exercise
+        CompletableFuture<Long> response = service.setrange(key, offset, str);
+        Long payload = response.get();
+
+        // verify
+        assertEquals(testResponse, response);
+        assertEquals(value, payload);
+    }
+
     @SneakyThrows
     @Test
     public void hget_success() {

java/client/src/test/java/glide/api/models/TransactionTests.java

@@ -48,6 +48,7 @@
 import static redis_request.RedisRequestOuterClass.RequestType.SCard;
 import static redis_request.RedisRequestOuterClass.RequestType.SMembers;
 import static redis_request.RedisRequestOuterClass.RequestType.SRem;
+import static redis_request.RedisRequestOuterClass.RequestType.SetRange;
 import static redis_request.RedisRequestOuterClass.RequestType.SetString;
 import static redis_request.RedisRequestOuterClass.RequestType.Strlen;
 import static redis_request.RedisRequestOuterClass.RequestType.TTL;
@@ -149,6 +150,11 @@ public void transaction_builds_protobuf_request(BaseTransaction<?> transaction)
         transaction.strlen("key");
         results.add(Pair.of(Strlen, ArgsArray.newBuilder().addArgs("key").build()));
 
+        transaction.setrange("key", 42, "str");
+        results.add(
+                Pair.of(
+                        SetRange, ArgsArray.newBuilder().addArgs("key").addArgs("42").addArgs("str").build()));
+
         transaction.hset("key", Map.of("field", "value"));
         results.add(
                 Pair.of(

java/integTest/src/test/java/glide/SharedCommandTests.java

@@ -430,6 +430,34 @@ public void strlen(BaseClient client) {
         assertTrue(exception.getCause() instanceof RequestException);
     }
 
+    @SneakyThrows
+    @ParameterizedTest
+    @MethodSource("getClients")
+    public void setrange(BaseClient client) {
+        String stringKey = UUID.randomUUID().toString();
+        String nonStringKey = UUID.randomUUID().toString();
+        String nonExistingKey = UUID.randomUUID().toString();
+
+        assertEquals(OK, client.set(stringKey, "Hello world").get());
+        assertEquals(11L, client.setrange(stringKey, 6, "GLIDE").get());
+        assertEquals("Hello GLIDE", client.get(stringKey).get());
+        // offset > len
+        assertEquals(20L, client.setrange(stringKey, 15, "GLIDE").get());
+        assertEquals("Hello GLIDE\0\0\0\0GLIDE", client.get(stringKey).get());
+        // new key
+        assertEquals(5L, client.setrange(nonExistingKey, 0, "GLIDE").get());
+        assertEquals("GLIDE", client.get(nonExistingKey).get());
+
+        assertEquals(1, client.lpush(nonStringKey, new String[] {"_"}).get());
+        Exception exception =
+                assertThrows(ExecutionException.class, () -> client.setrange(nonStringKey, 0, "_").get());
+        assertTrue(exception.getCause() instanceof RequestException);
+        exception =
+                assertThrows(
+                        ExecutionException.class, () -> client.setrange("foo", Integer.MAX_VALUE, "_").get());
+        assertTrue(exception.getCause() instanceof RequestException);
+    }
+
     @SneakyThrows
     @ParameterizedTest
     @MethodSource("getClients")

java/integTest/src/test/java/glide/TransactionTestUtilities.java

@@ -56,6 +56,7 @@ public static BaseTransaction<?> transactionTest(BaseTransaction<?> baseTransact
         baseTransaction.incrByFloat(key3, 0.5);
 
         baseTransaction.unlink(new String[] {key3});
+        baseTransaction.setrange(key3, 0, "GLIDE");
 
         baseTransaction.hset(key4, Map.of(field1, value1, field2, value2));
         baseTransaction.hget(key4, field1);
@@ -124,6 +125,7 @@ public static Object[] transactionTestResult() {
             0L,
             0.5,
             1L,
+            5L, // setrange(key3, 0, "GLIDE")
             2L,
             value1,
             true,