迭代EnumMap#entrySet

时间:2011-06-01 08:18:04

标签: java collections enums map

枚举Map#entrySet对所有Map实现都不起作用,特别是对于EnumMap,IdentityHashMap,这里是来自Josh Bloch的puzzler presentation (Puzzle 5)的示例代码 -

public class Size {

    private enum Sex { MALE, FEMALE }

    public static void main(String[] args) { 
        printSize(new HashMap<Sex, Sex>()); 
        printSize(new EnumMap<Sex, Sex>(Sex.class)); 
    }

    private static void printSize(Map<Sex, Sex> map) { 
        map.put(Sex.MALE,   Sex.FEMALE); 
        map.put(Sex.FEMALE, Sex.MALE); 
        map.put(Sex.MALE,   Sex.MALE); 
        map.put(Sex.FEMALE, Sex.FEMALE); 
        Set<Map.Entry<Sex, Sex>> set = 
            new HashSet<Map.Entry<Sex, Sex>>(map.entrySet()); 
        System.out.println(set.size()); 
    }
}

是的,这会产生错误的结果 -

应该是

 2 
 2

但产生

2 
1

但如果我尝试下面的代码 - ,它会产生正确的结果

更新
虽然结果集的大小为2,但条目相同。

public class Test{

 private enum Sex { MALE, FEMALE } 

    public static void main(String... args){
        printSize(new HashMap<Sex, String>());
        printSize(new EnumMap<Sex, String>(Sex.class));
    }


    private static void printSize(Map<Sex, String> map) {
        map.put(Sex.MALE,   "1");
        map.put(Sex.FEMALE, "2");
        map.put(Sex.MALE,   "3");
        map.put(Sex.FEMALE, "4");
        Set<Map.Entry<Sex, String>> set =
            new HashSet<Map.Entry<Sex, String>>(map.entrySet());
        System.out.println(set.size());
    }
}

我甚至尝试使用两种不同枚举类型作为键和值的上述代码。

只有当EnumMap具有与键和值相同的枚举时,这似乎才是问题。

我想知道为什么会这样?或者我错过了什么。为什么当ConcurrentHashMap长期修复后它没有修复?

3 个答案:

答案 0 :(得分:9)

查看EnumMap.EntryIterator.next()实施。这应该足以解决问题。

一个线索是结果集是:

[FEMALE=2, FEMALE=2]

这不是正确的结果。

您看到的效果是由于EnumMap.EntryIterator.hashCode()实现(这里是Map.Entry)。这是

h = key ^ value

这导致

生成的条目具有相同的哈希值
map.put(Sex.MALE,   Sex.MALE); 
map.put(Sex.FEMALE, Sex.FEMALE); 

稳定0。

map.put(Sex.MALE,   Sex.FEMALE); 
map.put(Sex.FEMALE, Sex.MALE);

这里是一个不稳定的(多次执行)int值。如果键和值哈希值相同,您将始终看到效果,因为:a ^ b == b ^ a。这会导致Entry的哈希值相同。

如果条目具有相同的哈希值,它们最终会出现在哈希表的同一个桶中,并且equals将始终工作,因为它们仍然是同一个对象。

有了这些知识,我们现在也可以与其他类型如Integer(我们知道hashCode实现)产生相同的效果:

map.put(Sex.MALE,   Integer.valueOf(Sex.MALE.hashCode())); 
map.put(Sex.FEMALE, Integer.valueOf(Sex.MALE.hashCode()));

[FEMALE=1671711, FEMALE=1671711]

奖金:EnumMap实现打破了equals()合约:

EnumMap<Sex, Object> enumMap = new EnumMap<Sex, Object>(Sex.class);
enumMap.put(Sex.MALE, "1");
enumMap.entrySet().iterator().next().equals(enumMap.entrySet().iterator());

抛出:

Exception in thread "main" java.lang.IllegalStateException: Entry was removed
    at java.util.EnumMap$EntryIterator.checkLastReturnedIndexForEntryUse(EnumMap.java:601)
    at java.util.EnumMap$EntryIterator.getValue(EnumMap.java:557)
    at java.util.EnumMap$EntryIterator.equals(EnumMap.java:576)
    at com.Test.main(Test.java:13)

答案 1 :(得分:2)

EnumMap.EntryIterator.next()返回this引用。您可以按如下方式验证它:

Iterator<? extends Map.Entry<Sex, Sex>> e = map.entrySet().iterator();
while (e.hasNext()) {
    Map.Entry<Sex, Sex> x = e.next();
    System.out.println(System.identityHashCode(x));
}

答案 2 :(得分:0)

问题不在于地图,而在于EntryIterator的实现和仅接受不相等元素的HashSet规范。

如果1和2地图应该有两个元素,您可以验证是否正在调用

map.entrySet().size();

'问题'在EnumMap类的EntryIterator实现中,因为这是一个谜题,试图找出原因。

PS。使用调试器。

编辑:

这是你真正做的是:

    Set<Map.Entry<Sex, Sex>> set =  new HashSet<Map.Entry<Sex, Sex>>();



    Iterator<Map.Entry<Sex, Sex>> e = entrySet.iterator();
    while (e.hasNext()) {
        set.add(e.next());
    }

请记住,HashSet是通过HashMap实现的,这些值基于哈希码和相等性添加到hashMap。

BTW在OP中解释的一切都与拼图相关联。该错误采用相同的方法,在第二次调用方法next()之后,更改工作方式并比较类类型而不是值return o == this;