From ac3e5ac5c6f4615029ed9335c55904f28ae9223c Mon Sep 17 00:00:00 2001 From: "Yangkai.Shen" <237497819@qq.com> Date: Mon, 5 Sep 2022 21:46:08 +0800 Subject: [PATCH] =?UTF-8?q?:sparkles:=20=E5=88=86=E5=B8=83=E5=BC=8F?= =?UTF-8?q?=E9=94=81=E6=A8=A1=E5=9D=97=E4=B9=8B=20=E5=9F=BA=E4=BA=8E=20Zoo?= =?UTF-8?q?keeper=E5=8E=9F=E7=94=9F=E5=AE=A2=E6=88=B7=E7=AB=AF=20=E5=AE=9E?= =?UTF-8?q?=E7=8E=B0=E5=88=86=E5=B8=83=E5=BC=8F=E9=94=81=20=E6=A1=88?= =?UTF-8?q?=E4=BE=8B=E5=AE=8C=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../demo-distributed-lock-zookeeper/README.md | 328 +++++++++++++++++++++ .../docker-compose.env.yml | 14 + .../demo-distributed-lock-zookeeper/pom.xml | 35 +++ .../lock/ZookeeperDistributedLockApplication.java | 21 ++ .../distributed/lock/autoconfigure/ZkClient.java | 77 +++++ .../autoconfigure/ZookeeperDistributedLock.java | 156 ++++++++++ .../ZookeeperDistributedLockAutoConfiguration.java | 25 ++ .../ZookeeperDistributedLockClient.java | 33 +++ .../src/main/resources/application.yml | 16 + 9 files changed, 705 insertions(+) create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/README.md create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/docker-compose.env.yml create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/ZookeeperDistributedLockApplication.java create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZkClient.java create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLock.java create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockAutoConfiguration.java create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockClient.java create mode 100644 demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/resources/application.yml diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/README.md b/demo-distributed-lock/demo-distributed-lock-zookeeper/README.md new file mode 100644 index 0000000..d689139 --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/README.md @@ -0,0 +1,328 @@ +## spring-boot-demo-distributed-lock-zookeeper + +> 此 demo 主要演示了 Spring Boot 如何基于 Zookeeper 原生客户端实现一个分布式锁，支持可重入 + +## 1.开发步骤 + +在 `demo-distributed-lock-api` 模块中，已经实现了基于 AOP 的分布式锁注解拦截、简单的扣减库存案例，因此本模块只需要实现以下两个接口即可。 +- `com.xkcoding.distributed.lock.api.DistributedLock` +- `com.xkcoding.distributed.lock.api.DistributedLockClient` + +### 1.1.添加依赖 + +```xml + + + com.xkcoding + common-tools + + + + com.xkcoding + demo-distributed-lock-api + 1.0.0-SNAPSHOT + + + + org.apache.zookeeper + zookeeper + ${zookeeper.version} + + + + org.projectlombok + lombok + true + + +``` + +### 1.2.代码实现 + +#### 1.2.1.ZookeeperDistributedLock + +> 基于 Zookeeper 实现分布式锁，主要是通过创建并监听临时有序节点 + ThreadLocal 数据结构实现可重入性 + +```java +public class ZookeeperDistributedLock extends DistributedLock { + + private final ZooKeeper zooKeeper; + + private final String lockRootPath; + + private final String lockNodePath; + + private final ThreadLocal LOCK_TIMES = new ThreadLocal<>(); + + protected ZookeeperDistributedLock(ZkClient client, String lockKey, long lockTime, TimeUnit timeUnit) { + super(lockKey, lockTime, timeUnit); + this.lockRootPath = client.getLockRootPath(); + this.zooKeeper = client.getZooKeeper(); + try { + // 直接创建临时顺序节点 + this.lockNodePath = this.zooKeeper.create(client.getLockRootPath() + "/" + lockKey + "-", null, ZooDefs.Ids.OPEN_ACL_UNSAFE, + CreateMode.EPHEMERAL_SEQUENTIAL); + } catch (KeeperException | InterruptedException e) { + throw new RuntimeException(e); + } + } + + @Override + public void lock() { + // 如果是锁重入，则次数加 1，然后直接返回 + Integer lockTimes = LOCK_TIMES.get(); + if (lockTimes != null && lockTimes > 0L) { + LOCK_TIMES.set(lockTimes + 1); + return; + } + + try { + // 当前节点是第一个临时节点，如果是，直接获得锁 + String previousLockNodePath = getPreviousNodePath(lockNodePath); + if (StrUtil.isBlank(previousLockNodePath)) { + LOCK_TIMES.set(1); + return; + } else { + // 如果不是第一个临时节点，则给它的前一个临时节点添加监听删除事件（即锁释放） + CountDownLatch wait = new CountDownLatch(1); + Stat stat = this.zooKeeper.exists(lockRootPath + "/" + previousLockNodePath, new Watcher() { + @Override + public void process(WatchedEvent event) { + if (Event.EventType.NodeDeleted == event.getType()) { + // 节点删除，通知获得锁，避免自旋判断影响性能 + wait.countDown(); + } + } + }); + // 未被删除，则需要阻塞 + if (stat != null) { + wait.await(); + } + // 获得锁，则重入次数设置为 1 + LOCK_TIMES.set(1); + } + } catch (Exception e) { + try { + TimeUnit.MILLISECONDS.sleep(50); + } catch (InterruptedException ex) { + throw new RuntimeException(ex); + } + // 重新尝试获取锁 + lock(); + } + + } + + @Override + public boolean tryLock() { + throw new UnsupportedOperationException("ZookeeperDistributedLock`s tryLock method is unsupported"); + } + + @Override + public boolean tryLock(long time, @NotNull TimeUnit unit) { + throw new UnsupportedOperationException("ZookeeperDistributedLock`s tryLock method is unsupported"); + } + + @Override + public void unlock() { + // -1 代表不使用乐观锁删除，类似 delete --force + Integer lockTimes = LOCK_TIMES.get(); + if (lockTimes != null && lockTimes > 0) { + // 计算剩下的重入次数 + int leftLockTime = lockTimes - 1; + LOCK_TIMES.set(leftLockTime); + // 剩下为 0 的时候，代表完全解锁，需要删除临时节点 + if (leftLockTime == 0) { + try { + this.zooKeeper.delete(lockNodePath, -1); + LOCK_TIMES.remove(); + } catch (InterruptedException | KeeperException e) { + throw new RuntimeException(e); + } + } + } + } + + /** + * 根据当前节点，获取前一个节点 + * + * @param currentNodePath 当前节点路径 + * @return 前一个节点 + */ + private String getPreviousNodePath(String currentNodePath) { + try { + // 截取当前临时节点编号 + long currentNodeSeq = Long.parseLong(StrUtil.subAfter(currentNodePath, "-", true)); + // 获取锁根路径下的所有临时节点 + List nodes = this.zooKeeper.getChildren(lockRootPath, false); + + if (CollUtil.isEmpty(nodes)) { + return null; + } + + // 用于标记遍历所有子节点时，离当前最近的一次编号 + long previousNodeSeq = 0L; + String previousNodePath = null; + for (String nodePath : nodes) { + // 获取每个临时节点编号 + long nodeSeq = Long.parseLong(StrUtil.subAfter(nodePath, "-", true)); + if (nodeSeq < currentNodeSeq && nodeSeq > previousNodeSeq) { + previousNodeSeq = nodeSeq; + previousNodePath = nodePath; + } + } + + return previousNodePath; + } catch (KeeperException | InterruptedException e) { + throw new RuntimeException(e); + } + + } +} +``` + +#### 1.2.2.ZkClient + +> 配置 Zookeeper 客户端连接 + +```java +@Slf4j +public class ZkClient { + /** + * 连接地址 + */ + @Getter + private final String connectServer; + + /** + * 节点根路径，默认是 /locks + */ + @Getter + private final String lockRootPath; + + @Getter + private ZooKeeper zooKeeper; + + private static final String DEFAULT_ROOT_PATH = "/locks"; + + public ZkClient(String connectServer) { + this.connectServer = connectServer; + this.lockRootPath = DEFAULT_ROOT_PATH; + } + + public ZkClient(String connectServer, String lockRootPath) { + this.connectServer = connectServer; + this.lockRootPath = lockRootPath; + } + + public void init() { + try { + this.zooKeeper = new ZooKeeper(connectServer, 3000, new Watcher() { + @Override + public void process(WatchedEvent watchedEvent) { + if (Event.KeeperState.SyncConnected == watchedEvent.getState() && Event.EventType.None == watchedEvent.getType()) { + log.info("===========> zookeeper connected <==========="); + } + } + }); + // 判断根节点是否存在，不存在则创建 + if (this.zooKeeper.exists(lockRootPath, false) == null) { + this.zooKeeper.create(lockRootPath, null, ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); + } + } catch (IOException | KeeperException | InterruptedException e) { + log.error("===========> zookeeper connect error <==========="); + throw new RuntimeException(e); + } + } + + public void destroy() { + if (this.zooKeeper != null) { + try { + zooKeeper.close(); + log.info("===========> zookeeper disconnected <==========="); + } catch (InterruptedException e) { + log.error("===========> zookeeper disconnect error <==========="); + throw new RuntimeException(e); + } + } + } +} +``` + +#### 1.2.3.ZookeeperDistributedLockClient + +> 获取一把 Zookeeper 分布式锁 + +```java +@RequiredArgsConstructor +public class ZookeeperDistributedLockClient implements DistributedLockClient { + private final ZkClient zkClient; + + /** + * 获取一把锁 + * + * @param lockKey 锁的标识 + * @param lockTime 锁的时间 + * @param timeUnit 锁的时间单位 + * @return 锁 + */ + @Override + public DistributedLock getLock(String lockKey, long lockTime, TimeUnit timeUnit) { + return new ZookeeperDistributedLock(zkClient, lockKey, lockTime, timeUnit); + } +} +``` + +#### 1.2.4.自动装配 + +> 替换 `demo-distributed-lock-api` 中的默认实现 + +```java +@Configuration(proxyBeanMethods = false) +public class ZookeeperDistributedLockAutoConfiguration { + @Bean(initMethod = "init", destroyMethod = "destroy") + public ZkClient zkClient() { + return new ZkClient("127.0.0.1:2181"); + } + + @Bean + public ZookeeperDistributedLockClient distributedLockClient(ZkClient zkClient) { + return new ZookeeperDistributedLockClient(zkClient); + } +} +``` + +## 2.测试 + +### 2.1.环境搭建 + +主要是 mysql 及 zookeeper 环境的搭建，这里我提供了 docker-compose 文件，方便同学们一键启动测试环境 + +```bash +$ cd demo-distributed-lock/demo-distributed-lock-zookeeper +$ docker compose -f docker-compose.env.yml up +``` + +### 2.2.测试流程 + +这里我通过 Apache Bench 进行模拟并发场景，我也构建了一个压测镜像 `xkcoding/ab:alpine-3.16.2` ，方便同学们进行快速测试 + +> 注意：每次启动项目，都会在重置库存，你也可以手动调用 `/demo/stock/reset` 接口重置 + +#### 2.2.1.测试无分布式锁下高并发请求是否会发生超卖 + +1. 把 `ZookeeperDistributedLockAutoConfiguration` 类全部注释掉，这将不会装配 Zookeeper 分布式锁 +2. 启动 `ZookeeperDistributedLockApplication` +3. 模拟 5000 请求数 100 并发的压测环境 `docker run -it --rm xkcoding/ab:alpine-3.16.2 ab -n 5000 -c 100 http://${替换成你电脑的内网IP}:8080/demo/stock/reduce` +4. 等待压测结束，前往数据库查看库存是否从 5000 减为 0 + +#### 2.2.2.测试 Zookeeper 分布式锁 + +1. 把 `ZookeeperDistributedLockAutoConfiguration` 类的注释解开，此时 Spring Boot 会自动装配我们的 Zookeeper 分布式锁 +2. 再次启动 `ZookeeperDistributedLockApplication` +3. 再次模拟 5000 请求数 100 并发的压测环境 `docker run -it --rm xkcoding/ab:alpine-3.16.2 ab -n 5000 -c 100 http://${替换成你电脑的内网IP}:8080/demo/stock/reduce` +4. 等待压测结束，前往数据库查看库存是否从 5000 减为 0 + +## 3.参考 + +- [Zookeeper官方 API 文档](https://zookeeper.apache.org/doc/r3.8.0/zookeeperProgrammers.html) diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/docker-compose.env.yml b/demo-distributed-lock/demo-distributed-lock-zookeeper/docker-compose.env.yml new file mode 100644 index 0000000..fcd4b62 --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/docker-compose.env.yml @@ -0,0 +1,14 @@ +version: "3.8" + +services: + mysql: + image: mysql:8.0.30 + ports: + - "3306:3306" + environment: + - MYSQL_ROOT_PASSWORD=root + - MYSQL_DATABASE=spring-boot-demo + zookeeper: + image: zookeeper:3.8.0 + ports: + - "2181:2181" diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/pom.xml b/demo-distributed-lock/demo-distributed-lock-zookeeper/pom.xml index dceb0a0..9ac4999 100644 --- a/demo-distributed-lock/demo-distributed-lock-zookeeper/pom.xml +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/pom.xml @@ -14,6 +14,41 @@ 17 + 3.8.0 + + + com.xkcoding + common-tools + + + + com.xkcoding + demo-distributed-lock-api + 1.0.0-SNAPSHOT + + + + org.apache.zookeeper + zookeeper + ${zookeeper.version} + + + + org.projectlombok + lombok + true + + + + + demo-distributed-lock-zookeeper + + + org.springframework.boot + spring-boot-maven-plugin + + + diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/ZookeeperDistributedLockApplication.java b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/ZookeeperDistributedLockApplication.java new file mode 100644 index 0000000..203fe1d --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/ZookeeperDistributedLockApplication.java @@ -0,0 +1,21 @@ +package com.xkcoding.distributed.lock; + +import org.springframework.boot.SpringApplication; +import org.springframework.boot.autoconfigure.SpringBootApplication; + +/** + *

