违背的青春
今天说下登录后获取用户菜单，流程就是根据用户的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上，有任何疑问可以联系我~
