opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-28 10:12:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix-exportTableData

Browse Source
This commit is contained in:
zgq
2024-06-21 18:29:12 +08:00
parent 9421c3d3f4
commit 2bfbbaeb87
9 changed files with 127 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -20,13 +20,13 @@ public class SqliteDBManage extends DefaultDBManage implements DBManage {
private void exportTables(Connection connection, String databaseName, String schemaName, AsyncContext asyncContext) throws SQLException {
try (ResultSet resultSet = connection.getMetaData().getTables(databaseName, null, null, new String[]{"TABLE", "SYSTEM TABLE"})) {
while (resultSet.next()) {
exportTable(connection,schemaName, resultSet.getString("TABLE_NAME"), asyncContext);
exportTable(connection, databaseName,schemaName, resultSet.getString("TABLE_NAME"), asyncContext);
}
}
}
private void exportTable(Connection connection, String schemaName, String tableName, AsyncContext asyncContext) throws SQLException {
private void exportTable(Connection connection, String databaseName, String schemaName, String tableName, AsyncContext asyncContext) throws SQLException {
String sql = String.format("SELECT sql FROM sqlite_master WHERE type='table' AND name='%s'", tableName);
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
if (resultSet.next()) {
@ -35,7 +35,7 @@ public class SqliteDBManage extends DefaultDBManage implements DBManage {
.append(resultSet.getString("sql")).append(";").append("\n");
asyncContext.write(sqlBuilder.toString());
if (asyncContext.isContainsData()) {
exportTableData(connection,schemaName, tableName, asyncContext);
exportTableData(connection, databaseName,schemaName, tableName, asyncContext);
}
}
}