在使用 ConcurrentHashMap 时,ConcurentHashMap 通过 CAS 保证了操作的线程安全。但是当我们需要进行多个操作和复合操作时,ConcurentHashMap 并不能保证操作的原子性:
get 和 put 分别是线程安全的,然而当我们先 get 后 put 操作时,从 get 到 put 的过程中间,其他线程可能已经修改了这个key对应的值。
ConcurrentHashMap 通过 compute 提供了单个 key 的原子操作,然而当我们需要操作多个 key 时,compute 无法支持。
现在我的业务场景需要将多个 key 原子地写入 ConcurentHashMap,当某个key已存在时,则所有 key 均不写入。我可以怎么样尽可能高性能地实现呢?
我考虑过:
- 整个 map 加锁,但是这样锁的粒度太大了,性能影响太大。
- 对需要操作的键加锁,这就不如直接不使用ConcurrentHashMap了。
使用CopyOnWriteArraySet保存已有的keys,写ConcurentHashMap之前,先判断Set是否存在需要写入的keys,如果不存在则立刻先写Set,再写Map