最新容易忽略的ConcurrentHashMap 线程不安全行为(2)

这段代码是用10个线程测试10以内各个整型随机数出现的次数,表面上看采用ConcurrentHashMap进行contain和put操作没有任何问题。但是仔细想下,尽管 containsKey和 put 两个方法都是原子的,但在jvm中并不是将这段代码做为单条指令来执行的,例如:假设连续生成2个随机数1,map的 containsKey 和 put 方法由线程A和B 同时执行 ,那么有可能会出现A线程还没有把 1 put进去时,B线程已经在进行if 的条件判断了,也就是如下的执行顺序:

A: map 正在放置随机数 1 进去

A 被挂起

B: 执行 map.containsKey(1) 返回false

B: 将随机数 1 放进 map

A: 将随机数 1 放进 map

map 中key 为1 的value值 还是为 1

这样会导致虽然生成了2次随机数 1 ,它的value值还是1,我们期望的结果应该是2,这并不是我们想要的结果。概括的说就是两个线程同时竞争map, 但他们对map访问顺序必须是先 containsKey 然后再 put 对象进去,即产生了竞态条件。解决方法当然就是同步了,现在我们将代码改成如下:

public class ThreadSafeTest {

public static Map<Integer,Integer> map=new ConcurrentHashMap<>();

public static void main(String[] args) {

ExecutorService pool1 = Executors.newFixedThreadPool(10);

for (int i = 0; i < 10; i++) {

pool1.execute(new Runnable() {

@Override

public void run() {

Random random=new Random();

int randomNum=random.nextInt(10);

countRandom(randomNum);

}

});

}

}

public static synchronized void countRandom(int randomNum){

if(map.containsKey(randomNum)){

map.put(randomNum,map.get(randomNum)+1);

}else{

map.put(randomNum,1);

}

}

}

上述代码在当前类中没有线程安全的问题,但依然有线程安全的危险,成员变量map依然有可能会在其他地方被更改,在java并发中属于无效的同步锁,将countRandom修改成如下即可:

public static void countRandom(int randomNum){

synchronized(map){

if(map.containsKey(randomNum)){

map.put(randomNum,map.get(randomNum)+1);

}else{

map.put(randomNum,1);

}

}

}

在上述代码中由于同步的原因,ConcurrentHashMap 即使换成HashMap 也可以,只要保证map的各个操作都是线程安全的即可。

写这篇文章也是我工作中经历的一个bug, 我目前是在从事酒店行业的房间预订工作,由于每一个房型会有多个不同的产品进行售卖,在通过接口获取数据时,需要将名称相同的房型合并成为一个产品进行展示售卖,例如以下数据:

{

“roomId”: 1,

“roomName”: “大床房”,

“price”: 1805

}, {

“roomId”: 2,

“roomName”: “大床房”,

“price”: 1705

}, {

“roomId”: 3,

“roomName”: “大床房”,

“price”: 1605

}

由于是面向C端用户需要实时展示各个房型产品的价格,所以采用了多线程并使用 ConcurrentHashMap ,其中key为房型名称roomName,value为3个房型产品的数据,所以我就在线程内部使用了如下代码:

if(map.containsKey(roomName)){

map.put(roomName, map.get(roomName)+roomData2);

}else{

map.put(roomName,roomData);

}

由于公司代码不便贴出来,用以上代码展示。逻辑就是若map中包含名称相同的产品则将其取出来放到一个 List中再 put 进去。结果就是当数据量大的时候,大床房的部分价格会被覆盖没有展示出来,导致我们的产品体验很差。最后的解决办法就是上面的采用 synchronized 关键字对map做同步,这样大床房的每一个价格都会展示出来,bug解决。


2019-04-02 更新

评论区中有人提到 可以使用 ConcurrentHashMap 的 putIfAbsent 方法 ,我们看下这个方法:

public V putIfAbsent(K key, V value) {

return putVal(key, value, true);

}

/** Implementation for put and putIfAbsent */

final V putVal(K key, V value, boolean onlyIfAbsent) {

if (key == null || value == null) throw new NullPointerException();

int hash = spread(key.hashCode());

int binCount = 0;

for (Node<K,V>[] tab = table;😉 {

Node<K,V> f; int n, i, fh;

if (tab == null || (n = tab.length) == 0)

tab = initTable();

else if ((f = tabAt(tab, i = (n - 1) & hash)) == null) {

if (casTabAt(tab, i, null,

new Node<K,V>(hash, key, value, null)))

break; // no lock when adding to empty bin

}

else if ((fh = f.hash) == MOVED)

tab = helpTransfer(tab, f);

else {

V oldVal = null;

synchronized (f) {

if (tabAt(tab, i) == f) {

if (fh >= 0) {

binCount = 1;

for (Node<K,V> e = f;; ++binCount) {

K ek;

if (e.hash == hash &&

((ek = e.key) == key ||

(ek != null && key.equals(ek)))) {

oldVal = e.val;

if (!onlyIfAbsent)

e.val = value;

break;

}

Node<K,V> pred = e;

if ((e = e.next) == null) {

pred.next = new Node<K,V>(hash, key,

value, null);

break;

}

}

}

else if (f instanceof TreeBin) {

Node<K,V> p;

binCount = 2;

if ((p = ((TreeBin<K,V>)f).putTreeVal(hash, key,

value)) != null) {

oldVal = p.val;

if (!onlyIfAbsent)

p.val = value;

}

}

}

}

if (binCount != 0) {

if (binCount >= TREEIFY_THRESHOLD)

treeifyBin(tab, i);

if (oldVal != null)

return oldVal;

break;

}

}

}

addCount(1L, binCount);

return null;

}

方法中 我们可以看到

if (oldVal != null)

return oldVal;

在并发 插入的时候若原来的值存在则直接返回,否则返回 null . 这个在某些场景下是合适的 ,但在我上面提到的场景是不合适的


  • 线程安全的扩展与发散

现在我们发散一下,上面只是讲了 ConcurrentHashMap 的线程不安全行为,但是任何线程安全的数据类型都有可能出现2个线程安全的方法放在一起使用导致线程不安全的行为,如List同步器 Vector 的contains 和 add行为 、线程安全的整型计数器 AtomicInteger 的 get 和 incrementAndGet 行为等等。

  • 归纳与总结
    最后的归纳总结,我们在多线程中使用线程安全类时也需要注意是否用到了2个及以上的同步方法,若用到了则需要将这多个方法使用同步变成原子操作,所谓的原子操作一定细化到jvm内部的指令,而不能单纯的以为一行代码就是原子操作。千万不要以为只要使用线程安全数据类型就万事大吉。

网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。

需要这份系统化资料的朋友,可以点击这里获取

一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!

  • 21
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值