From 99d90e76bae1ca8af099002ffaa3adbed3eaaeba Mon Sep 17 00:00:00 2001
From: SwallowGG <1558143046@qq.com>
Date: Wed, 11 Oct 2023 15:37:11 +0800
Subject: [PATCH] support select result data update

---
 .../api/controller/rdb/RdbDmlController.java  | 35 +++----------------
 1 file changed, 4 insertions(+), 31 deletions(-)

diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlController.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlController.java
index 59213c2b..1d59afef 100644
--- a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlController.java
+++ b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/RdbDmlController.java
@@ -65,38 +65,11 @@ public class RdbDmlController {
     @RequestMapping(value = "/execute_update", method = {RequestMethod.POST, RequestMethod.PUT})
     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", "");
+        DataResult<ExecuteResult>  result = dlTemplateService.executeUpdate(param);
+        if(!result.success()){
+            return DataResult.error(result.getErrorCode(),result.getErrorMessage());
         }
+       return DataResult.of(rdbWebConverter.dto2vo(result.getData()));
 
     }
     @RequestMapping(value = "/get_update_sql", method = {RequestMethod.POST, RequestMethod.PUT})