支持自动备份

core-lib · core-lib · commit 7dbc313e979f · 2019-08-02T16:39:54.000+08:00
diff --git a/src/main/java/io/sqlman/SqlDialectSupport.java b/src/main/java/io/sqlman/SqlDialectSupport.java
@@ -62,4 +62,14 @@ public interface SqlDialectSupport {
      */
     void unlock(Connection connection) throws SQLException;
 
+    /**
+     * 备份表
+     *
+     * @param connection 连接
+     * @param script     脚本
+     * @param ordinal    语句下标
+     * @throws SQLException SQL异常
+     */
+    void backup(Connection connection, SqlScript script, int ordinal) throws SQLException;
+
 }
diff --git a/src/main/java/io/sqlman/SqlSentence.java b/src/main/java/io/sqlman/SqlSentence.java
@@ -22,4 +22,10 @@ public interface SqlSentence {
      */
     String value();
 
+    /**
+     * 操作表名
+     *
+     * @return 操作表名
+     */
+    String table();
 }
diff --git a/src/main/java/io/sqlman/dialect/AbstractDialectSupport.java b/src/main/java/io/sqlman/dialect/AbstractDialectSupport.java
@@ -1,8 +1,6 @@
 package io.sqlman.dialect;
 
-import io.sqlman.SqlDialectSupport;
-import io.sqlman.SqlUtils;
-import io.sqlman.SqlVersion;
+import io.sqlman.*;
 
 import java.sql.Connection;
 import java.sql.PreparedStatement;
@@ -126,6 +124,22 @@ public void unlock(Connection connection) throws SQLException {
         connection.prepareStatement("DROP TABLE " + table.toUpperCase() + "_LOCK").executeUpdate();
     }
 
+    @Override
+    public void backup(Connection connection, SqlScript script, int ordinal) throws SQLException {
+        SqlSentence sentence = script.sentence(ordinal);
+        String table = sentence.table();
+        if (table == null || table.trim().isEmpty()) {
+            return;
+        }
+        try {
+            connection.prepareStatement("SELECT COUNT(*) FROM " + table).executeQuery();
+        } catch (SQLException e) {
+            return;
+        }
+        table = table + "_bak_" + script.version().replace('.', '_') + "$" + ordinal;
+        connection.prepareStatement("CREATE TABLE " + table + " AS SELECT * FROM " + sentence.table()).executeUpdate();
+    }
+
     public String getTable() {
         return table;
     }
diff --git a/src/main/java/io/sqlman/dialect/SQLServerDialectSupport.java b/src/main/java/io/sqlman/dialect/SQLServerDialectSupport.java
@@ -1,6 +1,8 @@
 package io.sqlman.dialect;
 
 import io.sqlman.SqlDialectSupport;
+import io.sqlman.SqlScript;
+import io.sqlman.SqlSentence;
 
 import java.sql.Connection;
 import java.sql.SQLException;
@@ -47,4 +49,19 @@ public void create(Connection connection) throws SQLException {
         connection.prepareStatement(ddl.toString()).executeUpdate();
     }
 
+    @Override
+    public void backup(Connection connection, SqlScript script, int ordinal) throws SQLException {
+        SqlSentence sentence = script.sentence(ordinal);
+        String table = sentence.table();
+        if (table == null || table.trim().isEmpty()) {
+            return;
+        }
+        try {
+            connection.prepareStatement("SELECT COUNT(*) FROM " + table).executeQuery();
+        } catch (SQLException e) {
+            return;
+        }
+        table = table + "_bak_" + script.version().replace('.', '_') + "$" + ordinal;
+        connection.prepareStatement("SELECT * INTO " + table + " FROM " + sentence.table()).executeUpdate();
+    }
 }
diff --git a/src/main/java/io/sqlman/script/DruidScriptResolver.java b/src/main/java/io/sqlman/script/DruidScriptResolver.java
@@ -1,7 +1,9 @@
 package io.sqlman.script;
 
 import com.alibaba.druid.sql.SQLUtils;
+import com.alibaba.druid.sql.ast.SQLObject;
 import com.alibaba.druid.sql.ast.SQLStatement;
+import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
 import com.alibaba.druid.sql.parser.ParserException;
 import com.alibaba.druid.util.JdbcUtils;
 import io.sqlman.*;
@@ -57,7 +59,16 @@ public SqlScript resolve(SqlSource source) throws IncorrectSyntaxException, IOEx
                 while (sql.endsWith(suffix)) {
                     sql = sql.substring(0, sql.length() - 1);
                 }
