玖叶教程网

前端编程开发入门

Redis系列四:“撸”一下Redis锁,附代码

本文目标

1. 熟悉乐观锁ABA概念

2. 理解掌握redis事务以及watch回滚;

3. 实战redis锁

乐观锁

乐观锁是一种不会阻塞其他线程并发的机制,它不会使用数据库的锁进行实现,它的设计里面由于不阻塞其他线程,所以并不会引发线程频繁挂起和恢复,这样便能够提高并发能力,所以也有人把它称为非阻塞锁,那么它的机制是怎么样的呢?

CAS 原理概述

在 CAS 原理中,对于多个线程共同的资源,先保存一个旧值(Old Value),比如进入线程后,查询当前存量为 100 个红包,那么先把旧值保存为 100,然后经过一定的逻辑处理。

当需要扣减红包的时候,先比较数据库当前的值和旧值是否一致,如果一致则进行扣减红包的操作,否则就认为它已经被其他线程修改过了,不再进行操作,CAS 原理流程如图 1 所示。

CAS 原理并不排斥并发,也不独占资源,只是在线程开始阶段就读入线程共享数据,保存为旧值。当处理完逻辑,需要更新数据的时候,会进行一次比较,即比较各个线程当前共享的数据是否和旧值保持一致。

如果一致,就开始更新数据;如果不一致,则认为该数据已经被其他线程修改了,那么就不再更新数据,可以考虑重试或者放弃。有时候可以重试,这样就是一个可重入锁,但是 CAS 原理会有一个问题,那就是 ABA 问题,下面先来讨论一下 ABA 问题。

ABA 问题

对于乐观锁而言,我们之前讨论了存在 ABA 的问题,那么什么是 ABA 问题呢?下面看看表 1 的两个线程发生的场景。

刻线程1线程2T0————初始化 X=AT1读入X=A————T2——读入X=A——T3处理线程 1 的业务逻辑X=B修改共享变量为 BT4处理线程 1 的业务逻辑处理线程 2 业务逻辑第一段此时线程1在 X=B 的情况下运行逻辑T5处理线程 1 的业务逻辑X=A还原变量为 AT6因为判断 X=A,所以更新数据处理线程 2 业务逻辑第二段此时线程 1 无法知道线程 2 是否修改过 X,引发业务逻辑错误T7——更新数据——

在 T3 时刻,由于线程 2 修改了 X=B,此时线程 1 的业务逻辑依旧执行,但是到了 T5 时刻,线程 2 又把 X 还原为 A,那么到了 T6 时刻,使用 CAS 原理的旧值判断,线程 1 就会认为 X 值没有被修改过,于是执行了更新。

我们难以判定的是在 T4 时刻,线程 1 在 X=B 的时候,对于线程 1 的业务逻辑是否正确的问题

由于 X 在线程 2 中的值改变的过程为 A->B->A,才引发这样的问题,因此人们形象地把这类问题称为 ABA 问题。

ABA 问题的发生,是因为业务逻辑存在回退的可能性。如果加入一个非业务逻辑的属性,比如在一个数据中加入版本号(version),对于版本号有一个约定,就是只要修改 X 变量的数据,强制版本号(version)只能递增,而不会回退,即使是其他业务数据回退,它也会递增,那么 ABA 问题就解决了,如表 2 所示。

表 2 用版本号消除 ABA 问题

时刻线程1线程2T0————初始化 X=A,version=0T1读入X=A——线程1旧值:version=0T2——读入X=A线程2旧值:version=0T3处理线程1的业务逻辑X=B修改共享变量为 B,version=1T4处理线程1的业务逻辑处理线程 2 业务逻辑第一段——T5——X=A还原变量为A,version=2T6判断 version == 0,由于线程 2 两次更新数据,导致数据 version=2,所以不再更新数据处理线程 2 业务逻辑第二段此时线程 1 知道旧值 version 和当前 version 不一致,将不更新数据T7——更新数据——

只是这个 version 变量并不存在什么业务逻辑,只是为了记录更新次数,只能递增,帮助我们克服 ABA 问题罢了,有了这些理论,我们就可以开始使用乐观锁来完成抢红包业务了。

redis 事务

