如何使用Stream API获取与指定数组中键匹配的Map条目？

嘿，这个需求用Stream API实现起来超顺手，我给你捋清楚步骤和最优写法：

核心思路 #

我们需要先把目标键的数组转换成集合（首推HashSet，能大幅提升查找性能），然后通过Stream过滤Map的条目，只保留键在集合中的部分，最后收集成新的Map即可。

完整代码示例 #

假设你的原始数据是这样的：

String[] str = {"ONE", "TWO", "THREE"};
Map<String, String> map = new HashMap<>();
// 给map填充测试数据
map.put("ONE", "1");
map.put("TWO", "2");
map.put("FOUR", "4");
map.put("THREE", "3");

筛选代码如下：

// 第一步：把数组转成HashSet，优化contains的查找效率
Set<String> targetKeys = new HashSet<>(Arrays.asList(str));

// 第二步：用Stream过滤并收集结果
Map<String, String> filteredMap = map.entrySet()
    .stream()
    // 过滤出键存在于目标集合中的条目
    .filter(entry -> targetKeys.contains(entry.getKey()))
    // 将过滤后的条目收集为新的Map
    .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

关键细节说明 #

• 为什么要转成HashSet？：如果直接用Arrays.asList(str).contains(entry.getKey())，每次查找都是线性遍历数组，当数组元素较多时性能会拉胯。而HashSet的contains是O(1)的哈希查找，效率提升非常明显。
• 如果数组里有null值？：如果你的数组包含null，且Map中也可能存在null键，当前写法依然有效；要是你想主动排除null键，可以在filter里加额外判断：filter(entry -> entry.getKey() != null && targetKeys.contains(entry.getKey()))。
• 是否可以直接用数组处理？：当然可以，但仅适合数组元素极少的场景，示例如下：

  Map<String, String> filteredMap = map.entrySet()
      .stream()
      .filter(entry -> Arrays.asList(str).contains(entry.getKey()))
      .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
  

  只是性能不如转成HashSet的版本，不推荐在数据量较大时使用。

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