linwe
2023-12-01 5d5cd3ad979a0b85ab8c7034dcad1663fdcf71be
src/main/java/org/springblade/modules/district/service/impl/DistrictServiceImpl.java
@@ -17,6 +17,9 @@
package org.springblade.modules.district.service.impl;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springblade.common.node.TreeIntegerNode;
import org.springblade.common.node.TreeStringNode;
import org.springblade.common.utils.NodeTreeUtil;
import org.springblade.modules.district.entity.DistrictEntity;
import org.springblade.modules.district.vo.DistrictVO;
import org.springblade.modules.district.mapper.DistrictMapper;
@@ -24,6 +27,8 @@
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;
import java.util.List;
/**
 * 小区表 服务实现类
@@ -39,5 +44,39 @@
      return page.setRecords(baseMapper.selectDistrictPage(page, district));
   }
   /**
    * 获取小区树
    * @param district
    * @return
    */
   @Override
   public Object getDistrictTree(DistrictVO district) {
      List<TreeStringNode> stringNodeTree = NodeTreeUtil.getStringNodeTree(baseMapper.getDistrictTree(district));
      stringNodeTree.forEach(node -> recursion(node));
      return stringNodeTree;
   }
   /**
    * 去除空的数据组
    * @param node
    */
   private void recursion(TreeStringNode node) {
      if (node.getChildren() != null && node.getChildren().size() > 0) {
         node.getChildren().forEach(node2 -> recursion(node2));
      } else {
         node.setChildren(null);
      }
   }
   /**
    * 小区表 自定义获取详情
    * @param district
    * @return
    */
   @Override
   public DistrictVO getDetail(DistrictVO district) {
      // 小区自定义获取详情查询并返回
      return baseMapper.getDetail(district);
   }
}