在 Redis 中,存在多个客户端同时向 Redis 系统发送命令的并发可能性,因此同一个数据,可能在不同的时刻被不同的线程所操纵,这样就出现了并发下的数据一致的问题。为了保证异性数据的安全性,Redis 为提供了事务方案。而 Redis 的事务是使用 MULTI-EXEC 的命令组合,使用它可以提供两个重要的保证:

  1. 事务是一个被隔离的操作,事务中的方法都会被 Redis 进行序列化并按顺序执行,事务在执行的过程中不会被其他客户端发生的命令所打断。
  2. 事务是一个原子性的操作,它要么全部执行,要么就什么都不执行。

Redis 通过MULTI 、EXEC、WATCH等命令来实现事务功能。事物提供了一种将多个命令请求打包,然后一次性、按顺序的执行多个命令的机制,并且在事物执行期间,服务器不会中断事务而去执行其他客户端的命令请求,它会将事物中所有的命令都执行完毕。

流程

  1. 开启事务
  2. 命令进入队列
  3. 执行事务。

Redis事物不支持回滚操作,所以事物队列中某个命令执行错误,整个事物也会继续执行下去。

# 开启事务
127.0.0.1:6379> MULTI
OK
# 入队
127.0.0.1:6379> set key1 v1
QUEUED
127.0.0.1:6379> INCR key1
QUEUED
127.0.0.1:6379> set key2 v2
QUEUED
127.0.0.1:6379> set key1 v3
QUEUED
# 执行
127.0.0.1:6379> EXEC
1) OK
# INCR 失败
2) (error) ERR value is not an integer or out of range
3) OK
4) OK
# key1 变为最后的v3
127.0.0.1:6379> GET key1
"v3"
# key2 set 成功
127.0.0.1:6379> GET key2
"v2"
44 90 62 4 56 43 96 24 81 89 35 84 68 11 73 99 57 95 49 25 34 75 76 41 28 86 53 69 59 64 45 3 78 19 82 48 36 91 12 94 32 9 15 37 63 70 98 92 14 6 87 88 18 2 29 33 5 30 10 39 8 83 7 80 55 31 40 47 71 58 50 93 77 51 42 20 61 54 65 74 22 16 17 79 67 21 52 100 23 26 66 27 1 13 97 38 60 0 46 85

watch

watch 命令可以决定事务是执行还是回滚,可以在 multi 命令之前使用 watch 命令监控某些键值对,然后使用 multi 命令开启事务,执行各类对数据结构进行操作的命令,这个时候这些命令就会进入队列。

如下图,在执行之前,更改了version的值,导致事务回滚,

是否可以避免 ABA 问题呢?如下图:

Redis 的机制是不会产生 ABA 问题的,这样就有利于在保证数据一致的基础上,提高高并发系统的数据读/写性能。

分布式锁

分布式锁的实现,目前常用的方案有以下三类:

  1. 数据库乐观锁;
  2. 基于分布式缓存实现的锁服务,典型代表有 Redis 和基于 Redis 的 RedLock;
  3. 基于分布式一致性算法实现的锁服务,典型代表有 ZooKeeper、Chubby 和 ETCD。

为了确保锁服务可用,通常,分布式锁需同时满足以下四个约束条件。

  1. 互斥性:在任意时刻,只有一个客户端能持有锁;
  2. 安全性:即不会形成死锁,当一个客户端在持有锁的期间崩溃而没有主动解锁的情况下,其持有的锁也能够被正确释放,并保证后续其它客户端能加锁;
  3. 可用性:就 Redis 而言,当提供锁服务的 Redis Master 节点发生宕机等不可恢复性故障时,Slave 节点能够升主并继续提供服务,支持客户端加锁和解锁;对基于分布式一致性算法实现的锁服务(如 ETCD)而言,当 Leader 节点宕机时,Follow 节点能够选举出新的 Leader 继续提供锁服务;
  4. 对称性:对于任意一个锁,其加锁和解锁必须是同一个客户端,即客户端 A 不能把客户端 B 加的锁给解了。

代码部分

上锁部分,

