Java Zookeeper - 分布式锁

小灰灰 2023-07-01 09:00 88阅读 0赞

## 前言 ##

本文来介绍下如何实现Zookeeper分布式锁，代码优化下就可以用了

## 分布式锁 ##

分布式锁是通过互诉等手段，来实现分布式系统之间同步访问共享资源，以保证数据的一致性。

\*\*锁定义：\*\*Zookeeper通过创建临时数据节点来表示一个锁，并且是顺序节点

\*\*获取锁：\*\*成功创建数据节点的客户端即为获取锁。未获取锁的客户端通过监听该节点变更情况，以便重新获取锁

**释放锁：**

*  当前客户端发生宕机的情况下，临时节点自动删除
 *  正常业务逻辑处理完毕后，客户端自动删除自己创建的临时节点

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly94aWFtZW4uYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70]

## 简单示例 ##

也就是不使用 Curator的方式实现，肯定是不推荐的，反正就是收藏下

*  [不使用第三方库][Link 1]

## Curator - 分布式锁 ##

它的实现也是通过创建临时的顺序节点机制

<table> 
 <thead> 
  <tr> 
   <th>类型</th> 
   <th>说明</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>InterProcessMutex</td> 
   <td>互斥锁</td> 
  </tr> 
 </tbody> 
</table>

zk客户端登录，创建节点

create /orderNo 1000

代码实现，轻松实现分布式锁 `这个只是演示代码，工作中按自己业务优化`

public class CreateLock { 
    
        public static void main(String[] args) throws Exception { 
            String zkServer = "127.0.0.1:2181";
            CuratorFramework curatorFramework = CuratorFrameworkFactory
                    .newClient(zkServer, new RetryNTimes(3, 1000));
            curatorFramework.start();
            System.out.println("Curator 连接成功..");
            // 互斥锁
            String orderNoPath = "/orderNo";
            InterProcessMutex interProcessMutex = new InterProcessMutex(
              curatorFramework, orderNoPath);
    
            // 批量创建订单，测试分布式锁
            for (int i = 0; i < 10; i++) { 
                new Thread(() -> { 
                    try { 
                        // 获得锁
                        interProcessMutex.acquire();
                        // 生成订单
                        int order = generateOrder(orderNoPath, curatorFramework);
                        System.out.println("当前订单号：" + order);
                        // 释放
                        interProcessMutex.release();
                    } catch (Exception e) { 
                        e.printStackTrace();
                    }
    
                }).start();
            }
    
        }
    
        public static int generateOrder(
          String orderNoPath, CuratorFramework curatorFramework) throws Exception { 
            // 订单号
            String currentNo = new String(curatorFramework.getData().forPath(orderNoPath));
            int order = Integer.parseInt(currentNo) + 1;
            currentNo = String.valueOf(order);
            curatorFramework.setData().forPath(orderNoPath, currentNo.getBytes());
            // 返回订单号
            return order;
        }
    }

## SpringBoot中使用 ##

请结合 [Java Zookeeper - Curator的使用][Java Zookeeper - Curator] 上一章补全代码

@Slf4j
    @RestController
    @RequiredArgsConstructor(onConstructor = @__(@Autowired))
    public class RestBizController { 
        private final CuratorFramework curatorFramework;
    
        @GetMapping("/zk")
        public String zk() { 
            String value = null;
            try { 
                byte[] bytes = curatorFramework.getData().forPath("/node_test");
                value = new String(bytes);
            } catch (Exception e) { 
                e.printStackTrace();
            }
            return value;
        }
    
        @GetMapping("/lock")
        public int getLock() throws Exception { 
            String orderNoPath = "/orderNo";
            InterProcessMutex interProcessMutex = new InterProcessMutex(curatorFramework, orderNoPath);
            // 获得锁
            interProcessMutex.acquire();
            // 生成订单
            int order = generateOrder(orderNoPath);
            log.info("当前订单号：" + order);
            // 释放
            interProcessMutex.release();
            // 返回
            return order;
        }
    
        public  int generateOrder(String orderNoPath) throws Exception { 
            // 订单号
            String currentNo = new String(this.curatorFramework.getData().forPath(orderNoPath));
            int order = Integer.parseInt(currentNo) + 1;
            currentNo = String.valueOf(order);
            curatorFramework.setData().forPath(orderNoPath, currentNo.getBytes());
            log.info("订单号：{}", order);
            // 估计延迟
            Thread.sleep(5000);
            // 返回订单号
            return order;
        }
    
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly94aWFtZW4uYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200121090911878.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly94aWFtZW4uYmxvZy5jc2RuLm5ldA==,size_16,color_FFFFFF,t_70
[Link 1]: https://download.csdn.net/download/qq_30889373/12116688
[Java Zookeeper - Curator]: https://xiamen.blog.csdn.net/article/details/104058450