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/service/DbMigrationInitializer.java
+++ b/springboot-database/src/main/java/com/tiesheng/database/service/DbMigrationInitializer.java
@@ -0,0 +1,137 @@
+package com.tiesheng.database.service;
+
+
+import cn.hutool.core.date.DateUtil;
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.core.lang.func.VoidFunc1;
+import cn.hutool.core.util.ArrayUtil;
+import cn.hutool.core.util.IdUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.db.Db;
+import cn.hutool.db.Entity;
+import com.tiesheng.database.config.DbMigrationConfig;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.web.servlet.ServletContextInitializer;
+import org.springframework.core.io.Resource;
+import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
+import org.springframework.stereotype.Component;
+
+import javax.servlet.ServletContext;
+import javax.sql.DataSource;
+import java.io.InputStream;
+import java.sql.SQLException;
+import java.util.List;
+import java.util.zip.Checksum;
+
+/**
+ * 数据库表版本对比
+ *
+ * @author hao
+ */
+@Component
+public class DbMigrationInitializer implements ServletContextInitializer {
+
+    @Autowired
+    DataSource dataSource;
+    @Autowired
+    DbMigrationConfig dbMigrationConfig;
+
+
+    @Override
+    public void onStartup(ServletContext servletContext) {
+        try {
+            startDeal();
+        } catch (Exception ignore) {
+        }
+    }
+
+
+    /**
+     * 开始处理数据
+     *
+     * @throws Exception
+     */
+    private void startDeal() throws Exception {
+        Db coreDb = Db.use(dataSource);
+        dbMigrationConfig.checkDbExists(coreDb);
+
+        PathMatchingResourcePatternResolver patternResolver = new PathMatchingResourcePatternResolver();
+        for (String location : dbMigrationConfig.getLocations()) {
+            Resource[] resources = patternResolver.getResources(location);
+            if (ArrayUtil.isEmpty(resources)) {
+                return;
+            }
+            for (Resource resource : resources) {
+                migrationByResource(resource, coreDb);
+            }
+        }
+    }
+
+
+    /**
+     * 根据资源合并数据库结构
+     *
+     * @param resource
+     * @param db
+     * @throws Exception
+     */
+    private void migrationByResource(Resource resource, Db db) throws Exception {
+        InputStream inputStream = resource.getInputStream();
+        String readUtf8 = IoUtil.readUtf8(inputStream);
+        String filename = resource.getFilename();
+        Checksum checksum = IoUtil.checksum(resource.getInputStream(), null);
+        Entity entity = fileChecksum(db, filename, checksum.getValue());
+        if (entity == null) {
+            return;
+        }
+        db.tx((VoidFunc1<Db>) parameter -> {
+            List<String> split = StrUtil.split(readUtf8, ";");
+            for (String sql : split) {
+                sql = sql.trim();
+                if (StrUtil.isEmpty(sql)) {
+                    continue;
+                }
+                if (StrUtil.containsAnyIgnoreCase(sql, StrUtil.splitToArray(dbMigrationConfig.getIgnoreSqls(), ","))) {
+                    continue;
+                }
+                try {
+                    parameter.execute(sql);
+                } catch (Exception ignore) {
+                }
+            }
+        });
+
+        entity.set("checksum", checksum.getValue());
+        entity.set("update_time", DateUtil.date());
+        db.update(entity, Entity.create(dbMigrationConfig.getTable()).set("id", entity.get("id")));
+    }
+
+
+    /**
+     * 检查文件是否执行
+     *
+     * @param db
+     * @param fileName
+     * @param checksum
+     */
+    private Entity fileChecksum(Db db, String fileName, long checksum) throws SQLException {
+        Entity entity = db.get(dbMigrationConfig.getTable(), "file", fileName);
+        if (entity == null) {
+            entity = new Entity(dbMigrationConfig.getTable());
+            entity.set("id", IdUtil.getSnowflakeNextId());
+            entity.set("create_time", DateUtil.date());
+            entity.set("update_time", DateUtil.date());
+            entity.set("is_deleted", 0);
+            entity.set("file", fileName);
+            entity.set("checksum", 0);
+            db.insert(entity);
+        }
+
+        if (entity.getLong("checksum") != checksum) {
+            return entity;
+        }
+        return null;
+    }
+
+
+}