是否可以使用一行将流收集到两个不同的集合？

Question

我有以下代码：（为简洁起见）

public void search(Predicate<String> predicate, Elements elements)
{
    List<SearchResult> searchResults = elements.stream()
        .filter(element -> predicate.test(element.ownText()))
        .map(element -> new SearchResult(element.ownText(), element.baseUri(),element.tagName()))
        .collect(Collectors.toList());
}

但是现在，我想要另一个列表，其中包含所有没有映射的过滤元素。 是否可以使用流来执行此操作，还是应该为此更改为 foreach 循环？

Answer 1

您可以收集过滤后的元素，然后使用此列表将元素映射到搜索结果：

public void search(Predicate<String> predicate, Elements elements) {
    List<Element> filteredElements = 
        elements.stream()
                .filter(element -> predicate.test(element.ownText()))
                .collect(Collectors.toList());

    List<SearchResult> searchResults = 
        filteredElements.stream()
                        .map(element -> new SearchResult(element.ownText(),element.baseUri(),element.tagName()))
                        .collect(Collectors.toList());
}

这不会比其他答案的解决方案花费更多时间，但没有副作用，线程安全且易于阅读和理解。

Answer 2

一个简单的方法是

List<Element> filteredElements = new ArrayList<>();
List<SearchResult> searchResults = elements.stream()
      .filter(element -> predicate.test(element.ownText()))
      .peek(filteredElements::add)
      .map(element -> 
         new SearchResult(element.ownText(),element.baseUri(),element.tagName()))
      .collect(Collectors.toList());

Answer 3

另一种方法是定义Supplier<Stream>：

public void search(Predicate<String> predicate, Elements elements)
{
    Supplier<Stream<Element>> supplier = () -> elements.stream()
           .filter(element -> predicate.test(element.ownText()));
    List<SearchResult> searchResults = supplier.get()
           .map(element -> new SearchResult(element.ownText(),element.baseUri(),element.tagName()))
           .collect(Collectors.toList());
    List<Element> elementList = supplier.get().collect(Collectors.toList());
}

请注意，使用这种方法实际上执行了两次过滤。

另一种（虽然在这种情况下不是很漂亮）解决方案是使用来自this answer 的pairing 收集器：

Collector<Element, ?, List<SearchResult>> c1 = 
    Collectors.mapping(element -> new SearchResult(element.ownText(),element.baseUri(),element.tagName()),
        Collectors.toList());
Collector<Element, ?, List<Element>> c2 = Collectors.toList();
Collector<Element, ?, Pair<List<SearchResult>, List<Element>>> pairCollector =
    pairing(c1, c2, Pair::new); // Assumes some Pair class existing

Pair<List<SearchResult>, List<Element>> result = elements.stream()
           .filter(element -> predicate.test(element.ownText()))
           .collect(pairing);

这些解决方案是通用的：它们允许对单个数据源执行两种不同的操作。但是在您的具体示例中，创建第一个过滤的非映射数据列表，然后在该列表上创建第二个流更容易：

List<Element> elementList = elements.stream()
           .filter(element -> predicate.test(element.ownText()))
           .collect(Collectors.toList());
List<SearchResult> searchResults = elementList.stream()
           .map(element -> new SearchResult(element.ownText(),element.baseUri(),element.tagName()))
           .collect(Collectors.toList());

Answer 4

您不能使用该流两次，因为在使用 collect 时它已关闭。 以下代码将抛出“java.lang.IllegalStateException：流已被操作或关闭”。因此，您需要创建两次流或使用 for-each 循环。

import java.util.Collections;
import java.util.List;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import java.util.stream.Stream;

class Test
{
    static class SearchResult
    {
        SearchResult(Object o,Object p,Object q) {}
    }

    static class Element
    {
        public String ownText() {return "text";}
        public String tagName() {return "tag";}
        public String baseUri() {return "base";}
    }

    public static void search(Predicate<String> predicate, List<Element> elements)
    {
        Stream<Element> stream = elements.stream().filter(element -> predicate.test(element.ownText()));
        List<Element> filtered = stream.collect(Collectors.toList());
        // next line will throw the IllegalStateException as above line has already consumed the stream
        List<SearchResult> results =    stream.map(element -> 
         new SearchResult(element.ownText(),element.baseUri(),element.tagName()))
                .collect(Collectors.toList());
    }

    public static void main(String[] args)
    {       
        search(s->true,Collections.singletonList(new Element()));
    }
}