Redis分佈式鎖的使用

 爲什麼要用到分佈式鎖？

在分佈式系統中，不同的系統需要使用到同一個資源的時候，需要對這個資源進行加鎖，使得在某一時刻只有一個系統操作資源，以此來保證數據的一致性。

兩個原子方法

setnx()：不存在時設置值返回true，存在則不進行設置並返回false（對應java的setIfAbsent()方法）

getset()：設置新值並返回舊值。

可以訪問Redis中文網站查看具體的介紹：http://www.redis.cn/commands/setnx.html

 

java代碼實現如下：

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

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.ValueOperations;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;


/**
 * @author river
 * 2020/1/16
 */
@Component
public class RedisDistributedLock {

    private static final int EXPIRE_TIME = 30;//設置默認過期時間爲30秒

    @Autowired
    StringRedisTemplate stringRedisTemplate;

    /**
     * 加鎖
     * @param key 加鎖對象
     * @return 加鎖成功返回true，失敗返回false
     */
    public boolean lock(String key) {
        return lock(key, EXPIRE_TIME);
    }

    /**
     * 加鎖
     *
     * @param key    加鎖鍵值
     * @param expire 過期時間，單位秒
     * @return 加鎖成功返回true，失敗返回false
     */
    public boolean lock(String key, int expire) {
        if (null == key) {
            System.out.println("key can not be null！");
            return false;
        }
        ValueOperations<String, String> valueOperations = stringRedisTemplate.opsForValue();
        final long currentTimeMillis = System.currentTimeMillis();
        final String expireTime = String.valueOf(currentTimeMillis + expire * 1000 + 1);
        //如果不存在就設置，設置成功後直接返回true
        Boolean flag = valueOperations.setIfAbsent(key, expireTime);
        if (null != flag && flag) {
            return true;
        }
        String savedTime = valueOperations.get(key);
        //鎖過期
        if (null != savedTime && Long.parseLong(savedTime) < currentTimeMillis) {
            String oldTime = valueOperations.getAndSet(key, expireTime);
            //舊的時間就是保存的時間時，加鎖成功，返回true
            return savedTime.equals(oldTime);
        }
        return false;
    }

    /**
     * 解鎖
     * @param key 解鎖對象
     */
    public void unlock(String key) {
        if (null == key)
            return;
        ValueOperations<String, String> valueOperations = stringRedisTemplate.opsForValue();
        String savedTime = valueOperations.get(key);
        //沒有過期就刪除鎖，過期了不進行操作
        if (!StringUtils.isEmpty(savedTime) && System.currentTimeMillis() < Long.parseLong(savedTime)) {
            valueOperations.getOperations().delete(key);
        }
    }
}