SpringBoot樹獲取方法總結

• 2022 年 2 月 14 日
• 筆記
• JAVA, springboot

最近項目中有需要獲取全國行政區劃省-市-區縣-鄉鎮、街道辦的樹狀結構數據，現將自己獲取樹的方法總結如下，有不到之處，敬請批評指正！

一、全國行政區劃數據的整理以及獲取

1. 獲取地址：//pan.baidu.com/s/19JxdWH1R4z_HrrodFJUfjw?pwd=7bey    提取碼:7bey 
2. 創建表

 二、java程式碼示例

• AreaTreeMapper.xml的編寫　

　　　　<select id="selectSymptomTreeNodeJson" resultType="com.boot.mybatis.entity.CategoryTreeNode">
　　　　　　SELECT dq_id as id,dq_name AS title,parent_id as parentId FROM ft_sys_dq
　　　　</select>

• CategoryTreeNode實體類的編寫
  

  • /**
    * @Project:
    * @Description:
    * @Auther: songwp
    * @Date: 2022/2/14 19:41
    **/
    @Data
    public class CategoryTreeNode {
    private BigInteger id;
    private String title;
    private BigInteger parentId;
    private List<CategoryTreeNode> children = new ArrayList<>();
    }

• AreaTreeMapper介面編寫

　　　　/**
　　　　* @Project:
　　　　* @Description:
　　　　* @Auther: songwp
　　　　* @Date: 2022/2/14 19:44
　　　　**/
　　　　@Mapper
　　　　public interface AreaTreeMapper {
　　　　　　List<CategoryTreeNode> selectSymptomTreeNodeJson();
　　　　}

• AreaTreeService業務介面編寫

　　　　/**
　　　　* @Project:
　　　　* @Description:
　　　　* @Auther: songwp
　　　　* @Date: 2022/2/14 19:52
　　　　**/
　　　　public interface AreaTreeService {

　　　　　　List<CategoryTreeNode> getSymptomTreeJson();
　　　　}

• AreaTreeServiceImpl的業務實現類編寫

  • /**
    * @Project:
    * @Description:
    * @Auther: songwp
    * @Date: 2022/2/14 19:53
    **/
    @Service
    public class AreaTreeServiceImpl implements AreaTreeService {
    
    @Resource
    AreaTreeMapper areaTreeMapper;
    
    @Override
    public List<CategoryTreeNode> getSymptomTreeJson() {
    //從資料庫獲取數據的list（一次性獲取的所有數據）
    List<CategoryTreeNode> list= areaTreeMapper.selectSymptomTreeNodeJson();
    List<CategoryTreeNode> resultList = new ArrayList<>();
    TreeUtil treeUtilNew = new TreeUtil();
    resultList = treeUtilNew.build(list); //進行封裝
    return resultList; //返回樹形結構json
    }
    }

• AreaTreeController的編寫　　

  • /**
    * @Project:
    * @Description:
    * @Auther: songwp
    * @Date: 2022/2/14 19:57
    **/
    @RestController
    public class AreaTreeController {
    
    @Autowired
    AreaTreeService areaTreeService;
    
    @GetMapping("/getTreeList")
    public List<CategoryTreeNode> getTreeList(){
    return areaTreeService.getSymptomTreeJson();
    }
    
    }

• TreeUtil的編寫

　　　　/**
　　　　* @Project:
　　　　* @Description: 構架樹結構樹形json--好用
　　　　* @Auther: songwp
　　　　* @Date: 2022/2/14 19:39
　　　　**/
　　　　public class TreeUtil {

　　　　/**
　　　　* 根據pid，構建樹節點
　　　　*/
　　　　public static <T extends CategoryTreeNode> List<T> build(List<T> treeNodes, Integer pid) {

　　　　　　List<T> treeList = new ArrayList<>();
　　　　　　for(T treeNode : treeNodes) {
　　　　　　　　if (pid.equals(treeNode.getParentId())) {
　　　　　　　　　　treeList.add(findChildren(treeNodes, treeNode));
　　　　　　　　}
　　　　　　}
　　　　return treeList;
　　　　}

　　　　/**
　　　　* 查找子節點
　　　　*/
　　　　private static <T extends CategoryTreeNode> T findChildren(List<T> treeNodes, T rootNode) {
　　　　　　for(T treeNode : treeNodes) {
　　　　　　　　if(rootNode.getId().equals(treeNode.getParentId())) {
　　　　　　　　　　rootNode.getChildren().add(findChildren(treeNodes, treeNode));
　　　　　　　　}
　　　　　　}
　　　　　　return rootNode;
　　　　}

　　　　/**
　　　　* 構建樹節點
　　　　*/
　　　　public static <T extends CategoryTreeNode> List<T> build(List<T> treeNodes) {
　　　　　　List<T> result = new ArrayList<>();
　　　　　　//list轉map
　　　　　　Map<BigInteger, T> nodeMap = new LinkedHashMap<>(treeNodes.size());
　　　　　　for(T treeNode : treeNodes){
　　　　　　　　nodeMap.put(treeNode.getId(), treeNode);
　　　　　　}
　　　　　　for(T node : nodeMap.values()) {
　　　　　　　　T parent = nodeMap.get(node.getParentId());
　　　　　　　　　　if(parent != null && !(node.getId().equals(parent.getId()))){
　　　　　　　　　　　　parent.getChildren().add(node);
　　　　　　　　　　　　continue;
　　　　　　　　　　　　}

　　　　　　　　　　result.add(node);
　　　　　　　　　　}
　　　　　　　　return result;
　　　　　　　　}
　　　　}

三、調用介面獲取數據如下：