使用基于键列表的 Streams 过滤器映射

Question

我有一个特殊问题，想知道 Java 8 Streams API 是否可以解决它。我知道这可以在使用 Streams API 之外完成，但如果可以使用 Streams 完成，我不想添加与尝试实现该目标相关的所有样板代码。我有地图

Map<String, String> greetings = new HashMap<>();
greetings.put("abc", "Hello");
greetings.put("def", "Goodbye");
greetings.put("ghi", "Ciao");
greetings.put("xyz", "Bonsoir");

还有一个键列表：

List<String> keys = Arrays.asList("def", "zxy");

将上述内容与 Streams API 一起使用，是否可以将其过滤为：

Map<String, String> filteredGreetings = new HashMap<>();
filteredGreetings.put("def", "Goodbye");
filteredGreetings.put("xyz", "Bonsoir");

希望这对我想要实现的目标有意义。

到目前为止，我只有在指定过滤地图的 keySet 的确切键时才能使用它，但是这只会返回一个条目集。我对完全过滤的地图感兴趣，我正在努力实现这一目标。

Answer 1

如果问题is not a typo中的输入和预期输出，也可以保留输入映射的key为：

Map<String, String> futureGreetings = new HashMap<>(greetings);
futureGreetings.keySet().retainAll(keys);

Answer 2

试试这个：

Map<String, String> result = keys.stream()
        .filter(greetings::containsKey)
        .collect(Collectors.toMap(Function.identity(), greetings::get));

或者反过来：

Map<String, String> result = greetings.entrySet().stream()
        .filter(e -> keys.contains(e.getKey()))
        .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

对于第二种方法，我建议将Set<String> 用于更大的keys 列表，因为它对于.contains() 具有O(1) 时间复杂度，因为它不包含任何重复：

Set<String> keys = new HashSet<>(Arrays.asList("def", "zxy"));

Answer 3

试试这个

filteredGreetings= keys.stream()
                  .collect(Collectors.toMap(Function.identity(),key->greetings.get(key)));

甚至对于缺少的密钥，您也可以使用getOrDefault() 方法。

... .collect(Collectors.toMap(Function.identity(),key->greetings.getOrDefault(key,"")));