Redis分布式锁

前言

工作将近两年的时间了，都没想着可以写些什么东西，但是光凭记忆还是容易遗忘的，现就总结下前面阶段学习的东西供后面进行查阅 。

---

 

关于Redis目前还了解的不太多，仅针对自己当前项目中用到的Redis分布式锁的实现进行讲解，后面有机会再专门记录下Redis相关的东西。

本篇主要使用Redis的setnx + get + getset命令来实现分布式锁的，主要步骤如下：

1、入redis的依赖

 

<dependency>
　　<groupId>org.springframework.boot</groupId>
　　<artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>

 

2、实现Redis的加锁和解锁的方法，这里使用StringRedisTemplate类来实现

加锁：

• 先使用setnx（get if not exist）命令来尝试获取锁，返回1则说明加锁成功，成功将值set进去；0则说明key以存在，加锁失败不做任何操作
• 如果没有获取到锁，则看锁是否过期了，如果锁过期了，则可以让新的请求来重新获取锁
• 新的请求过来，且锁已经过期，若在get和getset之间是原子的（没有其他的线程获取该锁），那个就可以获取锁 

解锁：

• 解锁就是直接将对应的key删除即可，但是前提是自己只能解自己加的锁

实现：

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;

@Component
@Slf4j
public class RedisLock
{
    @Autowired
    private StringRedisTemplate redisTemplate;

    /**
     * 加锁
     * @param key
     * @param value 当前时间+超时时间
     * @return
     */
    public boolean lock(String key, String value)
    {
        //首次请求，获取锁
        if (redisTemplate.opsForValue().setIfAbsent(key,value))
        {
            return true;
        }

        //没获取到锁，判断是否锁过期了
        String currentValue = redisTemplate.opsForValue().get(key);
        if(!StringUtils.isEmpty(currentValue) && Long.parseLong(currentValue) < System.currentTimeMillis())
        {
            //锁过期,说明可以供新的请求来重新获取锁
            String oldValue = redisTemplate.opsForValue().getAndSet(key, value);
            if(!StringUtils.isEmpty(oldValue) && oldValue.equals(currentValue))
            {
                //新请求get()的值与getset()返回的值相同，说明已经重新获取到了锁
                return true;
            }
        }

        return false;
    }

    /**
     * 解锁
     * @param key
     * @param value
     */
    public void unlock(String key, String value)
    {
        try
        {
            String currentValue = redisTemplate.opsForValue().get(key);
            if(!StringUtils.isEmpty(currentValue) && currentValue.equals(value))
            {
                //加锁和解锁是成对存在的，这里get()的值就是加锁时set()进去的值
                redisTemplate.opsForValue().getOperations().delete(key);
            }
        }catch(Exception e)
        {
            log.error("【redis分布式锁】解锁异常,{}", e);
        }
    }
}

 

3、调用锁

import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

@RunWith(SpringRunner.class)
@SpringBootTest
public class RedisLockTest
{
    /** 超时时间10s. */
    private static final int TIMEOUT = 10 * 1000;

    @Autowired
    private RedisLock redisLock;

    @Test
    public void testRedisLock() throws Exception
    {
        //加锁
        String key = "localKey";
        long time = System.currentTimeMillis() + TIMEOUT;
        if(!redisLock.lock(key, String.valueOf(time)))
        {
            throw new Exception("哎哟喂,人也太多了,换个姿势再试试~~");
        }

        //业务逻辑略

        //解锁
        redisLock.unlock(key, String.valueOf(time));
    }
}