在解决多线程发生”抢”资源的问题的时候我们可以实用synchronized来解决。但在分布式情况下,多个程序同时访问同一资源的情况,这时在方法上加synchronized明显是没有用处的。这个时候可以考虑使用zookeeper来解决分布式共享锁问题。
大体思路
我们可以在每个程序开始时就在zookeeper上注册节点,可以认为是注册了一把锁,然后指定规则,比如判断哪个锁的序号小(或者大)就让它先访问资源,其他程序进入等待状态。当资源访问完毕后,需要释放锁,并且再重新注册把锁一边下次访问,其他程序中监听到了锁释放(节点删除事件)则进行回调,获取所有节点并且判断自己的节点是否为最小(或最大)是的话则进行访问资源,否则进入等待状态。
程序实现
public class DistributeLock {
private static final String connectionString = "192.168.25.127:2181,"
+ "192.168.25.129:2181,"
+ "192.168.25.130:2181";
public static final Integer sessionTimeout = 2000;
public static ZooKeeper zkClient = null;
public static String parentNode="/parent";
public static String childNode="/child";
public static String path;
/**
* 获取连接
* @throws Exception
*/
public void getConnection() throws Exception{
zkClient = new ZooKeeper(connectionString, sessionTimeout, new Watcher() {
public void process(WatchedEvent event) {
//监听子节点发生变化事件
String p = event.getPath();
if(event.getType()==EventType.NodeChildrenChanged &&
event.getPath().equals(parentNode)){
//获取所有节点,并对父节点进行监听
try {
List<String> children = zkClient.getChildren(parentNode, true);
//对节点进行排序,最小的节点去访问共享资源
Collections.sort(children);
//判断自己的锁id是否最小,是的话直接访问资源
if (children.indexOf(path.substring((parentNode+"/").length()))==0){
getResource();
//释放锁后重新注册一把锁
path = zkClient.create(parentNode+childNode, null, Ids.OPEN_ACL_UNSAFE,
CreateMode.EPHEMERAL_SEQUENTIAL);
}
} catch (Exception e) {
e.printStackTrace();
}
}
}
});
//程序上来就注册一把锁
path = zkClient.create(parentNode+childNode, null, Ids.OPEN_ACL_UNSAFE,
CreateMode.EPHEMERAL_SEQUENTIAL);
Thread.sleep(300);
//获取所有锁,并对父节点监听
List<String> children = zkClient.getChildren(parentNode, true);
//如果只有自己这个节点,那么直接访问资源
if(children.size()==1){
//访问共享资源
getResource();
//释放锁后重新注册一把锁
path = zkClient.create(parentNode+childNode, null, Ids.OPEN_ACL_UNSAFE,
CreateMode.EPHEMERAL_SEQUENTIAL);
}
}
/**
* 共享资源的业务逻辑
*/
private void getResource() {
System.out.println("开始享用资源:"+path);
try {
Thread.sleep(2000);
} catch (InterruptedException e) {
e.printStackTrace();
}finally{
System.out.println("享用资源结束:"+path);
try {
//释放锁,-1表示所有版本
zkClient.delete(path, -1);
} catch (Exception e) {
}
}
}
public static void main(String[] args) throws Exception {
DistributeLock lock = new DistributeLock();
lock.getConnection();
System.in.read();
}
}
测试
先则zookeeper客户端创建父节点/parent
[zk: localhost:2181(CONNECTED) 24] create /parent null
Created /parent
[zk: localhost:2181(CONNECTED) 26] ls /
[app1, test, ide, servers, zookeeper, locks, parent]
[zk: localhost:2181(CONNECTED) 27]
程序启动三次(简单当做三个程序)
查看三个控制台输出
最后稳定的时候是每个程序依次访问共享资源。
注:程序并不是很健壮的只是为了简单演示,只要一直测肯定还能测出来一些问题的。