从 Java/Guava 中的某些键的映射中获取所有值？

Question

有没有一种聪明的方法可以从给定某些键的映射中获取所有值？

我想要一个这样的方法：

public static <K, V> Collection<V> getAll(Map<K, V> map, Collection<K> keys)

或者已经是番石榴的方式？

Answer 1

这取决于您希望该方法如何工作。例如，keys 中不在 map 中的元素应该被忽略 A) 还是应该 B) 在返回值集合中表示为 null 还是 C) 应该是一个错误？还要考虑您是否需要实时视图或包含值的单独集合。

对于A，我的偏好是：

Collection<V> values = Collections2.transform(
    Collections2.filter(keys, Predicates.in(map.keySet()),
    Functions.forMap(map));

这将结果限制为映射中实际存在的键的值，并且应该相对有效，即使映射比您设置的键集大得多。想。当然，您可能希望将该结果复制到另一个集合中，具体取决于您想用它做什么。

对于 B，您可以使用 @Michael Brewer-Davis 的解决方案，但 Functions.forMap(map, null) 除外。

对于C，您首先要检查map.keySet().containsAll(keys)，如果false则抛出错误，然后使用 @Michael Brewer-Davis 的解决方案...但请注意，除非您将结果复制到另一个集合中，否则从 map 中删除条目可能会导致代码出现 IllegalArgumentException在某个时刻使用返回的集合。

Answer 2

我同意skaffman的回答，只是不同意他的结论（我认为这比手动迭代更好）。

这里是拼写出来的：

public static <K, V> Collection<V> getAll(Map<K, V> map, Collection<K> keys) {
    return Maps.filterKeys(map, Predicates.in(keys)).values();
}

另外，这是一个非 Guava 版本（Java 8 或更高版本）：

public static <K, V> Collection<V> getAll(Map<K, V> map, Set<K> keys) {
    return map.entrySet()
            .stream()
            .filter(e -> keys.contains(e.getKey()))
            .map(Map.Entry::getValue)
            .collect(Collectors.toList());
}

Answer 3

我想您可以使用 Guava 的 Maps.filteredKeys() ，传入与您所需的键匹配的谓词，但这并不比手动迭代更好。

Answer 4

使用番石榴：
Collections2.transform(keys, Functions.forMap(map));

Answer 5

Java8 Streams：

keys.stream().map( map::get ).collect( Collectors.toList() );

或者如果您担心丢失键：

keys.stream().map( k -> map.getOrDefault( k, null ) ).filter(o -> o != null).collect( Collectors.toList() );

或者如果您需要多个线程来工作（很少见，请参阅@tkruse 评论）：

keys.parallelStream().map( map::get ).collect( Collectors.toList() );