使用ZooKeeper原生API实现分布式锁

分布式锁的引入

一个很典型的秒杀场景,或者说并发量非常高的场景下,对商品库存的操作,我用一个SpringBoot小项目模拟一下。

用到的技术知识:

  • SpringBoot
  • Redis
  • ZooKeeper

我提前将库存 stock 放在redis,初始值为288:

127.0.0.1:6379> set stock 288OK127.0.0.1:6379> get stock"288"复制代码

扣减库存的api:

@RequestMapping("/v1/reduce")public String reduceStock() {    String stockStr = redisTemplate.opsForValue().get("stock");    int stock = Integer.parseInt(stockStr);    if (stock > 0) {        int realStock = stock - 1;        redisTemplate.opsForValue().set("stock", String.valueOf(realStock));        System.out.println(Thread.currentThread().getName() + " 减库存成功,剩余库存:" + realStock);    } else {        System.out.println("不能再减了,没有库存了!");    }    return port + ": reduce stock end";}复制代码

单机环境下的高并发:

我用 Apache JMeter 模拟在同一时刻,有 500 个请求打到 /stock/v1/reduce 上进行减库存的操作:

Apache JMeter是Apache组织开发的基于Java的压力测试工具。用它很容易模拟出高并发场景。

Apache JMeter官网: jmeter.apache.org/download_jm…

启动SpringBoot项目,执行压测,运行结果:

500个并发,才扣减了5个!!!BOSS该找你事了!

不过这种情况我们程序员是不会让它出现的,加个 synchronized ,让每个线程拿到锁之后再去扣减库存:

代码实现:

