如何根据条目集过滤地图条目

Question

我正在使用 google guava 12 并有一张地图：

Map<OccupancyType, BigDecimal> roomPrice;

我有一套：

Set<OccupancyType> policy;

如何根据policy 过滤roomPrice map 中的条目并返回过滤后的映射？

filteredMap 需要具有来自policy 的所有值。如果 roomPrice 地图没有政策条目，我想输入默认值。

Answer 1

无需使用 Guava，Maps.filterKeys() 也可以为大型地图产生性能非常差的结果。

// (new map can be initialized to better value to avoid resizing)
Map<OccupancyType, BigDecimal> filteredMap = new HashMap<>(roomPrice.size());
for (OccupancyType key: policy) {
    // contains() and get() can usually be combined
    if (roomPrice.contains(key)) {
       filteredMap.put(key, roomPrice.get(key));
    }
}

Answer 2

既然你有一组键，你应该使用Maps.filterkeys()，Guava 还提供了一组非常好的谓词，你可以开箱即用。在您的情况下，Predicates.in() 之类的东西应该可以工作。

所以基本上你会得到：

Map<OccupancyType, BigDecimal> filteredMap
    = Maps.filterKeys(roomPrice, Predicates.in(policy));

希望对你有帮助。

Answer 3

• 覆盖并实现 equals 和 hashcode in OccupancyType。
• 循环通过roomPrice的密钥集并收集 过滤器中包含的元素。

类似这样的：

Map<OccupancyType, BigDecimal> filteredPrices = new HashMap<OccupancyType, BigDecimal>();
for(OccupancyType key : roomPrice.keySet()) {
    if(policy.contains(key) {
        filteredPrices.put(key, roomPrice.get(key));
    }
}

更新

好的，在 Google Guava 上阅读了一些内容后，您应该可以执行以下操作：

Predicate<OccupancyType> priceFilter = new Predicate<OccupancyType>() {
    public boolean apply(OccupancyType i) {
        return policy.contains(i);
    }
};

然后

return Maps.filterValues(roomPrice, priceFlter);

应该可以解决问题。