Redis实现分布式锁

本文主要分享一下redis的分布式锁；

　　可直接使用到项目中的,正常使用的;

 

先讲解一下 redis setIfAbsent的使用

如果为空就set值，并返回1
如果存在(不为空)不进行操作，并返回0

很明显，比get和set要好。因为先判断get，再set的用法，有可能会重复set值。

setIfAbsent 和 setnx

setIfAbsent 是java中的方法
setnx 是 redis命令中的方法

 

setnx 例子

redis> SETNX mykey "Hello"
(integer) 1
redis> SETNX mykey "World"
(integer) 0
redis> GET mykey
"Hello"

setIfAbsent 例子

BoundValueOperations boundValueOperations = this.redisTemplate.boundValueOps(redisKey);
flag = boundValueOperations.setIfAbsent(value); // flag 表示的是否set
boundValueOperations.expire(seconds, TimeUnit.SECONDS);

if(!flag){ // 重复
    repeatSerial.add(serialNo);
    continue;
}else{// 没有重复
    norepeatSerial.add(serialNo);
}

 

了解上面所述以后 直接看代码;

package com.example.demo.controller;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.test.context.junit4.SpringRunner;

import java.util.Collections;
import java.util.concurrent.TimeUnit;

/**
 * Description: RedisDemo <br>
 *
 * @author Mr.Liang
 * Date: 2020/9/27 10:25 <br>
 */
@SpringBootTest
@RunWith(SpringRunner.class)
public class RedisDemo {

    private static Logger log = LoggerFactory.getLogger(RedisDemo.class);

    @Autowired
    private RedisTemplate redisTemplate;

    /**
     * 锁时间 秒
     */
    private static final int LOCK_TIME = 60;

    @Test
    public void testLock() {

        String key = "demo";
        String value = "Lock";
        try {
            setLock(key, value, LOCK_TIME);
        } catch (Exception e) {
            //锁异常
            log.error("加锁异常{}", e);
        } finally {
            //释放锁
            releaseLock(key, value);
        }
    }

    /**
     * redis加分布式锁
     *
     * @param key
     * @param value
     * @param seconds 秒
     * @return
     */
    public boolean setLock(String key, Object value, long seconds) {
        boolean result = false;
        try {
            result = redisTemplate.opsForValue().setIfAbsent(key, value, seconds, TimeUnit.SECONDS);
        } catch (Exception e) {
            log.error("setLock发生异常", e);
        }
        return result;
    }


    /**
     * 释放锁
     *
     * @param lockKey
     * @param value
     * @return
     */
    public boolean releaseLock(String lockKey, String value) {
        //lua脚本
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] " +
                "then return redis.call('del', KEYS[1]) else return 0 end";
        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>(script, Long.class);
        Long result = (Long) redisTemplate.execute(redisScript, Collections.singletonList(lockKey), value);
        return result == 1;
    }
}

 

有问题欢迎指出交流；