6、Stream

1、介绍

目的：用于简化集合和数组操作的 API，结合了 Lambda 表达式
Stream 流思想的核心

• 先得到集合或者数组的 Stream 流（就是一根传送带）
• 把元素放上去
• 然后就用这个 Stream 流简化的 API 来方便的操作元素

List<String> names = new ArrayList<>();
Collections.addAll(names, "张无忌", "周芷若", "赵敏", "张强", "张三丰");

names.stream()
    .filter(name -> name.startsWith("张"))
    .filter(name -> name.length() == 3)
    .forEach(System.out::println);

Stream 流的三类方法

• 获取 Stream 流：创建一条流水线，并把数据放到流水线上准备进行操作
• 中间方法：流水线上的操作，一次操作完毕之后，还可以继续进行其他操作
• 终结方法：一个 Stream 流只能有一个终结方法，是流水线上的最后一个操作

2、获取 Stream 流

Stream 操作集合或者数组的第一步是先得到 Stream 流，然后才能使用流的功能
集合获取 Stream 流的方式：可以使用 Collection 接口中的默认方法 stream() 生成流

Map<String, Integer> map = new HashMap<>();

Set<String> keySet = map.keySet();
Stream<String> keyStream = keySet.stream();

Collection<Integer> values = map.values();
Stream<Integer> valuesStream = values.stream();

Set<Map.Entry<String, Integer>> entries = map.entrySet();
Stream<Map.Entry<String, Integer>> entryStream = entries.stream();

数组获取 Stream 流的方式

String[] names = {"张三", "李四", "王五"};

Stream<String> stringStream = Arrays.stream(names);

Stream<String> nameStream = Stream.of(names);

3、中间方法

中间方法也称为非终结方法，调用完成后返回新的 Stream 流可以继续使用，支持链式编程
在 Stream 流中无法直接修改集合、数组中的数据

names.stream().map(new Function<String, String>() {
    @Override
    public String apply(String name) {
        return "亲爱的".concat(name);
    }
}).forEach(System.out::println);

注意：中间方法会返回新的流，原来的流就不能用了，流只能使用一次

Stream<String> stringStream = names.stream();

stringStream.filter(name -> name.startsWith("张")); // 会返回新的流, 但是这里并没有接收新的流对象

stringStream.forEach(System.out::println);          // 原来的流已经不能用了, 这里会报错
// IllegalStateException: stream has already been operated upon or closed

4、终结方法

终结操作方法，调用完成后流就无法继续使用了，原因是不会返回 Stream 了

5、收集 Stream 流

就是把 Stream 流操作后的结果数据恢复到集合或者数组中去
Stream 流：方便操作集合 / 数组的手段
集合 / 数组：才是开发中的目的

Collectors 工具类提供了具体的收集方式

Object[] objects = names.stream().toArray(); // Object 数组

String[] strings = names.stream().toArray(new IntFunction<String[]>() {
    @Override
    public String[] apply(int length) {
        return new String[length];
    }
});
//String[] strings = names.stream().toArray(String[]::new); 指定类型数组

List<String> result = names.stream().filter(name -> name.length() == 3).collect(Collectors.toList()); // 把元素收集到 List 集合中
System.out.println(result);

// 把元素收集到 Map 集合中
List<UserInfo> userInfoList = userService.batchGetUserInfoByUserIds(userIdList);
Map<Long, UserInfo> userInfoMap = userInfoList.stream().collect(Collectors.toMap(UserInfo::getUserId, userInfo -> userInfo));

6、加工演示

// 需求: 把 names 列表转为 Student 数组
public class Student {
    private String name;
}

List<String> names = new ArrayList<>();
Collections.addAll(names, "张无忌", "周芷若", "赵敏", "张强", "张三丰");

// 方式一
List<Student> students = new ArrayList<>();
names.stream().map(new Function<String, Student>() {
    @Override
    public Student apply(String name) {
        return new Student(name);
    }
}).forEach(student -> students.add(student));
System.out.println(students);
// names.stream().map(name -> new Student(name)).forEach(student -> students.add(student));
// names.stream().map(Student::new).forEach(students::add);

// 方式二
List<Student> studentList = names.stream().map(Student::new).collect(Collectors.toList());
System.out.println(studentList);