最近在看分布式鎖的資料，看了 Josial L的《Redis in Action》的分布式鎖的章節(jié)。實現(xiàn)思路是利用springcloud結(jié)合redis實現(xiàn)分布式鎖。

一、簡介

一般來說，對數(shù)據(jù)進行加鎖時，程序先通過acquire獲取鎖來對數(shù)據(jù)進行排他訪問，然后對數(shù)據(jù)進行一些列的操作，最后需要釋放鎖。Redis 本身用 watch命令進行了加鎖，這個鎖是樂觀鎖。使用 watch命令對于頻繁訪問的鍵會引起性能的問題。

二、redis命令介紹

• SETNX命令（SET if Not eXists）

當且僅當 key 不存在，將 key 的值設(shè)為 value ，并返回1；若給定的 key 已經(jīng)存在，則 SETNX 不做任何動作，并返回0。

• SETEX命令

設(shè)置超時時間

• GET命令

返回 key 所關(guān)聯(lián)的字符串值，如果 key 不存在那么返回特殊值 nil 。

• DEL命令

刪除給定的一個或多個 key ,不存在的 key 會被忽略。

三、實現(xiàn)思路

由于redis的setnx命令天生就適合用來實現(xiàn)鎖的功能，這個命令只有在鍵不存在的情況下為鍵設(shè)置值。獲取鎖之后，其他程序再設(shè)置值就會失敗，即獲取不到鎖。獲取鎖失敗。只需不斷的嘗試獲取鎖，直到成功獲取鎖，或者到設(shè)置的超時時間為止。

另外為了防治死鎖，即某個程序獲取鎖之后，程序出錯，沒有釋放，其他程序無法獲取鎖，從而導(dǎo)致整個分布式系統(tǒng)無法獲取鎖而導(dǎo)致一系列問題，甚至導(dǎo)致系統(tǒng)無法正常運行。這時需要給鎖設(shè)置一個超時時間，即setex命令，鎖超時后，從而其它程序就可以獲取鎖了。

四、編碼實現(xiàn)

本文采用springboot結(jié)合redis 取實現(xiàn)的，所以你需要裝一個redis。

1. 首先引入創(chuàng)建springboot工程，引入redis 。

<dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>

        <!-- 開啟web-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        
        <!-- redis-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-redis</artifactId>
        </dependency>

2.創(chuàng)建一個鎖類

/**
 * 全局鎖，包括鎖的名稱
 * Created by fangzhipeng on 2017/4/1.
 */
public class Lock {
    private String name;
    private String value;

    public Lock(String name, String value) {
        this.name = name;
        this.value = value;
    }

    public String getName() {
        return name;
    }

    public String getValue() {
        return value;
    }

}

3.創(chuàng)建分布式鎖的具體方法,思路已經(jīng)說清楚了，代碼注釋也寫好了，就不講解了。

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
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 java.util.concurrent.TimeUnit;

/**
 * Created by fangzhipeng on 2017/4/1.
 */
@Component
public class DistributedLockHandler {

    private static final Logger logger = LoggerFactory.getLogger(DistributedLockHandler.class);
    private final static long LOCK_EXPIRE = 30 * 1000L;//單個業(yè)務(wù)持有鎖的時間30s，防止死鎖
    private final static long LOCK_TRY_INTERVAL = 30L;//默認30ms嘗試一次
    private final static long LOCK_TRY_TIMEOUT = 20 * 1000L;//默認嘗試20s

    @Autowired
    private StringRedisTemplate template;

    /**
     * 嘗試獲取全局鎖
     *
     * @param lock 鎖的名稱
     * @return true 獲取成功，false獲取失敗
     */
    public boolean tryLock(Lock lock) {
        return getLock(lock, LOCK_TRY_TIMEOUT, LOCK_TRY_INTERVAL, LOCK_EXPIRE);
    }

    /**
     * 嘗試獲取全局鎖
     *
     * @param lock    鎖的名稱
     * @param timeout 獲取超時時間 單位ms
     * @return true 獲取成功，false獲取失敗
     */
    public boolean tryLock(Lock lock, long timeout) {
        return getLock(lock, timeout, LOCK_TRY_INTERVAL, LOCK_EXPIRE);
    }

    /**
     * 嘗試獲取全局鎖
     *
     * @param lock        鎖的名稱
     * @param timeout     獲取鎖的超時時間
     * @param tryInterval 多少毫秒嘗試獲取一次
     * @return true 獲取成功，false獲取失敗
     */
    public boolean tryLock(Lock lock, long timeout, long tryInterval) {
        return getLock(lock, timeout, tryInterval, LOCK_EXPIRE);
    }

    /**
     * 嘗試獲取全局鎖
     *
     * @param lock           鎖的名稱
     * @param timeout        獲取鎖的超時時間
     * @param tryInterval    多少毫秒嘗試獲取一次
     * @param lockExpireTime 鎖的過期
     * @return true 獲取成功，false獲取失敗
     */
    public boolean tryLock(Lock lock, long timeout, long tryInterval, long lockExpireTime) {
        return getLock(lock, timeout, tryInterval, lockExpireTime);
    }


    /**
     * 操作redis獲取全局鎖
     *
     * @param lock           鎖的名稱
     * @param timeout        獲取的超時時間
     * @param tryInterval    多少ms嘗試一次
     * @param lockExpireTime 獲取成功后鎖的過期時間
     * @return true 獲取成功，false獲取失敗
     */
    public boolean getLock(Lock lock, long timeout, long tryInterval, long lockExpireTime) {
        try {
            if (StringUtils.isEmpty(lock.getName()) || StringUtils.isEmpty(lock.getValue())) {
                return false;
            }
            long startTime = System.currentTimeMillis();
            {
                if (!template.hasKey(lock.getName())) {
                    ValueOperations<String, String> ops = template.opsForValue();
                    ops.set(lock.getName(), lock.getValue(), lockExpireTime, TimeUnit.MILLISECONDS);
                    return true;
                } else {//存在鎖
                    logger.debug("lock is exist!！！");
                }
                if (System.currentTimeMillis() - startTime > timeout) {//嘗試超過了設(shè)定值之后直接跳出循環(huán)
                    return false;
                }
                Thread.sleep(tryInterval);
            }
            while (template.hasKey(lock.getName())) ;
        } catch (InterruptedException e) {
            logger.error(e.getMessage());
            return false;
        }
        return false;
    }

    /**
     * 釋放鎖
     */
    public void releaseLock(Lock lock) {
        if (!StringUtils.isEmpty(lock.getName())) {
            template.delete(lock.getName());
        }
    }

}

4.用法：

@Autowired
DistributedLockHandler distributedLockHandler;
Lock lock=new Lock("lockk","sssssssss);
if(distributedLockHandler.tryLock(lock){
    doSomething();
    distributedLockHandler.releaseLock();
}

五、注意點

在使用全局鎖時為了防止死鎖采用 setex命令，這種命令需要根據(jù)具體的業(yè)務(wù)具體設(shè)置鎖的超時時間。另外一個就是鎖的粒度性。比如在redis實戰(zhàn)中有個案列，為了實現(xiàn)買賣市場交易的功能，把整個交易市場都鎖住了，導(dǎo)致了性能不足的情況，改進方案只對買賣的商品進行加鎖而不是整個市場。

六、參考資料

Josiah.L 《reids in action》

基于Redis實現(xiàn)分布式鎖

關(guān)注我：

image