From bd77157fe2564b93afc776cebd944f2f0f536ba1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=A1=BE=E5=90=9B=E7=A6=B9?= <2673609096@qq.com> Date: Wed, 27 Aug 2025 17:26:58 +0800 Subject: [PATCH] =?UTF-8?q?farm=E6=A8=A1=E5=9D=97-=E6=A8=A1=E5=9E=8B?= =?UTF-8?q?=E7=BB=9F=E8=AE=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../ModelManagementController.java | 9 +++++++- .../ModelManagementMapper.java | 4 ++++ .../ModelManagementService.java | 4 ++++ .../ModelManagementServiceImpl.java | 22 ++++++++++++++----- .../modelmanagement/ModelManagementMapper.xml | 12 ++++++++++ 5 files changed, 45 insertions(+), 6 deletions(-) diff --git a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/controller/admin/modelmanagement/ModelManagementController.java b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/controller/admin/modelmanagement/ModelManagementController.java index a1db7abc..f965158f 100644 --- a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/controller/admin/modelmanagement/ModelManagementController.java +++ b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/controller/admin/modelmanagement/ModelManagementController.java @@ -4,7 +4,6 @@ import cn.iocoder.yudao.framework.apilog.core.annotation.ApiAccessLog; import cn.iocoder.yudao.framework.common.pojo.CommonResult; import cn.iocoder.yudao.framework.common.pojo.PageParam; import cn.iocoder.yudao.framework.common.pojo.PageResult; -import cn.iocoder.yudao.framework.common.util.object.BeanUtils; import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils; import cn.iocoder.yudao.module.farm.controller.admin.modelmanagement.vo.ModelManagementPageReqVO; import cn.iocoder.yudao.module.farm.controller.admin.modelmanagement.vo.ModelManagementRespVO; @@ -23,6 +22,7 @@ import org.springframework.web.bind.annotation.*; import java.io.IOException; import java.util.List; +import java.util.Map; import static cn.iocoder.yudao.framework.apilog.core.enums.OperateTypeEnum.EXPORT; import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success; @@ -110,6 +110,13 @@ public class ModelManagementController { return success(respVOList); } + @GetMapping("/get-count") + @Operation(summary = "模型统计") + public CommonResult> getCount() { + Map map = modelManagementService.getCount(); + return success(map); + } + @GetMapping("/export-excel") @Operation(summary = "导出模型管理 Excel") @ApiAccessLog(operateType = EXPORT) diff --git a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/dal/mysql/modelmanagement/ModelManagementMapper.java b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/dal/mysql/modelmanagement/ModelManagementMapper.java index 98b1dbe4..b49a9367 100644 --- a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/dal/mysql/modelmanagement/ModelManagementMapper.java +++ b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/dal/mysql/modelmanagement/ModelManagementMapper.java @@ -59,4 +59,8 @@ public interface ModelManagementMapper extends BaseMapperX { return selectJoinPage(reqVO, ModelManagementDO.class, query); } + + Long selectSum(Long tenantId); + + Long selectCountBySpecies(Long tenantId); } \ No newline at end of file diff --git a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementService.java b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementService.java index c323707d..cee3deda 100644 --- a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementService.java +++ b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementService.java @@ -8,6 +8,7 @@ import cn.iocoder.yudao.module.farm.dal.dataobject.modelmanagement.ModelManageme import jakarta.validation.Valid; import java.util.List; +import java.util.Map; /** * 模型管理 Service 接口 @@ -69,4 +70,7 @@ public interface ModelManagementService { PageResult getModelManagementPageRelease(@Valid ModelManagementPageReqVO pageReqVO); List getList(@Valid ModelManagementPageReqVO pageReqVO); + + Map getCount(); + } \ No newline at end of file diff --git a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java index 3234e1ae..bdadc5f5 100644 --- a/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java +++ b/yudao-module-farm/src/main/java/cn/iocoder/yudao/module/farm/service/modelmanagement/ModelManagementServiceImpl.java @@ -38,15 +38,13 @@ import jakarta.annotation.Resource; import org.springframework.stereotype.Service; import org.springframework.validation.annotation.Validated; -import java.util.ArrayList; -import java.util.Date; -import java.util.List; -import java.util.Map; +import java.util.*; import java.util.stream.Collectors; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; import static cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils.getLoginUser; -import static cn.iocoder.yudao.module.farm.enums.CommonConstants.*; +import static cn.iocoder.yudao.module.farm.enums.CommonConstants.COMMON_DISASTER; +import static cn.iocoder.yudao.module.farm.enums.CommonConstants.WEATHER_DISASTER; import static cn.iocoder.yudao.module.farm.enums.ErrorCodeConstants.MODEL_MANAGEMENT_NOT_EXISTS; /** @@ -435,4 +433,18 @@ public class ModelManagementServiceImpl implements ModelManagementService { return modelManagementMapper.selectList(lambdaQueryWrapper); } + @Override + public Map getCount() { + Map map = new HashMap<>(); + LambdaQueryWrapper lambdaQueryWrapper = new LambdaQueryWrapper<>(); + lambdaQueryWrapper.eq(ModelManagementDO::getTenantId, getLoginUser().getTenantId()); + Long l1 = modelManagementMapper.selectCount(lambdaQueryWrapper); + map.put("total", l1.toString()); // 模型总数 + Long l2 = modelManagementMapper.selectCountBySpecies(getLoginUser().getTenantId()); + map.put("species", l2.toString()); // 涉及物种 + Long l3 = modelManagementMapper.selectSum(getLoginUser().getTenantId()); + map.put("quote", l3.toString()); // 引用次数 + return map; + } + } \ No newline at end of file diff --git a/yudao-module-farm/src/main/resources/mapper/modelmanagement/ModelManagementMapper.xml b/yudao-module-farm/src/main/resources/mapper/modelmanagement/ModelManagementMapper.xml index 278417e3..fa392b86 100644 --- a/yudao-module-farm/src/main/resources/mapper/modelmanagement/ModelManagementMapper.xml +++ b/yudao-module-farm/src/main/resources/mapper/modelmanagement/ModelManagementMapper.xml @@ -9,4 +9,16 @@ 文档可见：https://www.iocoder.cn/MyBatis/x-plugins/ --> + + \ No newline at end of file