快捷導(dǎo)航

redis分布式鎖的原理及代碼實(shí)例

更新時(shí)間：2024年01月13日 09:33:12 作者：java架構(gòu)師-太陽(yáng)

這篇文章主要介紹了redis分布式鎖的原理及代碼實(shí)例,Redis作為一款高性能內(nèi)存數(shù)據(jù)庫(kù),其提供了一種非常實(shí)用的分布式鎖解決方案,可以幫助開(kāi)發(fā)人員輕松地實(shí)現(xiàn)分布式鎖功能,對(duì)于分布式系統(tǒng)的開(kāi)發(fā)和維護(hù),具有非常大的實(shí)用價(jià)值,需要的朋友可以參考下

Redis作為分布式鎖的原理

Redis作為一款高性能內(nèi)存數(shù)據(jù)庫(kù)，其提供了一種非常實(shí)用的分布式鎖解決方案。

可以幫助開(kāi)發(fā)人員輕松地實(shí)現(xiàn)分布式鎖功能，對(duì)于分布式系統(tǒng)的開(kāi)發(fā)和維護(hù)，具有非常大的實(shí)用價(jià)值。

Redis作為分布式鎖，其核心思想是利用Redis的原子性和單線程執(zhí)行特性，以及過(guò)期時(shí)間等特性，來(lái)保證鎖的正確性和高效性。

基本原理

當(dāng)多個(gè)客戶端同時(shí)請(qǐng)求獲取同一把鎖時(shí)，Redis會(huì)選擇其中的一個(gè)客戶端來(lái)獲取鎖，并將鎖的value設(shè)置為隨機(jī)值。

此時(shí)其他客戶端請(qǐng)求獲取鎖時(shí)，需要等待該客戶端釋放鎖后再?lài)L試獲取，否則獲取不到鎖。

代碼演示

下面我們來(lái)看一下使用Redis實(shí)現(xiàn)分布式鎖的代碼演示：

初始化Redis客戶端

import redis.clients.jedis.Jedis;

public class RedisLock {

    private Jedis jedis;

    public RedisLock(String host, int port) {
        this.jedis = new Jedis(host, port);
    }

    //關(guān)閉Jedis客戶端連接
    public void close() {
        jedis.close();
    }
}

獲取鎖的方法，如果獲取成功，返回true；否則返回false。

public boolean lock(String lockKey, String requestId, int expireTime) {
    String result = jedis.set(lockKey, requestId, "NX", "PX", expireTime);
    if ("OK".equals(result)) {
        return true;
    }
    return false;
}

釋放鎖的方法

public boolean unlock(String lockKey, String requestId) {
    String lockValue = jedis.get(lockKey);
    if (requestId.equals(lockValue)) {
        jedis.del(lockKey);
        return true;
    }
    return false;
}

使用鎖

public void testLock(String lockKey) {
    String requestId = UUID.randomUUID().toString();
    RedisLock redisLock = new RedisLock("localhost", 6379);
    try {
        boolean lock = redisLock.lock(lockKey, requestId, 3000);
        if (lock) {
            System.out.println("獲取鎖成功");
            //模擬業(yè)務(wù)處理
            Thread.sleep(2000);
        } else {
            System.out.println("獲取鎖失敗");
        }
    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        boolean unlock = redisLock.unlock(lockKey, requestId);
        if (unlock) {
            System.out.println("釋放鎖成功");
        } else {
            System.out.println("釋放鎖失敗");
        }
        redisLock.close();
    }
}

上述代碼中，我們首先初始化了一個(gè)Redis客戶端，然后定義了獲取鎖和釋放鎖的方法。

在testLock方法中，我們首先生成一個(gè)隨機(jī)的請(qǐng)求ID，然后調(diào)用了獲取鎖的方法來(lái)獲取鎖，如果獲取到了鎖，則打印“獲取鎖成功”，并模擬一段業(yè)務(wù)處理時(shí)間（用Thread.sleep方法來(lái)模擬），然后再調(diào)用釋放鎖的方法來(lái)釋放鎖。

當(dāng)多個(gè)線程同時(shí)調(diào)用testLock方法時(shí)，只有一個(gè)線程能夠成功獲取鎖，其他線程則會(huì)等待，直到獲取鎖的線程釋放鎖。如果獲取失敗，也會(huì)打印“獲取鎖失敗”。