Java8 中包含流的嵌套列表

list java nested-loops java-8 java-stream

2022-09-01 04:20:47

我有一个对象列表 A。此列表中的每个对象 A 都包含对象 B 的列表，对象 B 包含对象 C 的列表。对象C包含一个属性名称，我想用它来过滤java 8。

如何使用流在java 8中编写下面的代码来避免嵌套循环：

C c1 = null;
String name = "name1"
for (A a: listOfAObjects) {
    for (B b: a.getList()) {
        for (C c: b.getPr()) {
            if (c.getName().equalsIgnoreCase(name)) {
                c1= c;
                break;
            }
        }
    }
}

答案 1

你可以使用两个，然后你可以选择第一个，或者如果没有结果返回：flatMapfilternull

C c1 = listOfAObjects.stream()
        .flatMap(a -> a.getList().stream())
        .flatMap(b -> b.getPr().stream())
        .filter(c -> c.getName().equalsIgnoreCase(name))
        .findFirst()
        .orElse(null);

答案 2

您可以使用 .flatMap

我做了一个例子，其中包含一个列表：CompanyPerson

public static void main(String[] args) {
    List<Company> companies = Arrays.asList(
            new Company(Arrays.asList(new Person("Jon Skeet"), new Person("Linus Torvalds"))),
            new Company(Arrays.asList(new Person("Dennis Ritchie"), new Person("Bjarne Stroustrup"))),
            new Company(Arrays.asList(new Person("James Gosling"), new Person("Patrick Naughton")))
    );

    List<String> persons = companies.stream()
            .flatMap(company -> company.getPersons().stream())
            .map(Person::getName)
            .collect(Collectors.toList());

    System.out.println(persons);
}

输出：

[Jon Skeet， Linus Torvalds， Dennis Ritchie， Bjarne Stroustrup， James Gosling， Patrick Naughton]