perf：增加数据库备份

2023-02-20 18:25:36 +08:00
parent 7fcc407514
commit 1918dfbaa0
9 changed files with 86 additions and 21 deletions
--- a/springboot-database/src/main/java/com/tiesheng/database/config/DbBackupConfig.java
+++ b/springboot-database/src/main/java/com/tiesheng/database/config/DbBackupConfig.java
@@ -0,0 +1,65 @@
+package com.tiesheng.database.config;
+
+import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.util.RuntimeUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.extra.spring.SpringUtil;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
+
+import javax.annotation.PostConstruct;
+import java.util.Date;
+
+/**
+ * @author hao
+ */
+@Configuration
+@ConfigurationProperties(prefix = "tiesheng.db-backup")
+public class DbBackupConfig {
+
+    /**
+     * 数据库备份的路径
+     */
+    private String path = "/root/backup/";
+
+    /**
+     * 备份文件的时间格式
+     */
+    private String format = "yyyyMMdd";
+
+
+    @PostConstruct
+    public void dbBackup() {
+        String url = SpringUtil.getProperty("spring.datasource.url");
+        String username = SpringUtil.getProperty("spring.datasource.username");
+        String password = SpringUtil.getProperty("spring.datasource.password");
+
+        try {
+            RuntimeUtil.exec(StrUtil.format("mysqldump -u{} -p{} {} > {}{}.sql", username, password, username,
+                    getPath(), DateUtil.format(new Date(), format)
+            ));
+        } catch (Exception ignored) {
+        }
+
+    }
+
+    ///////////////////////////////////////////////////////////////////////////
+    // setter\getter
+    ///////////////////////////////////////////////////////////////////////////
+
+    public String getPath() {
+        return path;
+    }
+
+    public void setPath(String path) {
+        this.path = path;
+    }
+
+    public String getFormat() {
+        return format;
+    }
+
+    public void setFormat(String format) {
+        this.format = format;
+    }
+}
--- a/springboot-database/src/main/java/com/tiesheng/database/config/DbMigrationConfig.java
+++ b/springboot-database/src/main/java/com/tiesheng/database/config/DbMigrationConfig.java
@@ -0,0 +1,75 @@
+package com.tiesheng.database.config;
+
+
+import cn.hutool.core.collection.CollUtil;
+import cn.hutool.db.Db;
+import cn.hutool.db.Entity;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
+
+import java.sql.SQLException;
+import java.util.List;
+
+/**
+ * 数据库表版本对比
+ *
+ * @author hao
+ */
+@Configuration
+@ConfigurationProperties(prefix = "tiesheng.db-migration")
+public class DbMigrationConfig {
+
+    private String table = "core_config_db";
+    private List<String> locations = CollUtil.newArrayList("classpath*:db/migration/*.sql");
+    private String ignoreSqls = "drop table,delete from";
+
+    /**
+     * 检查数据库是否存在
+     *
+     * @param db
+     * @throws SQLException
+     */
+    public void checkDbExists(Db db) throws SQLException {
+        try {
+            db.count(Entity.create(getTable()));
+        } catch (SQLException ignored) {
+            db.execute("CREATE TABLE `" + getTable() + "`  (\n" +
+                    "  `id` varchar(50) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NOT NULL,\n" +
+                    "  `create_time` datetime(0) NOT NULL,\n" +
+                    "  `update_time` datetime(0) NOT NULL,\n" +
+                    "  `is_deleted` int(6) NOT NULL DEFAULT 0,\n" +
+                    "  `file` varchar(500) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '文件名称',\n" +
+                    "  `checksum` varchar(500) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NULL DEFAULT NULL COMMENT '文件checksum',\n" +
+                    "  PRIMARY KEY (`id`) USING BTREE\n" +
+                    ") ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '配置-数据库' ROW_FORMAT = Dynamic;");
+        }
+    }
+
+    ///////////////////////////////////////////////////////////////////////////
+    // setter\getter
+    ///////////////////////////////////////////////////////////////////////////
+
+    public String getTable() {
+        return table;
+    }
+
+    public void setTable(String table) {
+        this.table = table;
+    }
+
+    public List<String> getLocations() {
+        return locations;
+    }
+
+    public void setLocations(List<String> locations) {
+        this.locations = locations;
+    }
+
+    public String getIgnoreSqls() {
+        return ignoreSqls;
+    }
+
+    public void setIgnoreSqls(String ignoreSqls) {
+        this.ignoreSqls = ignoreSqls;
+    }
+}