add key commands and unit tests

Author: tuhuynh27

docs/src/guide/overview/commands.md

@@ -33,6 +33,9 @@ Implemented commands:
 - RENAME
 - EXPIRE
 - EXPIREAT
+- DUMP
+- RESTORE
+- TYPE
 
 </details>
 

server/build.gradle

@@ -25,6 +25,7 @@ dependencies {
 
     implementation 'com.google.guava:guava:31.0.1-jre'
     implementation 'org.reflections:reflections:0.10.2'
+    implementation 'org.apache.commons:commons-lang3:3.0'
 
     // Experimental
     implementation files('libs/keva-ioc-0.1.0-SNAPSHOT.jar')

server/src/main/java/dev/keva/server/command/impl/key/Dump.java

@@ -0,0 +1,35 @@
+package dev.keva.server.command.impl.key;
+
+import dev.keva.ioc.annotation.Autowired;
+import dev.keva.ioc.annotation.Component;
+import dev.keva.protocol.resp.reply.BulkReply;
+import dev.keva.protocol.resp.reply.Reply;
+import dev.keva.server.command.annotation.CommandImpl;
+import dev.keva.server.command.annotation.Execute;
+import dev.keva.server.command.annotation.ParamLength;
+import dev.keva.store.KevaDatabase;
+import lombok.val;
+
+import java.util.Base64;
+
+@Component
+@CommandImpl("dump")
+@ParamLength(1)
+public class Dump {
+    private final KevaDatabase database;
+
+    @Autowired
+    public Dump(KevaDatabase database) {
+        this.database = database;
+    }
+
+    @Execute
+    public Reply<?> execute(byte[] key) {
+        val got = database.get(key);
+        if (got == null) {
+            return BulkReply.NIL_REPLY;
+        }
+        String dumped = Base64.getEncoder().encodeToString(got);
+        return new BulkReply(dumped);
+    }
+}

server/src/main/java/dev/keva/server/command/impl/key/Restore.java

@@ -0,0 +1,46 @@
+package dev.keva.server.command.impl.key;
+
+import dev.keva.ioc.annotation.Autowired;
+import dev.keva.ioc.annotation.Component;
+import dev.keva.protocol.resp.reply.ErrorReply;
+import dev.keva.protocol.resp.reply.Reply;
+import dev.keva.protocol.resp.reply.StatusReply;
+import dev.keva.server.command.annotation.CommandImpl;
+import dev.keva.server.command.annotation.Execute;
+import dev.keva.server.command.annotation.ParamLength;
+import dev.keva.server.command.impl.key.manager.ExpirationManager;
+import dev.keva.store.KevaDatabase;
+import lombok.val;
+
+import java.math.BigInteger;
+import java.util.Base64;
+
+@Component
+@CommandImpl("restore")
+@ParamLength(type = ParamLength.Type.AT_LEAST, value = 3)
+public class Restore {
+    private final KevaDatabase database;
+    private final ExpirationManager expirationManager;
+
+    @Autowired
+    public Restore(KevaDatabase database, ExpirationManager expirationManager) {
+        this.database = database;
+        this.expirationManager = expirationManager;
+    }
+
+    @Execute
+    public Reply<?> execute(byte[] key, byte[] ttl, byte[] dump, byte[] replace) {
+        val old = database.get(key);
+        boolean isReplace = replace != null && new String(replace).equalsIgnoreCase("REPLACE");
+        if (old != null && !isReplace) {
+            return new ErrorReply("ERR Target key name is busy");
+        }
+        byte[] value = Base64.getDecoder().decode(dump);
+        database.put(key, value);
+        long expireTime = new BigInteger(ttl).longValue();
+        if (expireTime > 0) {
+            expirationManager.expireAfter(key, expireTime);
+        }
+        return StatusReply.OK;
+    }
+}

server/src/main/java/dev/keva/server/command/impl/key/Type.java

@@ -0,0 +1,56 @@
+package dev.keva.server.command.impl.key;
+
+import dev.keva.ioc.annotation.Autowired;
+import dev.keva.ioc.annotation.Component;
+import dev.keva.protocol.resp.reply.StatusReply;
+import dev.keva.server.command.annotation.CommandImpl;
+import dev.keva.server.command.annotation.Execute;
+import dev.keva.server.command.annotation.ParamLength;
+import dev.keva.store.KevaDatabase;
+import lombok.val;
+import org.apache.commons.lang3.SerializationException;
+import org.apache.commons.lang3.SerializationUtils;
+
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.LinkedList;
+
+@Component
+@CommandImpl("type")
+@ParamLength(1)
+public class Type {
+    private final KevaDatabase database;
+
+    @Autowired
+    public Type(KevaDatabase database) {
+        this.database = database;
+    }
+
+    @Execute
+    public StatusReply execute(byte[] key) {
+        val got = database.get(key);
+        if (got == null) {
+            return new StatusReply("none");
+        }
+        try {
+            Object ignored = SerializationUtils.deserialize(got);
+        } catch (SerializationException e) {
+            return new StatusReply("string");
+        }
+        Object value = SerializationUtils.deserialize(got);
+        try {
+            String ignored = (String) value;
+            return new StatusReply("string");
+        } catch (ClassCastException e) {
+            if (value instanceof HashMap) {
+                return new StatusReply("hash");
+            } else if (value instanceof LinkedList) {
+                return new StatusReply("list");
+            } else if (value instanceof HashSet) {
+                return new StatusReply("set");
+            } else {
+                return new StatusReply("unknown");
+            }
+        }
+    }
+}

server/src/test/java/dev/keva/server/core/AbstractServerTest.java

@@ -791,4 +791,37 @@ void setrange() {
         }
     }
 
