Redis分布式锁

先讲下流程:

1.在redis中设置一个锁setnx(lockkey , currenttime+timeout)。

2.返回1代表之前没有这个锁,那么就调用expire(lockkey)来重新设置超时时间,执行业务,接下来del(key),释放锁,最后结束。

3.返回0代表有这个锁,那么传统流程中就直接结束。优化后,调用get(lockkey)来获取这个锁的超时时间的value。

3.1接下来判断如果value不是空,并且已经超时的话,那么调用getset(lockkey,currenttime+timeout)方法,重新设置value值,同时会返回一个老的值,判断如果这个值为空,或者旧值和新值相等,那么代表这个锁没有变化,获取锁成功,调用第二步操作。

3.2如果条件不满足,那么结束任务。

Redis+Scheduled

模拟一个需求,用户提交订单后,若干时间后如果还没有支付,那么关闭这个订单。那么我们需要用定时任务做轮询,把超时的订单全部删除。

在分布式的情况下,多个程序,我只需要其中一个运行这个定时任务,不需要所有程序都做轮询,因此我们可以用到redis锁来管理。接下来写代码:

@Component
public class CloseOrderTask {

    @Scheduled(cron = "0 */1 * * * ?")
    public void closeOrder() throws InterruptedException {
        System.out.println("关闭订单定时任务启动");
        long timeout = Long.parseLong(PropertiesUtils.getProperty("lock.timeout", "5000"));
        Long setnxResult = RedisShardedPoolUtil.setnx("CLOSE_ORDER_TASK_LOCK" , String.valueOf(System.currentTimeMillis() + timeout));
        if (setnxResult != null && setnxResult.intValue() == 1){
            //设置成功,获取锁
            closeOrder("CLOSE_ORDER_TASK_LOCK");
        }else {
            System.out.println("没有获得锁");
        }
        System.out.println("关闭订单定时任务关闭");
    }

    private void closeOrder(String lockName) throws InterruptedException {
        //设置key的有效期50秒,防止死锁
        RedisShardedPoolUtil.expire(lockName , 50);
        System.out.println(Thread.currentThread().getName() + "获取到锁" + lockName);
        System.out.println("执行关闭订单操作");
        Thread.sleep(3000);
        RedisShardedPoolUtil.del(lockName);
        System.out.println(Thread.currentThread().getName() + "释放锁" + lockName);
    }
}
public static Long setnx(String key , String value){
        ShardedJedis jedis = null;
        Long result = null;
        try {
            jedis = RedisShardedPool.getJedis();
            result = jedis.setnx(key , value);
        } catch (Exception e) {
            e.printStackTrace();
            RedisShardedPool.returnBrokenResource(jedis);
            return result;
        }
        RedisShardedPool.returnResource(jedis);
        return result;
    }

每次请求都会去redis中找是否有锁,如果有的话就不做删除,如果没有的话就插入一条,并且设置超时时间。运行结果:

tomcat1:

tomcat2:

这样就实现了只有一个可以运行的目的,但是这里会有一个死锁的问题:假如程序运行到第9行,刚set进值,还没有设置超时时间的时候,程序崩掉了,那么这个锁就一直待在redis中了,那也就死锁了。 

这种情况我们可以在定时任务中加一个方法:

    @PreDestroy
    public void delLock(){
        RedisShardedPoolUtil.del("CLOSE_ORDER_TASK_LOCK");
    }

@PreDestroy在程序结束时运行,但是仅限于shutdown这种平和的方式,如果kill进程,这个也是没有用的,因此我们需要演进一下代码:

@Scheduled(cron = "0 */1 * * * ?")
    public void closeOrder2() throws InterruptedException {
        System.out.println("关闭订单定时任务启动");
        long timeout = Long.parseLong(PropertiesUtils.getProperty("lock.timeout", "5000"));
        Long setnxResult = RedisShardedPoolUtil.setnx("CLOSE_ORDER_TASK_LOCK" , String.valueOf(System.currentTimeMillis() + timeout));
        if (setnxResult != null && setnxResult.intValue() == 1){
            //设置成功,获取锁
            closeOrder("CLOSE_ORDER_TASK_LOCK");
        }else {
            //未获取到锁,继续判断时间戳,看是否可以重置并获取到锁
            String lockValueStr = RedisShardedPoolUtil.get("CLOSE_ORDER_TASK_LOCK");
            if (lockValueStr != null && System.currentTimeMillis() > Long.parseLong(lockValueStr)){
                String getSetResult = RedisShardedPoolUtil.getSet("CLOSE_ORDER_TASK_LOCK" , String.valueOf(System.currentTimeMillis() + timeout));
                //再次用当前时间戳getset
                //返回给定的key的旧值,通过判断旧值来决定是否可以获取锁
                if (lockValueStr == null || (lockValueStr != null && StringUtils.equals(lockValueStr , getSetResult))){
                    closeOrder("CLOSE_ORDER_TASK_LOCK");
                }else {
                    System.out.println(Thread.currentThread().getName() + "没有获得锁");
                }
            }else {
                System.out.println(Thread.currentThread().getName() + "没有获得锁");
            }
        }
        System.out.println("关闭订单定时任务关闭");
    }
    public static String getSet(String key , String value){
        ShardedJedis jedis = null;
        String result = null;
        try {
            jedis = RedisShardedPool.getJedis();
            result = jedis.getSet(key , value);
        } catch (Exception e) {
            e.printStackTrace();
            RedisShardedPool.returnBrokenResource(jedis);
            return result;
        }
        RedisShardedPool.returnResource(jedis);
        return result;
    }

