使用Zookeeper对进程间加锁

/**
 * 主要实现进程间的锁利用zookeeper的
 * 
 * @author yc
 * 
 */
public class MyLock {
private CountDownLatch runFlag = new CountDownLatch(1);
private ZooKeeper zk = null;
private String path = "/zk-lock";
volatile private String mychild;
volatile private String firstChild;


public MyLock(String hosts) {
try {
zk = new ZooKeeper(hosts, 5000, null);
Stat stat = zk.exists(path, null);
if (stat == null) {
zk.create(path, "lock".getBytes(), Ids.OPEN_ACL_UNSAFE,
CreateMode.PERSISTENT);
}
} catch (Exception e) {
// TODO 自动生成的 catch 块
e.printStackTrace();
}
}


// 试图获取锁
public boolean lock() {// 阻塞的方法
try {
if (tryLock())
return true;
zk.exists(firstChild, new Watcher() {// 在最新的子节点上添加接听事件


public void process(WatchedEvent event) {
// TODO 自动生成的方法存根
if (event.getType() == EventType.NodeDeleted) {// 节点删除掉
if (tryLock()) {
runFlag.countDown();
}else{
try {
zk.exists(firstChild, this);
} catch (Exception e) {
// TODO 自动生成的 catch 块
e.printStackTrace();
} 
}
}
}


});
runFlag.await();//阻塞方法
return true;


} catch (Exception e) {
return false;
}
}


// 需要抽取一个方法它获取不是阻塞要么是true要是false
private boolean tryLock() {
try {
mychild = zk.create(path + "/child-lock", "child".getBytes(),
Ids.OPEN_ACL_UNSAFE, CreateMode.EPHEMERAL_SEQUENTIAL);//创建节点

List<String> childs = zk.getChildren(path, null);//获取所有节点
int len = childs.size();
firstChild = childs.get(0);
for(int i=1;i<len;i++){
String temp = childs.get(i);
if(temp.compareTo(firstChild)<0)//获取最小的节点
firstChild = temp;
}


firstChild=path+"/"+ firstChild;
if (firstChild.equals(mychild))// 这表示是它创建的子节点
return true;
zk.delete(mychild, -1);// 删掉自己创建的节点（必须要有这句）
return false;
} catch (Exception e) {
return false;
}
}


// 释放锁
public void unLock() {
try {
zk.delete(mychild, -1);
} catch (InterruptedException e) {
// TODO 自动生成的 catch 块
e.printStackTrace();
} catch (KeeperException e) {
// TODO 自动生成的 catch 块
e.printStackTrace();
}
}
}