+ * 启动器 + *

+ * + * @author yangkai.shen + * @date Created in 2022-09-05 19:34 + */ +@SpringBootApplication +public class ZookeeperDistributedLockApplication { + + public static void main(String[] args) { + SpringApplication.run(ZookeeperDistributedLockApplication.class, args); + } + +} diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZkClient.java b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZkClient.java new file mode 100644 index 0000000..61fe104 --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZkClient.java @@ -0,0 +1,77 @@ +package com.xkcoding.distributed.lock.autoconfigure; + +import lombok.Getter; +import lombok.extern.slf4j.Slf4j; +import org.apache.zookeeper.*; + +import java.io.IOException; + +/** + *

+ * zookeeper 操作类 + *

+ * + * @author yangkai.shen + * @date 2022-09-05 19:50 + */ +@Slf4j +public class ZkClient { + /** + * 连接地址 + */ + @Getter + private final String connectServer; + + /** + * 节点根路径，默认是 /locks + */ + @Getter + private final String lockRootPath; + + @Getter + private ZooKeeper zooKeeper; + + private static final String DEFAULT_ROOT_PATH = "/locks"; + + public ZkClient(String connectServer) { + this.connectServer = connectServer; + this.lockRootPath = DEFAULT_ROOT_PATH; + } + + public ZkClient(String connectServer, String lockRootPath) { + this.connectServer = connectServer; + this.lockRootPath = lockRootPath; + } + + public void init() { + try { + this.zooKeeper = new ZooKeeper(connectServer, 3000, new Watcher() { + @Override + public void process(WatchedEvent watchedEvent) { + if (Event.KeeperState.SyncConnected == watchedEvent.getState() && Event.EventType.None == watchedEvent.getType()) { + log.info("===========> zookeeper connected <==========="); + } + } + }); + // 判断根节点是否存在，不存在则创建 + if (this.zooKeeper.exists(lockRootPath, false) == null) { + this.zooKeeper.create(lockRootPath, null, ZooDefs.Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT); + } + } catch (IOException | KeeperException | InterruptedException e) { + log.error("===========> zookeeper connect error <==========="); + throw new RuntimeException(e); + } + } + + public void destroy() { + if (this.zooKeeper != null) { + try { + zooKeeper.close(); + log.info("===========> zookeeper disconnected <==========="); + } catch (InterruptedException e) { + log.error("===========> zookeeper disconnect error <==========="); + throw new RuntimeException(e); + } + } + } +} diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLock.java b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLock.java new file mode 100644 index 0000000..0a395c8 --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLock.java @@ -0,0 +1,156 @@ +package com.xkcoding.distributed.lock.autoconfigure; + +import cn.hutool.core.collection.CollUtil; +import cn.hutool.core.util.StrUtil; +import com.xkcoding.distributed.lock.api.DistributedLock; +import org.apache.zookeeper.*; +import org.apache.zookeeper.data.Stat; +import org.jetbrains.annotations.NotNull; + +import java.util.List; +import java.util.concurrent.CountDownLatch; +import java.util.concurrent.TimeUnit; + +/** + *

+ * 基于 Zookeeper 实现的分布式锁 + *

+ * + * @author yangkai.shen + * @date 2022-09-05 19:35 + */ +public class ZookeeperDistributedLock extends DistributedLock { + + private final ZooKeeper zooKeeper; + + private final String lockRootPath; + + private final String lockNodePath; + + private final ThreadLocal LOCK_TIMES = new ThreadLocal<>(); + + protected ZookeeperDistributedLock(ZkClient client, String lockKey, long lockTime, TimeUnit timeUnit) { + super(lockKey, lockTime, timeUnit); + this.lockRootPath = client.getLockRootPath(); + this.zooKeeper = client.getZooKeeper(); + try { + // 直接创建临时顺序节点 + this.lockNodePath = this.zooKeeper.create(client.getLockRootPath() + "/" + lockKey + "-", null, ZooDefs.Ids.OPEN_ACL_UNSAFE, + CreateMode.EPHEMERAL_SEQUENTIAL); + } catch (KeeperException | InterruptedException e) { + throw new RuntimeException(e); + } + } + + @Override + public void lock() { + // 如果是锁重入，则次数加 1，然后直接返回 + Integer lockTimes = LOCK_TIMES.get(); + if (lockTimes != null && lockTimes > 0L) { + LOCK_TIMES.set(lockTimes + 1); + return; + } + + try { + // 当前节点是第一个临时节点，如果是，直接获得锁 + String previousLockNodePath = getPreviousNodePath(lockNodePath); + if (StrUtil.isBlank(previousLockNodePath)) { + LOCK_TIMES.set(1); + return; + } else { + // 如果不是第一个临时节点，则给它的前一个临时节点添加监听删除事件（即锁释放） + CountDownLatch wait = new CountDownLatch(1); + Stat stat = this.zooKeeper.exists(lockRootPath + "/" + previousLockNodePath, new Watcher() { + @Override + public void process(WatchedEvent event) { + if (Event.EventType.NodeDeleted == event.getType()) { + // 节点删除，通知获得锁，避免自旋判断影响性能 + wait.countDown(); + } + } + }); + // 未被删除，则需要阻塞 + if (stat != null) { + wait.await(); + } + // 获得锁，则重入次数设置为 1 + LOCK_TIMES.set(1); + } + } catch (Exception e) { + try { + TimeUnit.MILLISECONDS.sleep(50); + } catch (InterruptedException ex) { + throw new RuntimeException(ex); + } + // 重新尝试获取锁 + lock(); + } + + } + + @Override + public boolean tryLock() { + throw new UnsupportedOperationException("ZookeeperDistributedLock`s tryLock method is unsupported"); + } + + @Override + public boolean tryLock(long time, @NotNull TimeUnit unit) { + throw new UnsupportedOperationException("ZookeeperDistributedLock`s tryLock method is unsupported"); + } + + @Override + public void unlock() { + // -1 代表不使用乐观锁删除，类似 delete --force + Integer lockTimes = LOCK_TIMES.get(); + if (lockTimes != null && lockTimes > 0) { + // 计算剩下的重入次数 + int leftLockTime = lockTimes - 1; + LOCK_TIMES.set(leftLockTime); + // 剩下为 0 的时候，代表完全解锁，需要删除临时节点 + if (leftLockTime == 0) { + try { + this.zooKeeper.delete(lockNodePath, -1); + LOCK_TIMES.remove(); + } catch (InterruptedException | KeeperException e) { + throw new RuntimeException(e); + } + } + } + } + + /** + * 根据当前节点，获取前一个节点 + * + * @param currentNodePath 当前节点路径 + * @return 前一个节点 + */ + private String getPreviousNodePath(String currentNodePath) { + try { + // 截取当前临时节点编号 + long currentNodeSeq = Long.parseLong(StrUtil.subAfter(currentNodePath, "-", true)); + // 获取锁根路径下的所有临时节点 + List nodes = this.zooKeeper.getChildren(lockRootPath, false); + + if (CollUtil.isEmpty(nodes)) { + return null; + } + + // 用于标记遍历所有子节点时，离当前最近的一次编号 + long previousNodeSeq = 0L; + String previousNodePath = null; + for (String nodePath : nodes) { + // 获取每个临时节点编号 + long nodeSeq = Long.parseLong(StrUtil.subAfter(nodePath, "-", true)); + if (nodeSeq < currentNodeSeq && nodeSeq > previousNodeSeq) { + previousNodeSeq = nodeSeq; + previousNodePath = nodePath; + } + } + + return previousNodePath; + } catch (KeeperException | InterruptedException e) { + throw new RuntimeException(e); + } + + } +} diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockAutoConfiguration.java b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockAutoConfiguration.java new file mode 100644 index 0000000..76eb628 --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockAutoConfiguration.java @@ -0,0 +1,25 @@ +package com.xkcoding.distributed.lock.autoconfigure; + +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Configuration; + +/** + *

+ * 基于 Zookeeper 分布式锁自动装配类 + *

+ * + * @author yangkai.shen + * @date 2022-09-05 19:36 + */ +@Configuration(proxyBeanMethods = false) +public class ZookeeperDistributedLockAutoConfiguration { + @Bean(initMethod = "init", destroyMethod = "destroy") + public ZkClient zkClient() { + return new ZkClient("127.0.0.1:2181"); + } + + @Bean + public ZookeeperDistributedLockClient distributedLockClient(ZkClient zkClient) { + return new ZookeeperDistributedLockClient(zkClient); + } +} diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockClient.java b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockClient.java new file mode 100644 index 0000000..e1c29cb --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/java/com/xkcoding/distributed/lock/autoconfigure/ZookeeperDistributedLockClient.java @@ -0,0 +1,33 @@ +package com.xkcoding.distributed.lock.autoconfigure; + +import com.xkcoding.distributed.lock.api.DistributedLock; +import com.xkcoding.distributed.lock.api.DistributedLockClient; +import lombok.RequiredArgsConstructor; + +import java.util.concurrent.TimeUnit; + +/** + *

+ * 获得一把 Zookeeper 分布式锁 + *

+ * + * @author yangkai.shen + * @date 2022-09-05 19:36 + */ +@RequiredArgsConstructor +public class ZookeeperDistributedLockClient implements DistributedLockClient { + private final ZkClient zkClient; + + /** + * 获取一把锁 + * + * @param lockKey 锁的标识 + * @param lockTime 锁的时间 + * @param timeUnit 锁的时间单位 + * @return 锁 + */ + @Override + public DistributedLock getLock(String lockKey, long lockTime, TimeUnit timeUnit) { + return new ZookeeperDistributedLock(zkClient, lockKey, lockTime, timeUnit); + } +} diff --git a/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/resources/application.yml b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/resources/application.yml new file mode 100644 index 0000000..113bd15 --- /dev/null +++ b/demo-distributed-lock/demo-distributed-lock-zookeeper/src/main/resources/application.yml @@ -0,0 +1,16 @@ +server: + port: 8080 + servlet: + context-path: /demo +spring: + sql: + init: + continue-on-error: true + mode: always + schema-locations: + - "classpath:db/schema.sql" + datasource: + driver-class-name: com.mysql.cj.jdbc.Driver + url: jdbc:mysql://localhost:3306/spring-boot-demo + username: root + password: root