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

support select result data update

Browse Source
This commit is contained in:
SwallowGG
2023-10-11 15:37:11 +08:00
parent 2ef8e21e39
commit 99d90e76ba
1 changed files with 4 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlController.java
View File

@ -65,38 +65,11 @@ public class RdbDmlController {
@RequestMapping(value = "/execute_update", method = {RequestMethod.POST, RequestMethod.PUT}) @RequestMapping(value = "/execute_update", method = {RequestMethod.POST, RequestMethod.PUT})
public DataResult<ExecuteResultVO> executeSelectResultUpdate(@RequestBody DmlRequest request) { public DataResult<ExecuteResultVO> executeSelectResultUpdate(@RequestBody DmlRequest request) {
DlExecuteParam param = rdbWebConverter.request2param(request); DlExecuteParam param = rdbWebConverter.request2param(request);
Connection connection = Chat2DBContext.getConnection(); DataResult<ExecuteResult> result = dlTemplateService.executeUpdate(param);
if (connection != null) { if(!result.success()){
try { return DataResult.error(result.getErrorCode(),result.getErrorMessage());
boolean flag = true;
ExecuteResultVO executeResult = null;
connection.setAutoCommit(false);
ListResult<ExecuteResult> resultDTOListResult = dlTemplateService.execute(param);
List<ExecuteResultVO> resultVOS = rdbWebConverter.dto2vo(resultDTOListResult.getData());
if (!CollectionUtils.isEmpty(resultVOS)) {
for (ExecuteResultVO resultVO : resultVOS) {
if (!resultVO.getSuccess()) {
flag = false;
executeResult = resultVO;
break;
}
}
}
if (flag) {
connection.commit();
return DataResult.of(resultVOS.get(0));
}else {
connection.rollback();
return DataResult.of(executeResult);
}
} catch (Exception e) {
throw new RuntimeException(e);
}
} else {
return DataResult.error("connection error", "");
} }
return DataResult.of(rdbWebConverter.dto2vo(result.getData()));
} }
@RequestMapping(value = "/get_update_sql", method = {RequestMethod.POST, RequestMethod.PUT}) @RequestMapping(value = "/get_update_sql", method = {RequestMethod.POST, RequestMethod.PUT})