使用 Java Streams 从超集中获取子集

Question

我需要从给定的超级列表中获取子列表，如下所述。

假设我有一个“Person”对象的超级列表，如下所示：

List<Person> personsList = new ArrayList<>();
personsList.add(new Person("John", 28));
personsList.add(new Person("Paul", 29));
personsList.add(new Person("Adam", 30));
personsList.add(new Person("Peter", 31));
personsList.add(new Person("Pat", 32));

让Person类定义如下：

class Person {
    public String name;
    public int age;

    public Person(String name, int age) {
        this.name = name;
        this.age = age;
    }

    @Override
    public String toString() {
        return name + ": " + age;
    }
}

给定一个像这样的子列表：

String [] names = {"John", "Paul", "Adam"};
List<String> namesList = Arrays.asList(names);

我想获取名称为John、Paul 和Adam 的人员对象。

我是 Java 流的新手，这是我迄今为止为满足我的要求所做的尝试。

List<Person> finalList = personsList.stream().filter(p -> namesList.contains(p.name))
                .collect(Collectors.toCollection(() -> new ArrayList<Person>()));

我得到了想要的结果，但我想知道是否有任何其他有效的方法来实现这一点（我主要认为 namesList.contains(p.name) 有点低效）。

如果有更好的方法，请告诉我。

Answer 1

假设在数组中重复名称是没有意义的，最好的方法是在名称中的Set<T> 中找到名称。这很容易实现。将数组转换为 List 并在接受任何集合后立即将其包含在 HashSet<T> 的构造函数中。

String [] names = {"John", "Paul", "Adam"};
Set<String> set = new HashSet<>(Arrays.asList(names));

现在使用与您尝试相同的 Stream-API。仅使用静态方法Collectors.toList()。

List<Person> list= personsList.stream()
    .filter(i -> set.contains(i.getName()))
    .collect(Collectors.toList());

结果将仅包含 John、Paul 和 Adam 人员的对象。

Answer 2

将nameList 定义为 Set 会更有效，但如果列表仅包含 3 个名称，则差异不会真正明显。

所以是这样的：

Set<String> namesList = new HashSet<> (Arrays.asList("John", "Paul", "Adam"));

List<Person> finalList = personsList.stream()
                                    .filter(p -> namesList.contains(p.name))
                                    .collect(Collectors.toList());

注意：我还将收集器更改为 toList，因为这显然是您需要的。