其实通过代码就很清楚地看出流程了,里面用了多重防死锁,这是因为在分布式环境下,程序可能不止2台,那么所有的机器同时运行大概率会有类似问题,所以我们通过多重验证来判断是否获得锁。

 

 

### Redis 分布式锁的实现方式、使用方法及最佳实践 #### ### 1. Redis 分布式锁的基本原理 Redis 分布式锁的核心思想是利用 Redis 的原子性操作来确保锁的唯一性。通过 Redis 的 `SET` 命令,结合参数 `NX` 和 `EX`,可以在多线程环境下实现加锁和解锁的功能[^1]。此外,为了提高可用性,还可以采用 RedLock 算法或多实例部署的方式。 #### ### 2. Redis 分布式锁的实现方式 #### #### 2.1 单实例 Redis 实现分布式锁 单实例 Redis 实现分布式锁是最简单的实现方式。通过以下命令完成加锁和解锁操作: ```python import time import redis # 初始化 Redis 客户端 client = redis.StrictRedis(host='localhost', port=6379, db=0) lock_key = "distributed_lock" lock_value = "unique_identifier" # 加锁操作 def acquire_lock(): result = client.set(lock_key, lock_value, nx=True, ex=10) # 设置过期时间为 10 秒 return result is not None # 解锁操作 def release_lock(): script = """ if redis.call("get", KEYS[1]) == ARGV[1] then return redis.call("del", KEYS[1]) else return 0 end """ client.eval(script, 1, lock_key, lock_value) ``` 上述代码中,`nx=True` 确保只有当键不存在时才设置键值对,从而实现加锁功能。`ex=10` 参数为锁设置了 10 秒的过期时间,防止死锁的发生[^1]。 #### #### 2.2 多实例 Redis 实现分布式锁(RedLock 算法) 在高可用场景下,可以使用 RedLock 算法实现分布式锁。RedLock 算法通过多个 Redis 实例来确保锁的可靠性。以下是 RedLock 的伪代码实现: ```python import redis import time class RedLock: def __init__(self, redis_nodes): self.redis_nodes = [redis.StrictRedis(**node) for node in redis_nodes] def acquire_lock(self, lock_key, lock_value, ttl): quorum = len(self.redis_nodes) // 2 + 1 start_time = time.time() success_count = 0 for node in self.redis_nodes: if node.set(lock_key, lock_value, nx=True, px=ttl): success_count += 1 elapsed_time = time.time() - start_time validity_time = ttl - int(elapsed_time * 1000) if success_count >= quorum and validity_time > 0: return True, validity_time else: self.release_lock(lock_key, lock_value) return False, 0 def release_lock(self, lock_key, lock_value): for node in self.redis_nodes: try: script = """ if redis.call("get", KEYS[1]) == ARGV[1] then return redis.call("del", KEYS[1]) else return 0 end """ node.eval(script, 1, lock_key, lock_value) except Exception: pass ``` RedLock 算法要求在大多数 Redis 实例上成功加锁,并且整个过程的时间小于锁的有效期,才能认为加锁成功[^3]。 #### ### 3. Redis 分布式锁的最佳实践 #### #### 3.1 设置合理的锁超时时间 为了避免死锁问题,必须为锁设置一个合理的超时时间。如果锁持有者在超时时间内未完成任务,锁将自动释放[^1]。 #### #### 3.2 使用唯一的锁标识符 在加锁时,应为每个锁分配一个唯一的标识符(如 UUID),以便在解锁时验证锁的拥有者身份,防止误删其他线程的锁[^3]。 #### #### 3.3 防止 GC 停顿导致锁失效 Java 程序中的垃圾回收(GC)可能导致线程长时间暂停,从而使锁提前释放。为了解决这一问题,可以使用续租机制,在锁即将到期时主动延长锁的有效期。 #### #### 3.4 监控锁的竞争情况 在高并发场景下,可以通过监控锁的竞争情况来优化系统性能。例如,记录加锁失败的次数或等待时间,分析是否存在锁争用问题[^1]。 #### ### 4. 示例代码:基于 Redisson 的分布式锁实现 Redisson 是一个成熟的 Redis 客户端库,提供了丰富的分布式锁功能。以下是使用 Redisson 实现分布式锁的示例代码: ```java import org.redisson.Redisson; import org.redisson.api.RLock; import org.redisson.api.RedissonClient; import org.redisson.config.Config; public class RedissonLockExample { public static void main(String[] args) throws InterruptedException { Config config = new Config(); config.useSingleServer().setAddress("redis://127.0.0.1:6379"); RedissonClient redisson = Redisson.create(config); RLock lock = redisson.getLock("myDistributedLock"); lock.lock(); // 加锁 try { // 执行业务逻辑 System.out.println("Lock acquired, performing task..."); Thread.sleep(1000); // 模拟任务执行 } finally { lock.unlock(); // 解锁 System.out.println("Lock released."); } redisson.shutdown(); } } ``` Redisson 提供了多种锁类型,包括公平锁、可重入锁和红锁(RedLock),开发者可以根据实际需求选择合适的锁类型[^3]。 #### ### 5. 注意事项 - 在高并发场景下,应尽量减少锁的粒度,避免因锁竞争导致性能下降。 - 如果 Redis 实例发生故障,可能会导致锁丢失。因此,在关键业务场景下,建议使用哨兵模式或集群模式来提高 Redis 的可用性[^2]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值