Java Collection API增强功能系列之六 改进的 ConcurrentHashMap：归约、搜索、计数与 Set 视图详解

Java 8 改进的 ConcurrentHashMap：归约、搜索、计数与 Set 视图详解

Java 8 对 ConcurrentHashMap 进行了重大优化，不仅提升了并发性能，还引入了许多函数式编程方法，使其在处理高并发场景时更加高效和灵活。本文将深入解析 ConcurrentHashMap 的以下改进特性：并行归约、并行搜索、计数方法及Set 视图，并通过代码示例展示其应用场景。

---

一、并行归约（Reduce）

作用

ConcurrentHashMap 的 reduce 方法允许在多线程环境下对键、值或键值对进行并行聚合操作（如求和、求最大值等），充分利用多核 CPU 的性能优势。

核心方法

// 对键和值进行归约
<U> U reduce(long parallelismThreshold,
             BiFunction<? super K, ? super V, ? extends U> transformer,
             BiFunction<? super U, ? super U, ? extends U> reducer);

// 对值进行归约
V reduceValues(long parallelismThreshold,
               BiFunction<? super V, ? super V, ? extends V> reducer);

• 参数：
  • parallelismThreshold：并行处理的阈值，当元素数量超过此值时启用并行计算。
  • transformer：将键值对转换为中间结果的函数。
  • reducer：合并中间结果的函数。

示例：计算所有值的总和

ConcurrentHashMap<String, Integer> scores = new ConcurrentHashMap<>();
scores.put("Alice", 90);
scores.put("Bob", 85);
scores.put("Charlie", 95);

// 对所有值求和
int sum = scores.reduceValues(1, (v1, v2) -> v1 + v2);
System.out.println("总分: " + sum); // 输出: 总分: 270

---

二、并行搜索（Search）

作用

search 方法用于在 ConcurrentHashMap 中并行查找满足条件的键、值或键值对，一旦找到匹配项立即返回结果，避免遍历整个集合。

核心方法

// 搜索键值对
<U> U search(long parallelismThreshold,
             BiFunction<? super K, ? super V, ? extends U> searchFunction);

// 搜索值
V searchValues(long parallelismThreshold,
               Function<? super V, ? extends V> searchFunction);

示例：查找第一个大于 90 的分数

Integer result = scores.searchValues(1, v -> v > 90 ? v : null);
System.out.println("大于90的分数: " + result); // 输出: 大于90的分数: 95

---

三、计数方法（mappingCount）

作用

mappingCount 方法返回 Map 中的键值对数量，返回类型为 long。与传统的 size() 方法相比，它避免了 Integer 溢出问题，适用于大规模数据统计。

语法

long mappingCount();

示例

System.out.println("条目数: " + scores.mappingCount()); // 输出: 条目数: 3

对比 size() 方法

• size()：返回 int，可能因数据量过大导致溢出。
• mappingCount()：返回 long，更安全且适合高并发环境。

---

四、Set 视图的线程安全改进

ConcurrentHashMap 的 keySet、values 和 entrySet 返回的视图是线程安全的，并支持直接在视图上进行并发操作。

1. keySet 视图

ConcurrentHashMap.KeySetView<String, Integer> keys = scores.keySet();
keys.add("Dave"); // 等同于 scores.put("Dave", null);（需注意默认值）

2. entrySet 视图的并行遍历

scores.entrySet().forEach(entry -> 
    System.out.println(entry.getKey() + ": " + entry.getValue())
);

3. 线程安全操作示例

// 多线程环境下安全操作
Set<String> keys = scores.keySet();
new Thread(() -> keys.remove("Alice")).start();
new Thread(() -> keys.add("Eve")).start();

---

五、其他实用方法

1. forEach 并行遍历

scores.forEach(2, (k, v) -> 
    System.out.println(k + " -> " + v)
);

2. compute 和 merge

与 HashMap 类似，但保证原子性：

scores.compute("Alice", (k, v) -> v + 10); // Alice的分数变为100
scores.merge("Bob", 85, Integer::sum);     // 若存在则累加

---

六、注意事项

1. 并行阈值设置
   parallelismThreshold 控制并行计算的触发条件。若设为 Long.MAX_VALUE，则始终串行执行；设为 1 则强制并行。

2. 线程安全与一致性
   ConcurrentHashMap 的迭代器是“弱一致性”的，反映的是遍历开始时的状态，可能不包含后续修改。

3. 避免副作用
   在 reduce 或 search 函数中避免修改 Map，否则可能导致不可预测的行为。

---

七、完整示例代码

import java.util.concurrent.ConcurrentHashMap;

public class ConcurrentHashMapDemo {
    public static void main(String[] args) {
        ConcurrentHashMap<String, Integer> scores = new ConcurrentHashMap<>();
        scores.put("Alice", 90);
        scores.put("Bob", 85);
        scores.put("Charlie", 95);

        // 归约：计算总分
        int sum = scores.reduceValues(1, Integer::sum);
        System.out.println("总分: " + sum);

        // 搜索：查找第一个大于90的值
        Integer found = scores.searchValues(1, v -> v > 90 ? v : null);
        System.out.println("大于90的分数: " + found);

        // 计数
        System.out.println("条目数: " + scores.mappingCount());

        // Set视图操作
        scores.keySet().forEach(System.out::println); // 输出所有键
    }
}

输出：

总分: 270
大于90的分数: 95
条目数: 3
Alice
Bob
Charlie

---

八、总结

Java 8 对 ConcurrentHashMap 的改进使其在高并发场景下表现更卓越：

• 并行归约与搜索：利用多核性能加速聚合和查找。
• mappingCount：安全统计大规模数据。
• 线程安全的 Set 视图：支持直接并发操作。
• 函数式方法：如 forEach、compute，代码更简洁。

这些改进不仅提升了性能，还显著增强了代码的可读性和可维护性。在处理高并发数据时，合理利用 ConcurrentHashMap 的新特性，能让你的程序如虎添翼！