Merge pull request OtterMind#1183 from openai0229/issue#1135

Chat2DB-Pro · web-flow · commit c654faefd942 · 2024-02-26T19:35:50.000+08:00
complete exception throwing
diff --git a/chat2db-server/chat2db-server-domain/chat2db-server-domain-api/src/main/java/ai/chat2db/server/domain/api/service/ProcedureService.java b/chat2db-server/chat2db-server-domain/chat2db-server-domain-api/src/main/java/ai/chat2db/server/domain/api/service/ProcedureService.java
@@ -6,6 +6,8 @@
 import ai.chat2db.spi.model.Procedure;
 import jakarta.validation.constraints.NotEmpty;
 
+import java.sql.SQLException;
+
 public interface ProcedureService {
 
     /**
@@ -31,5 +33,5 @@ public interface ProcedureService {
      * @param procedure
      * @return
      */
-    ActionResult update(String databaseName, String schemaName, Procedure procedure);
+    ActionResult update(String databaseName, String schemaName, Procedure procedure) throws SQLException;
 }
diff --git a/chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/impl/ProcedureServiceImpl.java b/chat2db-server/chat2db-server-domain/chat2db-server-domain-core/src/main/java/ai/chat2db/server/domain/core/impl/ProcedureServiceImpl.java
@@ -8,6 +8,8 @@
 import ai.chat2db.spi.sql.Chat2DBContext;
 import org.springframework.stereotype.Service;
 
+import java.sql.SQLException;
+
 @Service
 public class ProcedureServiceImpl implements ProcedureService {
 
@@ -21,7 +23,7 @@ public DataResult<Procedure> detail(String databaseName, String schemaName, Stri
         return DataResult.of(Chat2DBContext.getMetaData().procedure(Chat2DBContext.getConnection(), databaseName, schemaName, procedureName));
     }
     @Override
-    public ActionResult update(String databaseName, String schemaName, Procedure procedure) {
+    public ActionResult update(String databaseName, String schemaName, Procedure procedure) throws SQLException {
         Chat2DBContext.getDBManage().updateProcedure(Chat2DBContext.getConnection(), databaseName, schemaName, procedure);
         return ActionResult.isSuccess();
     }
diff --git a/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/ProcedureController.java b/chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/rdb/ProcedureController.java
@@ -15,6 +15,8 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
+import java.sql.SQLException;
+
 @ConnectionInfoAspect
 @RequestMapping("/api/rdb/procedure")
 @RestController
@@ -39,7 +41,7 @@ public DataResult<Procedure> detail(@Valid ProcedureDetailRequest request) {
     }
 
     @PostMapping("/update")
-    public ActionResult update(@Valid @RequestBody ProcedureUpdateRequest request) {
+    public ActionResult update(@Valid @RequestBody ProcedureUpdateRequest request) throws SQLException {
         Procedure procedure = procedureConverter.request2param(request);
         return procedureService.update(request.getDatabaseName(), request.getSchemaName(), procedure);
     }
