java使用ConcurrentHashMap对key进行加锁防止线程阻塞

不恋水的雨

已于 2022-04-07 14:01:41 修改

阅读量4.2k

点赞数 2

分类专栏： java并发编程

于 2022-03-23 18:05:07 首次发布

本文链接：https://blog.csdn.net/qq_36635569/article/details/123692370

版权

并发控制内存锁 ConcurrentHashMap 线程安全高并发

关键词由CSDN通过智能技术生成

java并发编程专栏收录该内容

1 篇文章

订阅专栏

package com.zhou.util;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 根据key来锁定
 * @author lang.zhou
 * @date 2022/3/22 17:45
 */
public class KeyLock {
    private static Map<String,Object> lock = new ConcurrentHashMap<>();

    /**
     * 尝试对key加一个锁，返回加锁成功或失败
     */
    public static boolean lock(String key){
        Object my = new Object();
        Object o = lock.computeIfAbsent(key,l->my);
        return my == o;
    }
    public static void unlock(String key){
        lock.remove(key);
    }
}

上述代码在并发场景中，key值相同的线程会争抢锁，但只有一个线程能抢到锁，其他没有抢到锁的线程，不去等待锁释放，直接返回失败，可以避免高并发场景下线程阻塞而导致服务器资源耗尽。

if(KeyLock.lock(key)){
    try{
        //do something
    }finally {
        //加锁成功后，一定要在finally里面解锁，保证锁一定会释放
        KeyLock.unlock(key);
    }
}else{
    return "请稍后再试";
}

应当注意的是：这种内存中的锁，只在同一个jvm内生效。抢到锁的线程执行完一定要保证释放锁，否则其他线程将无法访问key值锁定的操作。没有抢到锁的线程不能执行释放锁的操作，否则会释放掉其他线程抢到的锁。