saver 声明完成

Author: kaixinbaba

pom.xml

@@ -51,6 +51,13 @@
             <artifactId>commons-lang3</artifactId>
             <version>3.11</version>
         </dependency>
+        <!-- https://mvnrepository.com/artifact/com.alibaba/fastjson -->
+        <dependency>
+            <groupId>com.alibaba</groupId>
+            <artifactId>fastjson</artifactId>
+            <version>1.2.74</version>
+        </dependency>
+
     </dependencies>
 
 

src/main/java/ginyu/common/Consoles.java

@@ -108,6 +108,9 @@ public static void debug(String format, Object... arguments) {
         debug(format(format, arguments));
     }
 
+    public static void info(Object msg) {
+        info(String.valueOf(msg));
+    }
     public static void info(String msg) {
         if (canLog(Level.INFO)) {
             if (enableColor) {

src/main/java/ginyu/core/Server.java

@@ -5,10 +5,13 @@
 import ginyu.db.Db;
 import ginyu.io.Communicator;
 import ginyu.io.NettyCommunicator;
+import ginyu.persist.Saver;
+import ginyu.persist.SnapshotSaver;
 import ginyu.task.CleanExpiredTask;
 import ginyu.task.WakeupTimeoutClientTask;
 import ginyu.utils.ConfigUtils;
 import lombok.Getter;
+import lombok.ToString;
 import org.apache.commons.io.FileUtils;
 
 import java.io.File;
@@ -24,6 +27,7 @@
  * @description:
  */
 @SuppressWarnings("all")
+@ToString
 public class Server {
 
     public static final Server INSTANCE = new Server();
@@ -36,6 +40,8 @@ public class Server {
     @Getter
     private Db db;
 
+    private Saver saver;
+
     private CleanExpiredTask cleanExpiredTask;
 
     private WakeupTimeoutClientTask wakeupTimeoutClientTask;
@@ -49,6 +55,7 @@ public void init(String[] args) {
         this.db = new Db(this.ginyuConfig.getDbSize());
         this.cleanExpiredTask = new CleanExpiredTask();
         this.wakeupTimeoutClientTask = new WakeupTimeoutClientTask();
+        this.saver = new SnapshotSaver();
     }
 
     private void printEnv() {
@@ -80,7 +87,6 @@ public void start() throws Exception {
         Consoles.info("start listening port {}", ginyuConfig.getPort());
         this.communicator.start(ginyuConfig);
         this.startTask();
-        Consoles.info("server started");
     }
 
     public void addClient(Client client) {
@@ -95,4 +101,8 @@ private void startTask() {
         this.cleanExpiredTask.start();
         this.wakeupTimeoutClientTask.start();
     }
+
+    public void save() {
+        this.saver.save();
+    }
 }

src/main/java/ginyu/db/Database.java

@@ -21,6 +21,7 @@ public class Database {
     @Getter
     private final Integer id;
 
+    @Getter
     private final Dict<String, RedisObject> dict;
 
     @Getter
@@ -39,6 +40,14 @@ public Database(Integer id) {
         this.timeoutSets = new ConcurrentSkipListSet<>();
     }
 
+    public Database(Database database) {
+        this.id = database.id;
+        this.dict = new Dict<>(database.getDict());
+        this.expired = new ConcurrentSkipListMap<>(database.getExpired());
+        this.blockingDict = new Dict<>();
+        this.timeoutSets = new ConcurrentSkipListSet<>();
+    }
+
     public RedisObject get(String key) {
         return this.dict.get(key);
     }

src/main/java/ginyu/db/Db.java

@@ -22,6 +22,14 @@ public Db(Integer dbSize) {
         }
     }
 
+    public Db(Db db) {
+        List<Database> databases = db.getDatabases();
+        this.databases = new ArrayList<>(databases.size());
+        for (Database database : databases) {
+            this.databases.add(new Database(database));
+        }
+    }
+
     public Database getDatabase(Integer id) {
         return this.databases.get(id);
     }

src/main/java/ginyu/object/Dict.java

@@ -8,4 +8,11 @@
  * @description: TODO redis底层双数组
  */
 public class Dict<K, V> extends ConcurrentHashMap<K, V> {
+
+    public Dict() {
+    }
+
+    public Dict(Dict<K, V> dict) {
+        super(dict);
+    }
 }

src/main/java/ginyu/persist/Saver.java

@@ -0,0 +1,14 @@
+package ginyu.persist;
+
+import ginyu.core.Server;
+
+public interface Saver {
+
+    void save();
+
+    default void load() {
+        load(null);
+    }
+
+    void load(String filePath);
+}

src/main/java/ginyu/persist/ServerForSaver.java

@@ -0,0 +1,15 @@
+package ginyu.persist;
+
+import ginyu.db.Db;
+import lombok.Data;
+
+/**
+ * @author: junjiexun
+ * @date: 2020/10/27 9:48 下午
+ * @description:
+ */
+@Data
+public class ServerForSaver {
+
+    private Db db;
+}

src/main/java/ginyu/persist/SnapshotSaver.java

@@ -0,0 +1,34 @@
+package ginyu.persist;
+
+import com.alibaba.fastjson.JSONObject;
+import ginyu.common.Consoles;
+import ginyu.core.Server;
+import ginyu.db.Db;
+import ginyu.event.Events;
+
+import java.util.function.Supplier;
+
+/**
+ * @author: junjiexun
+ * @date: 2020/10/27 9:47 下午
+ * @description:
+ */
+public class SnapshotSaver implements Saver {
+
+    @Override
+    public void save() {
+        Events.post((Supplier<Void>) () -> {
+            final Server server = Server.INSTANCE;
+            ServerForSaver serverForSaver = new ServerForSaver();
+            serverForSaver.setDb(new Db(server.getDb()));
+            String s = JSONObject.toJSONString(serverForSaver, true);
+            Consoles.info(s);
+            return null;
+        });
+    }
+
+    @Override
+    public void load(String filePath) {
+
+    }
+}

src/main/java/ginyu/persist/SnapshotSaverListener.java

@@ -0,0 +1,36 @@
+package ginyu.persist;
+
+import com.google.common.eventbus.Subscribe;
+import ginyu.common.Consoles;
+import ginyu.core.Client;
+import ginyu.core.ClientTimeoutWrapper;
+import ginyu.db.Database;
+import ginyu.event.BlockEvent;
+import ginyu.event.list.BlockByBPopEvent;
+import ginyu.event.list.PushEvent;
+import ginyu.object.ListObject;
+import ginyu.object.ObjectType;
+import ginyu.protocol.Arrays;
+import ginyu.protocol.Resp2;
+import ginyu.protocol.Validates;
+import ginyu.utils.BlockUtils;
+
+import java.util.Collections;
+import java.util.HashSet;
+import java.util.Set;
+import java.util.function.Supplier;
+
+/**
+ * @author: junjiexun
+ * @date: 2020/10/20 1:41 下午
+ * @description:
+ */
+@SuppressWarnings("all")
+public class SnapshotSaverListener {
+
+    @Subscribe
+    public void saveSnapshot(Supplier<Void> supplier) {
+        supplier.get();
+        Consoles.info("save done");
+    }
+}