Como posso obter uma lista de algumas propriedades de classe com o Java 8 Stream?


134

Eu tenho um List<Person>. Eu preciso obter um Listde uma propriedade de Person.

Por exemplo, eu tenho uma Personclasse:

class Person
{
    private String name;
    private String birthDate;
    public String getName() {
        return name;
    }
    public String getBirthDate() {
        return birthDate; 
    }
    Person(String name) {
        this.name = name;
    }
}

List<Person> personList = new ArrayList<>();
personList.add(new Person("David"));
personList.add(new Person("Joe"));
personList.add(new Person("Michel"));
personList.add(new Person("Barak"));

Quero obter uma lista de nomes com a StreamAPI, assim:

List<String> names = personList.stream().somecode().collect(Collectors.toList());
names.stream().forEach(System.out::println);

#David
#Joe
#Michel
#Barak

Este código não funciona:

public class Main 
{
    public static void main(String[] args) 
    {
        List<Person> personList = new ArrayList<>();
        Person person = new Person("Иван");
        person.getFriends().addAll(Arrays.asList("Друг 1", "Друг 2", "Друг 3"));
        personList.add(person);
        person = new Person("Федор");
        person.getFriends().addAll(Arrays.asList("Друг 4", "Друг 5", "Друг 6"));
        personList.add(person);
        person = new Person("Алексей");
        person.getFriends().addAll(Arrays.asList("Друг 7", "Друг 8", "Друг 9"));
        personList.add(person);
        person = new Person("Константин");
        person.getFriends().addAll(Arrays.asList("Друг 10", "Друг 11", "Друг 12"));

        List<String> friens = personList.stream().map(e->e.getFriends()).collect(Collectors.toList());

        friends.stream().forEach(System.out::println);
        //Друг 1
        //Друг 2
        //Друг 3
        //Друг 4
        //...

    }
}

class Person
{
    String name;
    List<String> friends;

    Person(String name) {
        this.name = name;
    }
    public String getName() {
        return name;
    }

    public List<String> getFriends() {
        return friends;
    }
}


Como posso obter Listuma propriedade com a StreamAPI?

Respostas:


310

Você pode usar map:

List<String> names = 
    personList.stream()
              .map(Person::getName)
              .collect(Collectors.toList());

EDIT:

Para combinar as listas de nomes de amigos, você precisa usar flatMap:

List<String> friendNames = 
    personList.stream()
              .flatMap(e->e.getFriends().stream())
              .collect(Collectors.toList());

Obrigado. Como posso mesclar muitas listas de strings que recebi com .map (Person :: getFriends);
precisa saber é o seguinte

1
@NCNecros Isso depende de qual é a entrada. Não entendo o que você está tentando fazer no código adicionado à pergunta.
Eran

@Eran Ele cuidará do caso quando a lista estiver vazia?
Joy

1
@NagabhushanSN Você pode iterar com o loop foreach, tradicional para loop ou while, mas deve usar algum loop.
Eran

1
@PAA adicionar uma chamada filtro antes flatMap:.filter(e->e.getFriends() != null)
Eran
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.