Java list 取差集

Posted 2020-12-07 学业未成

今天业务代码上遇到一个取两个List差集的问题,记录一下.

业务描述

有两个list存放的同一个类的对象,这里对象举例为TestDomain.其中一个list存放了所有的结果集,取名为tlist,另一个list存放了需要被tlist移除的对象,取名为clist.即取tlist与clist的差集.

解决方式

第一种方式,重写TestDomain的equals()和hashcode()方法,这里是用idea自动生成的.代码如下

List<TestDomain> tlist = new ArrayList<TestDomain>();
for (int i = 0; i < 10; i++) {
    tlist.add(TestDomain.Builder.aTestDomain()
            .withId(Long.valueOf(i))
            .withCategoryName("AA" + String.valueOf(i))
            .build());
}

List<TestDomain> clist = new ArrayList<TestDomain>();
for (int i = 4; i < 8; i++) {
    clist.add(TestDomain.Builder.aTestDomain()
            .withId(Long.valueOf(i))
            .withCategoryName("AA" + String.valueOf(i))
            .build());
}
tlist.removeAll(clist);

第二种方式,在不重写equals()和hashcode()方法时,增加2个list,新增的alist用来记录需要被移除的对象的id,即clist中对象的id,然后通过alist中的id找到tlist中对应的对象,并放到blist中,最后取tlist与blist的差集.代码如下：

List<Long> alist =  new ArrayList<>(clist.size());
for (TestDomain v:clist) {
    alist.add(v.getId());
}
List<TestDomain> blist = new ArrayList<>(clist.size());
for (TestDomain v:tlist){
    for (Long id :alist){
        if (v.getId()==id){
            blist.add(v);
        }
    }
}
tlist.removeAll(blist);

第三种方式,在不重写equals()和hashcode()方法时,使用lambda表达式实现.返回结果放在dlist中,

List<VarietyScope> finalCList = clist;
List<TestDomain> dlist = tlist.stream()
        .filter(item -> !finalCList.stream()
                .map(e -> e.getId())
                .collect(Collectors.toList())
                .contains(item.getId()))
        .collect(Collectors.toList());