Java中Map循环安全的删除数据的4中方法

文章目录

前言
一、使用Iterator删除
二、使用 removeIf（Java 8+）
三、遍历时记录需要删除的键（不推荐）
四、使用 Stream（Java 8+）
总结

前言

在 Java 中，遍历 HashMap 并删除数据时，直接使用 for-each 循环或 Iterator 的 remove 方法可能会导致 ConcurrentModificationException 异常。为了避免这个问题，需要使用 Iterator 来安全地删除数据。

以下是常见的4中删除方法

一、使用Iterator删除

这种事最安全的方式！也是最推荐的一种方法！

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;public class Main {public static void main(String[] args) {HashMap<String, Integer> map = new HashMap<>();map.put("A", 1);map.put("B", 2);map.put("C", 3);// 使用 Iterator 遍历并删除Iterator<Map.Entry<String, Integer>> iterator = map.entrySet().iterator();while (iterator.hasNext()) {Map.Entry<String, Integer> entry = iterator.next();if (entry.getValue() == 2) { // 删除值为 2 的条目iterator.remove(); // 安全删除}}System.out.println(map); // 输出: {A=1, C=3}}
}

说明：

使用iterator.remove() 方法删除当前的数据，并不是直接操作HahsMap
这种方法不会引发ConcurrentModificationException 报错

二、使用 removeIf（Java 8+）

如果你使用的是 Java 8 或更高版本，可以使用 removeIf 方法，这是一种更简洁的方式。

import java.util.HashMap;public class Main {public static void main(String[] args) {HashMap<String, Integer> map = new HashMap<>();map.put("A", 1);map.put("B", 2);map.put("C", 3);// 使用 removeIf 删除符合条件的条目map.entrySet().removeIf(entry -> entry.getValue() == 2);System.out.println(map); // 输出: {A=1, C=3}}
}

说明：

removeIf 是 Collection 接口的方法，HashMap 的 entrySet() 返回一个集合，因此可以直接使用。
代码更简洁，适合 Java 8 及以上版本。

三、遍历时记录需要删除的键（不推荐）

如果你不想使用 Iterator 或 removeIf，可以先记录需要删除的键，然后再统一删除。这种方式不推荐，因为需要额外的空间。

import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;public class Main {public static void main(String[] args) {HashMap<String, Integer> map = new HashMap<>();map.put("A", 1);map.put("B", 2);map.put("C", 3);// 记录需要删除的键Set<String> keysToRemove = new HashSet<>();for (Map.Entry<String, Integer> entry : map.entrySet()) {if (entry.getValue() == 2) {keysToRemove.add(entry.getKey());}}// 统一删除for (String key : keysToRemove) {map.remove(key);}System.out.println(map); // 输出: {A=1, C=3}}
}

说明：

需要额外的集合（如 HashSet）来存储需要删除的键。
不推荐，因为代码冗长且效率较低。

四、使用 Stream（Java 8+）

如果你喜欢函数式编程风格，可以使用 Stream 来过滤并生成新的 HashMap。

import java.util.HashMap;
import java.util.Map;
import java.util.stream.Collectors;public class Main {public static void main(String[] args) {HashMap<String, Integer> map = new HashMap<>();map.put("A", 1);map.put("B", 2);map.put("C", 3);// 使用 Stream 过滤并生成新的 HashMapMap<String, Integer> filteredMap = map.entrySet().stream().filter(entry -> entry.getValue() != 2).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));System.out.println(filteredMap); // 输出: {A=1, C=3}}
}