通過查看java底層api實現可以發現在hashset中比較2個值是否相等,他是比較的hash值和equals方法,所以我們只需要重寫這2個方法即可。
實例程序:
package com;
import java.util.HashSet;
import java.util.Set;
public class SetDemo {
public static void main(String[] args) {
User user0 = new User(1, "admin");
User user3 = new User(2, "user");
User user1 = new User(2, "user");
User user2 = new User(3, "admin");
String s1 = "hello";
String s2 = "hello";
// boolean equals = user.equals(admin);
System.out.println(s1==s2);
// System.out.println(equals);
Set<User> userSet = new HashSet<User>();
userSet.add(user3);
userSet.add(user2);
userSet.add(user0);
userSet.add(user1);
// 輸入結果 admin1 admin3 user2
for(User u : userSet){
System.out.print(u.username + u.id + "\t");
}
}
}
class User{
protected Integer id;
protected String username;
public User(Integer id, String username){
this.id = id;
this.username = username;
}
/**
* 如果對象類型是User 的話 則返回true 去比較hashCode值
*/
@Override
public boolean equals(Object obj) {
if(obj == null) return false;
if(this == obj) return true;
if(obj instanceof User){
User user =(User)obj;
// if(user.id = this.id) return true; // 只比較id
// 比較id和username 一致時才返回true 之後再去比較 hashCode
if(user.id == this.id && user.username.equals(this.username))
return true;
}
return false;
}
/**
* 重寫hashcode 方法,返回的hashCode 不一樣才認定爲不同的對象
*/
@Override
public int hashCode() {
// return id.hashCode(); // 只比較id,id一樣就不添加進集合
return id.hashCode() * username.hashCode();
}
}