Java高级之HashMap中的keySet()方法

2023-07-15 16:12| 来源: 网络整理| 查看: 265

keySet()方法的源码为：

public Set keySet() { Set ks = keySet; if (ks == null) { ks = new KeySet(); keySet = ks; } return ks; }

注释为：

/** * 获取HashMap的键的集合，以Set保存 * @return 返回key的集合 */ public Set keySet() { /* 说明： 1.可以看到其实该方法中，并没有将HashMap中的键添加到Set集合中，那么是如何实现的呢？ 2.但实际上，我们访问Set集合，根本就无法通过索引，而是需要通过迭代器Iterator才能访问到元素，foreach本质上也是迭代器 3.这里的 ks 就仅仅只是一个Set引用，指向HashMap内部类KeySet的一个实例，重点在于该实例拥有自己的迭代器，当我们在使用增强for循环时才会调用该迭代器，也才会输出我们想要的东西 */ // 获取keySet Set ks = keySet; // 判断ks是否为null,如果为null，表示为空 if (ks == null) { // 实例化KeySet对象 ks = new KeySet(); // 然后将ks设置给keySet keySet = ks; } // 返回ks return ks; }

为了理解该方法参考了：源码解析HashMap的keySet()方法

但发现还是无法深入理解，最后查看增强for循环，得到了灵感。

所谓的增强型for循环，即是如下格式的for循环，其中遍历的目标可以是数组，也可以是集合（较为常用）：

for(数据类型变量名 :遍历的目标){ // }

例如：

public class Demo { public static void main(String[] args) { int[] arr=new int[]{1,2,3,4,5}; for (int i : arr) { System.out.println(i); } } }

面对这样的代码即使进行debug调试，可能也看不出来什么，但我们可以查看它的字节码文件（即后缀为.class的文件），里面会得到我们想要的。

该Demo.class的内容为：

public class Demo { public Demo() { } public static void main(String[] args) { int[] arr = new int[]{1, 2, 3, 4, 5}; int[] var2 = arr; int var3 = arr.length; for(int var4 = 0; var4 < var3; ++var4) { int i = var2[var4]; System.out.println(i); } } }

可以看到如果增强型for循环如果遍历的目标是数组，那么最终编译使用的是for循环，得到变量的值进行输出。

那么下面来看看使用增强型for来遍历List集合

public class Demo { public static void main(String[] args) { List list=new ArrayList(); list.add("唐僧"); list.add("孙悟空"); // 使用增强型for遍历循环List集合 for (String s : list) { System.out.println(s); } } }

编译后的字节码文件Demo.class内容如下：

public class Demo { public Demo() { } public static void main(String[] args) { List list = new ArrayList(); list.add("唐僧"); list.add("孙悟空"); Iterator var2 = list.iterator(); while(var2.hasNext()) { String s = (String)var2.next(); System.out.println(s); } } }

可以看到如果增强型for循环如果遍历的目标是集合，那么最终编译使用的是迭代器，得到变量的值进行输出。

正是因为可以查看编译后的字节码，所以我们可以来查看keySet()方法运行后的结果，先运行下面的代码：

public class Demo { public static void main(String[] args) { Map map=new HashMap(); map.put("abc","123"); map.put("efg","456"); // 使用增强型for遍历循环List集合 Set keySet = map.keySet(); for (String key : keySet) { System.out.println(key); } } }

查看编译生成的字节码文件Demo.class

public class Demo { public Demo() { } public static void main(String[] args) { Map map = new HashMap(); map.put("abc", "123"); map.put("efg", "456"); Set keySet = map.keySet(); Iterator var3 = keySet.iterator(); while(var3.hasNext()) { String key = (String)var3.next(); System.out.println(key); } } }

那么查看iterator()方法的源码，即迭代器的源码，该类在Set接口中定义了方法，而在HashMap在类KeySet中实现了该方法。

而我们知道类KeySet在keySet()方法中实例化了

在iterator()方法中又实例化了KeyIterator这个类对象，那么查看该类

这是一个next()方法，表示获取下一个元素，在编译后的字节码文件中，我们看到就是通过next()方法得到键key的。

在next()方法中的返回语句中又调用了nextNode()方法，nextNode().key就是键值对中的键，即我们需要的结果。

nextNode()方法的源代码如下：

下面注释这个方法：

/** * 获取下一个结点 * * @return 返回下一个结点信息 */ final Node nextNode() { // 局部变量，保存哈希桶数组table Node[] t; // 局部变量，保存当前结点的下一个结点 Node e = next; // 如果修改次数不对，那么多线程修改了HashMap if (modCount != expectedModCount) // 则抛出异常 throw new ConcurrentModificationException(); // 如果下一个结点是null if (e == null) // 则抛出NoSuchElementException异常 throw new NoSuchElementException(); // 当前的链表遍历完了就开始遍历下一个链表 if ((next = (current = e).next) == null && (t = table) != null) { do { } while (index < t.length && (next = t[index++]) == null); } // 返回下一个结点 return e; }

实际上就是遍历循环的时候获取下一个结点，而nextNode().key就得到了键的值。

所以遍历循环map.keySet()在编译时分为了三步完成：

第一步：获取到keyset的迭代器。第二步：调用hasNext()进行while循环。第三步：通过next()获取键key的值

到此，已经完成了keySet()方法的解析，主要作用就是实例化KeySet类对象，该类带有一个迭代器方法，使得可以循环遍历通过keySet()方法得到的Set集合。

总结：keySet()方法本质上是实例化KeySet类对象，然后获得该类自带的iterator()方法，从而能够获得key的集合。

参考链接：

源码解析HashMap的keySet()方法Java基础之增强型for循环

【本文地址】

公司简介

联系我们