redis分布式锁的使用

水深无声 2023-10-14 15:44 30阅读 0赞

### 分布式锁工具类 ###

#### LockUtils.java ####

package com.shein.pmj.utils;
    
    import lombok.extern.slf4j.Slf4j;
    
    @Slf4j
    public final class LockUtils {
        
        /**
         * 分布式锁，根据业务情况传入缓存失效时间
         * PS:上面的tryGetLock，自增和设置失效时间不是原子的，如果redis自增后，
         * 系统正好挂了，此时还未设置时效时间，那么定时任务当天就无法执行了（出现概率较小）
         * <p>
         * 由于ShardedJedis没有执行命令的功能且没有原子删除key的功能，所以没有对应的解锁方法
         *
         * @param key   String
         * @param value 释放锁时用到，怎么识别当前是自己设置的锁
         * @param time  注意传入的单位是秒
         * @return 是否获取成功
         */
        public static boolean tryGetDistributedLock(String key, String value, int time) {
        
            return JedisUtil.setNxEx(key, value, time);
        }
    
        /**
         * 重试多次获取分布式锁
         *
         * @param key       String
         * @param value     value
         * @param time      过期时间，单位秒
         * @param times     重试次数
         * @param sleepTime 每次重试休眠时间
         * @return 是否获取成功
         * @throws InterruptedException InterruptedException
         */
        public static boolean retryGetDistributedLock(String key, String value, int time, int times, int sleepTime){
        
            boolean getDistributedLock = tryGetDistributedLock(key, value, time);
            if (!getDistributedLock && times > 1) {
        
                log.info("获取锁失败，将重试，key:" + key);
                try {
        
                    Thread.sleep(sleepTime * 1000);
                } catch (InterruptedException e) {
        
                    return false;
                }
                return retryGetDistributedLock(key, value, time, --times, sleepTime);
            }
            return getDistributedLock;
        }
    
        /**
         * 重试多次获取分布式锁,使用默认值（重试5次，每次休眠2秒）
         *
         * @param key   String
         * @param value value
         * @param time  过期时间，单位秒
         * @return 是否获取成功
         * @throws InterruptedException InterruptedException
         */
        public static boolean retryGetDistributedLockByDefaultValue(String key, String value, int time){
        
            return retryGetDistributedLock(key, value, time, 5, 2);
        }
        /**
         * 重试多次获取分布式锁
         *
         * @param key       String
         * @param value     value
         * @param time      过期时间，单位秒
         * @param times     重试次数
         * @param sleepTime 每次重试休眠时间
         * @param runnable runnable
         * @return 是否获取成功
         */
        public static boolean retryGetDistributedLock(String key, String value, int time, int times, int sleepTime, Runnable runnable){
        
            if (retryGetDistributedLock(key, value, time, times, sleepTime)) {
        
                try {
        
                    runnable.run();
                    return true;
                } catch (Exception e) {
        
                    log.error("分布式锁内执行失败 e:", e);
                    throw e;
                } finally {
        
                    JedisUtil.del(key);
                }
            }
            return false;
        }
    }

#### 使用实例 ####

@GetMapping("/initSupplierCountryCode")
        public AdusResponse initSupplierCountryCode(HttpServletRequest request,
                                                    @RequestParam(name = "supplierIds",required = false) List<Integer> supplierIds,
                                                    @RequestParam(name = "supplierSource",required = true) Integer supplierSource) {
        
            checkVisitSignature(request);
            String lockKey = request.getRequestURI();
            if (!LockUtils.tryGetDistributedLock(lockKey, lockKey, 30 * 60)) {
        
                throw new BusinessException("任务正在执行，请勿重复操作");
            }
     }