Stream distinct去重案例

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

上一篇:13 Python 3 - Dictionary


下一篇:Swift-Mirror源码解析