java - 如何使用 Java 8 Stream 展平列表的嵌套映射？

Question

我有一个看起来像这样的结构：

public class Category {
    private String tag;
    private String name;
    private String description;
    private List<Item> items;
}

Item看起来像这样

public class Item {
    private String itemTag;
    private String itemName;
    private String itemType;
    private Integer itemStatus;
    private List<Item> items;
}

这不是最好的设计——我知道，但我没有权力改变那个设计。

我正在尝试找到一种方法将此结构展平为单个结构Stream并找到一个Itemwith matching itemTag。使用此代码：

String tagToFind = "someTag";
List<Category> categories = getCategoriesList(); // <-- returns a list of Category
Item item = categories.stream()
                .flatMap(category -> category.getItems().stream())
                .filter(tagToFind.equals(item.getItemTag()))
                .findFirst();

但这仅搜索项目列表的一级。如果我想更深入，我可以简单地做：

Item item = categories.stream()
                .flatMap(category -> category.getItems().stream())
                .flatMap(item->item.getItems().stream()))
                .filter(tagToFind.equals(item.getItemTag()))
                .findFirst();

哪个工作正常。但我正在尝试找到一种更具可扩展性的方法来做到这一点，它可以像嵌套列表一样深入。有没有一种有效的方法来做到这一点？

标签： javarecursionjava-streamnested-lists

Answer 1

您需要一个单独的递归方法。你可以这样做：

Optional<Item> item = categories.stream()
        .flatMap(category -> category.getItems().stream())
        .flatMap(MyClass::flatMapRecursive)
        .filter(i -> tagToFind.equals(i.getItemTag()))
        .findFirst();

使用此flatMapRecursive()方法：

public Stream<Item> flatMapRecursive(Item item) {
    return Stream.concat(Stream.of(item), item.getItems().stream()
            .flatMap(MyClass::flatMapRecursive));
}

要考虑的另一件事：该flatMapRecursive()方法不进行空值检查，因此每个项目至少需要一个空列表，否则您将得到一个NullPointerException.

如果null值是可能的，items您可以使用以下方法防止这种情况Optional：

public Stream<Item> flatMapRecursive(Item item) {
    return Stream.concat(Stream.of(item), Optional.ofNullable(item.getItems())
            .orElseGet(Collections::emptyList)
            .stream()
            .flatMap(MyClass::flatMapRecursive));
}

items或者在使用它之前做空检查：

public Stream<Item> flatMapRecursive(Item item) {
    if (item.getItems() == null) {
        return Stream.empty();
    }
    return Stream.concat(Stream.of(item), item.getItems().stream()
            .flatMap(MyClass::flatMapRecursive));
}