如何优雅的删除HashMap元素

文章目录

1.数据准备
2.删除方式
- 2.1.使用增强 for 循环删除
- 2.2.使用 forEach 循环删除
- 2.3.使用 Iterator 迭代器删除
- 2.4 使用 removeIf 删除（推荐使用）
- 2.5.使用 Stream 删除（推荐使用）

1.数据准备

public Map<String, String> initMap = new HashMap<String, String>() {{put("user1", "刘零");put("user2", "郑一");put("user3", "吴二");put("user4", "张三");put("user5", "李四");put("user6", "王五");put("user7", "钱六");put("user8", "孙七");
}};

2.删除方式

2.1.使用增强 for 循环删除

/*** 使用 for 循环删除*/
public void remove1() {Set<Map.Entry<String, String>> entries = new CopyOnWriteArraySet<>(initMap.entrySet());for (Map.Entry<String, String> entry : entries) {if ("王五".equals(entry.getValue())) {initMap.remove(entry.getKey());}}System.out.println(initMap);
}

输出结果：
{user1=刘零, user2=郑一, user7=钱六, user8=孙七, user5=李四, user3=吴二, user4=张三}
通过HashMap的entrySet方法获取元素集合，然后再进行循环遍历，判断value值是否为需要删除的元素，再移除对应的Key。
需要注意增强的 for 循环底层使用的迭代器 Iterator，而 HashMap 是 fail-fast 原则的错误机制，所以遍历时删除元素会出现 java.util.ConcurrentModificationException 并发修改异常。我们可以使用CopyOnWriteArraySet封装一层避免出现并发修改异常。

fail-fast:为了将错误或异常情况尽早暴露出来，避免潜在的问题在后续代码中蔓延，提高系统的稳定性和可靠性。

2.2.使用 forEach 循环删除

/*** 使用 forEach 循环删除*/
public void remove2() {ConcurrentHashMap<String, String> map = new ConcurrentHashMap<>(initMap);map.forEach((k, v) -> {if ("王五".equals(v)) {map.remove(k);}});System.out.println(map);
}

输出结果：
{user1=刘零, user2=郑一, user7=钱六, user8=孙七, user5=李四, user3=吴二, user4=张三}
通过HashMap的forEach方法循环删除目标元素，同样的使用了ConcurrentHashMap封装避免出现并发修改异常。

2.3.使用 Iterator 迭代器删除

/*** 使用 Iterator 迭代器删除*/
@Test
public void remove3() {ConcurrentHashMap<String, String> resultMap = new ConcurrentHashMap<>();Iterator<Map.Entry<String, String>> iterator= new ConcurrentHashMap<>(initMap).entrySet().iterator();while (iterator.hasNext()) {Map.Entry<String, String> entry = iterator.next();if ("王五".equals(entry.getValue())) {iterator.remove();}else {resultMap.put(entry.getKey(),entry.getValue());}}System.out.println(resultMap);
}

输出结果：
{user1=刘零, user2=郑一, user7=钱六, user8=孙七, user5=李四, user3=吴二, user4=张三}
通过Iterator迭代删除元素不会出现并发修改异常，但由于HashMap是线程不安全的，这时如果多个线程同时修改HashMap数据也会出现并发修改异常，日常使用可以先用ConcurrentHashMap封装。

2.4 使用 removeIf 删除（推荐使用）

/*** 使用 removeIf 删除*/
public void remove4() {initMap.entrySet().removeIf(entry -> "王五".equals(entry.getValue()));System.out.println(initMap);
}

输出结果：
{user1=刘零, user2=郑一, user7=钱六, user8=孙七, user5=李四, user3=吴二, user4=张三}
通过entrySet获取元素然后使用removeIf方法删除目标数据；而removeIf的底层是通过Iterator迭代器实现的。所以也存在第三种方法同样的问题。

default boolean removeIf(Predicate<? super E> filter) {Objects.requireNonNull(filter);boolean removed = false;final Iterator<E> each = iterator();while (each.hasNext()) {if (filter.test(each.next())) {each.remove();removed = true;}}return removed;
}

2.5.使用 Stream 删除（推荐使用）

/*** 使用 Stream 删除*/
public void remove5() {Map<String, String> map = initMap.entrySet().stream().filter(entry -> !"王五".equals(entry.getValue())).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));System.out.println(map);
}