目录
1、zookeeper节点特点
- 临时节点(EPHEMERAL) :session链接断开就没了;不能创建子节点;不能同名
- 临时顺序节点(EPHEMERAL_SEQUENTIAL) :session链接断开就没了;不能创建子节点;同名节点会在后面添加上序号(分布式锁使用的好处)
- 持久节点(PERSISTENT):session断联、服务端重启还在;可以创建子节点,子节点可以临时也可以持久;不能同名
- 持久顺序节点(PERSISTENT_SEQUENTIAL):session断联、服务端重启还在;可以创建子节点,子节点可以临时也可以持久;同名节点会在后面添加上序号
1.1 图解
2、分布式锁实现
参考ReentrantLock实现的Lock接口,主要有tryLock(),lock(),unLock()方法
2.1 lock自定义
public interface Lock {
/**
* 获取锁
*/
void getLock();
/**
* 释放锁
*/
void unLock();
}
2.2 抽象方法,使用模板模式定义好获取锁的方法
public abstract class AbstractZkLock implements Lock {
//使用了模板方法,开放使用的只有getLock和unLock方法
@Override
public void getLock() {
if (tryLock()) {
System.out.println("获取锁");
} else {
//未获取到锁等待
waitLock();
//获取到锁之后,唤醒
getLock();
}
}
/**
* ②占有锁
*
* @return 获取锁返回true
*/
public abstract boolean tryLock();
/**
* 等待
*/
public abstract void waitLock();
}
2.3 具体的工具类锁
public class ZkDistributeLock extends AbstractZkLock {
//连接zk服务器
private ZkClient zkClient = new ZkClient("127.0.0.1:21810");
//作为锁的父节点
private static final String PATH_LOCK = "/zk_path_lock";
//要获取锁的前一个节点
private String beforePath = null;
//当前要获取锁的节点
private String currentPath = null;
private CountDownLatch countDownLatch;
ZkDistributeLock() {
//先创建一个持久节点作为分布式锁的父节点
if (!zkClient.exists(PATH_LOCK)) {
zkClient.createPersistent(PATH_LOCK, PATH_LOCK);
}
}
@Override
public boolean tryLock() {
//首次进来肯定是空的
if (currentPath == null || currentPath.length() == 0) {
//currentpath = /zk_path_lock/0000000144
currentPath = zkClient.createEphemeralSequential(PATH_LOCK + "/", PATH_LOCK);
}
//获取所有临时节点并排序0000000196,0000000196,0000000196
List<String> children = zkClient.getChildren(PATH_LOCK);
//对所有子节点排序
Collections.sort(children);
//判断当前节点是不是第一个节点
if (currentPath.equals(PATH_LOCK + "/" + children.get(0))) {
return true;
} else {
//找到当前节点的下标
System.out.println(children.toString());
//currentPath要去掉前面的PATH_LOCK + "/"字符串
int wz = Collections.binarySearch(children, currentPath.substring((PATH_LOCK + "/").length()));
System.out.println(wz);
beforePath = PATH_LOCK + '/' + children.get(wz - 1);
}
return false;
}
@Override
public void waitLock() {
IZkDataListener zkDataListener = new IZkDataListener() {
@Override
public void handleDataChange(String s, Object o) throws Exception {
}
@Override
public void handleDataDeleted(String s) throws Exception {
if (countDownLatch != null) {
countDownLatch.countDown();
}
}
};
//给排在前面的的节点增加数据删除的watcher,本质是启动另外一个线程去监听前置节点
zkClient.subscribeDataChanges(beforePath, zkDataListener);
try {
if (zkClient.exists(beforePath)) {
countDownLatch = new CountDownLatch(1);
countDownLatch.await();
}
} catch (InterruptedException e) {
e.printStackTrace();
}
zkClient.unsubscribeDataChanges(beforePath, zkDataListener);
}
@Override
public void unLock() {
zkClient.delete(currentPath);
//每次实例化的时候,都会创建一个连接,所以要关闭
zkClient.close();
}
}
2.4 使用测试类
public class TestMain extends Thread {
private ZkDistributeLock lock = new ZkDistributeLock();
//全局订单id
private static int count = 0;
SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd-HH-mm-ss");
public static void main(String[] args) {
for (int i = 0; i < 50; i++) {
new Thread(new TestMain()).start();
}
}
@Override
public void run() {
lock.getLock();
try {
String number = dateFormat.format(new Date()) + "-" + ++count;
System.out.println(Thread.currentThread().getName() + ",生成订单ID:" + number);
} catch (Exception e) {
e.printStackTrace();
} finally {
lock.unLock();
}
}
}
2.5 输出结果
获取锁
Thread-2,生成订单ID:2021-04-03-21-08-57-1
获取锁
Thread-5,生成订单ID:2021-04-03-21-08-57-2
获取锁
Thread-8,生成订单ID:2021-04-03-21-08-57-3
获取锁
Thread-11,生成订单ID:2021-04-03-21-08-57-4
获取锁
Thread-14,生成订单ID:2021-04-03-21-08-57-5
获取锁
Thread-17,生成订单ID:2021-04-03-21-08-57-6
获取锁
Thread-20,生成订单ID:2021-04-03-21-08-57-7
获取锁
Thread-23,生成订单ID:2021-04-03-21-08-57-8
获取锁
Thread-26,生成订单ID:2021-04-03-21-08-57-9
获取锁
Thread-29,生成订单ID:2021-04-03-21-08-57-10
获取锁
Thread-32,生成订单ID:2021-04-03-21-08-57-11
获取锁
Thread-35,生成订单ID:2021-04-03-21-08-57-12
获取锁
Thread-38,生成订单ID:2021-04-03-21-08-57-13
获取锁
Thread-41,生成订单ID:2021-04-03-21-08-57-14
获取锁
Thread-44,生成订单ID:2021-04-03-21-08-57-15
获取锁
Thread-47,生成订单ID:2021-04-03-21-08-57-16
获取锁
Thread-50,生成订单ID:2021-04-03-21-08-57-17
获取锁
Thread-53,生成订单ID:2021-04-03-21-08-57-18
获取锁
Thread-56,生成订单ID:2021-04-03-21-08-57-19
获取锁
Thread-59,生成订单ID:2021-04-03-21-08-57-20
获取锁
Thread-62,生成订单ID:2021-04-03-21-08-57-21
获取锁
Thread-65,生成订单ID:2021-04-03-21-08-57-22
获取锁
Thread-68,生成订单ID:2021-04-03-21-08-57-23
获取锁
Thread-71,生成订单ID:2021-04-03-21-08-57-24
获取锁
Thread-74,生成订单ID:2021-04-03-21-08-57-25
获取锁
Thread-77,生成订单ID:2021-04-03-21-08-57-26
获取锁
Thread-80,生成订单ID:2021-04-03-21-08-57-27
获取锁
Thread-83,生成订单ID:2021-04-03-21-08-57-28
获取锁
Thread-86,生成订单ID:2021-04-03-21-08-57-29
获取锁
Thread-89,生成订单ID:2021-04-03-21-08-57-30
获取锁
Thread-92,生成订单ID:2021-04-03-21-08-57-31
获取锁
Thread-95,生成订单ID:2021-04-03-21-08-57-32
获取锁
Thread-98,生成订单ID:2021-04-03-21-08-57-33
获取锁
Thread-101,生成订单ID:2021-04-03-21-08-57-34
获取锁
Thread-104,生成订单ID:2021-04-03-21-08-57-35
获取锁
Thread-107,生成订单ID:2021-04-03-21-08-57-36
获取锁
Thread-110,生成订单ID:2021-04-03-21-08-57-37
获取锁
Thread-113,生成订单ID:2021-04-03-21-08-57-38
获取锁
Thread-116,生成订单ID:2021-04-03-21-08-57-39
获取锁
Thread-119,生成订单ID:2021-04-03-21-08-57-40
获取锁
Thread-122,生成订单ID:2021-04-03-21-08-57-41
获取锁
Thread-125,生成订单ID:2021-04-03-21-08-57-42
获取锁
Thread-128,生成订单ID:2021-04-03-21-08-57-43
获取锁
Thread-131,生成订单ID:2021-04-03-21-08-57-44
获取锁
Thread-134,生成订单ID:2021-04-03-21-08-57-45
获取锁
Thread-137,生成订单ID:2021-04-03-21-08-57-46
获取锁
Thread-140,生成订单ID:2021-04-03-21-08-57-47
获取锁
Thread-143,生成订单ID:2021-04-03-21-08-57-48
获取锁
Thread-146,生成订单ID:2021-04-03-21-08-57-49
获取锁
Thread-149,生成订单ID:2021-04-03-21-08-57-50