-                SqlSentence sentence = new DruidSentence(index + 1, sql);
+                String table = null;
+                List<SQLObject> children = statement.getChildren();
+                for (SQLObject child : children) {
+                    if (child instanceof SQLExprTableSource) {
+                        SQLExprTableSource tableSource = (SQLExprTableSource) child;
+                        table = tableSource.getName().toString();
+                        break;
+                    }
+                }
+                SqlSentence sentence = new DruidSentence(index + 1, sql, table);
                 sentences.add(sentence);
             }
             return new DruidScript(source.name(), source.version(), source.parameters(), source.description(), sentences);
diff --git a/src/main/java/io/sqlman/script/DruidSentence.java b/src/main/java/io/sqlman/script/DruidSentence.java
@@ -11,8 +11,9 @@
 public class DruidSentence implements SqlSentence {
     private final int ordinal;
     private final String value;
+    private final String table;
 
-    public DruidSentence(int ordinal, String value) {
+    public DruidSentence(int ordinal, String value, String table) {
         if (ordinal < 1) {
             throw new IllegalArgumentException("ordinal must not lesser than 1");
         }
@@ -21,6 +22,7 @@ public DruidSentence(int ordinal, String value) {
         }
         this.ordinal = ordinal;
         this.value = value;
+        this.table = table;
     }
 
     @Override
@@ -33,6 +35,11 @@ public String value() {
         return value;
     }
 
+    @Override
+    public String table() {
+        return table;
+    }
+
     @Override
     public String toString() {
         return value;
diff --git a/src/main/java/io/sqlman/spring/JdbcManagerConfiguration.java b/src/main/java/io/sqlman/spring/JdbcManagerConfiguration.java
@@ -1,6 +1,8 @@
 package io.sqlman.spring;
 
 import io.sqlman.*;
+import io.sqlman.version.JdbcIsolation;
+import io.sqlman.version.JdbcMode;
 import io.sqlman.version.JdbcVersionManager;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
@@ -54,12 +56,16 @@ public JdbcVersionManager sqlmanJdbcVersionManager(ApplicationContext applicatio
         if (dataSource == null) {
             throw new IllegalStateException("no dataSource found in application context named: " + properties.getDataSource());
         }
+        JdbcIsolation defaultIsolation = properties.getDefaultIsolation();
+        JdbcMode defaultMode = properties.getDefaultMode();
         JdbcVersionManager jdbcVersionManager = new JdbcVersionManager(
                 dataSource,
                 scriptProvider,
                 scriptResolver,
                 dialectSupport,
-                loggerSupplier
+                loggerSupplier,
+                defaultIsolation,
+                defaultMode
         );
         jdbcVersionManager.upgrade();
         return jdbcVersionManager;
diff --git a/src/main/java/io/sqlman/spring/JdbcManagerProperties.java b/src/main/java/io/sqlman/spring/JdbcManagerProperties.java
@@ -1,5 +1,7 @@
 package io.sqlman.spring;
 
+import io.sqlman.version.JdbcIsolation;
+import io.sqlman.version.JdbcMode;
 import org.springframework.boot.context.properties.ConfigurationProperties;
 
 /**
@@ -15,11 +17,37 @@ public class JdbcManagerProperties extends AbstractManagerProperties {
      */
     private String dataSource = "dataSource";
 
+    /**
+     * default transaction isolation level
+     */
+    private JdbcIsolation defaultIsolation;
+
+    /**
+     * default mode: SAFETY or DANGER which means backup or not backup before the sql sentence being execute
+     */
+    private JdbcMode defaultMode;
+
     public String getDataSource() {
         return dataSource;
     }
 
     public void setDataSource(String dataSource) {
         this.dataSource = dataSource;
     }
+
+    public JdbcIsolation getDefaultIsolation() {
+        return defaultIsolation;
+    }
+
+    public void setDefaultIsolation(JdbcIsolation defaultIsolation) {
+        this.defaultIsolation = defaultIsolation;
+    }
+
+    public JdbcMode getDefaultMode() {
+        return defaultMode;
+    }
+
+    public void setDefaultMode(JdbcMode defaultMode) {
+        this.defaultMode = defaultMode;
+    }
 }
diff --git a/src/main/java/io/sqlman/version/JdbcInstruction.java b/src/main/java/io/sqlman/version/JdbcInstruction.java
@@ -11,6 +11,17 @@ public interface JdbcInstruction {
      * 原子性执行：即整个脚本作为一个事务整体，不可分割。
      */
     String INSTRUCTION_ATOMIC = "ATOMIC";
+
+    /**
+     * 安全模式：备份被操作的表
+     */
+    String INSTRUCTION_SAFETY = "SAFETY";
+
+    /**
+     * 危险模式：不备份被操作的表
+     */
+    String INSTRUCTION_DANGER = "DANGER";
+
     /**
      * 隔离级别：读未提交
      */
@@ -27,5 +38,4 @@ public interface JdbcInstruction {
      * 隔离级别：串行执行
      */
     String INSTRUCTION_SERIALIZABLE = "SERIALIZABLE";
-
 }
diff --git a/src/main/java/io/sqlman/version/JdbcMode.java b/src/main/java/io/sqlman/version/JdbcMode.java
@@ -0,0 +1,45 @@
+package io.sqlman.version;
+
+import java.util.Set;
+
+/**
+ * 模式
+ *
+ * @author Payne 646742615@qq.com
+ * 2019/8/2 15:07
+ */
+public enum JdbcMode {
+
+    /**
+     * 安全模式
+     */
+    SAFETY,
+    /**
+     * 危险模式
+     */
+    DANGER;
+
+    /**
+     * 从指令集中获取模式，如果没有模式指令则返回{@code null}
+     *
+     * @param instructions 指令集
+     * @return 对应模式或{@code null}当没有模式指令时
+     */
+    public static JdbcMode valueOf(Set<String> instructions) {
+        if (instructions == null || instructions.isEmpty()) {
+            return null;
+        }
+        JdbcMode mode = null;
+        for (JdbcMode value : values()) {
+            if (instructions.contains(value.name())) {
+                if (mode != null) {
+                    throw new IllegalArgumentException("multiple mode instructions");
+                } else {
+                    mode = value;
+                }
+            }
+        }
+        return mode;
+    }
+
+}
diff --git a/src/main/java/io/sqlman/version/JdbcVersionManager.java b/src/main/java/io/sqlman/version/JdbcVersionManager.java
@@ -17,6 +17,17 @@
  * 2019/5/22 16:15
  */
 public class JdbcVersionManager extends AbstractVersionManager implements SqlVersionManager, JdbcInstruction {
+
+    /**
+     * 缺省隔离界别
+     */
+    protected JdbcIsolation defaultIsolation;
+
+    /**
+     * 缺省模式
+     */
+    protected JdbcMode defaultMode = JdbcMode.DANGER;
+
     protected JdbcVersionManager() {
         super();
     }
@@ -35,6 +46,20 @@ public JdbcVersionManager(
         super(dataSource, sourceProvider, scriptResolver, dialectSupport, loggerSupplier);
     }
 
+    public JdbcVersionManager(
+            DataSource dataSource,
+            SqlSourceProvider sourceProvider,
+            SqlScriptResolver scriptResolver,
+            SqlDialectSupport dialectSupport,
+            SqlLoggerSupplier loggerSupplier,
+            JdbcIsolation defaultIsolation,
+            JdbcMode defaultMode
+    ) {
+        super(dataSource, sourceProvider, scriptResolver, dialectSupport, loggerSupplier);
+        this.defaultIsolation = defaultIsolation;
+        this.defaultMode = defaultMode;
+    }
+
     @Override
     public void upgrade() throws SQLException {
         SqlLogger logger = logger(this.getClass());
@@ -131,14 +156,35 @@ public Integer execute(Connection connection) throws SQLException {
                     try {
                         Set<String> instructions = script.instructions();
                         JdbcIsolation isolation = JdbcIsolation.valueOf(instructions);
+                        if (isolation == null) {
+                            isolation = defaultIsolation;
+                        }
                         if (isolation != null && connection.getTransactionIsolation() != isolation.level) {
                             connection.setTransactionIsolation(isolation.level);
                         }
 
+                        JdbcMode mode = JdbcMode.valueOf(instructions);
+                        if (mode == null) {
+                            mode = defaultMode;
+                        }
+                        if (mode == null) {
+                            mode = JdbcMode.DANGER;
+                        }
+                        if (mode == JdbcMode.SAFETY) {
+                            dialectSupport.backup(connection, script, ordinal);
+                        }
+
                         SqlSentence sentence = script.sentence(ordinal);
                         String sql = sentence.value();
 
-                        logger.info("Executing sentence {}/{} of script version {} under {} transaction isolation level : {}", ordinal, script.sqls(), script.version(), isolation, sql.replaceAll("\\s+", " "));
+                        logger.info(
+                                "Executing sentence {}/{} of script version {} under {} transaction isolation level : {}",
+                                ordinal,
+                                script.sqls(),
+                                script.version(),
+                                isolation != null ? isolation : "default",
+                                sql.replaceAll("\\s+", " ")
+                        );
 
                         PreparedStatement statement = connection.prepareStatement(sql);
                         int rows = statement.executeUpdate();
@@ -174,4 +220,19 @@ public Integer execute(Connection connection) throws SQLException {
         }
     }
 
+    public JdbcIsolation getDefaultIsolation() {
+        return defaultIsolation;
+    }
+
+    public void setDefaultIsolation(JdbcIsolation defaultIsolation) {
+        this.defaultIsolation = defaultIsolation;
+    }
+
+    public JdbcMode getDefaultMode() {
+        return defaultMode;
+    }
+
+    public void setDefaultMode(JdbcMode defaultMode) {
+        this.defaultMode = defaultMode;
+    }
 }
