快捷導航

redis分布式鎖的原理及代碼實例

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

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

Redis作為分布式鎖的原理

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

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

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

基本原理

當多個客戶端同時請求獲取同一把鎖時，Redis會選擇其中的一個客戶端來獲取鎖，并將鎖的value設(shè)置為隨機值。

此時其他客戶端請求獲取鎖時，需要等待該客戶端釋放鎖后再嘗試獲取，否則獲取不到鎖。

代碼演示

下面我們來看一下使用Redis實現(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();
    }
}

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

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

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