add DBTableSupportProvider

Author: xlorne

README.md

@@ -145,6 +145,13 @@ DBStreamContext.getInstance().addEventPusher(DBEventPusher pusher);
 DBStreamContext.getInstance().addListener(SQLExecuteListener listener);
 ```
 
+#### 数据库表支持判断
+
+```java
+// 添加 SQL 表执行判断
+DBStreamContext.getInstance().setDbTableSupportProvider(DBTableSupportProvider dbTableSupportProvider);
+```
+
 #### 元数据管理
 
 ```java

pom.xml

@@ -4,7 +4,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>com.codingapi.dbstream</groupId>
     <artifactId>dbstream-driver</artifactId>
-    <version>1.0.0</version>
+    <version>1.0.1</version>
 
     <url>https://github.com/codingapi/dbstream-driver</url>
     <name>dbstream-driver</name>

src/main/java/com/codingapi/dbstream/DBStreamContext.java

@@ -2,13 +2,17 @@
 
 import com.codingapi.dbstream.interceptor.SQLRunningContext;
 import com.codingapi.dbstream.listener.SQLExecuteListener;
+import com.codingapi.dbstream.provider.DBTableSupportProvider;
+import com.codingapi.dbstream.provider.DefaultDBTableSupportProvider;
 import com.codingapi.dbstream.scanner.DBMetaContext;
 import com.codingapi.dbstream.scanner.DBMetaData;
 import com.codingapi.dbstream.stream.DBEventContext;
 import com.codingapi.dbstream.stream.DBEventPusher;
 import lombok.Getter;
+import lombok.Setter;
 
 import java.util.List;
