JDK-Map源码浅析——IdentityHashMap

        java.util.IdentityHashMap 类也是一种哈希表实现的Map接口，该类不是 通用 Map 实现。此类实现 Map 接口时，它有意违反 Map的常规协定，比较键/值时使用引用相等性代替对象相等性，不使用hashcode，而只使用Object.equals()。

        这里相等性比较的是地址：key1==key2，所以只有两个对象的地址相等才认为相等，是最严格的相等条件。   

        在JVM中每个对象的地址都是唯一的，IdentityHashMap就是利用这个唯一地址判断相等的，即使对象深度拷贝，copy了所有属性，但是地址还是不同的。

 

        这里的hash实现也和散列链表HashMap的不同，IdentityHashMap 只是散列而没有链表。这里不用Entry来存放元素，key和value全部放在table中，挨着放：talbe[i]= key;  table[i + 1] = value;  所以遍历查找和读取的时候，下标都需要+ 2递增。

private transient Object[] table;

        这里 hash 只是负责简单定位，定位后还要遍历 table，插入的时候变得不准确，所以查找 效率要比HashMap低一些。

        

        IdentityHashMap 这种特殊实现，会在一些特别的场合用到，比如说区分地址不同的对象，避免重复执行，如JVM中的关闭钩子，里面就用到了这个IdentityHashMap 。              见关闭钩子源码《JVM关闭钩子（2） —— 源码浅析》      

 

接下来，我们看看Map中关键的put/get方面，其他方法和HashMap中的差不多，就不在这里讨论了。

 

        put / get

        添加元素的时候，会先hash定位，然后遍历table，检查是否有重复元素(校验地址相等)。如果有重复，则覆盖值；没有重复，则找到一个空位插入。     

        查找元素，也是同样一个逻辑，会先hash定位，然后遍历table，找到==的key，然后返回。  

当然，此处遍历的时候，下标是+2递增，因为key和value都挨着存放在table中。

public V put(K key, V value) {
        Object k = maskNull(key);
        Object[] tab = table;
        int len = tab.length;
        int i = hash(k, len);
        Object item;
        while ( (item = tab[i]) != null) {
            if (item == k) {  // 使用==来判断，严格的相等。所以可以出现相同的key
                V oldValue = (V) tab[i + 1];
                tab[i + 1] = value;
                return oldValue;
            }
            i = nextKeyIndex(i, len);     // i + 2
        }
        modCount++;
        tab[i] = k;          // 没有使用Entry来存放元素，而是挨着放在table[]中
        tab[i + 1] = value;
        if (++size >= threshold)
            resize(len); // len == 2 * current capacity.
        return null;
}
private static int nextKeyIndex(int i, int len) {
        return (i + 2 < len ? i + 2 : 0);
}

       

public V get(Object key) {
        Object k = maskNull(key);
        Object[] tab = table;
        int len = tab.length;
        int i = hash(k, len);
        while (true) {
            Object item = tab[i];
            if (item == k)
                return (V) tab[i + 1];
            if (item == null)
                return null;
            i = nextKeyIndex(i, len);
        }
}

— — 源码来自 JDK1.7

应用

        另外要注意的是，由于比较的地址，所以可能会出现我们平时觉得相等的Key或Value，但是IdentityHashMap 却认为是不相等的，而没有定位到，或者进行过滤。
这里举个很简单的小例子：

    public static void main(String[] args) {
        IdentityHashMap<String, Integer> iMap = new IdentityHashMap<String, Integer>();
        iMap.put("250", 250);
        
        String s1 = "2";
        String s2 = "50";
        System.out.println(iMap.get("250"));
        System.out.println("250".equals(s1 + s2));
        System.out.println(iMap.get(s1 + s2));
    }

        这里第一个iMap.get(“250”)能正常取出250，这里很容易理解；而第二个iMap.get(s1 + s2) 却返回null，即使两个变量相加等于"250"，但是在IdentityHashMap 中就认为不相等，所以无法定位元素，返回null。

        所以IdentityHashMap 是一个非主流的Map，平时场景不是特别符合都不要用。

 这里给大家留一问题：为什么上面两个变量相加没有定位到元素呢？应该怎么修改呢？大伙有兴趣可以讨论一下:)

          具体见我的《String字符常量池和intern() 》