Java 8流for循环

时间:2015-03-30 09:28:52

标签: java arrays java-8 java-stream

我是Java 8 Streams的新手,并希望将以下代码块转换为Java 8的Stream方式来做同样的事情。

编辑:更新类名以减少混淆。 (删除了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());
    }
}

所有帮助&amp;解释赞赏

2 个答案:

答案 0 :(得分:11)

您可以使用flatMap

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

编辑:

由于objects.get(1).getObjects()似乎返回List<Object>,因此需要进行强制转换。为安全起见,您还可以添加一个过滤器,以确保在投射之前Object的类型确实为BarBO

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());

编辑:

以下是已编辑问题的类名的答案:

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

答案 1 :(得分:0)

考虑你的类(在你提供的示例中并假设它至少编译),我真的不明白为什么flatMap两次不起作用:

 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());