From efc218a71a3d0784bbb05ce5ffdfffa693fdb8f4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=86=B7=E8=B0=83?= <3084898776@qq.com>
Date: Fri, 6 Sep 2024 14:51:41 +0800
Subject: [PATCH] =?UTF-8?q?09061451:=E5=AF=B9=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E7=BB=93=E6=9E=84=E7=9A=84=E6=95=B0=E6=8D=AE=E9=87=8F=E5=81=9A?=
 =?UTF-8?q?=E4=B8=80=E4=B8=AA=E5=AE=9E=E6=97=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../source/controller/DataValueController.java   |  3 +--
 .../source/controller/TableDataController.java   | 16 ++++++++++++++++
 .../source/service/Impl/TableServiceImpl.java    |  1 +
 3 files changed, 18 insertions(+), 2 deletions(-)

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 72ca741..9c6a52b 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
@@ -66,7 +66,6 @@ public class DataValueController {
         Integer i = dataValueService.addTableValue(dataValueModel);
         return Result.success(i);
     }
-
     /**
      * 根据基础表ID和表名查询数据
      *
@@ -76,7 +75,7 @@ public class DataValueController {
      */
     @PostMapping("/findTableValueByTableName")
     @Operation(summary = "根据基础表ID和SQL语句查询数据", description = "根据基础表ID和SQL语句查询数据")
-    public Result findTableValueByTableName(@RequestParam("basicId") Long basicId, @RequestParam("tableName") String tableName) throws SQLException {
+    public Result findTableValueByTableName(@RequestParam("basicId") Long basicId, @RequestParam("tableName") String tableName) {
         List<DataValue> dataValueList = dataValueService.findTableValueByTableName(basicId, tableName);
         return Result.success(dataValueList);
     }
diff --git a/muyu-source-server/src/main/java/com/muyu/source/controller/TableDataController.java b/muyu-source-server/src/main/java/com/muyu/source/controller/TableDataController.java
index 69c68b3..8b76bfc 100644
--- a/muyu-source-server/src/main/java/com/muyu/source/controller/TableDataController.java
+++ b/muyu-source-server/src/main/java/com/muyu/source/controller/TableDataController.java
@@ -1,6 +1,8 @@
 package com.muyu.source.controller;
 
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
 import com.dtflys.forest.springboot.annotation.ForestScannerRegister;
 import com.muyu.common.core.domain.Result;
 import com.muyu.common.core.utils.poi.ExcelUtil;
@@ -10,6 +12,8 @@ import com.muyu.common.security.annotation.RequiresPermissions;
 import com.muyu.common.security.utils.SecurityUtils;
 import com.muyu.source.domain.Structure;
 import com.muyu.source.domain.TableData;
+import com.muyu.source.domain.TableInfo;
+import com.muyu.source.service.StructureService;
 import com.muyu.source.service.TableDataService;
 import com.muyu.source.service.TableInfoService;
 import io.swagger.v3.oas.annotations.Operation;
@@ -44,6 +48,8 @@ public class TableDataController extends BaseController {
     private TableDataService tableDataService;
     @Autowired
     private TableInfoService tableInfoService;
+    @Autowired
+    private StructureService structureService;
     /**
      * 根据childrenId查询表结构
      *
@@ -63,6 +69,16 @@ public class TableDataController extends BaseController {
      */
     @GetMapping("/selectStructureById/{id}")
     public Result<List<Structure>> selectStructureById(@PathVariable("id") Integer id) {
+        //根据id查询表结构中的数据量
+        long count = structureService.count(new LambdaQueryWrapper<>() {{
+            eq(Structure::getTableId, id);
+        }});
+        if (count!=0L) {
+            boolean update = tableInfoService.update(new LambdaUpdateWrapper<>() {{
+                eq(TableInfo::getId, id)
+                        .set(TableInfo::getDataNum, count);
+            }});
+        }
         List<Structure> structureList =tableInfoService.selectTableInfoById(id);
         return success(structureList);
     }
diff --git a/muyu-source-server/src/main/java/com/muyu/source/service/Impl/TableServiceImpl.java b/muyu-source-server/src/main/java/com/muyu/source/service/Impl/TableServiceImpl.java
index 379f3fa..cca472f 100644
--- a/muyu-source-server/src/main/java/com/muyu/source/service/Impl/TableServiceImpl.java
+++ b/muyu-source-server/src/main/java/com/muyu/source/service/Impl/TableServiceImpl.java
@@ -45,6 +45,7 @@ public class TableServiceImpl extends ServiceImpl<TableInfoMapper, TableInfo> im
         List<Structure> structureList = structureMapper.selectList(new LambdaQueryWrapper<>(Structure.class) {{
             eq(Structure::getTableId, id);
         }});
+
         return structureList;
     }