@RequestMapping("/v2/reduce")public String reduceStockV2() {    //加线程同步    synchronized (this) {        String stockStr = redisTemplate.opsForValue().get("stock");        int stock = Integer.parseInt(stockStr);        if (stock > 0) {            int realStock = stock - 1;            redisTemplate.opsForValue().set("stock", String.valueOf(realStock));            System.out.println(Thread.currentThread().getName() + " 减库存成功,剩余库存:" + realStock);        } else {            System.out.println("不能再减了,没有库存了!");        }    }    return port + ": reduce stock end";}复制代码

压测结果:

synchronized只能帮我们解决单进程内的线程安全问题,而事实上,在分布式环境下,我们一个服务,比如这个扣减库存的服务,可能存在于多个服务器(多个tomcat进程)中,再使用 synchronized 就无法解决分布式服务下的高并发问题了。

我们来模拟一下分布式环境下的高并发问题,我本地开启两个扣减库存的服务,一个8080端口,一个8090端口,使用Nginx做反向代理:

我的Nginx服务器的IP地址为 192.168.134.135 ,我开启 200个线程循环4次 ,相当于一共有 800 次访问 http://192.168.134.135/stock/v2/reduce ,如此就可以模拟出分布式下的高并发场景了。

开始压测:

压测结果:

服务1:

服务2:

发现 synchronized 根本没用,很多线程还是重复扣减了库存!!!

这种时候就需要 分布式锁 来解决这个问题了。

使用ZooKeeper实现分布式锁

本案例采用zk自己的api实现分布式锁。当然还可以使用第三方提供的api实现zk分布式锁,比如 Curator (现在Curator也归属于Apache了)。

Curator官网:curator.apache.org/

后续会考虑出一个Curator版本的zk分布式锁的实现。本文提供的方式掌握了,其他版本就不在话下了,哈哈!

为什么zk能实现分布式锁呢?我们之前已经了解过zk拥有类似于Linux文件系统的数据结构,还有一套事件监听机制。

zk的数据结构中,每个节点还可以存数据,这个就比较厉害了。还可以创建顺序节点,节点带编号。临时节点还有随session存在而存在的特性,所以当客户端失去连接的时候session消亡,临时节点也就消失了。

还有,zk的事件监听机制。zk上所有的节点,持久节点,顺序节点,临时顺序节点等都可以对它进行监听,当某一个节点上发生了变化,比如当节点被创建了、数据更新、 节点被删除 等,这些 事件都会被监听 到,同时 触发回调 ,那么我们在代码中就能够做一些相应的处理。

这里其实有个问题,如果我们只关注 /lock 节点的话,并发量一高会带来通信压力,因为很多client都watch了 /lock 节点,当 /lock 节点发生变化,这些client一窝蜂的进行事件回调争抢锁,压力就出现了。

zk的 临时顺序节点 能帮我们解决这个问题,我们只监听顺序节点的前一个节点,看它是不是顺序最小的,让顺序最小的获得锁!

因此,基于zk的 临时顺序节点事件监听 ,我们就可以实现 分布式锁

代码实现

总体框架,三大步:

  1. 争抢锁
  2. 做自己爱做的事情
  3. 释放锁

先把这个架子搭起来:

@RequestMapping("/v3/reduce")public String reduceStockV3() {    try {        //1. 抢锁        tryLock        //2. 做自己爱做的事        //比如减库存        //3. 释放锁        releaseLock    } catch (Exception e) {        e.printStackTrace();    }    return port + ": reduce stock end";}复制代码

zk API的响应式编程很爽,前文 使用ZooKeeper客户端原生API实现分布式配置中心 就是用的是响应式编程。

响应式编程很好理解,就是对事件加监听,当完成某个事件的时候,就出发相应的回调函数,zk的很多api都提供了方法的异步调用版本。

前文分析,实现分布式锁的流程是: 抢锁就是创建临时有序节点,监听创建成功后,获取根节点(连接zk集群时候的根节点)下的所有孩子节点,然后比较当前节点时候为第一个(要排好序),若是第一个,则抢锁成功,做自己的业务,然后释放锁-删除节点,当然删除节点也会触发回调。

因此,核心代码逻辑都在 监听回调 里,抽象出一个 WatchAndCallback 出来:

public class WatchAndCallback implements Watcher, AsyncCallback.StringCallback, AsyncCallback.Children2Callback, AsyncCallback.StatCallback {    //肯定得有zookeeper    private final ZooKeeper zk;    //创建的节点名称,节点监听它的前一个顺序节点时需要用到    private String pathName;    //也得有CountDownLatch,用来保证程序阻塞与继续执行    private final CountDownLatch latch = new CountDownLatch(1);    //线程名称,辅助观察    private final String threadName;    public WatchAndCallback(ZooKeeper zk, String threadName) {        this.zk = zk;        this.threadName = threadName;    }    //Children2Callback    //此回调用于检索节点的子节点和stat    //处理异步调用的结果    // List children 给定路径上节点的子节点的无序数组,基于watch前一个节点,最小节点获得锁的机制,需要给children排个序    @Override    public void processResult(int rc, String path, Object ctx, List children, Stat stat) {        //实现分布式锁的核心            }    //StatCallback//    处理异步调用的结果//    成功, rc is KeeperException.Code.OK.//    失败, rc is set to the corresponding failure code in KeeperException.    @Override    public void processResult(int rc, String path, Object ctx, Stat stat) {        //TODO    }    //创建节点时的回调 StringCallback    //此回调用于检索节点的名称    // name: 创建的znode的名称。如果成功,名称和路径通常相等,除非创建了顺序节点。    @Override    public void processResult(int rc, String path, Object ctx, String name) {            }    //Watcher    //如果某个线程释放锁了,也就是节点被删除了,也要触发监听    //如果不是释放锁而是某个节点本身出问题了,zk也会删除node,也需要一个监听    @Override    public void process(WatchedEvent watchedEvent) {            }}复制代码

我把分布式锁的代码逻辑都放在 WatchAndCallback 类中。

  1. 抢锁,tryLock
/** * 抢锁---zk节点具有互斥性,当已存在该节点时会创建失败,所以创建临时节点可以知道是否抢锁成功 * @author 行百里者 * @create 2020/9/18 15:37 **/public void tryLock() {    System.out.println(threadName + " 试图抢锁");    //创建临时序列节点,data就设置为当前线程名字即可,实际业务可设置为用户id    //创建节点也有callback,传一个this即可,会触发调用processResult(int rc, String path, Object ctx, String name)    zk.create("/stock",            threadName.getBytes(),            ZooDefs.Ids.OPEN_ACL_UNSAFE,            CreateMode.EPHEMERAL_SEQUENTIAL,            this,            "create node ctx");    //异步的,要await一下    try {        latch.await();    } catch (InterruptedException e) {        e.printStackTrace();    }}复制代码
  1. 减库存,保存lucky(实际场景中的订单信息)

在 controller 中,也就是说当一个客户抢锁成功,就要对数据做相应的改变:

//2. 做自己爱做的事int stock = Integer.parseInt(redisTemplate.opsForValue().get("stock"));if (stock > 0) {    int realStock = stock - 1;    redisTemplate.opsForValue().set("stock", String.valueOf(realStock));    //同时lucky+1    redisTemplate.opsForValue().increment("lucky");} else {    System.out.println("库存不足");}复制代码
  1. 给别人机会,释放锁,releaseLock
//也就是删除临时节点public void releaseLock() {    try {        zk.delete(pathName, -1);        System.out.println(threadName + " 流程走完了,释放锁");    } catch (InterruptedException e) {        e.printStackTrace();    } catch (KeeperException e) {        e.printStackTrace();    }}复制代码

压力测试

还是用JMeter模拟高并发场景,一次并发800个请求打到nginx上,nginx反向代理到两个tomcat。

首先开启两个tomcat:

初始化库存仍设置为288(redis),同时在设置一个 lucky ,表示有多少人抢到了库存中的数据,这个数据最终为288才是正确的:

127.0.0.1:6379> set stock 288OK127.0.0.1:6379> get stock"288"127.0.0.1:6379> set lucky 0OK127.0.0.1:6379> get lucky"0"复制代码

初始化zk lock根节点:

[zk: localhost:2181(CONNECTED) 5] ls /lock[]复制代码

压力测试:

看一下后台redis中存储的stock和lucky数据:

127.0.0.1:6379> get stock"0"127.0.0.1:6379> get lucky"288"127.0.0.1:6379> 复制代码

stock为0说明没有超卖,lucky=288说明也没有同一个库存扣减多次的情况。

ok,zk实现分布式锁就是这么完美!

小结

zk实现分布式锁:

  1. 争抢锁,只有一个能获得锁
  2. 获得锁的人,如果故障了,导致死锁,用 zookeeper 实现分布式锁解决,基于zk的特征,可以创建临时节点,产生一个session,它如果挂了,session会消失,然后释放锁,因此 ZooKeeper能规避死锁
  3. 获得锁的人成功了,释放锁
  4. 锁被释放/删除,别人怎么知道?方式1:for循环,主动轮询,发送心跳,这种方式的弊端:延迟(实时性不强),压力(服务很多,都去轮询访问某一把锁)方式2:watch 解决延迟问题,此法也有弊端:通信压力(watch完了很多服务器回调去抢锁)方式3: 临时顺序节点 + 事件监听机制 (序列节点+watch) watch谁?watch前一个节点,最小的获得锁,一旦最小的释放锁,此法的效率成本是 ZooKeeper只给第二个node发事件回调

发表评论
留言与评论(共有 0 条评论) “”
   
验证码:

相关文章

推荐文章