wuxw
2025-02-28 c1413aeb9fe5d736af38e25285ac8d88074a89bf
service-report/src/main/java/com/java110/report/cmd/community/QueryCommunityUnitTreeCmd.java
@@ -11,6 +11,7 @@
import com.java110.intf.report.IReportCommunityInnerServiceSMO;
import com.java110.utils.exception.CmdException;
import com.java110.utils.util.ListUtil;
import com.java110.utils.util.StringUtil;
import com.java110.vo.ResultVo;
import org.springframework.beans.factory.annotation.Autowired;
@@ -49,7 +50,7 @@
        JSONObject communityInfo = null;
        for(UnitDto tmpUnitDto:unitDtos){
            if(!hasInCommunitys(tmpUnitDto,communitys)){
            if (!hasInCommunity(tmpUnitDto, communitys)) {
                communityInfo = new JSONObject();
                communityInfo.put("id","c_"+tmpUnitDto.getCommunityId());
                communityInfo.put("communityId",tmpUnitDto.getCommunityId());
@@ -60,11 +61,114 @@
            }
        }
        JSONObject community = null;
        for (int cIndex = 0; cIndex < communitys.size(); cIndex++) {
            community = communitys.getJSONObject(cIndex);
            // find floor data in unitDtos
            findFloor(community, unitDtos);
        }
        context.setResponseEntity(ResultVo.createResponseEntity(communitys));
    }
    private boolean hasInCommunitys(UnitDto tmpUnitDto,JSONArray communitys) {
    /**
     * find community floor data
     *
     * @param community current community
     * @param unitDtos  all units data
     */
    private void findFloor(JSONObject community, List<UnitDto> unitDtos) {
        JSONArray floors = community.getJSONArray("children");
        JSONObject floorInfo = null;
        for (UnitDto tmpUnitDto : unitDtos) {
            if (!hasInFloor(tmpUnitDto, floors)) {
                floorInfo = new JSONObject();
                floorInfo.put("id", "f_" + tmpUnitDto.getFloorId());
                floorInfo.put("floorId", tmpUnitDto.getFloorId());
                floorInfo.put("text", tmpUnitDto.getFloorNum()+"栋");
                floorInfo.put("icon", "/img/floor.png");
                floorInfo.put("children", new JSONArray());
                floors.add(floorInfo);
            }
        }
        JSONObject floor = null;
        for (int cIndex = 0; cIndex < floors.size(); cIndex++) {
            floor = floors.getJSONObject(cIndex);
            // find floor data in unitDtos
            findUnit(floor, unitDtos);
        }
    }
    private void findUnit(JSONObject floor, List<UnitDto> unitDtos) {
        JSONArray units = floor.getJSONArray("children");
        JSONObject unitInfo = null;
        for (UnitDto tmpUnitDto : unitDtos) {
            if (!hasInUnit(tmpUnitDto, units)) {
                unitInfo = new JSONObject();
                unitInfo.put("id", "u_" + tmpUnitDto.getUnitId());
                unitInfo.put("unitId", tmpUnitDto.getUnitId());
                unitInfo.put("text", tmpUnitDto.getUnitNum()+"单元");
                unitInfo.put("icon", "/img/unit.png");
                units.add(unitInfo);
            }
        }
    }
    private boolean hasInUnit(UnitDto unitDto, JSONArray units) {
        JSONObject unit = null;
        for (int cIndex = 0; cIndex < units.size(); cIndex++) {
            unit = units.getJSONObject(cIndex);
            if (!unit.containsKey("unitId")) {
                continue;
            }
            if (StringUtil.isEmpty(unit.getString("unitId"))) {
                continue;
            }
            if (unit.getString("unitId").equals(unitDto.getUnitId())) {
        return true;
    }
        }
        return false;
    }
    private boolean hasInFloor(UnitDto tmpUnitDto, JSONArray floors) {
        JSONObject floor = null;
        for (int cIndex = 0; cIndex < floors.size(); cIndex++) {
            floor = floors.getJSONObject(cIndex);
            if (!floor.containsKey("floorId")) {
                continue;
            }
            if (StringUtil.isEmpty(floor.getString("floorId"))) {
                continue;
            }
            if (floor.getString("floorId").equals(tmpUnitDto.getFloorId())) {
                return true;
            }
        }
        return false;
    }
    private boolean hasInCommunity(UnitDto tmpUnitDto, JSONArray communitys) {
        JSONObject community = null;
        for (int cIndex = 0; cIndex < communitys.size(); cIndex++) {
            community = communitys.getJSONObject(cIndex);
            if (!community.containsKey("communityId")) {
                continue;
            }
            if (StringUtil.isEmpty(community.getString("communityId"))) {
                continue;
            }
            if (community.getString("communityId").equals(tmpUnitDto.getCommunityId())) {
                return true;
            }
        }
        return false;
    }
}