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

When the sql execution is abnormal, the error is reported separately

Browse Source
This commit is contained in:
JiaJu Zhuang
2023-07-03 21:32:28 +08:00
parent 8201cf375f
commit 4ecdbc87b2
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/impl/DlTemplateServiceImpl.java
View File

@ -11,6 +11,7 @@ import com.alibaba.druid.sql.PagerUtils;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
import com.alibaba.druid.sql.parser.ParserException;
import com.alibaba.druid.sql.parser.SQLParserUtils;
import ai.chat2db.server.domain.api.param.DlCountParam;
@ -67,7 +68,19 @@ public class DlTemplateServiceImpl implements DlTemplateService {
String sqlType = SqlTypeEnum.UNKNOWN.getCode();
// 解析sql分页
SQLStatement sqlStatement = SQLUtils.parseSingleStatement(sql, dbType);
SQLStatement sqlStatement;
try {
sqlStatement = SQLUtils.parseSingleStatement(sql, dbType);
} catch (ParserException e) {
log.warn("解析sql失败:{}", sql, e);
ExecuteResult executeResult = ExecuteResult.builder()
.success(Boolean.FALSE)
.sql(sql)
.message(e.getMessage())
.build();
result.add(executeResult);
continue;
}
// 是否需要代码帮忙分页
boolean autoLimit = false;
if (sqlStatement instanceof SQLSelectStatement) {