From f8f07247aea5cb558803885bb15f04f0fedc65ce Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B0=91=E5=B9=B4=E6=A2=A6=E4=B8=8E=E7=A0=96?=
 <2847127106@qq.com>
Date: Fri, 6 Sep 2024 10:19:32 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=9F=A5=E8=AF=A2=E6=9D=A1?=
 =?UTF-8?q?=E6=95=B0=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/DataValueController.java       | 13 ++++++++++++
 .../muyu/source/service/DataValueService.java |  2 ++
 .../service/Impl/DataValueServiceImpl.java    | 20 +++++++++++++++++++
 3 files changed, 35 insertions(+)

diff --git a/muyu-source-server/src/main/java/com/muyu/source/controller/DataValueController.java b/muyu-source-server/src/main/java/com/muyu/source/controller/DataValueController.java
index 38cb001..b47c558 100644
--- a/muyu-source-server/src/main/java/com/muyu/source/controller/DataValueController.java
+++ b/muyu-source-server/src/main/java/com/muyu/source/controller/DataValueController.java
@@ -41,6 +41,19 @@ public class DataValueController {
         return Result.success(dataValueList);
     }
 
+    /**
+     * 根据基础表ID和SQL语句查询条数
+     *
+     * @param dataValueModel 基础表ID和sql语句
+     * @return DataValue{kltv}
+     */
+    @PostMapping("/getTableValueTotal")
+    @Operation(summary = "根据基础表ID和SQL语句新增数据", description = "根据基础表ID和SQL语句新增数据")
+    public Result getTableValueTotal(@RequestBody DataValueModel dataValueModel) {
+        Integer i = dataValueService.getTableValueTotal(dataValueModel);
+        return Result.success(i);
+    }
+
     /**
      * 根据基础表ID和SQL语句查询数据
      *
diff --git a/muyu-source-server/src/main/java/com/muyu/source/service/DataValueService.java b/muyu-source-server/src/main/java/com/muyu/source/service/DataValueService.java
index 7a2a4a7..77dd5f0 100644
--- a/muyu-source-server/src/main/java/com/muyu/source/service/DataValueService.java
+++ b/muyu-source-server/src/main/java/com/muyu/source/service/DataValueService.java
@@ -21,4 +21,6 @@ public interface DataValueService extends IService<DataValue> {
     List<DataValue> findTableValueByTableName(Long basicId, String tableName) throws SQLException;
 
     Integer addTableValue(DataValueModel dataValueModel);
+
+    Integer getTableValueTotal(DataValueModel dataValueModel);
 }
diff --git a/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java b/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java
index 026e3e5..2ea4dae 100644
--- a/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java
+++ b/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java
@@ -140,6 +140,26 @@ public class DataValueServiceImpl extends ServiceImpl<DataValueMapper, DataValue
             throw new RuntimeException(e);
         }
     }
+
+    @Override
+    public Integer getTableValueTotal(DataValueModel dataValueModel) {
+        MysqlQuery mysqlQuery = new MysqlQuery();
+        mysqlQuery.setDataSourceId(String.valueOf(dataValueModel.getBasicId()));
+        DataSource dataSource = dataSourceService.getById(dataValueModel.getBasicId());
+        MysqlPool mysqlPool = new MysqlPool(dataSource);
+        mysqlPool.init();
+        Connection conn = mysqlPool.getConn();
+        try (Statement statement = conn.createStatement();
+             ResultSet resultSet = statement.executeQuery(dataValueModel.getSql())) {
+            if (resultSet.next()) {
+                return resultSet.getInt(1);
+            }
+        } catch (SQLException e) {
+            throw new RuntimeException(e);
+        }
+        return 0;
+    }
+
     private static final int THREAD_POOL_SIZE = 10;
     private final ExecutorService executorService = Executors.newFixedThreadPool(THREAD_POOL_SIZE);