如何用Java Stream/Collectors筛选字段匹配另一列表的对象并按序返回

现有代码问题 #

你目前的双层循环实现时间复杂度为O(nm)*（n为stores长度，m为userStores长度），数据量较大时性能损耗会非常明显，写法也偏冗余，完全可以用Stream + 哈希查找优化。

优化实现（Stream版本） #

首先你需要先给UserStores类的storeNo字段添加Getter方法，之后可以按以下方式实现：

import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;

private static List<UserStores> getModifiedUserStoresList(
        List<String> stores, List<UserStores> userStores
) {
    // 先把userStores转成 门店编号->UserStores对象 的哈希表，O(m)复杂度
    Map<String, UserStores> storeMap = userStores.stream()
            .collect(Collectors.toMap(
                    UserStores::getStoreNo, // key为门店编号
                    item -> item, // value为对象本身
                    (oldVal, newVal) -> oldVal // 如果有重复门店编号，保留先出现的对象，可按需调整
            ));
    // 按stores的顺序遍历取值，O(n)复杂度
    return stores.stream()
            .map(storeMap::get)
            .filter(Objects::nonNull) // 过滤掉stores里存在但userStores里没有的门店编号，不需要可以删除
            .collect(Collectors.toList());
}

实现说明 #

• 整体时间复杂度降到O(n + m)，远优于原有嵌套循环实现，数据量越大性能优势越明显
• 直接遍历stores列表取值，返回结果天然和stores的顺序完全一致，符合需求
• 如果业务允许stores中存在userStores没有的门店编号，且需要在结果中保留对应位置的空值，删掉filter(Objects::nonNull)即可
• 如果门店编号存在重复的场景，toMap的第三个合并函数可以根据业务调整，比如取最新值可以改成(oldVal, newVal) -> newVal

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