添加小区过滤条件

linwei

2024-01-29 29aaaa6553042d3f46fa0f77786c911a375312bd

添加小区过滤条件

 src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java

@@ -87,6 +87,33 @@
     */
    @Override
    public Object getDistrictTree(DistrictVO district) {
        // 判断角色，物业角色只能查询当前小区的
        String userRole = AuthUtil.getUserRole();
        if (userRole.contains("wygly") || userRole.contains("wyxmjl")) {
            if (district.getFilterFlag().equals(1)) {
                // 查询小区id
                IPropertyDistrictUserService propertyDistrictUserService = SpringUtils.getBean(IPropertyDistrictUserService.class);
                List<String> districtIds = propertyDistrictUserService.selectPropertyDistrictByUserId(AuthUtil.getUserId());
                // 通过用户机构查询用户的物业公司
                // 通过用户机构查询用户的物业公司
                IPropertyCompanyService bean = SpringUtil.getBean(IPropertyCompanyService.class);
                PropertyCompanyEntity one = bean.getOne(Wrappers.<PropertyCompanyEntity>lambdaQuery().eq(PropertyCompanyEntity::getDeptId, AuthUtil.getDeptId()));
                if (one != null) {
                    IPropertyCompanyDistrictService bean2 = SpringUtils.getBean(IPropertyCompanyDistrictService.class);
                    // 通过物业公司，查询小区
                    List<PropertyCompanyDistrictEntity> list = bean2.list(Wrappers.<PropertyCompanyDistrictEntity>lambdaQuery()
                        .eq(PropertyCompanyDistrictEntity::getPropertyCompanyId, one.getId()));
                    if (list.size() > 0) {
                        List<String> collect = list.stream().map(i -> i.getDistrictId()).collect(Collectors.toList());
                        districtIds.addAll(collect);
                    }
                }
                district.setDistrictIdList(districtIds);
                if (districtIds.size() == 0) {
                    return new ArrayList<>();
                }
            }
        }
        Map<String, TreeStringNode> districtTree = baseMapper.getDistrictTree(district);
        List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(districtTree);
        stringNodeTree.forEach(node -> recursion(node));

 src/main/java/org/springblade/modules/district/vo/DistrictVO.java

@@ -61,4 +61,7 @@
     */
    private String regionCode;

    // 是否过滤0 否 1 是
    private Integer filterFlag = 0;

}