opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-29 02:32:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

export info

Browse Source
This commit is contained in:
SwallowGG
2024-06-26 22:54:02 +08:00
parent 9a6e89dfb4
commit 0152ec4ead
2 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -14,11 +14,13 @@ import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.Date;
import static cn.hutool.core.date.DatePattern.NORM_DATETIME_PATTERN;
@Slf4j
public class MysqlDBManage extends DefaultDBManage implements DBManage {
@Override
public void exportDatabase(Connection connection, String databaseName, String schemaName, AsyncContext asyncContext) throws SQLException {
asyncContext.write(String.format(EXPORT_TITLE, DateUtil.formatDate(new Date())));
asyncContext.write(String.format(EXPORT_TITLE, DateUtil.format(new Date(),NORM_DATETIME_PATTERN)));
exportTables(connection, databaseName, schemaName, asyncContext);
asyncContext.setProgress(50);
exportViews(connection, databaseName, asyncContext);
@ -54,14 +56,14 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
}
private void exportTables(Connection connection, String databaseName, String schemaName, AsyncContext asyncContext) throws SQLException {
asyncContext.write("SET FOREIGN_KEY_CHECKS=0;\n");
asyncContext.write("SET FOREIGN_KEY_CHECKS=0;");
try (ResultSet resultSet = connection.getMetaData().getTables(databaseName, null, null, new String[]{"TABLE", "SYSTEM TABLE"})) {
while (resultSet.next()) {
String tableName = resultSet.getString("TABLE_NAME");
exportTable(connection, databaseName, schemaName, tableName, asyncContext);
}
}
asyncContext.write("SET FOREIGN_KEY_CHECKS=1;\n");
asyncContext.write("SET FOREIGN_KEY_CHECKS=1;");
}
@ -70,7 +72,7 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage {
try (ResultSet resultSet = connection.createStatement().executeQuery(sql)) {
if (resultSet.next()) {
StringBuilder sqlBuilder = new StringBuilder();
asyncContext.write(String.format(EXPORT_TITLE, tableName));
asyncContext.write(String.format(TABLE_TITLE, tableName));
sqlBuilder.append("DROP TABLE IF EXISTS ").append(format(tableName)).append(";").append("\n")
.append(resultSet.getString("Create Table")).append(";").append("\n");
asyncContext.write(sqlBuilder.toString());