Stream distinct去重案例

系统管理员 2022-10-16 07:51 243阅读 0赞

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());
        }
    }