Question

我正在尝试使用flatmap与Stream API进行嵌套循环，但我似乎无法弄明白。例如，我想重新创建以下循环：

sessionRegistry.getAllPrincipals()

我可以这样做，但这看起来很难看：

List<String> xs = Arrays.asList(new String[]{ "one","two", "three"});
List<String> ys = Arrays.asList(new String[]{"four", "five"});

System.out.println("*** Nested Loop ***");
for (String x : xs)
    for (String y : ys)
        System.out.println(x + " + " + y);

Flatmap看起来很有希望，但是如何在外部循环中访问变量？

System.out.println("*** Nested Stream ***");
xs.stream().forEach(x ->
    ys.stream().forEach(y -> System.out.println(x + " + " + y))
);

输出：

System.out.println("*** Flatmap *** ");
xs.stream().flatMap(x -> ys.stream()).forEach(y -> System.out.println("? + " + y));

Answer 1

您必须在flatMap阶段创建所需的元素，例如：

xs.stream().flatMap(x -> ys.stream().map(y -> x + " + " + y)).forEach(System.out::println);

Answer 2

通常，不需要flatMap：

xs.forEach(x -> ys.stream().map(y ->  x + " + " + y).forEach(System.out::println)); // X
xs.forEach(x -> ys.forEach(y -> System.out.println(x + " + " + y))); // V

以及此处不需要Stream API。

是的，它看起来很漂亮，但只有这么幼稚的任务。您只为每个元素创建/关闭一个新流，以将它们合并到结果流中。所有这些只是为了打印出来的？

相比之下，forEach提供了单线解决方案而没有任何性能成本（内部标准foreach）。

Answer 3

基本上，这是这些列表的笛卡尔积。我会先将它们合并到一个列表中：

List<String> xs = Arrays.asList(new String[]{ "one","two", "three"});
List<String> ys = Arrays.asList(new String[]{"four", "five"});
List<List<String>> input = Arrays.asList(xs, ys);

然后创建一个列表流，每个列表将映射到自己的流，并将这些内容保存到Supplier：

Supplier<Stream<String>> result = input.stream() // Stream<List<String>>
                .<Supplier<Stream<String>>>map(list -> list::stream) // Stream<Supplier<Stream<String>>>

然后减少这些供应商流，并为属于供应商的字符串流生成笛卡尔积，如下所示：

.reduce((sup1, sup2) -> () -> sup1.get().flatMap(e1 -> sup2.get().map(e2 -> e1 + e2)))

Reduce返回可选，所以为了处理缺席值，我将返回一个空字符串流：

.orElse(() -> Stream.of(""));

毕竟我们只需要获取供应商价值（这将是一串字符串）并打印出来：

s.get().forEach(System.out::println);

整个方法看起来像：

public static void printCartesianProduct(List<String>... lists) {
        List<List<String>> input = asList(lists);
        Supplier<Stream<String>> s = input.stream()
                // Stream<List<String>>
                .<Supplier<Stream<String>>>map(list -> list::stream)
                // Stream<Supplier<Stream<String>>>
                .reduce((sup1, sup2) -> () -> sup1.get()
                        .flatMap(e1 -> sup2.get().map(e2 -> e1 + e2)))
                .orElse(() -> Stream.of(""));

        s.get().forEach(System.out::println);
    }

用Java 8 flatmap替换嵌套循环

3 个答案: