权限之权限模块树的书写（重点）

第一步：权限模块树形结构

package com.mall.dto;

import com.google.common.collect.Lists;
import com.mall.model.SysAclModule;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;
import org.springframework.beans.BeanUtils;
import java.util.List;

@Getter
@Setter
@ToString
public class AclModuleLevelDto extends SysAclModule {
    //自己下面包含自己
    private List<AclModuleLevelDto> aclModuleList= Lists.newArrayList();

    public static AclModuleLevelDto adapt(SysAclModule aclModule){
        AclModuleLevelDto dto=new AclModuleLevelDto();
        //使用cope bean的工具
        BeanUtils.copyProperties(aclModule,dto);
        return dto;
    }
}

第二步：等级帮助类

package com.mall.util;

import org.apache.commons.lang3.StringUtils;

public class LevelUtil {

    public final static String SEPARATOR=".";

    public final static String ROOT="0";

    public static String calculateLevel(String parentLevel,int parentId){
        if(StringUtils.isBlank(parentLevel)){
            return ROOT;
        }else{
            return StringUtils.join(parentLevel,SEPARATOR,parentId);
        }
    }
}

 

第三步：

package com.mall.service;

import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.Lists;
import com.google.common.collect.Multimap;
import com.mall.dao.DepartMapper;
import com.mall.dao.SysaclModuleMapper;
import com.mall.dto.AclModuleLevelDto;
import com.mall.dto.DeptLevelDto;
import com.mall.model.Depart;
import com.mall.model.SysAclModule;
import com.mall.util.LevelUtil;
import org.apache.commons.collections.CollectionUtils;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

@Service
public class SysTreeService {
    @Resource
    private DepartMapper departMapper;//注入部门，DeptLevelDto是作为部门层级的类

    @Resource
    private SysaclModuleMapper sysaclModuleMapper;//注入权限模块

    public List<AclModuleLevelDto> aclModuleTree(){
        //第一步：获取所有的权限模块
        List<SysAclModule> aclModuleList=sysaclModuleMapper.getAllAclModule();
        //第二步：定义我们的树形结构
        List<AclModuleLevelDto> dtoList=Lists.newArrayList();
        //第三步：将我们从数据库取出来的数据遍历放到我们定义的dtoList里面去
        for(SysAclModule aclModule:aclModuleList){
            dtoList.add(AclModuleLevelDto.adapt(aclModule));
        }
        //根据当前的结构适配一颗权限树
        return aclModuleListToTree(dtoList);
    }

    public List<AclModuleLevelDto> aclModuleListToTree(List<AclModuleLevelDto> dtoList){
        if(CollectionUtils.isEmpty(dtoList)){
            return Lists.newArrayList();
        }
        //level->[aclmodule1,aclmodule2...]
        Multimap<String,AclModuleLevelDto> levelAclModuleMap= ArrayListMultimap.create();
        List<AclModuleLevelDto> rootList=Lists.newArrayList();//这是一级部门
        for(AclModuleLevelDto dto:dtoList){//将传过来的部门进行遍历
            levelAclModuleMap.put(dto.getLevel(),dto);//将它存在levelDeptMap里面，等级为键，部门为值
            if(LevelUtil.ROOT.equals(dto.getLevel())){//如何获得到的部门的level为0，表示该部门为一级部门
                rootList.add(dto);//将一级部门放入到一级的List里面
            }
        }
        //通过比较器来进行排序
        Collections.sort(rootList,aclModuleSeqComparator);
        //开始传入首层元素
        transformAclModuleTree(rootList,LevelUtil.ROOT,levelAclModuleMap);

        return rootList;
    }
    //权限模块的比较器的处理
    public Comparator<AclModuleLevelDto> aclModuleSeqComparator=new Comparator<AclModuleLevelDto>() {
        @Override
        public int compare(AclModuleLevelDto o1, AclModuleLevelDto o2) {
            return o1.getSeq()-o2.getSeq();
        }
    };
    //转化成一颗树的方法(将每一个点做成树形结构，每个点就是一棵树)
    public void  transformAclModuleTree(List<AclModuleLevelDto> dtoList,String level,Multimap<String,AclModuleLevelDto> levelAclModuleMap){
        for(int i=0;i<dtoList.size();i++){
            AclModuleLevelDto dto=dtoList.get(i);
            //得到当前要处理的层级
            String nextLevel=LevelUtil.calculateLevel(level,dto.getId());
            List<AclModuleLevelDto> tempList = (List<AclModuleLevelDto> )levelAclModuleMap.get(nextLevel);
            if(CollectionUtils.isNotEmpty(tempList)){
                //进行排序
                Collections.sort(tempList,aclModuleSeqComparator);
                //将处理好的list传给AclModuleLevelDto
                dto.setAclModuleList(tempList);
                //处理它的下一层
                transformAclModuleTree(tempList,nextLevel,levelAclModuleMap);
            }
        }
    }
}