CODEWITHSUNDEEP
javaarraysjava-8java-stream
henrik
henrik

Reputation: 1618

Java 8 stream for-loop

Im new to Java 8 Streams and would like to convert following code-block to Java 8's Stream way of doing the same thing.

Edit: Updates the class-names to be less confusing. (Removed Foo, Bar, Baz...)

ArrayList<PriceList> priceLists = new ArrayList<PriceList>();

// I'm casting to a type-safe List from getObjects() 
// -which is a function I dont have access to. Is there a nice 
// solution to embed this in the stream-syntax?
List<PriceListGroup> distObjects = (List<PriceListGroup>) objects.get(1).getObjects();

for(PriceListGroup group : distObjects) {
    Set<Affiliate> affiliates = group.getAffiliates();
    for(Affiliate affiliate : affiliates) {
        priceLists.add(affiliate.getPriceList());
    }
}

All help & explanation appreciated

Upvotes: 6

Views: 4987

Answers (2)

Eran
Eran

Reputation: 393781

You can do it with flatMap :

List<FooBO> list1 = objects.get(1).getObjects().stream()
                                  .flatMap (b -> b.getAffiliates().stream())
                                  .map(BazBo::getPriceList)
                                  .collect(Collectors.toList());

Edit :

Since objects.get(1).getObjects() seems to return a List<Object>, a cast is required. To be safe, you can also add a filter that makes sure the type of the Objects is indeed BarBO prior to the cast :

List<FooBO> list1 = objects.get(1).getObjects().stream()
                                  .filter (o -> (o instanceof BarBo))
                                  .map (o -> (BarBO)o)
                                  .flatMap (b -> b.getAffiliates().stream())
                                  .map(BazBo::getPriceList)
                                  .collect(Collectors.toList());

EDIT :

Here's an answer with the class names of the edited question:

List<PriceList> priceLists = 
    distObjects.stream()
               .flatMap (g -> g.getAffiliates().stream())
               .map(Affiliate::getPriceList)
               .collect(Collectors.toList());

Upvotes: 11

Eugene
Eugene

Reputation: 120848

Considering your classes (in the example that you provided and assuming it compiles at least), I don't really understand why flatMap two times would not work:

 List<BarBO> input = new ArrayList<>();

    List<FooBO> result = input.stream()
            .flatMap((BarBO token) -> {
                return token.getAffiliats().stream();
            })
            .flatMap((BazBO token) -> {
                return token.getPriceList().stream();
            })
            .collect(Collectors.toList());

Upvotes: 0

Related Questions