+import java.util.Properties;
 
 /**
  * DBStream 对外提供的能力服务
@@ -18,6 +22,10 @@ public class DBStreamContext {
     @Getter
     private final static DBStreamContext instance = new DBStreamContext();
 
+    @Setter
+    private DBTableSupportProvider dbTableSupportProvider;
+
+
     private DBStreamContext() {
 
     }
@@ -31,6 +39,19 @@ public void addListener(SQLExecuteListener sqlExecuteListener) {
         SQLRunningContext.getInstance().addListener(sqlExecuteListener);
     }
 
+    /**
+     * 是否支持SQL 拦截代理分析
+     * @param info 数据库连接信息
+     * @param tableName 数据库名称
+     * @return 是否支持
+     */
+    public boolean support(Properties info, String tableName) {
+        if (dbTableSupportProvider == null) {
+            this.dbTableSupportProvider = new DefaultDBTableSupportProvider();
+        }
+        return dbTableSupportProvider.support(info, tableName);
+    }
+
 
     /**
      * 添加DB事件推送

src/main/java/com/codingapi/dbstream/interceptor/SQLExecuteState.java

@@ -8,10 +8,7 @@
 import lombok.Setter;
 
 import java.sql.*;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
 /**
  * 数据库的执行状态
@@ -218,6 +215,14 @@ public List<Map<String, Object>> getStatementGenerateKeys(DbTable dbTable) {
     }
 
 
+    public Properties getDriverProperties(){
+        if (metaData == null) {
+            return null;
+        }
+        return this.metaData.getProperties();
+    }
+
+
     public String getJdbcUrl() {
         if (metaData == null) {
             return null;

src/main/java/com/codingapi/dbstream/listener/SQLDeleteExecuteListener.java

@@ -1,5 +1,6 @@
 package com.codingapi.dbstream.listener;
 
+import com.codingapi.dbstream.DBStreamContext;
 import com.codingapi.dbstream.interceptor.SQLExecuteState;
 import com.codingapi.dbstream.parser.DeleteDBEventParser;
 import com.codingapi.dbstream.scanner.DbTable;
@@ -28,7 +29,7 @@ public void before(SQLExecuteState executeState) throws SQLException {
                 Delete delete = (Delete) parserStatement;
                 Table table = delete.getTable();
                 DbTable dbTable = executeState.getDbTable(table.getName());
-                if (dbTable != null) {
+                if (dbTable != null && DBStreamContext.getInstance().support(executeState.getDriverProperties(), dbTable.getName())) {
                     DeleteDBEventParser dataParser = new DeleteDBEventParser(executeState, delete, table, dbTable);
                     dataParser.prepare();
                     threadLocal.set(dataParser);

src/main/java/com/codingapi/dbstream/listener/SQLInsertExecuteListener.java

@@ -1,5 +1,6 @@
 package com.codingapi.dbstream.listener;
 
+import com.codingapi.dbstream.DBStreamContext;
 import com.codingapi.dbstream.interceptor.SQLExecuteState;
 import com.codingapi.dbstream.parser.InsertDBEventParser;
 import com.codingapi.dbstream.scanner.DbTable;
@@ -28,7 +29,7 @@ public void before(SQLExecuteState executeState) throws SQLException {
                 Insert insert = (Insert) parserStatement;
                 Table table = insert.getTable();
                 DbTable dbTable = executeState.getDbTable(table.getName());
-                if (dbTable != null) {
+                if (dbTable != null && DBStreamContext.getInstance().support(executeState.getDriverProperties(), dbTable.getName())) {
                     InsertDBEventParser dataParser = new InsertDBEventParser(executeState, insert, table, dbTable);
                     dataParser.prepare();
                     threadLocal.set(dataParser);

src/main/java/com/codingapi/dbstream/listener/SQLUpdateExecuteListener.java

@@ -1,5 +1,6 @@
 package com.codingapi.dbstream.listener;
 
+import com.codingapi.dbstream.DBStreamContext;
 import com.codingapi.dbstream.interceptor.SQLExecuteState;
 import com.codingapi.dbstream.parser.UpdateDBEventParser;
 import com.codingapi.dbstream.scanner.DbTable;
@@ -28,7 +29,7 @@ public void before(SQLExecuteState executeState) throws SQLException {
                 Update update = (Update) parserStatement;
                 Table table = update.getTable();
                 DbTable dbTable = executeState.getDbTable(table.getName());
-                if (dbTable != null) {
+                if (dbTable != null && DBStreamContext.getInstance().support(executeState.getDriverProperties(), dbTable.getName())) {
                     UpdateDBEventParser dataParser = new UpdateDBEventParser(executeState, update, table, dbTable);
                     dataParser.prepare();
                     threadLocal.set(dataParser);

src/main/java/com/codingapi/dbstream/parser/InsertDBEventParser.java

@@ -6,12 +6,16 @@
 import com.codingapi.dbstream.stream.DBEvent;
 import com.codingapi.dbstream.stream.EventType;
 import com.codingapi.dbstream.utils.SQLParamUtils;
+import net.sf.jsqlparser.expression.Expression;
+import net.sf.jsqlparser.expression.JdbcParameter;
+import net.sf.jsqlparser.expression.operators.relational.ExpressionList;
 import net.sf.jsqlparser.schema.Column;
 import net.sf.jsqlparser.schema.Table;
 import net.sf.jsqlparser.statement.Statement;
 import net.sf.jsqlparser.statement.insert.Insert;
 import net.sf.jsqlparser.statement.select.PlainSelect;
 import net.sf.jsqlparser.statement.select.Select;
+import net.sf.jsqlparser.statement.select.Values;
 
 import java.sql.SQLException;
 import java.util.ArrayList;
@@ -100,7 +104,7 @@ private List<DBEvent> loadDefaultInsertEvent() {
                 Object value = params.get(i);
                 String column = insertColumns.get(i);
                 DbColumn dbColumn = dbTable.getColumnByName(column);
-                if (dbColumn != null) {
+                if (dbColumn != null && !dbColumn.isPrimaryKey()) {
                     event.set(dbColumn.getName(), value);
                 }
             }
@@ -111,11 +115,17 @@ private List<DBEvent> loadDefaultInsertEvent() {
 
 
     private List<String> loadInsertColumns() {
-        List<String> columns = new ArrayList<>();
-        for (Column column : this.insert.getColumns()) {
-            columns.add(column.getColumnName());
+        ExpressionList<?> values =  this.insert.getValues().getExpressions();
+        ExpressionList<Column> columns = this.insert.getColumns();
+        List<String> columnList = new ArrayList<>();
+        for (int i=0;i<columns.size();i++) {
+            Column column = columns.get(i);
+            Expression expression = values.get(i);
+            if(expression instanceof JdbcParameter) {
+                columnList.add(column.getColumnName());
+            }
         }
-        return columns;
+        return columnList;
     }
 
 

src/main/java/com/codingapi/dbstream/provider/DBTableSupportProvider.java

@@ -0,0 +1,11 @@
+package com.codingapi.dbstream.provider;
+
+import java.util.Properties;
+
+/**
+ * 数据库表支持判断
+ */
+public interface DBTableSupportProvider {
+
+    boolean support(Properties info,String tableName);
+}

src/main/java/com/codingapi/dbstream/provider/DefaultDBTableSupportProvider.java

@@ -0,0 +1,11 @@
+package com.codingapi.dbstream.provider;
+
+import java.util.Properties;
+
+public class DefaultDBTableSupportProvider implements DBTableSupportProvider {
+
+    @Override
+    public boolean support(Properties info, String tableName) {
+        return true;
+    }
+}