Spring Security （二)：获取用户权限菜单树

违背的青春

今天说下登陆后获取用户菜单，流程就是根据用户的token而后经过SecurityContextHolder来获取用户信息，根据用户来查询菜单树，其实最主要的就是怎么把后台查询出的数据转换成树形菜单，其实也不难，先后端均可以作，由于树形仍是比较常见的因此能够作成一个通用的类来使用，使用时只须要传入数据就行了。

GenTree

这个就是把数据转成树形的方法了。

public class GenTree {


    /**
     * 递归根节点
     * @param nodes
     * @return
     */
    public static Set<Menu> genRoot(Set<Menu> nodes){
        Set<Menu> root = new HashSet<>();
        //遍历数据
        nodes.forEach(menu -> {
            //当父id是0的时候应该是根节点
            System.out.println(menu.getPer_paerent_id());
            if(menu.getPer_paerent_id() == 0){
                root.add(menu);
            }
        });
        //这里是子节点的建立方法
        root.forEach(menu -> {
            genChildren(menu,nodes);
        });
        //返回数据
        return root;
    }

    /**
     * 递归子节点
     * @param menu
     * @param nodes
     * @return
     */
    private static Menu genChildren(Menu menu, Set<Menu> nodes) {
        //遍历传过来的数据
        for (Menu menu1 :nodes){
            //若是数据中的父id和上面的per_id一致应该就放children中去
            if(menu.getPer_id().equals(menu1.getPer_paerent_id())){
                //若是当前节点的子节点是空的则初始化，若是不为空就加进去
                if(menu.getChildren() == null){
                    menu.setChildren(new ArrayList<Menu>());
                }
                menu.getChildren().add(genChildren(menu1,nodes));
            }
        }
        //返回数据
        return menu;
    }

   
}
复制代码

上面的方法其实也就是几个forEach，递归判断，放入数据。 其中Menu的代码以下，这个就是你要获得菜单的格式了，能够根据本身须要进行改造。

@Data
public class Menu {
    private Integer per_id;
    private Integer per_paerent_id;
    private String per_name;
    private String per_resource;
    private List<Menu> children;
    public Menu(Integer per_id,Integer per_paerent_id,String per_name,String per_resource){
        this.per_id = per_id;
        this.per_paerent_id = per_paerent_id;
        this.per_name = per_name;
        this.per_resource = per_resource;
    }
}

复制代码

而后就是Mapper、Service等的编写了，在Controller层中经过SecurityContextHolder来得到用户信息。

//使用Spring Security 获取用户信息
        UserDetails userDetails = (UserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
复制代码

其实以前的项目应该改一下，我把token放到msg里了，修改了一下，

而后咱们就能够拿着这个 token查询菜单树了，

代码已经更新到 github上，有任何疑问能够联系我~

你们能够关注个人我的公众号~