From d0b0e00e40226700a0ffa1f1671a8dc273610f67 Mon Sep 17 00:00:00 2001
From: linwe <872216996@qq.com>
Date: Mon, 18 Dec 2023 14:28:10 +0800
Subject: [PATCH] 菜单分类优化

---
 src/main/java/org/springblade/modules/doorplateAddress/service/impl/DoorplateAddressServiceImpl.java |  127 ++++++++++++++++++++++++++++++++++++++---
 1 files changed, 116 insertions(+), 11 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 2957ec3..2a6d590 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
@@ -24,8 +24,13 @@
 import org.apache.logging.log4j.util.Strings;
 import org.springblade.common.constant.DictConstant;
 import org.springblade.common.node.TreeStringNode;
+import org.springblade.common.utils.ComplexNumberStringComparator;
 import org.springblade.common.utils.NodeTreeUtil;
 import org.springblade.core.secure.utils.AuthUtil;
+import org.springblade.modules.category.dto.CategoryLabelDTO;
+import org.springblade.modules.category.service.ICategoryLabelService;
+import org.springblade.modules.district.entity.DistrictEntity;
+import org.springblade.modules.district.service.IDistrictService;
 import org.springblade.modules.doorplateAddress.entity.DoorplateAddressEntity;
 import org.springblade.modules.doorplateAddress.mapper.DoorplateAddressMapper;
 import org.springblade.modules.doorplateAddress.service.IDoorplateAddressService;
@@ -33,7 +38,9 @@
 import org.springblade.modules.doorplateAddress.vo.DoorplateAddressVOTree;
 import org.springblade.modules.doorplateAddress.vo.FuncNode;
 import org.springblade.modules.grid.entity.GridEntity;
+import org.springblade.modules.grid.entity.GridmanEntity;
 import org.springblade.modules.grid.service.IGridService;
+import org.springblade.modules.grid.service.IGridmanService;
 import org.springblade.modules.house.entity.HouseEntity;
 import org.springblade.modules.house.service.IHouseRentalService;
 import org.springblade.modules.house.service.IHouseService;
@@ -41,9 +48,13 @@
 import org.springblade.modules.house.vo.HouseParam;
 import org.springblade.modules.house.vo.HouseRentalVO;
 import org.springblade.modules.house.vo.HouseholdVO;
+import org.springblade.modules.place.entity.PlaceEntity;
 import org.springblade.modules.place.service.IPlaceService;
+import org.springblade.modules.system.entity.Region;
+import org.springblade.modules.system.service.IRegionService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 import java.util.*;
 import java.util.stream.Collectors;
@@ -71,6 +82,18 @@
 
 	@Autowired
 	private IGridService gridService;
+
+	@Autowired
+	private IGridmanService gridmanService;
+
+	@Autowired
+	private IRegionService regionService;
+
+	@Autowired
+	private IDistrictService districtService;
+
+	@Autowired
+	private ICategoryLabelService iCategoryLabelService;
 
 
 	@Override
