From 25a7e54610be96501134d9a0292a52e57d08d4ca Mon Sep 17 00:00:00 2001
From: junjun <junjie.xia@fit2cloud.com>
Date: Thu, 16 Nov 2023 11:49:44 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E6=A0=B9=E6=8D=AE?=
 =?UTF-8?q?=E6=95=B0=E6=8D=AE=E9=9B=86ID=E8=8E=B7=E5=8F=96count=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/dataease/dataset/manage/DatasetDataManage.java     | 8 ++++++++
 .../io/dataease/dataset/server/DatasetDataServer.java     | 5 +++++
 .../main/java/io/dataease/api/dataset/DatasetDataApi.java | 3 +++
 3 files changed, 16 insertions(+)

diff --git a/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetDataManage.java b/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetDataManage.java
index 34672a43a5..f575012f4c 100644
--- a/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetDataManage.java
+++ b/core/core-backend/src/main/java/io/dataease/dataset/manage/DatasetDataManage.java
@@ -210,6 +210,14 @@ public class DatasetDataManage {
         return map;
     }
 
+    public Long getDatasetTotal(Long datasetGroupId) throws Exception {
+        DatasetGroupInfoDTO dto = datasetGroupManage.get(datasetGroupId, null);
+        if (StringUtils.equalsIgnoreCase(dto.getNodeType(), "dataset")) {
+            return getDatasetTotal(dto);
+        }
+        return 0L;
+    }
+
     public Long getDatasetTotal(DatasetGroupInfoDTO datasetGroupInfoDTO) throws Exception {
         Map<String, Object> sqlMap = datasetSQLManage.getUnionSQLForEdit(datasetGroupInfoDTO, null);
         String sql = (String) sqlMap.get("sql");
diff --git a/core/core-backend/src/main/java/io/dataease/dataset/server/DatasetDataServer.java b/core/core-backend/src/main/java/io/dataease/dataset/server/DatasetDataServer.java
index 4298a46b0a..c6e41b1e4d 100644
--- a/core/core-backend/src/main/java/io/dataease/dataset/server/DatasetDataServer.java
+++ b/core/core-backend/src/main/java/io/dataease/dataset/server/DatasetDataServer.java
@@ -46,4 +46,9 @@ public class DatasetDataServer implements DatasetDataApi {
     public List<String> getFieldEnum(List<Long> ids) throws Exception {
         return datasetDataManage.getFieldEnum(ids);
     }
+
+    @Override
+    public Long getDatasetCount(DatasetGroupInfoDTO datasetGroupInfoDTO) throws Exception {
+        return datasetDataManage.getDatasetTotal(datasetGroupInfoDTO.getId());
+    }
 }
diff --git a/sdk/api/api-base/src/main/java/io/dataease/api/dataset/DatasetDataApi.java b/sdk/api/api-base/src/main/java/io/dataease/api/dataset/DatasetDataApi.java
index eacb18c023..7b2213d534 100644
--- a/sdk/api/api-base/src/main/java/io/dataease/api/dataset/DatasetDataApi.java
+++ b/sdk/api/api-base/src/main/java/io/dataease/api/dataset/DatasetDataApi.java
@@ -28,4 +28,7 @@ public interface DatasetDataApi {
 
     @PostMapping("enumValue")
     List<String> getFieldEnum(@RequestBody List<Long> ids) throws Exception;
+
+    @PostMapping("getDatasetCount")
+    Long getDatasetCount(@RequestBody DatasetGroupInfoDTO datasetGroupInfoDTO) throws Exception;
 }