java - 嵌套 ArrayList 到单维

Question

我有一些看起来像这样的代码。

class A {}
class B extends A {
    private String name; // assume there's a getter
}
class C extends A {
    private List<B> items = new ArrayList<>(); // assume a getter
}

在另一个类中，我有一个 ArrayList ( ArrayList<A> )。我正在尝试映射此列表以获取所有名称。

List<A> list = new ArrayList<>();
// a while later
list.stream()
    .map(a -> {
        if (a instanceof B) {
            return ((B) a).getName();
        } else {
            C c = (C) a;
            return c.getItems().stream()
                .map(o::getName);
        }
    })
    ...

这里的问题是我最终得到了这样的东西(用于视觉目的的 JSON)。

["name", "someName", ["other name", "you get the idea"], "another name"]

如何映射此列表，以便最终得到以下结果？

["name", "someName", "other name", "you get the idea", "another name"]

Answer 1

使用 flatMap :

list.stream()
    .flatMap(a -> {
        if (a instanceof B) {
            return Stream.of(((B) a).getName());
        } else {
            C c = (C) a;
            return c.getItems().stream().map(o::getName);
        }
    })
    ...

这将产生一个 Stream<String>所有的名字，没有嵌套。