SpringBoot 使用 Redis(lettuce) 分布式锁

最新推荐文章于 2024-08-01 02:10:00 发布

热血蜗牛

最新推荐文章于 2024-08-01 02:10:00 发布

阅读量2.6k

点赞数 2

分类专栏： Spring boot Redis 分布式

本文链接：https://blog.csdn.net/u010663021/article/details/108342989

版权

Spring boot 同时被 3 个专栏收录

13 篇文章 1 订阅

订阅专栏

Redis

6 篇文章 0 订阅

订阅专栏

分布式

1 篇文章 0 订阅

订阅专栏

一、引入依赖

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

二、redis(lettuce) 配置

spring:
  redis:
    # Redis默认情况下有16个分片，这里配置具体使用的分片。默认是索引为0的分片。redis集群只有db0
    database: 0
    # Redis服务器地址
    host: 192.168.249.154
    # Redis服务器连接端口
    port: 6379
    # Redis服务器连接密码（默认为空）
    password:
    # 连接超时时间（毫秒）
    timeout: 2000s
    # 配置文件中添加 lettuce.pool 相关配置，则会使用到lettuce连接池
    lettuce:
      pool:
        # 连接池最大阻塞等待时间（使用负值表示没有限制） 默认 -1
        max-wait: 60s
        # 连接池中的最大空闲连接 默认 8
        max-idle: 8
        # 连接池中的最小空闲连接 默认 0
        min-idle: 5
        # 连接池最大连接数（使用负值表示没有限制） 默认 8
        max-active: 8

三、Redis配置类

@Configuration
public class RedisConfig {
    @Bean
    public RedisTemplate<String, Object> redisTemplate(LettuceConnectionFactory redisConnectionFactory) {
        RedisTemplate<String, Object> redisTemplate = new RedisTemplate<>();
        redisTemplate.setKeySerializer(new StringRedisSerializer());
        redisTemplate.setValueSerializer(new GenericJackson2JsonRedisSerializer());
        redisTemplate.setConnectionFactory(redisConnectionFactory);
        return redisTemplate;
    }
}

四、Redis分布式锁工具类

@Component
public class RedisLockUtil {
    @Autowired
    private RedisTemplate<String, Object> redisTemplate;

    private static final byte[] SCRIPT_RELEASE_LOCK = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end".getBytes();

    /**
     * 尝试获取分布式锁
     *
     * @param key       键
     * @param requestId 请求ID
     * @param expire    锁的有效时间（秒）
     */
    public synchronized Boolean tryLock(String key, String requestId, long expire) {
        return redisTemplate.execute((RedisCallback<Boolean>) redisConnection -> redisConnection.set(key.getBytes(), requestId.getBytes(), Expiration.from(expire, TimeUnit.SECONDS), RedisStringCommands.SetOption.SET_IF_ABSENT));
    }

    /**
     * 释放分布式锁
     *
     * @param key       键
     * @param requestId 请求ID
     */
    public synchronized Boolean releaseLock(String key, String requestId) {
        return redisTemplate.execute((RedisCallback<Boolean>) redisConnection -> redisConnection.eval(SCRIPT_RELEASE_LOCK, ReturnType.BOOLEAN, 1, key.getBytes(), requestId.getBytes()));
    }
}

五、使用示例

//获取锁
Boolean lock = redisLockUtil.tryLock(rKeyLock, requestId, 10);
if (null != lock && lock) {
    try {
        //TODO 业务处理
        
    } finally {
        //释放锁
        Boolean release = redisLockUtil.releaseLock(rKeyLock, requestId);
    }
} else {
    log.warn("redis锁获取失败|{}|{}", rKeyLock, lock);
}