Wie werden zwei Felder eines Objekts in derselben Liste gesammelt?

Question

Ich habe ein Objekt von Waren, die zwei Eigenschaften hat: firstCategoryId und secondCategoryId. Ich habe eine Warenliste und möchte alle Kategorie-IDs (einschließlich firstCategoryId und secondCategoryId) abrufen.

Meine aktuelle Lösung ist:

List<Integer> categoryIdList = goodsList.stream().map(g->g.getFirstCategoryId()).collect(toList()); 
categoryIdList.addAll(goodsList.stream().map(g->g.getSecondCategoryId()).collect(toList())); 

Gibt es eine bequemere Art und Weise habe ich alle CategoryIDs in einer einzigen Anweisung zu bekommen?

Answer 1

Sie können es mit einer einzigen Stream Pipeline flatMap mit:

List<Integer> cats = goodsList.stream() 
           .flatMap(c->Stream.of(c.getFirstCategoryID(),c.getSecondCategoryID())) 
           .collect(Collectors.toList());