如何使用Java 8 Stream API过滤集合中的列表

Question

class A {
    List<B> b;
}

class B{
    boolean flag;

    B(boolean flag) {
        this.flag = flag;
    }
    //getter setter
}

B b1 = new B(true);
B b2 = new B(true);
B b3 = new B(false);

List<B> list1 = new ArrayList<>();
list1.add(b1);
list1.add(b2);
list1.add(b3);

Set<A> a = new HashSet<>();
a.add(list1);

有一组对象A。

对象A包含对象B的列表。B具有布尔属性名'flag'使用Java 8流API，如何过滤上面的集合，使其包含布尔标志为B的B列表。 即b3应该从list1中删除。 为了简单起见，只需添加单个值。

Answer 1

你有一个类， A包含列表B ; 目标是在每个A删除所有符合条件的B

第一个解决方案是制作A集合的流，并将每个A映射到新的A ，其中过滤了B的列表。 假设有一个构造函数A(List<B> b)和适当的吸气剂（类A List<B> getB() ；类B boolean isFlag() ），则可能有：

Set<A> set = new HashSet<>();

Set<A> filtered =
    set.stream()
       .map(a -> new A(a.getB().stream().filter(B::isFlag).collect(Collectors.toList())))
       .collect(Collectors.toSet());

如果您可以就地修改B列表，而不是创建一个新的列表，则可能有第二种解决方案。 在这种情况下，您可能会：

set.forEach(a -> a.getB().removeIf(b -> !b.isFlag()));

这将删除所有B其中该标志为false每个A 。

Answer 2

    @org.junit.Test
    public void test() {
        B b1 = new B(true);
        B b2 = new B(true);
        B b3 = new B(false);

        List<B> list1 = new ArrayList();
        list1.add(b1);
        list1.add(b2);
        list1.add(b3);

        Set<A> a = new HashSet();
        a.add(new A(list1));

        a.stream().forEach(aa -> aa.setB(aa.getB().stream().filter(B::isFlag).collect(Collectors.toList())));

        System.out.println(a);
    }

输出：

[A{b=[B{flag=true}, B{flag=true}]}]

Answer 3

a.stream().forEach(l -> l.removeIf(b -> !b.getFlag()));