keySet 与entrySet 遍历HashMap性能差别

keySet 与entrySet 遍历HashMap性能差别

• 博客分类： 
• Java

 

一.问题发现 
今天,在写完代码后用Find Bugs扫锚了一下,发现类中一处代码中有提示如下内容: 
       

Java代码  

1. Map<String, EventChain> map = ContextHolder.getContext().getEventChains();  
2.         for (Iterator<String> iter = map.keySet().iterator(); iter.hasNext();) {  
3.             String key = iter.next();  
4.             EventChain eventChain = eventChains.get(key);  
5.          }  

makes inefficient use of keySet iterator instead of entrySet iterator 
意思是说用keySet 方式遍历Map的性能不如entrySet性能好.起初想不明白,索性仔细看下代码: 

二.常用的遍历HashMap的两种方法 

1.第一种方式 

Java代码  

1. Iterator<String> keySetIterator = keySetMap.keySet().iterator();  
2.         while (keySetIterator.hasNext()) {  
3.             String key = keySetIterator.next();  
4.             String value = keySetMap.get(key);  
5.               
6.         }  

2.第二种方式 

Java代码  

1. Iterator<Entry<String, String>> entryKeyIterator = entrySetMap.entrySet()  
2.                 .iterator();  
3.         while (entryKeyIterator.hasNext()) {  
4.             Entry<String, String> e = entryKeyIterator.next();  
5.             String value=e.getValue();  
6.         }  

三.性能比较 

    到底第二种方式的性能比第一种方式的性能高多少呢,通过一个简单的测试类可以看一下,测试代码如下: 

Java代码  

1. public class HashMapTest {  
2.     public static void main(String[] args) {  
3.   
4.         HashMap<String, String> keySetMap = new HashMap<String, String>();  
5.         HashMap<String, String> entrySetMap = new HashMap<String, String>();  
6.   
7.         for (int i = 0; i < 1000; i++) {  
8.             keySetMap.put("" + i, "keySet");  
9.         }  
10.         for (int i = 0; i < 1000; i++) {  
11.             entrySetMap.put("" + i, "entrySet");  
12.         }  
13.   
14.         long startTimeOne = System.currentTimeMillis();  
15.         Iterator<String> keySetIterator = keySetMap.keySet().iterator();  
16.         while (keySetIterator.hasNext()) {  
17.             String key = keySetIterator.next();  
18.             String value = keySetMap.get(key);  
19.             System.out.println(value);  
20.         }  
21.   
22.         System.out.println("keyset spent times:"  
23.                 + (System.currentTimeMillis() - startTimeOne));  
24.   
25.         long startTimeTwo = System.currentTimeMillis();  
26.   
27.         Iterator<Entry<String, String>> entryKeyIterator = entrySetMap  
28.                 .entrySet().iterator();  
29.         while (entryKeyIterator.hasNext()) {  
30.             Entry<String, String> e = entryKeyIterator.next();  
31.             System.out.println(e.getValue());  
32.         }  
33.         System.out.println("entrySet spent times:"  
34.                 + (System.currentTimeMillis() - startTimeTwo));  
35.   
36.     }  
37. }  

通过测试发现,第二种方式的性能通常要比第一种方式高一倍. 

四.原因分析: 

  通过查看源代码发现,调用这个方法keySetMap.keySet()会生成KeyIterator迭代器,其next方法只返回其key值. 

Java代码  

1. private class KeyIterator extends HashIterator<K> {  
2.        public K next() {  
3.            return nextEntry().getKey();  
4.        }  
5.    }  

而调用entrySetMap.entrySet()方法会生成EntryIterator 迭代器,其next方法返回一个Entry对象的一个实例,其中包含key和value. 

Java代码  

1. private class EntryIterator extends HashIterator<Map.Entry<K,V>> {  
2.        public Map.Entry<K,V> next() {  
3.            return nextEntry();  
4.        }  
5.  }  

二者在此时的性能应该是相同的,但方式一再取得key所对应的value时,此时还要访问Map的这个方法,这时,方式一多遍历了一次table. 

Java代码  

1. public V get(Object key) {  
2.         Object k = maskNull(key);  
3.         int hash = hash(k);  
4.         int i = indexFor(hash, table.length);  
5.         Entry<K,V> e = table[i];   
6.         while (true) {  
7.             if (e == null)  
8.                 return null;  
9.             if (e.hash == hash && eq(k, e.key))   
10.                 return e.value;  
11.             e = e.next;  
12.         }  
13.     }  

这个方法就是二者性能差别的主要原因.