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上，有任何疑問能夠聯繫我~

你們能夠關注個人我的公衆號~