From fa9441abe6326540763d38fdc712fef3f6ee2983 Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Fri, 17 Nov 2023 18:02:49 +0800
Subject: [PATCH] 统计代码

---
 src/main/java/org/springblade/modules/doorplateAddress/service/impl/DoorplateAddressServiceImpl.java |  136 +++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 124 insertions(+), 12 deletions(-)

diff --git a/src/main/java/org/springblade/modules/doorplateAddress/service/impl/DoorplateAddressServiceImpl.java b/src/main/java/org/springblade/modules/doorplateAddress/service/impl/DoorplateAddressServiceImpl.java
index dadb38f..32e3943 100644
--- a/src/main/java/org/springblade/modules/doorplateAddress/service/impl/DoorplateAddressServiceImpl.java
+++ b/src/main/java/org/springblade/modules/doorplateAddress/service/impl/DoorplateAddressServiceImpl.java
@@ -16,12 +16,14 @@
  */
 package org.springblade.modules.doorplateAddress.service.impl;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import liquibase.repackaged.org.apache.commons.lang3.StringUtils;
 import org.springblade.common.constant.DictConstant;
 import org.springblade.common.node.TreeNode;
+import org.springblade.common.utils.NodeTreeUtil;
 import org.springblade.core.secure.utils.AuthUtil;
 import org.springblade.modules.doorplateAddress.entity.DoorplateAddressEntity;
 import org.springblade.modules.doorplateAddress.mapper.DoorplateAddressMapper;
@@ -29,7 +31,10 @@
 import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVO;
 import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
 import org.springblade.modules.doorplateAddress.vo.FuncNode;
+import org.springblade.modules.grid.service.IGridService;
+import org.springblade.modules.house.entity.HouseEntity;
 import org.springblade.modules.house.service.IHouseRentalService;
+import org.springblade.modules.house.service.IHouseService;
 import org.springblade.modules.house.service.IHouseholdService;
 import org.springblade.modules.house.vo.HouseParam;
 import org.springblade.modules.house.vo.HouseRentalVO;
@@ -54,10 +59,16 @@
 	private IPlaceService placeService;
 
 	@Autowired
+	private IHouseService houseService;
+
+	@Autowired
 	private IHouseholdService householdService;
 
 	@Autowired
 	private IHouseRentalService houseRentalService;
+
+	@Autowired
+	private IGridService gridService;
 
 
 	@Override
@@ -73,14 +84,23 @@
 	 */
 	@Override
 	public Object getFuncList(Integer type,String roleName) {
+		HouseParam houseParam = new HouseParam();
+		List<String> stringList = new ArrayList<>();
+		if (null!=roleName && !roleName.equals("")){
+			houseParam.setRoleName(roleName);
+			if (roleName.equals("网格员")){
+				// 查询对应的房屋地址code
+				stringList = gridService.getAddressCodeListByUserId(AuthUtil.getUserId());
+			}
+		}
 		List<TreeNode> list = new ArrayList<>();
 		if (null!=type) {
 			// 如果是网格管理员,系统管理员
 			if (type==1){
 				// 查询街道
-				List<TreeNode> townList = baseMapper.getRegionListByGroupTwon();
+				List<TreeNode> townList = baseMapper.getRegionListByGroupTwon(houseParam,stringList);
 				// 查询社区
-				List<TreeNode> neiList = baseMapper.getRegionListByGroupNei();
+				List<TreeNode> neiList = baseMapper.getRegionListByGroupNei(houseParam,stringList);
 				// 遍历
 				for (TreeNode treeNode : townList) {
 					// 遍历
@@ -91,7 +111,7 @@
 						}
 					}
 				}
-				// 查询区域数据,当前只有西市街道数据
+				// 查询区域数据
 				return townList;
 			}
 			// 如果是居民
