JAVA代码优雅实现之Stream流（二）

前言：上一篇文章笔者介绍了大部分的Stream流，这篇文章笔者接着介绍剩余的Stream流。

一、必会方法

1.1 sorted方法

/**
 * 按照年龄从小到大排列
 */
@Test
public void testSorted(){
    Class studentClass = initTestData();

    studentClass.getStudents().stream()
            .distinct()
            .sorted((student1, student2) -> student1.getAge().compareTo(student2.getAge()))
            .forEach(student -> System.out.println(student));
}

上面两个重载方法，笔者只使用了一个，对于另一个空餐的sorted方法，如果我们想使用这个，那么我们需要在流元素对应的类（也就是Student类）中实现Comparator接口，不然使用sorted方法就会报错，因为不实现Comparator接口怎么这道Student的比较规则是什么呢？

2.2 min方法

    /**
     * 获取年龄最小的学生的个人信息
     */
    @Test
    public void testMin(){
        Class studentClass = initTestData();
        Optional<Student> minValue = studentClass.getStudents().stream()
                .min((student1, student2) -> student1.getAge().compareTo(student2.getAge()));

        minValue.ifPresent(student -> System.out.println(student));
    }

2.3 max方法

    /**
     * 获取年龄最大的学生的个人信息
     */
    @Test
    public void testMax(){
        Class studentClass = initTestData();
        Optional<Student> maxValue = studentClass.getStudents().stream()
                .max((student1, student2) -> student1.getAge().compareTo(student2.getAge()));

        maxValue.ifPresent(student -> System.out.println(student));
    }

2.4 collect方法

    /**
     * 筛选出所有年龄大于20岁的学生
     */
    @Test
    public void testCollect(){
        Class studentClass = initTestData();

        List<Student> newStudentClass = studentClass.getStudents().stream()
                .distinct()
                .filter(student -> student.getAge().compareTo(20) > 0)
                .collect(Collectors.toList());

        //遍历筛选后得到的集合
        newStudentClass.stream()
                .forEach(student -> System.out.println(student));
    }

这个方法非常重要，即开始时将集合变成Stream流处理，然后使用collect方法将处理过的stream流重新变成集合。

2.5 findFirst方法

    /**
     * 找到第一个年龄大于20岁的学生
     */
    @Test
    public void testFindFirst(){
        Class studentClass = initTestData();

        Student student1 = studentClass.getStudents().stream()
                .distinct()
                .filter(student -> student.getAge().compareTo(20) > 0)
                .findFirst().orElse(null);

        System.out.println(student1);
    }

2.6 flatMap方法

    /**
     * 统计学生选的所有科目
     */
    @Test
    public void testFlatMap(){
        Class studentClass = initTestData();

        List<String> subjects = studentClass.getStudents().stream()
                .flatMap(student -> student.getCourses().stream().map(course -> course.getSubject()))
                .distinct().collect(Collectors.toList());

        subjects.stream().forEach(subject -> System.out.println(subject));
    }

看一下使用map的效果

    /**
     * 统计学生选的所有科目:使用map
     */
    @Test
    public void testMapV2(){
        Class studentClass = initTestData();

        List<Stream<String>> courses = studentClass.getStudents().stream()
                .map(student -> student.getCourses().stream().map(course -> course.getSubject()))
                .distinct().collect(Collectors.toList());

        System.out.println(courses);
    }

可以看到使用map后通过collect收集后的元素是list<Stream<String>>集合，最后打印输出明显不是我们想要的。

对于这个方法，可能有朋友会稍微有点晕，下面画图演示一下。

其实通俗点来讲就是flatMap是扁平化处理，即flatMap可以将多个流合并成一个流之后进行处理。即如果集合内部有集合，使用flatMap可以将集合内部的集合展开，然后所有展开后的元素合并成一个流进行处理。