java集群环境下redis分布式锁实现

清疚 2022-08-28 14:50 344阅读 0赞

# 背景 #

在很多互联网产品应用中，有些场景需要加锁处理，比如：秒杀，全局递增ID，楼层生成等等。大部分的解决方案是基于DB实现的，Redis为单进程单线程模式，采用队列模式将并发访问变成串行访问，且多客户端对Redis的连接并不存在竞争关系。其次Redis提供一些命令SETNX，GETSET，可以方便实现分布式锁机制

# 使用redis相关命令介绍 #

stringRedisTemplate.opsForValue().setIfAbsent(key, value)   如果key存在就返回0，不存在则设置并返回1

stringRedisTemplate.opsForValue().getAndSet(key, value)  获取旧值并设置新值

# 代码 #

## 分布式锁工具类 ##

package cn.hsa.pss.imbz.utils;
    
    import cn.hsa.hsaf.core.framework.web.exception.BusinessException;
    import lombok.extern.slf4j.Slf4j;
    import org.apache.commons.lang3.StringUtils;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.data.redis.core.RedisTemplate;
    import org.springframework.stereotype.Component;
    
    /**
     * 使用Redis进行分布式锁
     * @author lsj
     * @date 2021/7/23 16:07
     */
    
    @Component
    @Slf4j
    public class RedisLockCommonUtil {
        @Autowired
        private RedisTemplate<String, String> stringRedisTemplate;
    
        /**
         * Redis加锁的操作
         *
         * @param key
         * @param value 当前时间撮作为value
         * @return
         */
        public Boolean tryLock(String key, String value) {
            if (stringRedisTemplate.opsForValue().setIfAbsent(key, value)) {
                log.info("【redis分布式锁】获取成功!");
                return true;
            }
            //如果获取锁的客户端端执行时间过长，进程被kill掉，或者因为其他异常崩溃，导致无法释放锁，就会造成死锁。
            // 所以，需要对加锁要做时效性检测。因此，我们在加锁时，把当前时间戳作为value存入此锁中，
            // 通过当前时间戳和Redis中的时间戳进行对比，如果超过一定差值，认为锁已经时效，防止锁无限期的锁下去
            String currentValue = stringRedisTemplate.opsForValue().get(key);
            //当时间相差1秒表示锁已经时效
            if(StringUtils.isNotEmpty(currentValue) && (Long.valueOf(value)-Long.valueOf(currentValue))>1000){
                log.info("【redis分布式锁】由于外部原因死锁!");
                //设置并获取就值
                String oldValue = stringRedisTemplate.opsForValue().getAndSet(key, value);
               //如果oldValue=currentValue 说明中间没有其他客户端竞争,则 否则表示已经被其他客户端抢走锁
                if (StringUtils.isNotEmpty(oldValue) && oldValue.equals(currentValue)) {
                    log.info("【redis分布式锁】解除死锁成功并获取到锁!");
                    return true;
                }
            }
            log.info("【redis分布式锁】获取锁失败!已被其它客户端占用!");
            return false;
        }
    
    
        /**
         * Redis解锁的操作
         *
         * @param key
         * @param value
         */
        public void unlock(String key, String value) {
            String currentValue = stringRedisTemplate.opsForValue().get(key);
            try {
                if (StringUtils.isNotEmpty(currentValue) && currentValue.equals(value)) {
                    stringRedisTemplate.opsForValue().getOperations().delete(key);
                    log.info("【redis分布式锁】解锁成功");
                }
            } catch (Exception e) {
                e.printStackTrace();
                log.info("【redis分布式锁】 解锁失败!");
                throw new BusinessException("【redis分布式锁】 解锁失败!");
            }
        }
    }

## 使用者示例 ##

/**
         * 修改统计信息
         */
        public void updateScoStatustical(String drugScoId, String drugName, String drugType, String rewProfInfoId) {
            String key = Commonstant.DRUG_LOCK + drugScoId;
            long time = System.currentTimeMillis();
            try {
                //如果加锁失败
                if (!redisLockCommonUtil.tryLock(key, String.valueOf(time))) {
                    throw new BusinessException("网络信号差,请重新尝试");
                }
                //对药品表统计信息
                updateDrugScoStatistical(drugScoId, drugName, drugType);
                //修改专家表统计信息
                updateRewProfScoStatistical(rewProfInfoId, drugType);
            } catch (Exception e) {
                e.printStackTrace();
                throw new BusinessException(e.getMessage());
            } finally {
                //解锁
                redisLockCommonUtil.unlock(key, String.valueOf(time));
            }
        }

# 流程图 #

![watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6ZOB5oao5oaoYg_size_20_color_FFFFFF_t_70_g_se_x_16][]

[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA6ZOB5oao5oaoYg_size_20_color_FFFFFF_t_70_g_se_x_16]: /images/20220828/83356d74dfd04decb68397f8f1ef8fed.png