SpringBoot树获取方法总结

来源： https://www.cnblogs.com/songweipeng/p/15894317.html

最近项目中有需要获取全国行政区划省-市-区县-乡镇、街道办的树状结构数据，现将自己获取树的方法总结如下

一、全国行政区划数据的整理以及获取

创建表如下：


二、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;
　　　　　　　　}
　　　　}

三、调用接口获取数据如下：