Java 中对Set接口如何实现存储的自定义对象去重

た入场券 2023-03-14 08:04 48阅读 0赞

# Set #

首先我们都知道实现Set不可以存储重复的对象，如果在存储的时候出现了重复的，就不会存入set，实现Set中都是不重复的元素。  
但我们在实际开发应用中，往往会存储自定义的对象，而该对象很有可能不只一个属性，需求有时候只是需要这个对象中的一个属性或者其中几个做为唯一不重复的属性，意思就是该属性不能重复，其他的属性重复也不会判断为一个对象，这时候我们应该怎么做呢？  
这就需要来看一下set是怎样来判断对象重复的了：  
它的底层逻辑是这样的  
在调用add()存储对象时，先判断Hash码是否相同，不相同直接添加进来；如果Hash码相同，则再通过eqauls()判断是否相同，不相同才添加，相同则不添加。

上边所提到了Hash码，是通过对象的hashCode()计算得出的，具体算法可以先不了解。  
意思就是，首先得判断哈希码，就得重写hashCode()方法，举个栗子：

## 创建Book类 ##

创建一个Book类，属性如下

## 重写hashCode()和equals() ##

咱们重写默认的hashCode()和equals()看是什么样子：

@Override
        public boolean equals(Object o) {
        
            if (this == o) return true;
            if (o == null || getClass() != o.getClass()) return false;
            Book book = (Book) o;
            return number == book.number &&
                    Float.compare(book.price, price) == 0 &&
                    Objects.equals(name, book.name) &&
                    Objects.equals(info, book.info) &&
                    Objects.equals(author, book.author) &&
                    Objects.equals(publicationDate, book.publicationDate) &&
                    Objects.equals(simpleDateFormat, book.simpleDateFormat);
        }
    
        @Override
        public int hashCode() {
        
            return Objects.hash(name, number, info, author, publicationDate, price, simpleDateFormat);
        }

看！就算不懂原理看了这个也就明白啦！  
hashCode()通过我们所有的属性去计算哈希码  
而equals()比较了我们所有的属性都要相同才返回true

假设！我们只要求numbers是唯一的就好了，那我们就要先  
1、hashCode()只生成通过numbers属性的哈希码对比  
2、equals()只比较numbers是否相同

代码如下：

@Override
        public boolean equals(Object o) {
        
            if (this == o) {
        
                return true;
            }
            if (o != null && getClass() == o.getClass()) {
        
                Book book = (Book) o;
                return number == book.number;
            } else {
        
                return false;
            }
        }
    
        @Override
        public int hashCode() {
        
            return Objects.hash(number);
        }

现在的话，这个Books类就是以number为唯一标识，只有对象的number不一样才能存储的set集合

## 完整代码 ##

定义了全参的构造方便存数据  
总结一下完整的Book类！

package com.kaikeba.bookmanagementsysment;
    
    import java.text.ParseException;
    import java.text.SimpleDateFormat;
    import java.util.Date;
    import java.util.Objects;
    
    /**
     * 存储书籍信息
     * 图书名称:name
     * 图书编号:number
     * 图书简介:info
     * 图书作者:author
     * 出版日期:publicationDate
     * 图书价格:price
     * @author Administrator
     */
    public class Book {
        
         private String name;
         private int number;
         private String info;
         private String author;
         private Date publicationDate;
         private float price;
    
        SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyy年MM月dd日");
        public Book(String name, int number, String info, String author, String publicationDateStr, float price) throws ParseException {
        
            this.name = name;
            this.number = number;
            this.info = info;
            this.author = author;
            this.publicationDate = simpleDateFormat.parse(publicationDateStr);
            this.price = price;
        }
    
        @Override
        public String toString() {
        
    
            return "【书名】：" +
                    "《" + name + '》' +
                    ", 【编号】：" + number +
                    ", 【简介】：'" + info + '\'' +
                    ", 【作者】：" + author  +
                    ", 【出版日期】：" + simpleDateFormat.format(publicationDate) +
                    ", 【价格】：" + price +"元\n";
        }
    
        /**
         * 当编号相同则视为这两个对象相同
         * @param o
         * @return
         */
        @Override
        public boolean equals(Object o) {
        
            if (this == o) {
        
                return true;
            }
            if (o != null && getClass() == o.getClass()) {
        
                Book book = (Book) o;
                return number == book.number;
            } else {
        
                return false;
            }
        }
    
        @Override
        public int hashCode() {
        
            return Objects.hash(number);
        }
    }

## 结果 ##

现在最后一步创建一个set集合，这里选用hashset

public class Test {
        
        public static void main(String[] args) throws ParseException {
        
            //创建几个Book对象
            Book book1 = new Book("西游记",1111,"讲述了师徒四人。。。","吴承恩","1922年1月1日", (float) 99.9);
            Book book2 = new Book("水浒传",2222,"讲述了绿林好汉。。。","？？？","1922年1月2日", (float) 88.8);
            Book book3 = new Book("Java—从入门到放弃",3333,"讲述了学习Java的心路历程","小黄","2020年5月1日", (float) 0.5);
            //创建一个存储Book的HashSet集合
            Set<Book> bookSet = new HashSet<>();
            //存入数据
            bookSet.add(book1);
            bookSet.add(book2);
            bookSet.add(book3);
            //打印
            System.out.println(bookSet);
    
        }
    }

看看运行结果呢，可以尝试修改numbers，重复的就不会存储咯！

"C:\Program Files\Java\jdk-14\bin\java.exe" "-javaagent:D:\Program Files\JetBrains\IntelliJ IDEA 2020.1\lib\idea_rt.jar=59338:D:\Program Files\JetBrains\IntelliJ IDEA 2020.1\bin" -Dfile.encoding=UTF-8 -classpath C:\Users\Administrator\bookmanagementsystem\out\production\bookmanagementsystem com.kaikeba.bookmanagementsysment.Test
    [【书名】：《Java—从入门到放弃》, 【编号】：3333, 【简介】：'讲述了学习Java的心路历程', 【作者】：小黄, 【出版日期】：2020年05月01日, 【价格】：0.5元
    , 【书名】：《西游记》, 【编号】：1111, 【简介】：'讲述了师徒四人。。。', 【作者】：吴承恩, 【出版日期】：1922年01月01日, 【价格】：99.9元
    , 【书名】：《水浒传》, 【编号】：2222, 【简介】：'讲述了绿林好汉。。。', 【作者】：？？？, 【出版日期】：1922年01月02日, 【价格】：88.8元
    ]
    
    Process finished with exit code 0

大家可以去尝试一下，毕竟实践出真知嘛！