opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-30 03:03:13 +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:34:04 +08:00
parent 35f15835f1
commit 283073b0e9
6 changed files with 135 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,10 +63,46 @@ public class RdbDmlController {
* @return
*/
@RequestMapping(value = "/execute_update", method = {RequestMethod.POST, RequestMethod.PUT})
public DataResult<String> executeUpdate(@RequestBody SelectResultUpdateRequest request) {
public DataResult<ExecuteResultVO> executeSelectResultUpdate(@RequestBody DmlRequest request) {
DlExecuteParam param = rdbWebConverter.request2param(request);
Connection connection = Chat2DBContext.getConnection();
if (connection != null) {
try {
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", "");
}
}
@RequestMapping(value = "/get_update_sql", method = {RequestMethod.POST, RequestMethod.PUT})
public DataResult<String> getUpdateSelectResultSql(SelectResultUpdateRequest request) {
UpdateSelectResultParam param = rdbWebConverter.request2param(request);
return dlTemplateService.updateSelectResult(param);
}