From 88655595f109f42c9d6f2825da936f556f5b2d5d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E6=B5=B7?= <10402852@qq.com>
Date: Fri, 10 Jan 2025 15:10:07 +0800
Subject: [PATCH] 'commit'

---
 .../YunXiaoTools/Controller/DataBaseController.java   |  1 +
 .../com/dsideal/YunXiaoTools/Cron4j/LingDianTask.java | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Controller/DataBaseController.java b/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Controller/DataBaseController.java
index 76c4506e..0610b79c 100644
--- a/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Controller/DataBaseController.java
+++ b/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Controller/DataBaseController.java
@@ -28,6 +28,7 @@ public class DataBaseController extends Controller {
     /**
      * 还原mysql数据
      */
+    // http://10.10.21.20:9008/restoreMysql
     public void restoreMysql() {
         //获取最后一次上传的数据库文件
         String key = CommonUtil.getLastDataBaseZip();
diff --git a/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Cron4j/LingDianTask.java b/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Cron4j/LingDianTask.java
index 491e59aa..3e6afe77 100644
--- a/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Cron4j/LingDianTask.java
+++ b/YunXiaoTools/src/main/java/com/dsideal/YunXiaoTools/Cron4j/LingDianTask.java
@@ -1,7 +1,9 @@
 package com.dsideal.YunXiaoTools.Cron4j;
 
 import com.dsideal.YunXiaoTools.Service.MysqlBackupService;
+import com.dsideal.YunXiaoTools.Service.MysqlRestoreService;
 import com.dsideal.YunXiaoTools.Start;
+import com.dsideal.YunXiaoTools.Utils.CommonUtil;
 import com.jfinal.plugin.cron4j.ITask;
 import lombok.SneakyThrows;
 
@@ -14,6 +16,15 @@ public class LingDianTask implements ITask {
             MysqlBackupService bm = new MysqlBackupService();
             bm.doAction();
         }
+        //如果是写入侧
+        else {
+            //获取最后一次上传的数据库文件
+            String key = CommonUtil.getLastDataBaseZip();
+            //还原数据库
+            MysqlRestoreService mr = new MysqlRestoreService();
+            int total = mr.restore(mr.jdbcUrl, mr.username, mr.password, key);
+            System.out.println("恭喜，成功还原数据表:" + total + "个。");
+        }
     }
 
     public void stop() {