opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-08-02 21:50:43 +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());

6
chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/model/AsyncContext.java
View File

@ -2,14 +2,18 @@ package ai.chat2db.spi.model;
import ai.chat2db.server.tools.common.model.Context;
import ai.chat2db.server.tools.common.util.ContextUtils;
import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import lombok.extern.slf4j.Slf4j;
import java.io.File;
import java.io.PrintWriter;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import static cn.hutool.core.date.DatePattern.NORM_DATETIME_PATTERN;
@Slf4j
public class AsyncContext {
@ -35,6 +39,7 @@ public class AsyncContext {
this.containsData = containsData;
createWriter();
asyncCallBack(context);
info("start:"+ DateUtil.format(new Date(),NORM_DATETIME_PATTERN));
}
private void createWriter() {
@ -108,6 +113,7 @@ public class AsyncContext {
public void finish() {
finish = true;
this.progress = 100;
info("finish:"+ DateUtil.format(new Date(),NORM_DATETIME_PATTERN));
if (writer != null) {
writer.flush();
writer.close();