Java 8 根据属性值对列表去重

爱被打了一巴掌 2022-05-31 06:29 347阅读 0赞

对列表的去重处理，Java 8 在 Stream 接口上提供了类似于 SQL 语句那样的 distinct() 方法，不过它也只能基于对象整体比较来去重，即通过 equals/hashCode 方法。distinct 方法的功效与以往的 new ArrayList(new HashSet(books)) 差不多。用起来是

List<Book> unique = book.stream().distinct().collect(Collectors.toList())

并且这种去重方式需要在模型类中同时实现 equals 和 hashCode 方法。

回到实际项目中来，我们很多时候的需求是要根据对象的某个属性来去重。比如接下来的一个实例，一个 books 列表中存在 ID 一样，name 却不同的 book, 我们认为这是重复的，所以需要根据 book 的 id 属性对行去重。在 collect 的时候用到的方法是 collectinAndThen(…), 下面是简单代码:

/** * @author wzx * @time 2018/2/9 */
    public class Deduplication { 
        public static void main(String[] args) {
            List<Book> books = Arrays.asList(
                    new Book(12, "Sun Java"),
                    new Book(12, "Oracle Java"),
                    new Book(15, "Scala")
            );
            List<Book> unique = books.stream().collect(
                    Collectors.collectingAndThen(Collectors.toCollection(
                            () -> new TreeSet<>(Comparator.comparing(o -> o.id))), ArrayList::new)
            );
            unique.forEach(book -> System.out.printf("book[id: %s, name: %s]\n", book.id, book.name));
        }
    }
    
    class Book {
        public final Integer id;
        public final String name;
    
        public Book(Integer id, String name) {
            this.id = id;
            this.name = name;
        }
    }

执行后打印出

book[id: 12, name: Sun Java]
    book[id: 15, name: Scala]

> [https://unmi.cc/java-8-list-deduplication-with-lambda/][https_unmi.cc_java-8-list-deduplication-with-lambda]

[https_unmi.cc_java-8-list-deduplication-with-lambda]: https://unmi.cc/java-8-list-deduplication-with-lambda/