如何使用Java Streams按id()将User列表转换为Map<Long, User>

嘿，这事儿好办！用Java 8+的Stream API结合Collectors.toMap()就能轻松实现你的需求，我给你写个完整的示例：

完整实现代码 #

首先是你提供的User类（我调整了格式让它更易读）：

public final class User {
    private final Long id;
    private final String name;

    public User(final Long id, final String name) {
        this.id = id;
        this.name = name;
    }

    public Long id() {
        return this.id;
    }

    public String name() {
        return this.name;
    }

    // 可选：重写toString方便调试输出
    @Override
    public String toString() {
        return "User{id=" + id + ", name='" + name + '\'' + '}';
    }
}

然后是核心的转换逻辑：

import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

public class UserMapExample {
    public static void main(String[] args) {
        // 创建目标User列表
        List<User> userList = List.of(
                new User(1L, "First"),
                new User(2L, "Second")
        );

        // 转换为Map<Long, User>
        Map<Long, User> userMap = userList.stream()
                .collect(Collectors.toMap(
                        User::id,  // 提取Map的键：调用User的id()方法
                        Function.identity()  // 提取Map的值：直接使用User对象本身
                ));

        // 测试输出验证结果
        userMap.forEach((id, user) -> System.out.println(id + " -> " + user));
    }
}

关键部分解释 #

• userList.stream()：将List转换为Stream，开启流式操作的入口
• Collectors.toMap(keyMapper, valueMapper)：
  • 第一个参数User::id是键映射器，用方法引用简化了user -> user.id()的写法，负责从每个User对象中提取Map的键（即id值）
  • 第二个参数Function.identity()是值映射器，它会直接返回输入的对象本身，等价于user -> user，这里就是把User对象作为Map的值

处理重复id的特殊情况 #

如果你的列表中可能存在id重复的User对象，默认的toMap()会抛出IllegalStateException。这时可以添加第三个参数处理冲突：

// 冲突时保留第一个出现的User对象
Map<Long, User> userMap = userList.stream()
        .collect(Collectors.toMap(
                User::id,
                Function.identity(),
                (existingUser, newUser) -> existingUser
        ));

// 冲突时保留最后一个出现的User对象
Map<Long, User> userMap = userList.stream()
        .collect(Collectors.toMap(
                User::id,
                Function.identity(),
                (existingUser, newUser) -> newUser
        ));

这样就能完美实现你想要的转换啦！

内容的提问来源于stack exchange，提问作者Roman Cherepanov