java 8 stream返回Map类型数据

原创

mob64ca12e1881c 2024-01-24 08:52:46 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e1881c的原创作品，请联系作者获取转载授权，否则将追究法律责任

Java 8 Stream返回Map类型数据

在Java 8中，引入了Stream API，提供了一种更加简洁和易于使用的方式来处理集合数据。Stream API提供了很多便捷的方法，可以让我们对集合进行过滤、映射、排序等操作。除了这些基本操作外，Stream API还提供了一种便捷的方式来将集合数据转换为Map类型数据。

在本文中，我们将介绍如何使用Java 8 Stream API来返回Map类型数据，并通过代码示例详细说明其用法。

Stream API简介

在开始之前，让我们先简单了解一下Stream API。Stream代表一个来自特定源的元素序列，可以支持各种操作，例如过滤、映射、排序等。Stream API的主要优点是可以提供一种流式处理集合数据的方式，在处理大型数据集时更加高效。

返回Map类型数据的示例

假设我们有一个包含员工信息的列表，每个员工对象具有id和name属性。我们希望将这个列表转换为一个Map，其中key是员工的id，value是员工的name。下面是一个使用Stream API实现的示例代码：

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

public class StreamExample {
    public static void main(String[] args) {
        List<Employee> employees = List.of(
                new Employee(1, "Alice"),
                new Employee(2, "Bob"),
                new Employee(3, "Charlie")
        );

        Map<Integer, String> employeeMap = employees.stream()
                .collect(Collectors.toMap(Employee::getId, Employee::getName));

        System.out.println(employeeMap);
    }
}

class Employee {
    private int id;
    private String name;

    public Employee(int id, String name) {
        this.id = id;
        this.name = name;
    }

    public int getId() {
        return id;
    }

    public String getName() {
        return name;
    }
}

在上面的代码中，我们首先创建了一个包含三个员工对象的列表。然后，我们使用Stream API的collect方法来将这个列表转换为一个Map。toMap方法接受两个参数，第一个参数是key的提取函数，用于提取员工对象的id，第二个参数是value的提取函数，用于提取员工对象的name。

运行上述代码，我们将得到以下输出：