如何使用一组字符串过滤地图值？

Question

我必须使用 Set 字符串过滤 Map 值，我可以让它工作（一位合作伙伴建议使用 anyMatch() 代替我在这里所做的，但我看不出如何) 我想知道你对这个算法有什么看法，如果可以改进它，也许使用另一个Stream 函数甚至contains() 方法，我也不确定我是否可以避免直接迭代Set（对于每个循环）。

ProductsResponse serviceResponse = (obtained from backend) ...;
Set<String> productIds = (some code to collect the strings from another API) ...;
//Here starts the filtering process 
serviceResponse.setProducts(serviceResponse.getProducts().entrySet().stream()
          .filter(product -> {
             for (String productId: productIds) {
                if (product.getKey().startsWith(productId)) {
                    return true;
                }
             }
             return false;
           }).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)));

重要提示：我使用Set过滤Map中的产品，Set中的字符串格式与Map，因此使用startsWith() 比较它们是安全的

更新 1：添加涉及的类定义。

class ProductsResponse() {

  private Map<String, ProductResource> products;
}

class ProductResource () {

   private String productId;
   private String name;
   private Double price;
}

Answer 1

如果productIds 集合中没有两个值a 和b where a.startsWith(b)，那么您可以通过将集合设为TreeSet 来大大提高性能。

TreeSet<String> productIds = (some code to collect the strings from another API) ...;

.filter(product -> {
   String id = productIds.floor(product.getKey());
   return (id != null && product.getKey().startsWith(id));
})

或者：

.filter(product -> Optional.ofNullable(productIds.floor(product.getKey()))
                   .map(product.getKey()::startsWith).orElse(false))

这会将性能从 O(n*m) 更改为 O(n*log(m))，其中 m 是 productIds 的大小。

---

更新

如果在productIds 集合中a.startsWith(b) 中有有 值a 和b，那么您需要一些额外的逻辑。

例如如果集合包含G 和GED，并且您正在检查它是否包含GET 的前缀，那么floor() 将返回GED，因此您必须去掉最后一个字符并使用@ 重复查找987654342@，现在返回G，将其作为有效前缀。

所以我们需要添加一个循环来重新检查：

.filter(product -> {
   String candidate = product.getKey();
   while ((candidate = productIds.floor(candidate)) != null) {
      if (product.getKey().startsWith(candidate))
         return true;
      candidate = candidate.substring(0, candidate.length() - 1);
   }
   return false;
})

它会稍微减慢查找速度，但仍然比完整的顺序搜索要好得多。

Answer 2

您可以通过流式传输集合并使用 anyMatch 来避免内部 for 循环：

serviceResponse.setProducts(serviceResponse.getProducts()
    .entrySet().stream()
    .filter(product -> productIds.stream().anyMatch(productId -> product.getKey().startsWith(productId)))
    .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)));

为简洁起见，您也可以使用方法参考：

serviceResponse.setProducts(serviceResponse.getProducts()
    .entrySet().stream()
    .filter(product -> productIds.stream().anyMatch(product.getKey()::startsWith))
    .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue)));