@@ -170,10 +190,11 @@
 	public Object getHousesList(HouseParam houseParam) {
 		List<TreeNode> list = new ArrayList<>();
 		Map<String, Object> map = new HashMap<>(2);
+		List<String> stringList = getHouseCodeList(houseParam);
 		// 查小区,场所
 		if (houseParam.getType()==1){
 			// 根据社区名称查询小区集合
-			list  = baseMapper.getDistrictList(houseParam.getName(),houseParam.getCode());
+			list  = baseMapper.getDistrictList(houseParam,stringList);
 		}
 		// 查楼栋,街路巷
 		if (houseParam.getType()==2){
@@ -193,12 +214,13 @@
 	 * @return
 	 */
 	private Map<String, Object> getHouseLevelData(HouseParam houseParam, Map<String, Object> map) {
+		List<String> stringList = getHouseCodeList(houseParam);
 		// 判断地址类型
 		if (houseParam.getAddressType()==1) {
 			List<FuncNode> aoiList = new ArrayList<>();
 			List<FuncNode> shopList = new ArrayList<>();
 			// 查询户室及住户相关信息,单元中包含住户,或者和单元平级的 商铺
-			List<FuncNode> householdList = getUnitHouseholdList(houseParam.getName(), houseParam.getCode());
+			List<FuncNode> householdList = getUnitHouseholdList(houseParam,stringList);
 			// 遍历
 			if (householdList.size()>0){
 				for (FuncNode funcNode : householdList) {
@@ -216,7 +238,7 @@
 			return map;
 		}else{
 			// 根据街路巷编号查询街路巷门牌名称集合
-			List<FuncNode> doorplateNameList = baseMapper.getDoorplateNameList(houseParam.getName(), houseParam.getCode());
+			List<FuncNode> doorplateNameList = baseMapper.getDoorplateNameList(houseParam,stringList);
 			map.put("aoiList",new ArrayList<>());
 			map.put("shopList",doorplateNameList);
 			// 返回
@@ -231,12 +253,14 @@
 	 * @return
 	 */
 	private Object getBuildLevelData(HouseParam houseParam,Map<String, Object> map) {
+		List<String> stringList = getHouseCodeList(houseParam);
 		// 判断 code 长度,如果 code 长度大于 12 则为小区查楼栋/商铺,否则则按社区查街路巷
 		if (houseParam.getCode().length()>12){
 			List<TreeNode> aoiList = new ArrayList<>();
 			List<TreeNode> shopList = new ArrayList<>();
 			// 根据社区名称查询楼栋或者商铺的集合
-			List<TreeNode> list = baseMapper.getBuildingList(houseParam.getName(),houseParam.getCode());
+
+			List<TreeNode> list = baseMapper.getBuildingList(houseParam,stringList);
 			// 排序  StringUtils.getDigits(X.getName()) 取出数字排序
 			List<TreeNode> sortList = list.stream().sorted(Comparator.comparing(X -> StringUtils.getDigits(X.getName()))).collect(Collectors.toList());
 			if (list.size()>0){
@@ -255,7 +279,7 @@
 			return map;
 		}else {
 			// 查询街路巷
-			List<TreeNode> list  = baseMapper.getStreetRuList(houseParam.getName(),houseParam.getCode());
+			List<TreeNode> list  = baseMapper.getStreetRuList(houseParam,stringList);
 			map.put("aoiList",new ArrayList<>());
 			map.put("shopList",list);
 			// 返回
@@ -265,14 +289,14 @@
 
 	/**
 	 * 查询户室及住户相关信息,单元中包含住户
-	 * @param name
-	 * @param code
+	 * @param houseParam
+	 * @param stringList
 	 * @return
 	 */
-	private List<FuncNode> getUnitHouseholdList(String name, String code) {
+	private List<FuncNode> getUnitHouseholdList(HouseParam houseParam,List<String> stringList) {
 		List<FuncNode> list = new ArrayList<>();
 		// 查询户室及住户相关信息,单元中包含住户
-		List<FuncNode> funcNodes  = baseMapper.getUnitHouseholdList(name,code);
+		List<FuncNode> funcNodes  = baseMapper.getUnitHouseholdList(houseParam,stringList);
 		// 遍历
 		List<FuncNode> aoiNodes  = new ArrayList<>();
 		List<FuncNode> shopNodes  = new ArrayList<>();
@@ -434,4 +458,92 @@
 		}
 		return 0;
 	}
+
+	/**
+	 * 查询社区信息
+	 * @param name
+	 * @return
+	 */
+	@Override
+	public List<DoorplateAddressEntity> getAllDoorplateAddress(String name) {
+		return baseMapper.getAllDoorplateAddress(name);
+	}
+
+	/**
+	 * 获取房屋树
+	 * @param houseParam
+	 * @return
+	 */
+	@Override
+	public List<DoorplateAddressVOTree> getHouseTree(HouseParam houseParam) {
+		List<String> stringList = getHouseCodeList(houseParam);
+		// 根据社区居委会编号获取对应的小区/楼栋/单元/户室
+		return NodeTreeUtil.getAddressNodeTree(baseMapper.getHouseTree(houseParam,stringList));
+	}
+
+	/**
+	 * 根据角色获取地址编号集合
+	 * @param houseParam
+	 * @return
+	 */
+	private List<String> getHouseCodeList(HouseParam houseParam) {
+		List<String> stringList = new ArrayList<>();
+		if (null != houseParam.getRoleName() && !houseParam.getRoleName().equals("")) {
+			if (houseParam.getRoleName().equals("网格员")) {
+				// 查询对应的房屋地址code
+				stringList = gridService.getAddressCodeListByUserId(AuthUtil.getUserId());
+			}
+		}
+		return stringList;
+	}
+
+	/**
+	 * 数据处理
+	 * @return
+	 */
+	@Override
+	public Object dataHandle() {
+		// 处理户室数据
+		handleHouseData();
+
+		return null;
+	}
+
+	/**
+	 * 处理房屋数据
+	 */
+	private void handleHouseData() {
+		// 查询所有户室数据
+		List<DoorplateAddressEntity> list = baseMapper.getHouseList();
+		// 需要新增的房屋 list
+		List<HouseEntity> houseList = new ArrayList<>();
+		// 处理户室数据
+		for (DoorplateAddressEntity doorplateAddressEntity : list) {
+			// 查询是否已存在,存在就插入,不存在则插入
+			QueryWrapper<HouseEntity> wrapper = new QueryWrapper<>();
+			wrapper.eq("house_code",doorplateAddressEntity.getAddressCode())
+				.eq("is_deleted",0);
+			HouseEntity one = houseService.getOne(wrapper);
+			if (null==one){
+				HouseEntity houseEntity = new HouseEntity();
+				houseEntity.setHouseCode(doorplateAddressEntity.getAddressCode());
+				houseEntity.setDistrictCode(doorplateAddressEntity.getAoiCode());
+				houseEntity.setDistrictName(doorplateAddressEntity.getAoiName());
+				houseEntity.setHouseName(doorplateAddressEntity.getAddressName());
+				houseEntity.setFloor(Integer.parseInt(doorplateAddressEntity.getFloor()));
+				houseEntity.setBuilding(doorplateAddressEntity.getBuildingName());
+				houseEntity.setUnit(doorplateAddressEntity.getUnitName());
+				houseEntity.setRoom(doorplateAddressEntity.getHouseName());
+				houseEntity.setBuildingNo(doorplateAddressEntity.getBuildingCode());
+				houseEntity.setCreateUser(AuthUtil.getUserId().toString());
+				houseEntity.setCreateTime(new Date());
+				houseEntity.setUpdateUser(AuthUtil.getUserId().toString());
+				houseEntity.setUpdateTime(new Date());
+				// 加入集合
+				houseList.add(houseEntity);
+			}
+		}
+		// 批量插入
+		houseService.saveBatch(houseList);
+	}
 }

--
Gitblit v1.9.3