锁接口
package lock;
public interface Lock {
//获取到锁的资源
public void getLock();
// 释放锁
public void unLock();
}
锁抽象类
package lock;
import org.I0Itec.zkclient.ZkClient;
/**
* @Auther: 段金良
* @Date: 2019/4/27 0027 15:06
* @Description:
*/
public abstract class ZookeeperAbstractLock {
// zk连接地址
private static final String CONNECTSTRING = "127.0.0.1:2181";
// 创建zk连接
protected ZkClient zkClient = new ZkClient(CONNECTSTRING);
protected static final String PATH = "/lock";
public void getLock() {
if (tryLock()) {
System.out.println("##获取lock锁的资源####");
} else {
// 等待
waitLock();
// 重新获取锁资源
getLock();
}
}
// 获取锁资源
abstract boolean tryLock();
// 等待
abstract void waitLock();
public void unLock() {
if (zkClient != null) {
zkClient.close();
System.out.println("释放锁资源...");
}
}
}
//子类
package lock;
import org.I0Itec.zkclient.IZkDataListener;
import java.util.concurrent.CountDownLatch;
/**
* @Auther: 段金良
* @Date: 2019/4/27 0027 15:08
* @Description:
*/
public class ZookeeperDistrbuteLock extends ZookeeperAbstractLock {
private CountDownLatch countDownLatch = null;
boolean tryLock() {
try {
zkClient.createEphemeral(PATH);
return true;
} catch (Exception e) {
return false;
}
}
void waitLock() {
IZkDataListener izkDataListener = new IZkDataListener() {
public void handleDataDeleted(String path) throws Exception {
// 唤醒被等待的线程
if (countDownLatch != null) {
countDownLatch.countDown();
}
}
public void handleDataChange(String path, Object data) throws Exception {
}
};
// 注册事件
zkClient.subscribeDataChanges(PATH, izkDataListener);
if (zkClient.exists(PATH)) {
countDownLatch = new CountDownLatch(1);
try {
countDownLatch.await();
} catch (Exception e) {
e.printStackTrace();
}
}
// 删除监听
zkClient.unsubscribeDataChanges(PATH, izkDataListener);
}
}
package lock;
import java.text.SimpleDateFormat;
import java.util.Date;
/**
* @Auther: 段金良
* @Date: 2019/4/27 0027 15:02
* @Description:
*/
public class OrderService implements Runnable {
private static int count =0;
private OrderNumGenerator orderNumGenerator = new OrderNumGenerator();
ZookeeperAbstractLock lock = new ZookeeperDistrbuteLock();
public void run() {
lock.getLock();
//业务代码
SimpleDateFormat simpt = new SimpleDateFormat("yyyy-MM-dd-HH-mm-ss");
System.out.println(simpt.format(new Date()) + "-" + ++count);
lock.unLock();
}
public void getNumber() {
String number = orderNumGenerator.getNumber();
System.out.println(Thread.currentThread().getName() + ",生成订单ID:" + number);
}
public static void main(String[] args) {
System.out.println("####生成唯一订单号###");
for (int i = 0; i < 100; i++) {
new Thread(new OrderService()).start();
}
}
}