快捷導(dǎo)航

Java結(jié)合redistemplate使用分布式鎖案例講解

更新時(shí)間：2024年08月26日 10:39:41 作者：&loopy&

在Java中使用RedisTemplate結(jié)合Redis來實(shí)現(xiàn)分布式鎖是一種常見的做法,特別適用于微服務(wù)架構(gòu)或多實(shí)例部署的應(yīng)用程序中,以確保數(shù)據(jù)的一致性和避免競態(tài)條件,下面給大家分享使用Spring Boot和RedisTemplate實(shí)現(xiàn)分布式鎖的案例,感興趣的朋友一起看看吧

在Java中使用RedisTemplate結(jié)合Redis來實(shí)現(xiàn)分布式鎖是一種常見的做法，特別適用于微服務(wù)架構(gòu)或多實(shí)例部署的應(yīng)用程序中，以確保數(shù)據(jù)的一致性和避免競態(tài)條件。以下是一個(gè)簡單的使用Spring Boot和RedisTemplate實(shí)現(xiàn)分布式鎖的案例。

1. 引入依賴

首先，確保你的Spring Boot項(xiàng)目中已經(jīng)包含了Redis相關(guān)的依賴。如果是Maven項(xiàng)目，可以在pom.xml中添加如下依賴：

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

2. 配置Redis

在application.properties或application.yml中配置Redis服務(wù)器的連接信息：

# application.properties 示例  
spring.redis.host=localhost  
spring.redis.port=6379  
spring.redis.password= # 如果Redis設(shè)置了密碼，請?zhí)顚?/pre>

3. 實(shí)現(xiàn)分布式鎖

使用Redis實(shí)現(xiàn)分布式鎖的關(guān)鍵在于利用Redis的原子操作（如SETNX或SET命令的NX、PX選項(xiàng)）來確保鎖的互斥性。在Spring Boot中，我們可以通過自定義服務(wù)或使用RedisTemplate直接操作Redis。

下面是一個(gè)簡單的分布式鎖實(shí)現(xiàn)：

import org.springframework.beans.factory.annotation.Autowired;  
import org.springframework.data.redis.core.RedisTemplate;  
import org.springframework.data.redis.core.script.DefaultRedisScript;  
import org.springframework.stereotype.Service;  
import java.util.Collections;  
import java.util.UUID;  
import java.util.concurrent.TimeUnit;  
@Service  
public class RedisLockService {  
    @Autowired  
    private RedisTemplate<String, Object> redisTemplate;  
    private static final String LOCK_PREFIX = "lock:";  
    /**  
     * 嘗試獲取鎖  
     * @param key 鎖的key  
     * @param expireTime 鎖的過期時(shí)間，單位秒  
     * @return 是否獲取鎖成功  
     */  
    public boolean tryLock(String key, long expireTime) {  
        String requestId = UUID.randomUUID().toString();  
        String result = redisTemplate.opsForValue().setIfAbsent(LOCK_PREFIX + key, requestId, expireTime, TimeUnit.SECONDS);  
        return "OK".equals(result);  
    }  
    /**  
     * 釋放鎖  
     * @param key 鎖的key  
     * @param requestId 請求的唯一標(biāo)識  
     * @return 是否釋放鎖成功  
     */  
    public boolean releaseLock(String key, String requestId) {  
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";  
        Long result = redisTemplate.execute(new DefaultRedisScript<>(script, Long.class), Collections.singletonList(LOCK_PREFIX + key), requestId);  
        return result != null && result > 0;  
    }  
}

4. 使用分布式鎖

在你的業(yè)務(wù)邏輯中，可以這樣使用RedisLockService：

@Autowired  
private RedisLockService redisLockService;  
public void criticalSection() {  
    String lockKey = "yourLockKey";  
    try {  
        if (redisLockService.tryLock(lockKey, 10)) { // 嘗試獲取鎖，10秒過期  
            // 執(zhí)行你的關(guān)鍵代碼  
            // ...  
        }  
    } finally {  
        redisLockService.releaseLock(lockKey, requestId); // 釋放鎖，requestId需要是之前嘗試加鎖時(shí)生成的UUID  
    }  
}

注意：這里的requestId需要是在嘗試加鎖時(shí)生成的，并在釋放鎖時(shí)使用相同的值。實(shí)際使用中，你可能需要在調(diào)用tryLock方法之前或方法中生成requestId，并在調(diào)用releaseLock時(shí)傳遞它。