java - 根据两个List的内容生成SortedMap

Question

我正在尝试使用直接来自两个不同源的两个ArrayList的流来生成SortedMap。我的目标是让 SortedMap 将第二个列表中的 Double 属性存储为键，将第一个列表中的对象存储为值(如果这些对象具有匹配的属性)，由辅助对象检查。

到目前为止，我可以使用以下方法完成它:

SortedMap<Double, FirstObject> myMap = new TreeMap<>(Double::compareTo);

List<FirstObject> myList = firstDao.get(someId).stream()
    .filter(firstobject -> secondDao.get(firstObject.getObjectId())
        .stream()
        .anyMatch(secondObject -> {
            if (Helper.check(secondObject).matches()) {
                myMap.put(
                    secondObject.getEfficiency(), firstObject
                );
            }
            return Helper.check(secondObject).matches();
    }))
    .collect(Collectors.toList());

我用该代码生成的 myList 没有用处，但到目前为止，这是我能够填充 Map 的唯一方法。

有没有办法可以直接填充到 SortedMap 而无需生成该列表？

Answer 1

您需要使用collect，而不是创建会导致副作用的过滤器。

实现此目的的方法之一是创建一个中间 Map，它将 firstDao 返回的每个对象与 secondDao 返回的匹配对关联起来。

然后在中间映射的条目上创建一个流。过滤掉带有空键(没有匹配对)的条目。然后应用collect(Collectors.toMap())。

NavigableMap<Double, FirstObject> myMap = firstDao.get(tableId).stream()
    .collect(Collectors.toMap( // creates a map `Map<Optional<Double>,FirstObject>`
        firstObject -> secondDao.get(firstObject.getObjectId()).stream()
            .filter(secondObject -> firstObject.getAttribute().equals(secondObject.getAttribute()))
            .findFirst()
            .map(FirstObject::getEfficiency),
        Function.identity(),
        (left, right) -> left // duplicated keys would appear when there are more then one object having no matching pair (the key would be an empty optional), so we need to guard against that case by providing a merge function to resulve duplicates
    ))
    .entrySet().stream()
    .filter(entry -> entry.getKey().isPresent())
    .collect(Collectors.toMap(
        entry -> entry.getKey().get(),    // extracting a key (efficiency)
        Map.Entry::getValue,              // extracting a value
        (left, right) -> { throw new AssertionError("duplicates are not expected"); }, // since the source is an intermediate map we don't expect duplicates
        TreeMap::new
    ));

解决此问题的另一种更简洁的方法是使用 Collector.of() 创建一个自定义收集器 (整体逻辑保持不变):

NavigableMap<Double, FirstObject> myMap = firstDao.get(tableId).stream()
    .collect(Collector.of(
        TreeMap::new,                                          // mutable container of the collector
        (NavigableMap<Double, FirstObject> map, FirstObject firstObject) -> 
            secondDao.get(firstObject.getObjectId()).stream()  // population the mutable container
                .filter(secondObject -> next.getAttribute().equals(secondObject.getAttribute()))
                .findFirst()
                .map(FirstObject::getEfficiency)
                .ifPresent(efficiency -> map.put(efficiency, firstObject)),
        (left, right) -> { left.putAll(right); return left; } // merging containers in parallel, I'm assuming that there's no duplicates and hence there's no special policy
    ));

旁注:当您需要 TreeMap 时，最好使用 NavigableMap 作为抽象类型。此接口(interface)扩展了 SortedMap 并提供了许多 SortedMap 无法访问的方法。

---

解决 OP 发表的评论

问题中提供的代码创建了通过firstDao和以下filter<的方式检索的对象流<如果谓词很重要， 将通过以效率添加新条目(或替换现有条目的值)来更新映射myMap > 使用 secondDao.get() 作为源创建的流中第一次遇到匹配的对象。

    .filter(firstobject -> secondDao.get().stream()
        .anyMatch(secondObject -> {
            if (someCondition) {
                myMap.put(secondObject.getEfficiency(), firstObject);
            }
            return someCondition;
    }))

anyMatch - 是一个短路操作，如果有相同的id但不同的效率，它们将被忽略。

上述解决方案中的这段代码的行为方式完全相同:findFirst 将选择将评估 someCondition 的第一个对象为 true (之后会单独处理具有空可选值的情况，并且可选值的 equals/hashCode 会委托(delegate)给 equals/hashCode 实现其值，因此结果中没有差异的余地)。

firstObject -> secondDao.get().stream()
    .filter(secondObject -> someCondition)
    .findFirst()
    .map(FirstObject::getEfficiency)

我认为这是预期的行为，因为没有提到问题中提供的代码在某些方面出现故障，并且没有样本数据(可能会得出相反的结论)。

因此，上述解决方案未产生预期结果的说法与问题的当前状态相矛盾，因为它是基于原始代码的逻辑。

直到期望的结果 未指定(通过提供示例数据，或描述原始代码的问题)，我认为没有可能改进答案。