@@ -148,6 +171,10 @@
 		List<TreeStringNode> placeNodeList = placeService.selectPlaceNodeList(AuthUtil.getUserId());
 		for (TreeStringNode treeNode : placeNodeList) {
 			treeNode.setAddressType(2);
+			CategoryLabelDTO categoryLabelDTO = new CategoryLabelDTO();
+			categoryLabelDTO.setPlaceId(treeNode.getId());
+			List<CategoryLabelDTO> categoryLabelDTOS = iCategoryLabelService.selectCategoryLabelList(categoryLabelDTO);
+			treeNode.setCategoryLabelList(categoryLabelDTOS);
 //			if (DictConstant.SMALL_DOORPLATE.equals(treeNode.getDoorplateType()) ||
 //				(DictConstant.centre_DOORPLATE.equals(treeNode.getDoorplateType()) &&
 //					treeNode.getAddressLevel().equals(1))) {
@@ -220,14 +247,17 @@
 	 */
 	private void getGridInfoByGridman(HouseParam houseParam) {
 		if (houseParam.getRoleName().equals("网格员")) {
-			QueryWrapper<GridEntity> wrapper = new QueryWrapper<>();
+			QueryWrapper<GridmanEntity> wrapper = new QueryWrapper<>();
 			wrapper.eq("is_deleted", 0)
 				.eq("user_id", AuthUtil.getUserId());
-			List<GridEntity> list = gridService.list(wrapper);
+			List<GridmanEntity> list = gridmanService.list(wrapper);
 			if (list.size() > 0) {
-				GridEntity gridEntity = list.get(0);
-				if (!Strings.isBlank(gridEntity.getCommunityName())) {
-					houseParam.setCommunityName(gridEntity.getCommunityName().split("居民委员会")[0]);
+				GridmanEntity gridmanEntity = list.get(0);
+				GridEntity gridEntity = gridService.getById(gridmanEntity.getGridId());
+				// 查询居委会
+				Region region = regionService.getById(gridEntity.getCommunityCode());
+				if (null != region) {
+					houseParam.setCommunityName(region.getName());
 				}
 				if (!Strings.isBlank(gridEntity.getGridName())) {
 					houseParam.setGridName(gridEntity.getGridName());
@@ -309,10 +339,15 @@
 				List<TreeStringNode> aoiList = new ArrayList<>();
 				List<TreeStringNode> shopList = new ArrayList<>();
 				// 根据社区名称查询楼栋或者商铺的集合
-
 				List<TreeStringNode> list = baseMapper.getBuildingList(houseParam, stringList);
 				// 排序  StringUtils.getDigits(X.getName()) 取出数字排序
-				List<TreeStringNode> sortList = list.stream().sorted(Comparator.comparing(X -> StringUtils.getDigits(X.getName()))).collect(Collectors.toList());
+				List<TreeStringNode> sortList = list.stream().
+					sorted(new Comparator<TreeStringNode>() {
+						@Override
+						public int compare(TreeStringNode o1, TreeStringNode o2) {
+							return ComplexNumberStringComparator.compare(o1.getName(),o2.getName());
+						}
+					}).collect(Collectors.toList());
 				if (list.size() > 0) {
 					for (TreeStringNode treeNode : sortList) {
 						if (treeNode.getAddressType() == 1) {
@@ -551,12 +586,12 @@
 	}
 
 	/**
-	 * 数据处理
+	 * 房屋数据处理
 	 * @return
 	 */
 	@Override
-	public Object dataHandle() {
-		// 处理户室数据
+	public Object houseDataHandle() {
+		// 处理房屋数据
 		handleHouseData();
 
 		return null;
@@ -566,7 +601,7 @@
 	 * 处理房屋数据
 	 */
 	private void handleHouseData() {
-		// 查询所有户室数据
+		// 查询所有户室数据(未入库的)
 		List<DoorplateAddressEntity> list = baseMapper.getHouseList();
 		// 需要新增的房屋 list
 		List<HouseEntity> houseList = new ArrayList<>();
@@ -592,6 +627,8 @@
 				houseEntity.setCreateTime(new Date());
 				houseEntity.setUpdateUser(AuthUtil.getUserId().toString());
 				houseEntity.setUpdateTime(new Date());
+				// 设置来源 1:地址总表  2:国控采集
+				houseEntity.setSource(1);
 				// 加入集合
 				houseList.add(houseEntity);
 			}
@@ -599,4 +636,72 @@
 		// 批量插入
 		houseService.saveBatch(houseList);
 	}
+
+	/**
+	 * 小区数据处理
+	 * @return
+	 */
+	@Override
+	public Object aoiDataHandle() {
+		// 查询所有的地址表id集合
+		List<Long> aoiCodeList = baseMapper.getAoiCodeList();
+		// 查询小区集合
+		List<DoorplateAddressEntity> list = baseMapper.getAoiList(aoiCodeList);
+		// 创建小区集合对象
+		List<DistrictEntity> aoiList = new ArrayList<>();
+		// 将小区数据保存到小区表中
+		for (DoorplateAddressEntity addressEntity : list) {
+			// 查询小区是否已存在,不存在则插入,否则不新增
+			QueryWrapper<DistrictEntity> wrapper = new QueryWrapper<>();
+			wrapper.eq("aoi_code",addressEntity.getAoiCode())
+			.eq("is_deleted",0);
+			DistrictEntity one = districtService.getOne(wrapper);
+			if (null==one) {
+				DistrictEntity districtEntity = new DistrictEntity();
+				districtEntity.setCommunityCode(addressEntity.getNeiCode());
+				districtEntity.setAoiCode(addressEntity.getAoiCode());
+				districtEntity.setName(addressEntity.getAoiName());
+				districtEntity.setAddress(addressEntity.getAddressName());
+				districtEntity.setLng(addressEntity.getX());
+				districtEntity.setLat(addressEntity.getY());
+				// 加入集合
+				districtService.save(districtEntity);
+//				aoiList.add(districtEntity);
+			}
+		}
+		// 批量插入
+//		districtService.saveBatch(aoiList);
+		// 返回
+		return null;
+	}
+
+	/**
+	 * 场所数据处理
+	 * @return
+	 */
+	@Override
+	@Transactional(rollbackFor = Exception.class)
+	public Object placeDataHandle() {
+		// 查询所有的地址表和场所表差集集合
+		List<DoorplateAddressEntity> list = baseMapper.getPlaceList();
+		// 创建场所集合对象
+		List<PlaceEntity> placeList = new ArrayList<>();
+		// 将场所数据保存到场所表中
+		for (DoorplateAddressEntity addressEntity : list) {
+			PlaceEntity placeEntity = new PlaceEntity();
+			placeEntity.setHouseCode(addressEntity.getAddressCode());
+			placeEntity.setPlaceName(addressEntity.getPoi());
+			placeEntity.setLng(addressEntity.getX());
+			placeEntity.setLat(addressEntity.getY());
+			placeEntity.setLocaltion(addressEntity.getAddressName());
+			// 设置来源( 1:地址总表  2:国控采集)
+			placeEntity.setSource(1);
+			// 加入集合
+			placeList.add(placeEntity);
+		}
+		// 批量插入
+		placeService.saveBatch(placeList);
+		// 返回
+		return null;
+	}
 }

--
Gitblit v1.9.3