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

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

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)限名稱(chēng)",
  `url` varchar(64) DEFAULT NULL COMMENT "資源url",
  `type` int(11) NOT NULL COMMENT "權(quán)限類(lèi)型，1:模塊，2：菜單，3：url資源",
  `parent_id` int(11) NOT NULL DEFAULT "0" COMMENT "上級(jí)資源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)建時(shí)間",
  `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT "更新時(shí)間",
  `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ì)依據(jù)數(shù)據(jù)庫(kù)的表來(lái)進(jìn)行設(shè)計(jì)，構(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)限名稱(chēng) */
    private String name;

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

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

    /** 資源類(lèi)型 */
    private Integer type;

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

    /** 層級(jí) */
    private Integer level;

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

    /** 排序 */
    private Integer sort;

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

    /** 更新時(shí)間 */
    private Date updateTime;

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

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

我只提供一個(gè)實(shí)現(xiàn)思路，具體可根據(jù)自己業(yè)務(wù)實(shí)現(xiàn)，通過(guò)hasMap方式，一個(gè)是性能有了較大的提升，另一個(gè)不用擔(dān)心爆棧的風(fēng)險(xiǎn)。下面維護(hù)關(guān)系已經(jīng)給出，可進(jìn)行適當(dāng)?shù)母脑靵?lái)實(shí)現(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;

    /**
     * 初始化多叉樹(shù)
     * @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)建多叉樹(shù)
     */
    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ù)層級(jí)獲取多叉樹(shù)
     * @param level
     * @return
     */
    List getPermissionsByLevel(Integer level){
        return this.permissions.parallelStream().filter(permission -> permission.getLevel() == level).collect(Collectors.toList());
    }

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

    /**
     * 向多叉樹(shù)添加子節(jié)點(diǎn)
     * @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)限選擇

樹(shù)形表格