Set针对复杂对象去重问题
在项目中我们经常使用set，因其可以去重特性，平时使用较多的是基础数据类型，Set
创建的对象去覆盖set中type相同的对象，于是想到Set这个集合类型，并且重写了自定义对象的equals()和hashCode()方法，但调试阶段发现结果并非所想。
以下代码是自定义的Bean：
@AllArgsConstructor
@ToString
public class Foo implements Serializable {
private static final long serialVersionUID = -3968061057233768716L;
@Getter
@Setter
private int type;
@Getter
@Setter
private String name;
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
Foo foo = (Foo) o;
return type == foo.type;
}
@Override
public int hashCode() {
return Objects.hash(type);
}
}
接下来写个单元测试吧
public class SetTest {
@Test
public void test1() {
Set set = new HashSet<>();
Foo foo = new Foo(1, "1");
set.add(foo);
foo = new Foo(2, "2");
set.add(foo);
foo = new Foo(1, "3");
set.add(foo);
// 输出结果是什么
System.out.println("set = " + set);
}
第一感觉会输出什么，set.size()=2是毫无疑问的，重点是type=1的对象对应的name是1还是3？当时我用脑子意想出来的name=3。可能会有童鞋认为输出的结果name=1，恭喜你答对了，确实
是name=1。
set = [Foo(type=1, name=1), Foo(type=2, name=2)]
为什么不是3呢，3呢，3呢？接下来我们分析下，我们都知道当我们向HashMap中放入相同key和不同val放入到map中，map会保留最新的val（hashMap源码分析网上有很多解析，这里不再赘
述了，自行Google下），可见map其实也是保证key唯一的，不会出现key对应多个val的值（限：jdk中自带HashMap，Guava中Multimap是可以key对应多个val的，get(key)时返回的是val的集
合），HashSet就是利用HashMap中key唯一不重复的特性来实现的。
jdk1.8源码
public class HashSet
extends AbstractSet
implements Set, Cloneable, java.io.Serializable
{
static final long serialVersionUID = -5024744406713321676L;
private transient HashMap map;
// Dummy value to associate with an Object in the backing Map
private static final Object PRESENT = new Object();
/**
* Constructs a new, empty set; the backing HashMap instance has
* default initial capacity (16) and load factor (0.75).
*/
public HashSet() {
map = new HashMap<>();
}
-----------add部分 begin--------------------
/**
* Adds the specified element to this set if it is not already present.
* More formally, adds the specified element e to this set if
* this set contains no element e2 such that
* (e==null ? e2==null : e.equals(e2)).
* If this set already contains the element, the call leaves the set
* unchanged and returns false.
*
* @param e element to be added to this set
* @return true if this set did not already contain the specified
* element
*/
// 其实这里是有返回值的
public boolean add(E e) {
return map.put(e, PRESENT)==null;
}
-----------add部分 end--------------------
调用add()实际就是调用HashMap的put方法，注意此时put的val是固定式**PRESENT**是final修饰的一个对象，针对所有key都一样，因为我们是set所有val对我们来说没啥用。但是我们注意到其
实add()方法是有返回值的，当调用map.put()方法返回的对象是null的时候则代表新增成功，set中以前不存在此元素。当返回的对象not null的时候代表set中有此对象，而我们关注的其实只是key，
并不关注这个val。
public V put(K key, V value) {
return putVal(hash(key), key, value, false, true);
}
-----------
final V putVal(int hash, K key, V value, boolean onlyIfAbsent,
boolean evict) {
Node[] tab; Node p; int n, i;
if ((tab = table) == null || (n = tab.length) == 0)
n = (tab = resize()).length;
if ((p = tab[i = (n - 1) & hash]) == null)
tab[i] = newNode(hash, key, value, null);
else {
Node e; K k;
// 重点在这里
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))
e = p;
else if (p instanceof TreeNode)
e = ((TreeNode)p).putTreeVal(this, tab, hash, key, value);
else {
for (int binCount = 0; ; ++binCount) {
if ((e = p.next) == null) {
p.next = newNode(hash, key, value, null);
if (binCount >= TREEIFY_THRESHOLD - 1) // -1 for 1st
treeifyBin(tab, hash);
break;
}
if (e.hash == hash &&
((k = e.key) == key || (key != null && key.equals(k))))
break;
p = e;
}
}
// 当我们第二次加入type=1时代码肯定就走到了这里，return oldValue其实也就是returun我们传入的PRESENT not nul
if (e != null) { // existing mapping for key
V oldValue = e.value;
if (!onlyIfAbsent || oldValue == null)
e.value = value;
afterNodeAccess(e);
return oldValue;
}
}
++modCount;
if (++size > threshold)
resize();
afterNodeInsertion(evict);
return null;
}
重点在这里
重点在这里
重点在这里
if (p.hash == hash &&
((k = p.key) == key || (key != null && key.equals(k))))
e = p;
hash相等是肯定的，因为我们重写了Foo对象的hashCode方法，只跟type有关，所有p.hash==hash为true，大括号中(k = p.key) == key部门为false因为对象地址不同，但是
key != null && key.equals(k)是为true的，因为重写了equals()方法也只跟type有关，都为1，所以e = p了，流程判断语句结束，return一个not null的value。其实putVal()流程控制来看，只要我们的key
是存在的都会返回一个not null的val，而新插入的key返回值都是null。
代码调试截图走一波
最后没办法先remove()后add()。。。
注：代码中使用了lombok插件，非常好用哈哈，墙裂推荐。一个@Data搞定所有，不要告诉我idea代码自动生成，想一下，当你开发代码时给对象增加一个属性是不是还要给新属性生成get和set方法，还要toString()是不是也要改，而你加了@Data属性，这些都不需要考虑了。