add

chat2db-server/chat2db-plugins/chat2db-clickhouse/src/main/java/ai/chat2db/plugin/clickhouse/ClickHouseDBManage.java

@@ -1,47 +1,18 @@
 package ai.chat2db.plugin.clickhouse;
 
+import java.sql.Connection;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class ClickHouseDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
-    }
+public class ClickHouseDBManage extends DefaultDBManage implements DBManage {
+
 
     @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "DROP TABLE IF EXISTS " + databaseName + "." + tableName;
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection, sql, resultSet -> null);
     }
 
 

chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/DB2DBManage.java

@@ -1,47 +1,16 @@
 package ai.chat2db.plugin.db2;
 
+import java.sql.Connection;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class DB2DBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
-
-    }
+public class DB2DBManage extends DefaultDBManage implements DBManage {
 
     @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "DROP TABLE " + tableName;
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
     }
 }

chat2db-server/chat2db-plugins/chat2db-db2/src/main/java/ai/chat2db/plugin/db2/DB2MetaData.java

@@ -1,11 +1,12 @@
 package ai.chat2db.plugin.db2;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-import java.sql.SQLException;
-
 public class DB2MetaData extends DefaultMetaService implements MetaData {
     private String functionSQL
             = "CREATE FUNCTION tableSchema.ufn_GetCreateTableScript( @schema_name NVARCHAR(128), @table_name NVARCHAR"
@@ -37,17 +38,17 @@ public class DB2MetaData extends DefaultMetaService implements MetaData {
 
 
     @Override
-    public String tableDDL(String databaseName, String schemaName, String tableName) {
+    public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
         try {
             System.out.println(functionSQL);
-            SQLExecutor.getInstance().executeSql(functionSQL.replace("tableSchema", schemaName), resultSet -> null);
+            SQLExecutor.getInstance().executeSql(connection, functionSQL.replace("tableSchema", schemaName), resultSet -> null);
         } catch (Exception e) {
             //log.error("创建函数失败", e);
         }
 
         String ddlSql = "SELECT " + schemaName + ".ufn_GetCreateTableScript('" + schemaName + "', '" + tableName
                 + "') AS sql";
-        return SQLExecutor.getInstance().executeSql(ddlSql, resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection, ddlSql, resultSet -> {
             try {
                 if (resultSet.next()) {
                     return resultSet.getString("sql");

chat2db-server/chat2db-plugins/chat2db-dm/src/main/java/ai/chat2db/plugin/dm/DMDBManage.java

@@ -1,47 +1,17 @@
 package ai.chat2db.plugin.dm;
 
+import java.sql.Connection;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class DMDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class DMDBManage extends DefaultDBManage implements DBManage {
 
-    }
 
     @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "DROP TABLE IF EXISTS " +tableName;
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
     }
 }

chat2db-server/chat2db-plugins/chat2db-dm/src/main/java/ai/chat2db/plugin/dm/DMMetaData.java

@@ -1,19 +1,20 @@
 package ai.chat2db.plugin.dm;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.sql.SQLExecutor;
 import ai.chat2db.spi.util.SqlUtils;
 
-import java.sql.SQLException;
-
 public class DMMetaData extends DefaultMetaService implements MetaData {
-    public String tableDDL(String databaseName, String schemaName, String tableName) {
+    public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
         String selectObjectDDLSQL = String.format(
                 "select dbms_metadata.get_ddl(%s, %s, %s) AS \"sql\" from dual",
                 SqlUtils.formatSQLString("TABLE"), SqlUtils.formatSQLString(tableName),
                 SqlUtils.formatSQLString(schemaName));
-        return SQLExecutor.getInstance().executeSql(selectObjectDDLSQL, resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection,selectObjectDDLSQL, resultSet -> {
             try {
                 if (resultSet.next()) {
                     return resultSet.getString("sql");

chat2db-server/chat2db-plugins/chat2db-h2/src/main/java/ai/chat2db/plugin/h2/H2DBManage.java

@@ -1,47 +1,17 @@
 package ai.chat2db.plugin.h2;
 
+import java.sql.Connection;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class H2DBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class H2DBManage extends DefaultDBManage implements DBManage {
 
-    }
 
     @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "DROP TABLE " +tableName;
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
     }
 }

chat2db-server/chat2db-plugins/chat2db-h2/src/main/java/ai/chat2db/plugin/h2/H2Meta.java

@@ -1,10 +1,5 @@
 package ai.chat2db.plugin.h2;
 
-import ai.chat2db.spi.MetaData;
-import ai.chat2db.spi.jdbc.DefaultMetaService;
-import ai.chat2db.spi.sql.SQLExecutor;
-import jakarta.validation.constraints.NotEmpty;
-
 import java.sql.Connection;
 import java.sql.ResultSet;
 import java.sql.ResultSetMetaData;
@@ -13,16 +8,19 @@ import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
+import ai.chat2db.spi.MetaData;
+import ai.chat2db.spi.jdbc.DefaultMetaService;
+import jakarta.validation.constraints.NotEmpty;
+
 public class H2Meta extends DefaultMetaService implements MetaData {
     @Override
-    public String tableDDL(@NotEmpty String databaseName, String schemaName, @NotEmpty String tableName) {
-        return getDDL(databaseName, schemaName, tableName);
+    public String tableDDL(Connection connection, @NotEmpty String databaseName, String schemaName, @NotEmpty String tableName) {
+        return getDDL(connection,databaseName, schemaName, tableName);
     }
 
 
-    private String getDDL(String databaseName, String schemaName, String tableName) {
+    private String getDDL(Connection connection, String databaseName, String schemaName, String tableName) {
         try {
-            Connection connection = SQLExecutor.getInstance().getConnection();
             // 查询表结构信息
             ResultSet columns = connection.getMetaData().getColumns(databaseName, schemaName, tableName, null);
             List<String> columnDefinitions = new ArrayList<>();

chat2db-server/chat2db-plugins/chat2db-hive/src/main/java/ai/chat2db/plugin/hive/HiveDBManage.java

@@ -1,47 +1,17 @@
 package ai.chat2db.plugin.hive;
 
+import java.sql.Connection;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class HiveDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class HiveDBManage extends DefaultDBManage implements DBManage {
 
-    }
 
     @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "drop table if exists " +tableName;
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
     }
 }

chat2db-server/chat2db-plugins/chat2db-kingbase/src/main/java/ai/chat2db/plugin/kingbase/KingBaseDBManage.java

@@ -1,47 +1,16 @@
 package ai.chat2db.plugin.kingbase;
 
+import java.sql.Connection;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class KingBaseDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
-
-    }
+public class KingBaseDBManage extends DefaultDBManage implements DBManage {
 
     @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "drop table if exists " +tableName;
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
     }
 }

chat2db-server/chat2db-plugins/chat2db-mariadb/src/main/java/ai/chat2db/plugin/mariadb/MariaDBManage.java

@@ -1,47 +1,17 @@
 package ai.chat2db.plugin.mariadb;
 
+import java.sql.Connection;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class MariaDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class MariaDBManage extends DefaultDBManage implements DBManage {
 
-    }
 
     @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "DROP TABLE " +tableName ;
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
     }
 }

chat2db-server/chat2db-plugins/chat2db-mongodb/src/main/java/ai/chat2db/plugin/mongodb/MongodbManage.java

@@ -1,45 +1,8 @@
 package ai.chat2db.plugin.mongodb;
 
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 
-public class MongodbManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class MongodbManage extends DefaultDBManage implements DBManage {
 
-    }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
 }

chat2db-server/chat2db-plugins/chat2db-mongodb/src/main/java/ai/chat2db/plugin/mongodb/MongodbMetaData.java

@@ -3,6 +3,7 @@ package ai.chat2db.plugin.mongodb;
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 
+
 public class MongodbMetaData extends DefaultMetaService implements MetaData {
 
 }

chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java

@@ -1,58 +1,32 @@
 package ai.chat2db.plugin.mysql;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 import org.springframework.util.StringUtils;
 
-import java.sql.SQLException;
-
-public class MysqlDBManage implements DBManage {
+public class MysqlDBManage extends DefaultDBManage implements DBManage {
     @Override
-    public void connectDatabase(String database) {
+    public void connectDatabase(Connection connection, String database) {
         if (StringUtils.isEmpty(database)) {
             return;
         }
         try {
-            SQLExecutor.getInstance().execute("use `" + database + "`;");
+            SQLExecutor.getInstance().execute(connection,"use `" + database + "`;");
         } catch (SQLException e) {
             throw new RuntimeException(e);
         }
     }
 
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
 
-    }
 
     @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
+    public void dropTable(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "DROP TABLE "+ format(tableName);
-        SQLExecutor.getInstance().executeSql(sql, resultSet -> null);
+        SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> null);
     }
 
     public static String format(String tableName) {

chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlMetaData.java

@@ -1,18 +1,19 @@
 package ai.chat2db.plugin.mysql;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.sql.SQLExecutor;
 import jakarta.validation.constraints.NotEmpty;
 
-import java.sql.SQLException;
-
 public class MysqlMetaData extends DefaultMetaService implements MetaData {
     @Override
-    public String tableDDL(@NotEmpty String databaseName, String schemaName, @NotEmpty String tableName) {
+    public String tableDDL(Connection connection, @NotEmpty String databaseName, String schemaName, @NotEmpty String tableName) {
         String sql = "SHOW CREATE TABLE " + format(databaseName) + "."
                 + format( tableName);
-        return SQLExecutor.getInstance().executeSql(sql, resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> {
             try {
                 if (resultSet.next()) {
                     return resultSet.getString("Create Table");

chat2db-server/chat2db-plugins/chat2db-oceanbase/src/main/java/ai/chat2db/plugin/oceanbase/OceanBaseDBManage.java

@@ -1,45 +1,8 @@
 package ai.chat2db.plugin.oceanbase;
 
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 
-public class OceanBaseDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class OceanBaseDBManage extends DefaultDBManage implements DBManage {
 
-    }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
 }

chat2db-server/chat2db-plugins/chat2db-oracle/src/main/java/ai/chat2db/plugin/oracle/OracleDBManage.java

@@ -1,45 +1,8 @@
 package ai.chat2db.plugin.oracle;
 
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 
-public class OracleDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class OracleDBManage extends DefaultDBManage implements DBManage {
 
-    }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
 }

chat2db-server/chat2db-plugins/chat2db-oracle/src/main/java/ai/chat2db/plugin/oracle/OracleMetaData.java

@@ -1,17 +1,18 @@
 package ai.chat2db.plugin.oracle;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-import java.sql.SQLException;
-
 public class OracleMetaData extends DefaultMetaService implements MetaData {
     @Override
-    public String tableDDL(String databaseName, String schemaName, String tableName) {
+    public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "select dbms_metadata.get_ddl('TABLE','"+tableName+"') as sql from dual,"
                 + "user_tables where table_name = '" + tableName + "'";
-        return SQLExecutor.getInstance().executeSql(sql, resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> {
             try {
                 if (resultSet.next()) {
                     return resultSet.getString("sql");

chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLDBManage.java

@@ -1,54 +1,19 @@
 package ai.chat2db.plugin.postgresql;
 
-import ai.chat2db.spi.DBManage;
-import ai.chat2db.spi.sql.Chat2DBContext;
-import ai.chat2db.spi.sql.ConnectInfo;
-import ai.chat2db.spi.sql.SQLExecutor;
-
+import java.sql.Connection;
 import java.sql.SQLException;
 
-public class PostgreSQLDBManage implements DBManage {
+import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
+import ai.chat2db.spi.sql.SQLExecutor;
+
+public class PostgreSQLDBManage extends DefaultDBManage implements DBManage {
     @Override
-    public void connectDatabase(String database) {
+    public void connectDatabase(Connection connection, String database) {
         try {
-            SQLExecutor.getInstance().execute("SELECT pg_database_size('"+database+"');");
+            SQLExecutor.getInstance().execute(connection,"SELECT pg_database_size('"+database+"');");
         } catch (SQLException e) {
             throw new RuntimeException(e);
         }
     }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
 }

chat2db-server/chat2db-plugins/chat2db-postgresql/src/main/java/ai/chat2db/plugin/postgresql/PostgreSQLMetaData.java

@@ -1,11 +1,12 @@
 package ai.chat2db.plugin.postgresql;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-import java.sql.SQLException;
-
 public class PostgreSQLMetaData extends DefaultMetaService implements MetaData {
     private String functionSQL =
             " CREATE OR REPLACE FUNCTION showcreatetable(namespace character varying, tablename character "
@@ -163,10 +164,10 @@ public class PostgreSQLMetaData extends DefaultMetaService implements MetaData {
                     + "        $BODY$ LANGUAGE plpgsql";
 
     @Override
-    public String tableDDL(String databaseName, String schemaName, String tableName) {
-        SQLExecutor.getInstance().executeSql(functionSQL.replaceFirst("tableSchema", schemaName), resultSet -> null);
+    public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
+        SQLExecutor.getInstance().executeSql(connection,functionSQL.replaceFirst("tableSchema", schemaName), resultSet -> null);
         String ddlSql = "select showcreatetable('" + schemaName + "','" + tableName + "') as sql";
-        return SQLExecutor.getInstance().executeSql(ddlSql, resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection,ddlSql, resultSet -> {
             try {
                 if (resultSet.next()) {
                     return resultSet.getString("sql");

chat2db-server/chat2db-plugins/chat2db-presto/src/main/java/ai/chat2db/plugin/presto/PrestoDBManage.java

@@ -1,45 +1,7 @@
 package ai.chat2db.plugin.presto;
 
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 
-public class PrestoDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
-
-    }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
+public class PrestoDBManage extends DefaultDBManage implements DBManage {
 }

chat2db-server/chat2db-plugins/chat2db-redis/src/main/java/ai/chat2db/plugin/redis/RedisDBManage.java

@@ -1,45 +1,8 @@
 package ai.chat2db.plugin.redis;
 
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 
-public class RedisDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
+public class RedisDBManage extends DefaultDBManage implements DBManage {
 
-    }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
 }

chat2db-server/chat2db-plugins/chat2db-redis/src/main/java/ai/chat2db/plugin/redis/RedisMetaData.java

@@ -1,28 +1,23 @@
 package ai.chat2db.plugin.redis;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.List;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.model.Database;
 import ai.chat2db.spi.model.Table;
 import ai.chat2db.spi.sql.SQLExecutor;
-import jakarta.validation.constraints.NotEmpty;
 import org.apache.commons.lang3.StringUtils;
 
-import java.sql.SQLException;
-import java.util.ArrayList;
-import java.util.List;
-
 public class RedisMetaData extends DefaultMetaService implements MetaData {
-    @Override
-    public String tableDDL(@NotEmpty String databaseName, String schemaName, @NotEmpty String tableName) {
-        return "";
-    }
-
 
     @Override
-    public List<Database> databases() {
+    public List<Database> databases(Connection connection) {
         List<Database> databases = new ArrayList<>();
-        return SQLExecutor.getInstance().executeSql("config get databases", resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection,"config get databases", resultSet -> {
             try {
                 if (resultSet.next()) {
                     Object count = resultSet.getObject(2);
@@ -41,8 +36,8 @@ public class RedisMetaData extends DefaultMetaService implements MetaData {
     }
 
     @Override
-    public List<Table> tables(String databaseName, String schemaName, String tableName) {
-        return SQLExecutor.getInstance().executeSql("scan 0 MATCH * COUNT 1000", resultSet -> {
+    public List<Table> tables(Connection connection, String databaseName, String schemaName, String tableName) {
+        return SQLExecutor.getInstance().executeSql(connection,"scan 0 MATCH * COUNT 1000", resultSet -> {
             List<Table> tables = new ArrayList<>();
             try {
                 while (resultSet.next()) {

chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteDBManage.java

@@ -1,45 +1,7 @@
 package ai.chat2db.plugin.sqlite;
 
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 
-public class SqliteDBManage implements DBManage {
-    @Override
-    public void connectDatabase(String database) {
-
-    }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
+public class SqliteDBManage extends DefaultDBManage implements DBManage {
 }

chat2db-server/chat2db-plugins/chat2db-sqlite/src/main/java/ai/chat2db/plugin/sqlite/SqliteMetaData.java

@@ -1,5 +1,9 @@
 package ai.chat2db.plugin.sqlite;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.List;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.model.Database;
@@ -7,14 +11,11 @@ import ai.chat2db.spi.model.Schema;
 import ai.chat2db.spi.sql.SQLExecutor;
 import com.google.common.collect.Lists;
 
-import java.sql.SQLException;
-import java.util.List;
-
 public class SqliteMetaData extends DefaultMetaService implements MetaData {
     @Override
-    public String tableDDL(String databaseName, String schemaName, String tableName) {
+    public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
         String sql = "SELECT sql FROM sqlite_master WHERE type='table' AND name='" + tableName + "'";
-        return SQLExecutor.getInstance().executeSql(sql, resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection,sql, resultSet -> {
             try {
                 if (resultSet.next()) {
                     return resultSet.getString("sql");
@@ -26,12 +27,12 @@ public class SqliteMetaData extends DefaultMetaService implements MetaData {
         });
     }
     @Override
-    public List<Database> databases() {
+    public List<Database> databases(Connection connection) {
         return Lists.newArrayList(Database.builder().name("main").build());
     }
 
     @Override
-    public List<Schema> schemas(String databaseName) {
+    public List<Schema> schemas(Connection connection,String databaseName) {
         return Lists.newArrayList();
     }
 }

chat2db-server/chat2db-plugins/chat2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/SqlServerDBManage.java

@@ -1,52 +1,19 @@
 package ai.chat2db.plugin.sqlserver;
 
+import java.sql.Connection;
 import java.sql.SQLException;
 
 import ai.chat2db.spi.DBManage;
+import ai.chat2db.spi.jdbc.DefaultDBManage;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-public class SqlServerDBManage implements DBManage {
+public class SqlServerDBManage extends DefaultDBManage implements DBManage {
     @Override
-    public void connectDatabase(String database) {
+    public void connectDatabase(Connection connection, String database) {
         try {
-            SQLExecutor.getInstance().execute("use [" + database + "];");
+            SQLExecutor.getInstance().execute(connection,"use [" + database + "];");
         } catch (SQLException e) {
             throw new RuntimeException(e);
         }
     }
-
-    @Override
-    public void modifyDatabase(String databaseName, String newDatabaseName) {
-
-    }
-
-    @Override
-    public void createDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void dropDatabase(String databaseName) {
-
-    }
-
-    @Override
-    public void createSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void dropSchema(String databaseName, String schemaName) {
-
-    }
-
-    @Override
-    public void modifySchema(String databaseName, String schemaName, String newSchemaName) {
-
-    }
-
-    @Override
-    public void dropTable(String databaseName, String schemaName, String tableName) {
-
-    }
 }

chat2db-server/chat2db-plugins/chat2db-sqlserver/src/main/java/ai/chat2db/plugin/sqlserver/SqlServerMetaData.java

@@ -1,11 +1,12 @@
 package ai.chat2db.plugin.sqlserver;
 
+import java.sql.Connection;
+import java.sql.SQLException;
+
 import ai.chat2db.spi.MetaData;
 import ai.chat2db.spi.jdbc.DefaultMetaService;
 import ai.chat2db.spi.sql.SQLExecutor;
 
-import java.sql.SQLException;
-
 public class SqlServerMetaData extends DefaultMetaService implements MetaData {
     private String functionSQL
             = "CREATE FUNCTION tableSchema.ufn_GetCreateTableScript( @schema_name NVARCHAR(128), @table_name NVARCHAR"
@@ -37,17 +38,17 @@ public class SqlServerMetaData extends DefaultMetaService implements MetaData {
     
 
     @Override
-    public String tableDDL(String databaseName, String schemaName, String tableName) {
+    public String tableDDL(Connection connection, String databaseName, String schemaName, String tableName) {
         try {
             System.out.println(functionSQL);
-            SQLExecutor.getInstance().executeSql(functionSQL.replace("tableSchema", schemaName), resultSet -> null);
+            SQLExecutor.getInstance().executeSql(connection,functionSQL.replace("tableSchema", schemaName), resultSet -> null);
         } catch (Exception e) {
             //log.error("创建函数失败", e);
         }
 
         String ddlSql = "SELECT " + schemaName + ".ufn_GetCreateTableScript('" + schemaName + "', '" + tableName
                 + "') AS sql";
-        return SQLExecutor.getInstance().executeSql(ddlSql, resultSet -> {
+        return SQLExecutor.getInstance().executeSql(connection,ddlSql, resultSet -> {
             try {
                 if (resultSet.next()) {
                     return resultSet.getString("sql");