diff --git a/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java b/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java index abdf1583..56566a0a 100644 --- a/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java +++ b/chat2db-server/chat2db-plugins/chat2db-mysql/src/main/java/ai/chat2db/plugin/mysql/MysqlDBManage.java @@ -48,8 +48,10 @@ public class MysqlDBManage extends DefaultDBManage implements DBManage { if (resultSet.next()) { asyncContext.write(String.format(FUNCTION_TITLE, functionName)); StringBuilder sqlBuilder = new StringBuilder(); - sqlBuilder.append("DROP FUNCTION IF EXISTS ").append(functionName).append(";").append("\n") - .append(resultSet.getString("Create Function")).append(";").append("\n"); + sqlBuilder.append("DROP FUNCTION IF EXISTS ").append(functionName).append(";").append("\n"); + + sqlBuilder.append("delimiter ;;").append("\n").append(resultSet.getString("Create Function")).append(";;") + .append("\n").append("delimiter ;").append("\n\n"); asyncContext.write(sqlBuilder.toString()); } } diff --git a/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/model/AsyncContext.java b/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/model/AsyncContext.java index 9c68430e..210a7c34 100644 --- a/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/model/AsyncContext.java +++ b/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/model/AsyncContext.java @@ -39,7 +39,7 @@ public class AsyncContext { this.containsData = containsData; createWriter(); asyncCallBack(context); - info("start:"+ DateUtil.format(new Date(),NORM_DATETIME_PATTERN)); + info("start:" + DateUtil.format(new Date(), NORM_DATETIME_PATTERN)); } private void createWriter() { @@ -57,8 +57,8 @@ public class AsyncContext { int n = 1; while (!finish) { // 更新时间逐渐变长避免频繁更新 - Thread.sleep(2000 * n); callUpdate(); + Thread.sleep(2000 * n); if (n < 300) { n++; } @@ -110,10 +110,17 @@ public class AsyncContext { info.append(message + "\n"); } + public void stop(){ + this.finish = true; + } + public void finish() { finish = true; this.progress = 100; - info("finish:"+ DateUtil.format(new Date(),NORM_DATETIME_PATTERN)); + info("finish:" + DateUtil.format(new Date(), NORM_DATETIME_PATTERN)); + if (writeFile != null) { + info("file:" + writeFile.getAbsolutePath()); + } if (writer != null) { writer.flush(); writer.close();