网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
一、Demo 概述
1、关于 zookeeper “命名服务协调”
Zookeeper 能保证数据的强一致性,用户任何时候都可以相信集群中每个节点的数据都是相同的。一个用户创建一个节点作为锁,另一个用户检测该节点,如果存在,代表别的用户已经锁住,如果不存在,则可以创建一个节点,代表拥有一个锁。
2、Demo 设计
分布式锁本质,就是多个资源竞争者对一份资源的排他占有
- 我们设置多个线程,分别在同一 path 下创建节点
- 没个线程获取当前 path 下子节点,看最小子节点是否为自身,是则加锁成功(更好的方式是用 Watcher 对前一个地址监控,这里图方便用子节点排序取最小的方式 )
- 线程加锁成功后,执行任务,执行完毕后解锁
3、Demo 前提
二、代码实现
1、引用 Maven 依赖
<!-- 选择对应的Zookeeper版本 -->
<dependency>
<groupId>org.apache.zookeeper</groupId>
<artifactId>zookeeper</artifactId>
<version>3.7.0</version>
</dependency>
2、ConnectionWatcher 类创建 Zookeeper 连接
import java.io.IOException;
import java.util.concurrent.CountDownLatch;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooKeeper;
public class ConnectionWatcher implements Watcher {
private final CountDownLatch connectedSignal = new CountDownLatch(1);
private static final int SESSION\_TIMEOUT = 5000;
protected ZooKeeper zk;
public void connect(String hosts) throws IOException, InterruptedException {
zk = new ZooKeeper(hosts, SESSION\_TIMEOUT, this);
connectedSignal.await();
}
@Override
public void process(WatchedEvent event) {
if (event.getState() == Event.KeeperState.SyncConnected) {
connectedSignal.countDown();
}
}
public void close() throws InterruptedException {
zk.close();
}
}
3、ActiveKeyValueStore 类读写 Zookeeper 数据
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.List;
import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZooDefs;
import org.apache.zookeeper.data.Stat;
public class ActiveKeyValueStore extends ConnectionWatcher {
private static final Charset CHARSET = StandardCharsets.UTF\_8;
int state = 0;
/\*\*
\* 写入节点数据
\*
\* @param path 节点地址
\* @param value 数据值
\* @throws InterruptedException 中断异常
\* @throws KeeperException ZooKeeper异常
\*/
public void write(String path, String value) throws InterruptedException, KeeperException {
Stat stat = zk.exists(path, false);
if (stat == null) {
if (value == null) {
zk.create(path, null,
ZooDefs.Ids.OPEN\_ACL\_UNSAFE, CreateMode.PERSISTENT);
} else {
zk.create(path, value.getBytes(CHARSET),
ZooDefs.Ids.OPEN\_ACL\_UNSAFE, CreateMode.PERSISTENT);
}
} else {
if (value == null) {
zk.setData(path, null, -1);
} else {
zk.setData(path, value.getBytes(CHARSET), -1);
}
}
}
public boolean lock(String path, String name) throws InterruptedException, KeeperException {
boolean flag = tryLock(path, name);
if (flag) {
state++;
}
return flag;
}
public boolean tryLock(String path, String name) throws InterruptedException, KeeperException {
String lockPath = path + "/" + name;
zk.create(lockPath, null, ZooDefs.Ids.OPEN\_ACL\_UNSAFE, CreateMode.EPHEMERAL);
List<String> waits = readChildren(path, null);
Collections.sort(waits);
if (waits.get(0).equals(name)) {
return true;
}
CountDownLatch latch = new CountDownLatch(1);
for (int i = 0; i < waits.size(); i++) {
String cur = waits.get(i);
if (!cur.equalsIgnoreCase(name)) {
continue;
}
String prePath = path + "/" + waits.get(i - 1);
zk.exists(prePath, new Watcher() {
@Override
public void process(WatchedEvent event) {
latch.countDown();
}
});
break;
}
latch.await();
return true;
}
public boolean unlock(String path, String name) {
if (state > 1) {
state--;
return true;
}
String lockPath = path + "/" + name;
try {
Stat stat = zk.exists(lockPath, false);
int version = stat.getVersion();
zk.delete(lockPath, version);
state--;
return true;
} catch (Exception e) {
System.out.println("unlock:" + lockPath + " ,exception,");
}
return false;
}
/\*\*
\* 获取所有子节点
\*
\* @param path 节点地址
\* @param watcher watcher
\* @return 所有子节点
\* @throws InterruptedException 中断异常
\* @throws KeeperException ZooKeeper异常
\*/
public List<String> readChildren(String path, Watcher watcher) throws InterruptedException, KeeperException {
List<String> childrens = null;
if (watcher == null) {
childrens = zk.getChildren(path, false);
} else {
childrens = zk.getChildren(path, watcher, null);
}
return childrens;
}
}
4、ZkLock 类实现分布式锁
网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
444117)]
网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!