如何使用Java 8 Stream按ID匹配替换List集合中的自定义对象

实现思路 #

最优方案的核心是先将otherEntityList转为以id为key、Entity对象为value的Map，将匹配查询的时间复杂度降到O(1)，再遍历mainEntityList完成替换，整体时间复杂度为O(m+n)（m为otherEntityList长度，n为mainEntityList长度），远优于嵌套循环的O(mn)。

前置Entity类定义 #

假设你的Entity类结构如下：

class Entity {
    private Integer id;
    private String name;

    public Entity(Integer id, String name) {
        this.id = id;
        this.name = name;
    }

    // getter
    public Integer getId() {
        return id;
    }
    public String getName() {
        return name;
    }
    // 其他setter、toString等方法省略
}

核心代码实现 #

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

public class ListReplaceDemo {
    public static void main(String[] args) {
        // 示例数据
        List<Entity> mainEntityList = List.of(new Entity(1, "abc"), new Entity(2, "xyz"));
        List<Entity> otherEntityList = List.of(new Entity(2, "value"));

        // 第一步：将other列表转为id为键的Map
        Map<Integer, Entity> otherIdMap = otherEntityList.stream()
                .collect(Collectors.toMap(
                        Entity::getId,
                        Function.identity(),
                        // 若other列表存在重复id，保留后出现的对象，可按需调整合并逻辑
                        (oldEntity, newEntity) -> newEntity
                ));

        // 第二步：生成替换后的新列表（符合Stream不可变设计，推荐使用）
        List<Entity> updatedMainList = mainEntityList.stream()
                .map(entity -> otherIdMap.getOrDefault(entity.getId(), entity))
                .collect(Collectors.toList());

        // 若必须修改原mainEntityList，可不用Stream直接调用List.replaceAll（写法更简洁）
        // mainEntityList.replaceAll(entity -> otherIdMap.getOrDefault(entity.getId(), entity));
    }
}

输出验证 #

替换后的updatedMainList和预期完全一致：

[{id:1, name:"abc"}, {id:2, name:"value"}]

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