package com.gblfy.gxts;
import com.google.common.collect.Lists;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.junit.Before;
import org.junit.Test;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
/**
* 案例3:
* 权限管理功能模块。查询某用户所有角色所包含的权限名称。
*/
public class CaseThree {
/**
* 角色模型
*/
@Data
@AllArgsConstructor
@NoArgsConstructor
class Role {
/**
* 权限列表
*/
private List<Permission> permissionList;
}
@Data
@AllArgsConstructor
@NoArgsConstructor
class Permission {
//权限名称
private String name;
}
/**
* 用户角色列表
*/
List<Role> roleList;
//初始化数据
@Before
public void init() {
//声明用户角色列表
roleList = new ArrayList<>();
//声明一个管理员角色
Role adminRole = new Role();
//声明一个管理员权限列表
ArrayList<Permission> adminPermissionList = Lists.newArrayList(
new Permission("删除"),
new Permission("查看"),
new Permission("导出")
);
adminRole.setPermissionList(adminPermissionList);
//声明一个用户角色
Role userRole = new Role();
//声明一个用户权限列表
ArrayList<Permission> userPermissionList = Lists.newArrayList(
new Permission("新建"),
new Permission("修改"),
new Permission("删除"),
new Permission("查看")
);
userRole.setPermissionList(userPermissionList);
//将权限放入角色列表中
roleList.add(adminRole);
roleList.add(userRole);
}
/**
* 查找去重后权限列表
*/
@Test
public void findNoScoreStudentList() {
roleList.stream()
.flatMap(role -> role.permissionList.stream())
.peek(permission ->
System.out.println("新的流元素:" + permission))
.distinct()
// .forEach(permission -> System.out.println(permission));
.collect(Collectors.toList());
}
}