Support RESET command (redis#2952)

ihaohong · Jun 14, 2022 · f45adc2 · f45adc2
1 parent d6ba80a
commit f45adc2
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 1 deletion.
diff --git a/src/main/java/redis/clients/jedis/Jedis.java b/src/main/java/redis/clients/jedis/Jedis.java
@@ -9176,6 +9176,12 @@ public String lolwut(LolwutParams lolwutParams) {
     return connection.getBulkReply();
   }
 
+  @Override
+  public String reset() {
+    connection.sendCommand(Command.RESET);
+    return connection.getStatusCodeReply();
+  }
+
   @Override
   public StreamEntryID xadd(final String key, final StreamEntryID id, final Map<String, String> hash) {
     checkIsInMultiOrPipeline();

diff --git a/src/main/java/redis/clients/jedis/Protocol.java b/src/main/java/redis/clients/jedis/Protocol.java
@@ -237,7 +237,7 @@ public static enum Command implements ProtocolCommand {
     XADD, XLEN, XDEL, XTRIM, XRANGE, XREVRANGE, XREAD, XACK, XGROUP, XREADGROUP, XPENDING, XCLAIM,
     XAUTOCLAIM, XINFO, BITFIELD_RO, ROLE, FAILOVER, GEOSEARCH, GEOSEARCHSTORE, EVAL_RO, EVALSHA_RO,
     LOLWUT, EXPIRETIME, PEXPIRETIME, FUNCTION, FCALL, FCALL_RO, LMPOP, BLMPOP, ZMPOP, BZMPOP,
-    COMMAND, @Deprecated STRALGO;
+    COMMAND, @Deprecated STRALGO, RESET;
 
     private final byte[] raw;
 

diff --git a/src/main/java/redis/clients/jedis/commands/ServerCommands.java b/src/main/java/redis/clients/jedis/commands/ServerCommands.java
@@ -221,4 +221,6 @@ public interface ServerCommands {
   String lolwut();
 
   String lolwut(LolwutParams lolwutParams);
+
+  String reset();
 }
diff --git a/src/test/java/redis/clients/jedis/commands/jedis/AllKindOfValuesCommandsTest.java b/src/test/java/redis/clients/jedis/commands/jedis/AllKindOfValuesCommandsTest.java
@@ -20,6 +20,7 @@
 
 import redis.clients.jedis.HostAndPort;
 import redis.clients.jedis.Jedis;
+import redis.clients.jedis.Transaction;
 import redis.clients.jedis.args.ExpiryOption;
 import redis.clients.jedis.exceptions.JedisException;
 import redis.clients.jedis.params.CommandListFilterByParams;
@@ -1131,4 +1132,31 @@ public void commandList() {
     }
   }
 
+  @Test
+  public void reset() {
+    // response test
+    String status = jedis.reset();
+    assertEquals("RESET", status);
+
+    // auth reset
+    String counter = "counter";
+    Exception ex1 = assertThrows(JedisDataException.class, () -> {
+      jedis.set(counter, "1");
+    });
+    assertEquals("NOAUTH Authentication required.", ex1.getMessage());
+
+    // multi reset
+    jedis.auth("foobared");
+    jedis.set(counter, "1");
+
+    Transaction trans = jedis.multi();
+    trans.incr(counter);
+    jedis.reset();
+
+    Exception ex2 = assertThrows(JedisDataException.class, trans::exec);
+    assertEquals("EXECABORT Transaction discarded because of: NOAUTH Authentication required.", ex2.getMessage());
+
+    jedis.auth("foobared");
+    assertEquals("1", jedis.get(counter));
+  }
 }