提交

CODE/smart-community/property-api/src/main/java/com/community/huiju/service/impl/BuildingTreeServiceImpl.java

@@ -604,14 +604,25 @@ public class BuildingTreeServiceImpl implements BuildingTreeServiceI {
 				tpPhase.setName(tpBuildingOwnerInfo.getPhaseName());
 				tpPhase.setCommunityId(communityId);
 				tpPhase.setCreateDate(LocalDateTime.now());
-				tpPhaseMapper.insert(tpPhase);
-				//重置Last
-				lastPhaseName = tpBuildingOwnerInfo.getPhaseName();
-				lastBuildingName = "";
-				lastUnitName = "";
-				lastLevelName = "";
-				lastRoomNoName = "";
-				lastPhaseId = tpPhase.getId();
+
+				QueryWrapper<TpPhase> phaseQueryWrapper = new QueryWrapper<>();
+				phaseQueryWrapper.eq("community_id", communityId);
+				phaseQueryWrapper.eq("`NAME`", tpBuildingOwnerInfo.getPhaseName());
+				TpPhase selectOne = tpPhaseMapper.selectOne(phaseQueryWrapper);
+				// 判断是否存在库里
+				if (null != selectOne) {
+					lastPhaseName = selectOne.getName();
+					lastPhaseId = selectOne.getId();
+				} else {
+					tpPhaseMapper.insert(tpPhase);
+					//重置Last
+					lastPhaseName = tpBuildingOwnerInfo.getPhaseName();
+					lastBuildingName = "";
+					lastUnitName = "";
+					lastLevelName = "";
+					lastRoomNoName = "";
+					lastPhaseId = tpPhase.getId();
+				}
 			}
 			//插入楼栋数据
 			if (!tpBuildingOwnerInfo.getBuildingName().equals(lastBuildingName)){
@@ -621,9 +632,20 @@ public class BuildingTreeServiceImpl implements BuildingTreeServiceI {
 				tpBuilding.setPhaseName(lastPhaseName);
 				tpBuilding.setName(tpBuildingOwnerInfo.getBuildingName());
 				tpBuilding.setCreateDate(LocalDateTime.now());
-				tpBuildingMapper.insert(tpBuilding);
-				lastBuildingName = tpBuildingOwnerInfo.getBuildingName();
-				lastBuildingId = tpBuilding.getId();
+
+				QueryWrapper<TpBuilding> buildingQueryWrapper = new QueryWrapper<>();
+				buildingQueryWrapper.eq("community_id", communityId);
+				buildingQueryWrapper.eq("phase_id", lastPhaseId);
+				buildingQueryWrapper.eq("`name`", tpBuildingOwnerInfo.getBuildingName());
+				TpBuilding selectOne = tpBuildingMapper.selectOne(buildingQueryWrapper);
+				if (null != selectOne) {
+					lastBuildingName = selectOne.getName();
+					lastBuildingId = selectOne.getId();
+				} else {
+					tpBuildingMapper.insert(tpBuilding);
+					lastBuildingName = tpBuildingOwnerInfo.getBuildingName();
+					lastBuildingId = tpBuilding.getId();
+				}
 			}
 			//插入单元信息
 			if (!tpBuildingOwnerInfo.getUnitName().equals(lastUnitName)){
@@ -635,9 +657,22 @@ public class BuildingTreeServiceImpl implements BuildingTreeServiceI {
 				tpUnit.setBuildingName(lastBuildingName);
 				tpUnit.setName(tpBuildingOwnerInfo.getUnitName());
 				tpUnit.setCreateDate(LocalDateTime.now());
-				tpUnitMapper.insert(tpUnit);
-				lastUnitName = tpBuildingOwnerInfo.getUnitName();
-				lastUnitId = tpUnit.getId();
+
+				QueryWrapper<TpUnit> unitQueryWrapper = new QueryWrapper<>();
+				unitQueryWrapper.eq("community_id", communityId);
+				unitQueryWrapper.eq("phase_id", lastPhaseId);
+				unitQueryWrapper.eq("building_id", lastBuildingId);
+				unitQueryWrapper.eq("`name`",tpBuildingOwnerInfo.getUnitName());
+				TpUnit selectOne = tpUnitMapper.selectOne(unitQueryWrapper);
+				// 校验是否存在库里
+				if (null != selectOne) {
+					lastUnitName = selectOne.getName();
+					lastUnitId = selectOne.getId();
+				} else {
+					tpUnitMapper.insert(tpUnit);
+					lastUnitName = tpBuildingOwnerInfo.getUnitName();
+					lastUnitId = tpUnit.getId();
+				}
 			}
 			//插入楼层信息
 			if (!tpBuildingOwnerInfo.getLevelName().equals(lastLevelName)){
@@ -651,9 +686,23 @@ public class BuildingTreeServiceImpl implements BuildingTreeServiceI {
 				tpLevel.setUnitName(lastUnitName);
 				tpLevel.setName(tpBuildingOwnerInfo.getLevelName());
 				tpLevel.setCreateDate(LocalDateTime.now());
-				tpLevelMapper.insert(tpLevel);
-				lastLevelName = tpBuildingOwnerInfo.getLevelName();
-				lastLevelId = tpLevel.getId();
+
+				QueryWrapper<TpLevel> levelQueryWrapper = new QueryWrapper<>();
+				levelQueryWrapper.eq("community_id", communityId);
+				levelQueryWrapper.eq("phase_id", lastPhaseId);
+				levelQueryWrapper.eq("building_id", lastBuildingId);
+				levelQueryWrapper.eq("unit_id", lastUnitId);
+				levelQueryWrapper.eq("`name`", tpBuildingOwnerInfo.getLevelName());
+				TpLevel selectOne = tpLevelMapper.selectOne(levelQueryWrapper);
+				// 校验是否存在库里
+				if (null != selectOne) {
+					lastLevelName = selectOne.getName();
+					lastLevelId = selectOne.getId();
+				} else {
+					tpLevelMapper.insert(tpLevel);
+					lastLevelName = tpBuildingOwnerInfo.getLevelName();
+					lastLevelId = tpLevel.getId();
+				}
 			}
 			//插入房间号信息
 			if (!tpBuildingOwnerInfo.getRoomNoName().equals(lastRoomNoName)){
@@ -669,7 +718,18 @@ public class BuildingTreeServiceImpl implements BuildingTreeServiceI {
 				tpRoomNo.setLevelName(lastLevelName);
 				tpRoomNo.setName(tpBuildingOwnerInfo.getRoomNoName());
 				tpRoomNo.setCreateDate(LocalDateTime.now());
-				tpRoomNoMapper.insert(tpRoomNo);
+
+				QueryWrapper<TpRoomNo> roomNoQueryWrapper = new QueryWrapper<>();
+				roomNoQueryWrapper.eq("community_id", communityId);
+				roomNoQueryWrapper.eq("phase_id", lastPhaseId);
+				roomNoQueryWrapper.eq("building_id", lastBuildingId);
+				roomNoQueryWrapper.eq("unit_id", lastUnitId);
+				roomNoQueryWrapper.eq("level_id", lastLevelId);
+				roomNoQueryWrapper.eq("`name`", tpBuildingOwnerInfo.getRoomNoName());
+				TpRoomNo selectOne = tpRoomNoMapper.selectOne(roomNoQueryWrapper);
+				if (null == selectOne) {
+					tpRoomNoMapper.insert(tpRoomNo);
+				}
 			}
 		}
 	}