摘要：經(jīng)過分析和思考，我決定不采用遞歸的方式來編寫樹形數(shù)據(jù)的處理，最終選用來維護樹節(jié)點之間的關(guān)系。以權(quán)限樹為例，做一個樹形數(shù)據(jù)工具類的設(shè)計。

? 在一些管理系統(tǒng)中一般都會用到，會用到一些樹形數(shù)據(jù)，例如部門組織以及權(quán)限等數(shù)據(jù)，都得生成樹形數(shù)據(jù)，需要寫一些樹形數(shù)據(jù)生成工具，一般使用遞歸的方式，性能低下還可能會導(dǎo)致爆棧。經(jīng)過分析和思考，我決定不采用遞歸的方式來編寫樹形數(shù)據(jù)的處理，最終選用hasMap來維護樹節(jié)點之間的關(guān)系。以權(quán)限樹為例，做一個樹形數(shù)據(jù)工具類的設(shè)計。

SET FOREIGN_KEY_CHECKS=0;

-- ----------------------------
-- Table structure for permission
-- ----------------------------
DROP TABLE IF EXISTS `permission`;
CREATE TABLE `permission` (
  `id` int(10) unsigned NOT NULL AUTO_INCREMENT COMMENT "權(quán)限id",
  `name` varchar(32) NOT NULL COMMENT "權(quán)限名稱",
  `url` varchar(64) DEFAULT NULL COMMENT "資源url",
  `type` int(11) NOT NULL COMMENT "權(quán)限類型，1:模塊，2：菜單，3：url資源",
  `parent_id` int(11) NOT NULL DEFAULT "0" COMMENT "上級資源id",
  `icon` varchar(64) DEFAULT NULL COMMENT "菜單圖標(biāo)",
  `sort` int(11) DEFAULT NULL COMMENT "排序",
  `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT "創(chuàng)建時間",
  `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT "更新時間",
  `operator` varchar(32) DEFAULT NULL COMMENT "操作者",
  `level` int(11) NOT NULL DEFAULT "0",
  `code` varchar(32) DEFAULT NULL,
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=19 DEFAULT CHARSET=utf8;

java bean的設(shè)計依據(jù)數(shù)據(jù)庫的表來進行設(shè)計，構(gòu)造方法以及get、set方式使用lombok注解。

package com.lk.permission.common.pojo;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;
import lombok.experimental.Accessors;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;

//@SuppressWarnings("serail")
@ToString
@NoArgsConstructor
@AllArgsConstructor
@Data
@Accessors(chain = true)
public class Permission {

    /** 權(quán)限ID */
    private Integer id;

    /** 權(quán)限名稱 */
    private String name;

    /** 權(quán)限編碼 */
    private String code;

    /** 菜單圖標(biāo) */
    private String icon;

    /** 資源類型 */
    private Integer type;

    /** 資源地址 */
    private String url;

    /** 層級 */
    private Integer level;

    /** 上層ID */
    private Integer parentId;

    /** 排序 */
    private Integer sort;

    /** 創(chuàng)建時間 */
    private Date createTime;

    /** 更新時間 */
    private Date updateTime;

    /** 操作人員 */
    private String operator;

    /** 下級權(quán)限 */
    private List subPermissions = new ArrayList<>();
}

我只提供一個實現(xiàn)思路，具體可根據(jù)自己業(yè)務(wù)實現(xiàn)，通過hasMap方式，一個是性能有了較大的提升，另一個不用擔(dān)心爆棧的風(fēng)險。下面維護關(guān)系已經(jīng)給出，可進行適當(dāng)?shù)母脑靵韺崿F(xiàn)自己具體業(yè)務(wù)需求。

package com.lk.permission.system.service.impl;

import com.lk.permission.common.pojo.Permission;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class TreeService {

    private Map map;

    private List permissions;

    /**
     * 初始化多叉樹
     * @param permissions
     */
    TreeService(List permissions){

        this.permissions = permissions;
        this.map = new HashMap<>();

        for (Permission permission : permissions){
            this.map.put(permission.getId(),permission);
        }
        createTree();
    }

    /**
     * 創(chuàng)建多叉樹
     */
    private void createTree(){
        for (Permission permission : this.permissions){
            if (this.map.containsKey(permission.getParentId())){
                this.map.get(permission.getParentId()).getSubPermissions().add(permission);
                System.out.println(permission.toString());
            }
        }
    }

    /**
     * 根據(jù)層級獲取多叉樹
     * @param level
     * @return
     */
    List getPermissionsByLevel(Integer level){
        return this.permissions.parallelStream().filter(permission -> permission.getLevel() == level).collect(Collectors.toList());
    }

    /**
     * 根據(jù)樹的id獲取多叉樹
     * @param id
     * @return
     */
    Permission getPermissionById(Integer id){
        return this.map.get(id);
    }

    /**
     * 向多叉樹添加子節(jié)點
     * @param permission
     */
    public void addChild(Permission permission){
        if (this.map.containsKey(permission.getParentId())){
            ((Permission)this.map.get(permission.getParentId())).getSubPermissions().add(permission);
        }
        this.map.put(permission.getId(),permission);
    }


}

權(quán)限選擇

樹形表格