+    @Test
+    void dumpAndRestore() {
+        try {
+            val set1 = jedis.set("key1", "Hello World");
+            assertEquals("OK", set1);
+            val dump1 = jedis.dump("key1");
+            assertNotNull(dump1);
+            val restore1 = jedis.restore("key2", 0, dump1);
+            assertEquals("OK", restore1);
+            val key2 = jedis.get("key2");
+            assertEquals("Hello World", key2);
+        } catch (Exception e) {
+            fail(e);
+        }
+    }
+
+    @Test
+    void type() {
+        try {
+            val set1 = jedis.set("key1", "Hello World");
+            assertEquals("OK", set1);
+            val type1 = jedis.type("key1");
+            assertEquals("string", type1);
+            val type2 = jedis.type("key2");
+            assertEquals("none", type2);
+            val hashSet = jedis.hset("key3", "field1", "value1");
+            assertEquals(1, hashSet.intValue());
+            val type3 = jedis.type("key3");
+            assertEquals("hash", type3);
+        } catch (Exception e) {
+            fail(e);
+        }
+    }
 }

store/src/main/java/dev/keva/store/impl/OnHeapDatabaseImpl.java

@@ -198,7 +198,7 @@ public boolean hdel(byte[] key, byte[] field) {
     public int lpush(byte[] key, byte[]... values) {
         lock.lock();
         try {
-            byte[] value = map.get(key).getBytes();
+            byte[] value = map.get(new BytesKey(key)).getBytes();
             LinkedList<BytesValue> list;
             list = value == null ? new LinkedList<>() : (LinkedList<BytesValue>) SerializationUtils.deserialize(value);
             for (byte[] v : values) {
@@ -216,7 +216,7 @@ public int lpush(byte[] key, byte[]... values) {
     public int rpush(byte[] key, byte[]... values) {
         lock.lock();
         try {
-            byte[] value = map.get(key).getBytes();
+            byte[] value = map.get(new BytesKey(key)).getBytes();
             LinkedList<BytesValue> list;
             list = value == null ? new LinkedList<>() : (LinkedList<BytesValue>) SerializationUtils.deserialize(value);
             for (byte[] v : values) {