Java list 取差集

今天業務代碼上遇到一個取兩個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<TestDomain> finalCList = clist;
List<TestDomain> dlist = tlist.stream()
        .filter(item -> !finalCList.stream()
                .map(e -> e.getId())
                .collect(Collectors.toList())
                .contains(item.getId()))
        .collect(Collectors.toList());