From a9a2d0f1d8d271985869dd2b26ac29217191ab51 Mon Sep 17 00:00:00 2001
From: zhongrj <646384940@qq.com>
Date: Tue, 10 Dec 2024 11:51:58 +0800
Subject: [PATCH] 应急空间新增字典,查询修改

---
 src/main/java/org/springblade/modules/yw/service/impl/RiskSourceServiceImpl.java |   23 ++++++++++++++++++++++-
 1 files changed, 22 insertions(+), 1 deletions(-)

diff --git a/src/main/java/org/springblade/modules/yw/service/impl/RiskSourceServiceImpl.java b/src/main/java/org/springblade/modules/yw/service/impl/RiskSourceServiceImpl.java
index d9d1b3e..60ed811 100644
--- a/src/main/java/org/springblade/modules/yw/service/impl/RiskSourceServiceImpl.java
+++ b/src/main/java/org/springblade/modules/yw/service/impl/RiskSourceServiceImpl.java
@@ -3,6 +3,7 @@
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import org.apache.logging.log4j.util.Strings;
+import org.springblade.common.cache.DictBizCache;
 import org.springblade.core.tool.utils.BeanUtil;
 import org.springblade.modules.yw.entity.FirmInfo;
 import org.springblade.modules.yw.entity.RiskSourceEntity;
@@ -16,6 +17,7 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 
 import java.util.List;
+import java.util.Map;
 import java.util.Objects;
 
 /**
@@ -38,7 +40,15 @@
 	 */
 	@Override
 	public IPage<RiskSourceVO> selectRiskSourcePage(IPage<RiskSourceVO> page, RiskSourceVO riskSource) {
-		return page.setRecords(baseMapper.selectRiskSourcePage(page, riskSource));
+		List<RiskSourceVO> riskSourceVOS = baseMapper.selectRiskSourcePage(page, riskSource);
+		for (RiskSourceVO riskSourceVO : riskSourceVOS) {
+			if (!Strings.isBlank(riskSourceVO.getCategory())){
+				riskSourceVO.setCategory(
+					DictBizCache.getValues("industry_category",riskSourceVO.getCategory())
+				);
+			}
+		}
+		return page.setRecords(riskSourceVOS);
 	}
 
 	/**
@@ -105,4 +115,15 @@
 		}
 		return null;
 	}
+
+	/**
+	 * 风险源统计查询
+	 * @param riskSource
+	 * @return
+	 */
+	@Override
+	public List<Map<String, Object>> getStatisticData(RiskSourceVO riskSource) {
+		// 风险源统计查询
+		return baseMapper.getStatisticData(riskSource);
+	}
 }

--
Gitblit v1.9.3