/**
     * 上锁
     * @param lockName 锁名字
     * @param randomVale 随机值, 为解锁是校验持有者
     * @param expireTime 过期时间,单位秒, 过期后自动删除
     * !保证上锁的安全--一条命令保证安全性
     */
    public async lock(lockName: string, randomVale: string, expire: number) {
        const start = Date.now();
        const self = this;

        return (async function intranetLock(this: any): Promise<any> {
            try {
                // 核心代码
                //* 1. 如果key不存在,则设置key-value, 且value的唯一性,保证删除的时候同样可以校验
                // * 2. 如果key存在,NX 保证不做任何操作
                const result = await self.redis.set(lockName, randomVale, self.expiryMode, expire | self.lockLeaseTime, self.setMode);
                // 上锁成功
                if (result === 'OK') {
                    console.log(`${lockName} ${randomVale} 上锁成功`);
                    return true;
                }
                // 锁超时
                if (Math.floor((Date.now() - start) / 1000) > self.lockTimeout) {
                    console.log(`${lockName} ${randomVale} 上锁重试超时结束`);
                    return false;
                }
                // 循环等待重试
                console.log(`${lockName} ${randomVale} 等待重试`);
                await sleep(3);
                console.log(`${lockName} ${randomVale} 开始重试`);
                return intranetLock();
            } catch (err) {
                throw new Error(err);
            }
        })();

    }
    44 90 62 4 56 43 96 24 81 89 35 84 68 11 73 99 57 95 49 25 34 75 76 41 28 86 53 69 59 64 45 3 78 19 82 48 36 91 12 94 32 9 15 37 63 70 98 92 14 6 87 88 18 2 29 33 5 30 10 39 8 83 7 80 55 31 40 47 71 58 50 93 77 51 42 20 61 54 65 74 22 16 17 79 67 21 52 100 23 26 66 27 1 13 97 38 60 0 46 85


解锁部分

/**
     * 解锁
     */
    public async unLock(lockName: string, randomValue: string) {
        try {
            // Watch 命令监控锁。监控锁对应的 key(lockName),事务开启后,如果其它的客户端对这个 Key 进行了更改,
            // 那么本次事务会被取消而不会执行 redis.watch(lockName)。
            await this.redis.watch(lockName);
            const value = await this.redis.get(lockName);
            // 如果获取到对应的值,与 randomValue 相等
            if (randomValue === value) {
                // 开启事务,删除锁
                const result = await this.redis.multi().del(lockName).exec();
                if (DistributedLock.RELEASE_SUCCESS === result.length) {
                    console.log('解锁成功');
                    return true;
                }
            } else {
                console.log('value 不相等,解锁失败');
            }
        } catch (error) {
            console.log('解锁error', error);

        }
    }
44 90 62 4 56 43 96 24 81 89 35 84 68 11 73 99 57 95 49 25 34 75 76 41 28 86 53 69 59 64 45 3 78 19 82 48 36 91 12 94 32 9 15 37 63 70 98 92 14 6 87 88 18 2 29 33 5 30 10 39 8 83 7 80 55 31 40 47 71 58 50 93 77 51 42 20 61 54 65 74 22 16 17 79 67 21 52 100 23 26 66 27 1 13 97 38 60 0 46 85

返回值

?  5锁 git:(main) ? ts-node index.ts
name1 b9ccabe0-2ec7-11eb-b965-933ad584a1de 上锁成功
name1 b9cdbd50-2ec7-11eb-b965-933ad584a1de 等待重试
name1 b9cdbd50-2ec7-11eb-b965-933ad584a1de 开始重试
name1 b9cdbd50-2ec7-11eb-b965-933ad584a1de 等待重试
解锁成功
unLock:  name1 b9ccabe0-2ec7-11eb-b965-933ad584a1de true
name1 b9cdbd50-2ec7-11eb-b965-933ad584a1de 开始重试
name1 b9cdbd50-2ec7-11eb-b965-933ad584a1de 上锁成功
解锁成功
unLock:  name1 b9cdbd50-2ec7-11eb-b965-933ad584a1de true

具体的实例代码:https://github.com/simuty/Integration/blob/main/Redis/

发表评论:

控制面板
您好,欢迎到访网站!
  查看权限
网站分类
最新留言