jdk8中的StreamAPI

Posted 2021-01-02 psy-code

1.实体类

package com.zy.model;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;

import java.io.Serializable;

@Data
@Builder
@AllArgsConstructor
public class Employee implements Serializable {

    private Integer id;
    private String name;
    private Integer age;
    private double salary;

}

备注,该实体类采用了JetBrains的插件,IDEA编译器需要先导入插件(链接:http://plugins.jetbrains.com/plugin/6317-lombok-plugin)

并在pom.xml中引入Jar包

    <dependency>
      <groupId>org.projectlombok</groupId>
      <artifactId>lombok</artifactId>
      <version>1.16.20</version>
      <scope>provided</scope>
    </dependency>

2.Stream 的操作步骤

/*
* 
* 1. 创建 Stream
*
* 2. 中间操作
*
* 3. 终止操作
*/

2.1创建Stream

 

package com.zy.stream;

import com.zy.model.Employee;
import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;


public class TestStreamAPI01 {


    // 1.创建stream流的方式
    @Test
    public void test1(){
        // 1.1第一种方式:通过Collection系列集合提供的Stream或paralleStream
        List<Object> list = new ArrayList<>();
        // 串行流,按顺序
        Stream<Object> streamList1 = list.stream();
        // 并行流,不按顺序
        Stream<Object> parallelStream = list.parallelStream();

        // 1.2第二种方式:通过Arrays中的静态方法获取数组流
        Employee[] employee = new Employee[10];
        Stream<Employee> streamArray = Arrays.stream(employee);

        // 1.3第三种方式:通过Stream类中的of()方法创建流
        Stream<String> stringStream = Stream.of("a", "b", "c");

        // 1.4创建无限流
        // 1.4.1第一种方式:迭代
        Stream.iterate(0, (x) -> (x + 1))
                .limit(20)
                .forEach(System.out::println);
        System.out.println("=================");
        // 1.4.2第二种方式:生成
        Stream.generate(()->Math.random())
                .limit(10)
                .forEach(System.out::println);
    }

}

 

2.2Stream的中间操作

package com.zy.stream;

import com.zy.model.Employee;
import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;


public class TestStreamAPI02 {


    // 2.中间操作
    /**
     *
     * 筛选与切片
     * filter--接收Lambda,从流中排除某些元素
     * limit(n)--截断流,使其元素不超过给定数量n
     * skip(n)--跳过元素,返回一个扔掉了前n个元素的流,若元素不足n个,则返回null,与limit(n)互补
     * distinct--筛选,通过流所生成的hashCode(),和equals()去除重复元素
     *
     * map--接收Lambda表达式,将元素转换成其他形式或提取信息,接收一个函数作为参数,该函数会应用到每个元素上,并将其映射成一个新的元素.
     * flatMap--接收一个函数作为参数,将流中的每个值都换成另一个流,然后把所有流凑成一个流
     * map与flatMap的关系类似于List中的add()与addALl()
     *
     * sorted--自然排序
     *
     */

    List<Employee> emps = Arrays.asList(
            new Employee(102, "李四", 59, 6666.66),
            new Employee(101, "张三", 18, 9999.99),
            new Employee(103, "王五", 28, 3333.33),
            new Employee(104, "赵六", 8, 7777.77),
            new Employee(104, "赵六", 38, 7777.77),
            new Employee(104, "赵六", 38, 7777.77),
            new Employee(105, "田七", 38, 5555.55)
    );

    // 筛选元素
    @Test
    public void fn1(){
        // 内部迭代,由StreamAPI完成
        Stream<Employee> employeeStream = emps.stream()
                .filter((e) -> e.getAge() > 30);
        // 一次性执行完所有内容,即"惰性求值"
        employeeStream.forEach(System.out::println);
    }

    // 只选前几个元素
    @Test
    public void fn2(){
        emps.stream()
                .filter((e)->e.getSalary()>6000)
                .limit(3)
                .forEach(System.out::println);
    }

    // 跳过前几个元素
    @Test
    public void fn3(){
        emps.stream()
                .filter((e)->e.getAge()>20)
                .skip(2)
                .forEach(System.out::println);

    }

    // 去重
    @Test
    public void fn4(){
        emps.stream()
                .distinct()
                .forEach(System.out::println);
    }

    // map
    @Test
    public void fn5(){
        emps.stream()
                .map(Employee::getName)
                .forEach(System.out::println);
    }

    @Test
    public void fn6(){
        List<String> list = Arrays.asList("aaa", "bbb", "ccc", "ddd");
        Stream<Character> characterStream = list.stream()
                .flatMap(TestStreamAPI02::filterCharacter);
        characterStream.forEach(System.out::println);
    }

    public static Stream<Character> filterCharacter(String str){
        List<Character> list = new ArrayList<>();
        for(Character character : str.toCharArray()){
            list.add(character);
        }
        return list.stream();
    }

    // 排序
    @Test
    public void fn7(){
        List<String> list = Arrays.asList("b", "c", "a", "m", "d");
        list.stream()
                .sorted()
                .forEach(System.out::println);
    }

    @Test
    public void fn8(){
        emps.stream()
                .sorted((e1,e2)->{
                    if (e1.getAge().equals(e2.getAge())){
                        return e1.getName().compareTo(e2.getName());
                    }else {
                        return -e1.getAge().compareTo(e2.getAge());
                    }
                })
                .forEach(System.out::println);
    }
}

2.3Stream的终止操作

package com.zy.stream;

import com.zy.model.Employee;
import org.junit.Test;

import java.util.Arrays;
import java.util.List;
import java.util.Optional;

public class TestStreamAPI03 {

    //3. 终止操作
    /*
        allMatch——检查是否匹配所有元素
        anyMatch——检查是否至少匹配一个元素
        noneMatch——检查是否没有匹配的元素
        findFirst——返回第一个元素
        findAny——返回当前流中的任意元素
        count——返回流中元素的总个数
        max——返回流中最大值
        min——返回流中最小值
     */
    List<Employee> emps = Arrays.asList(
            new Employee(102, "李四", 59, 6666.66),
            new Employee(101, "张三", 18, 9999.99),
            new Employee(103, "王五", 28, 3333.33),
            new Employee(104, "赵六", 8, 7777.77),
            new Employee(104, "赵六", 38, 7777.77),
            new Employee(104, "赵六", 38, 7777.77),
            new Employee(105, "田七", 38, 5555.55)
    );

    // 终止操作集锦
    @Test
    public void fn1(){
        boolean b = emps.stream()
                .allMatch((e) -> e.getAge() > 7);
        System.out.println("allMatch()================="+b);

        boolean b1 = emps.stream()
                .anyMatch((e) -> e.getAge() > 70);
        System.out.println("anyMatch()================="+b1);

        boolean b2 = emps.stream()
                .noneMatch((e) -> e.getAge() == 9);
        System.out.println("noMatch()================"+b2);

        Optional<Employee> optional = emps.stream()
                .sorted((e1, e2) -> e1.getAge().compareTo(e2.getAge()))
                .findFirst();
        System.out.println("findFirst()================" + optional.get());

        Optional<Employee> any = emps.stream()
                .filter((e)->e.getAge()>30)
                .findAny();
        System.out.println("findAny()============="+any.get());

        long count = emps.stream()
                .count();
        System.out.println("count()========="+count);

        Optional<Employee> max = emps.stream()
                .max((e1, e2) -> Double.compare(e1.getSalary(), e2.getSalary()));
        System.out.println("max()==================="+max);

        Optional<Employee> min = emps.stream()
                .min((e1, e2) -> Double.compare(e1.getSalary(), e2.getSalary()));
        System.out.println("min()====================="+min);


    }
}