Java 8: map、filter、reduce 学习笔记

map

private static void map(List<User> userList) {
   println("map begin");
    // 调用 setupUser 方法， 返回 Object[]
    Object[] resultList1 = userList.stream().map(Map::setupUser).toArray();
    
    // 使用 Lambda 表达式， 返回 Object[]
    Object[] resultList2 = userList.stream().map((User user) -> {
        println(user.toString());
        user.name = user.name.replace("s", "S");
        return user;
    }).toArray();
    
    // 使用 Lambda 表达式， 返回 List<User>
    List<User> resultList3 = userList.stream().map((User user) -> {
        println(user.toString());
        user.name = user.name.replace("s", "S");
        return user;
    }).collect(Collectors.toList());
    
    println("map end");
    for (int i = 0; i < resultList1.size(); i++) {
        println(resultList.get(i).toString());
    }
}

// 控制台输出如下：
map begin
id: 1, name: zs
id: 2, name: ls
id: 3, name: ww
id: 4, name: ws
map end
id: 1, name: zS
id: 2, name: lS
id: 3, name: ww
id: 4, name: wS

TIPS

• Lambda 表达式中返回的是什么类型，map 后的集合中就返回什么类型

filter

private static void filter(ArrayList<User> userList) {
    // 获取 id 是偶数，并且 name 包含 w 的用户（过滤掉 id 是奇数，并且 name 不包含 w 的用户）
    List<User> resultList = userList.stream()
                .filter(user -> user.id % 2 == 0)
                .filter(user -> user.name.contains("w"))
                .collect(Collectors.toList());
    for (int i = 0; i < resultList.length; i++) {
        println(resultList[i].toString());
    }
}

// 控制台输出如下：
id: 4, name: ws

reduce

private static void reduce(List<User> userList) {
    println("reduce begin");
    Optional<User> result = userList.stream().reduce((user, user2) -> {
        // 测试代码，完全没有意义
        // 聚合操作，每次遍历，让 id 相加、name 拼接
        user.id += user2.id;
        user.name += user2.name;
        return user;
    });
    println("reduce end");
    println(result.get().toString());
}

// 控制台输出如下：
id: 10, name: zslswwws

TIPS

• reduce 返回的是 Optional 修饰的值

准备工作

User 类

class User {
    int id;
    String name;
    User(int id, String name) {
        this.id = id;
        this.name = name;
    }
    @Override
    public String toString() {
        return "id: " + id + ", name: " + name;
    }
}

User 集合

private List<User> setupData() {
    List<User> userList = new ArrayList<>();
    userList.add(new User(1, "zs"));
    userList.add(new User(2, "ls"));
    userList.add(new User(3, "ww"));
    userList.add(new User(4, "ws"));
    return userList;
}

map 相关

User setupUser(User user) {
    user.name = user.name.replace("s", "S");
    // 这里返回 User，map 操作后，返回的就是 User 集合
    // 如果这里返回 name，则 map 操作后，返回的就是 String 集合
    return user;
}