Redis分布式锁

使用方式:在你项目的类中通过@Resource,注入RedisLockUtil类，然后调用如下所示：

@Scheduled(cron = "0 0/30 * * * ?")
public void statisticDayLive() {

    boolean lock = redisLockUtil.lock(DAYLIVELOCKNAME, DAYLIVELOCKNAME, 5 * 60 * 1000L, 3, 1L);
    try {
        if (lock) {
            System.err.println("获取锁成功");
            mvsDailyLiveService.statisticDayLive();
        } else {
            log.error("获取日活统计分布式锁失败");
        }
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        redisLockUtil.release(DAYLIVELOCKNAME, DAYLIVELOCKVALUE);
    }
}

package com.baec.antiviral.server.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;

import java.time.Duration;
import java.util.Arrays;
import java.util.concurrent.TimeUnit;


@Slf4j
@Component
public class RedisLockUtil {
    @Autowired
    private StringRedisTemplate redisTemplate;
    @Autowired
    private DefaultRedisScript<Long> redisScript;

    private static final Long SUCCESS = 1L;


    /**
     *  加锁
     * @param key       分布式锁key
     * @param value     分布式锁value
     * @param expireTime    锁定时间    毫秒
     * @param tryTimes      尝试次数
     * @param sleepMillis   休眠时间    毫秒
     * @return
     */
    public boolean lock(String key,String value,Long expireTime,int tryTimes,Long sleepMillis){
        boolean success;int count=0;
        while(!(success=tryLock(key,value,expireTime))&&count++<tryTimes){
            try {
                Thread.sleep(sleepMillis);
            } catch (InterruptedException e) {
                log.error("RedisLock|lock fail，key={}",key,e);
                break;
            }
        }

        return success;
    }

    /**
     *
     * @param key
     * @param value
     * @param expireTime
     * @return
     */
    public boolean tryLock(String key, String value, Long expireTime) {
        try{
            return redisTemplate.opsForValue().setIfAbsent(key,value, Duration.ofMillis(expireTime));
        }catch (Exception e){
            log.error("RedisLock|tryLock fail, key={}",key,e);
            return false;
        }
    }


    /**
     * 释放锁
     * @param key
     * @param value
     * @return
     */
    public boolean release(String key,String value){
        try {
            Long result = redisTemplate.execute(redisScript, Arrays.asList(key),value);
            return SUCCESS.equals(result);
        } catch (Exception e) {
            log.error("RedisLock|releaseLock fail lockKey={}|value={}", key, value, e);
        }

        return false;
    }



    @Bean
    public DefaultRedisScript<Long> defaultRedisScript() {
        DefaultRedisScript<Long> defaultRedisScript = new DefaultRedisScript<>();
        defaultRedisScript.setResultType(Long.class);
        defaultRedisScript.setScriptText("if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end");
        return defaultRedisScript;
    }

}