linwe
2024-08-09 8b7258c9427882bb1798f1502eaa35184c6e374e
src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java
@@ -16,29 +16,36 @@
 */
package org.springblade.modules.district.service.impl;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.cache.SysCache;
import org.springblade.common.node.TreeIntegerNode;
import org.springblade.common.node.TreeStringNode;
import org.springblade.common.param.CommonParamSet;
import org.springblade.common.utils.NodeTreeUtil;
import org.springblade.common.utils.RoleUtil;
import org.springblade.common.utils.SpringUtils;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.core.tool.utils.SpringUtil;
import org.springblade.modules.district.entity.DistrictEntity;
import org.springblade.modules.district.vo.DistrictVO;
import org.springblade.modules.district.mapper.DistrictMapper;
import org.springblade.modules.district.service.IDistrictService;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springblade.modules.district.vo.DistrictVO;
import org.springblade.modules.grid.service.IGridService;
import org.springblade.modules.grid.vo.GridVO;
import org.springblade.modules.property.entity.PropertyCompanyDistrictEntity;
import org.springblade.modules.property.entity.PropertyCompanyEntity;
import org.springblade.modules.property.service.IPropertyCompanyDistrictService;
import org.springblade.modules.property.service.IPropertyCompanyService;
import org.springblade.modules.property.service.IPropertyDistrictUserService;
import org.springblade.modules.system.entity.Dept;
import org.springblade.modules.system.service.IDeptService;
import org.springblade.modules.task.vo.TaskReportForRepairsVO;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
 * 小区表 服务实现类
@@ -54,14 +61,16 @@
   @Override
   public IPage<DistrictVO> selectDistrictPage(IPage<DistrictVO> page, DistrictVO district) {
      List<String> regionChildCodesList = SysCache.getRegionChildCodesByDeptId(AuthUtil.getDeptId());
      Integer isAdministrator = AuthUtil.isAdministrator()==true?1:2;
      List<DistrictVO> districtVOS = baseMapper.selectDistrictPage(page, district,regionChildCodesList,isAdministrator);
      CommonParamSet invoke = new CommonParamSet<>().invoke(DistrictVO.class, district);
      List<DistrictVO> districtVOS = baseMapper.selectDistrictPage(page,
         district,
         invoke.getRegionChildCodesList(),
         invoke.getIsAdministrator());
      // 遍历
      for (DistrictVO districtVO : districtVOS) {
         // 设置对应的网格名称
         List<GridVO> gridVO = gridService.getGridListByAoiCode(districtVO.getAoiCode());
         if (gridVO.size()>0) {
         if (gridVO.size() > 0) {
            StringBuilder builder = new StringBuilder();
            for (GridVO vo : gridVO) {
               builder.append(vo.getGridName()).append(",");
@@ -77,18 +86,43 @@
   /**
    * 获取小区树
    *
    * @param district
    * @return
    */
   @Override
   public Object getDistrictTree(DistrictVO district) {
      // String userRole = AuthUtil.getUserRole();
      // if (userRole.equals("wygly")) {
      //    IPropertyDistrictUserService bean = SpringUtils.getBean(IPropertyDistrictUserService.class);
      //    List<String> strings = bean.selectPropertyDistrictByUserId(AuthUtil.getUserId());
      //    district.setDistrictIdList(strings);
      // }
      Map<String, TreeStringNode> districtTree = baseMapper.getDistrictTree(district);
      // 判断角色,物业角色只能查询当前小区的
      String userRole = AuthUtil.getUserRole();
      CommonParamSet invoke = new CommonParamSet<>().invoke(DistrictVO.class, district);
      if (RoleUtil.isProperty(userRole)) {
         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()).last("limit 1"));
            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,invoke.getRegionChildCodesList(),invoke.getIsAdministrator());
      List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(districtTree);
      stringNodeTree.forEach(node -> recursion(node));
      return stringNodeTree;
@@ -97,6 +131,7 @@
   /**
    * 去除空的数据组
    *
    * @param node
    */
   private void recursion(TreeStringNode node) {
@@ -109,6 +144,7 @@
   /**
    * 小区表 自定义获取详情
    *
    * @param district
    * @return
    */