利用curator实现的zookeeper分布式锁服务

Curator是Netflix开源的一套ZooKeeper客户端框架，用它来操作zookeeper更加简单方便，按Curator官方所比喻的，guava to JAVA， Curator to Zookeeper，Curator采用了fluent风格的代码，非常简洁。     
主要看下，使用curator操作zookeeper的一些基础例子： 
主要的功能： 
1，在zk上添加，或更新数据 
2，删除zk节点上数据 
3，读取某个节点上的数据 
4，上传一些本地文件到zk节点上 
5，检查zookeeper上是否存在某个节点路径 
核心代码如下： 
package com.qin.curator.zk;
import java.io.File;
import java.util.List;
import org.apache.commons.io.FileUtils;
import org.apache.curator.CuratorZookeeperClient;
import org.apache.curator.RetryPolicy;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.framework.CuratorFrameworkFactory.Builder;
import org.apache.curator.framework.api.CreateBuilder;
import org.apache.curator.retry.ExponentialBackoffRetry;
import org.apache.zookeeper.WatchedEvent;
import org.apache.zookeeper.Watcher;
import org.apache.zookeeper.ZKUtil;
import framework.CrudExamples;
/**
 * @author qindongliang
 * curator操作zookeeper的
 * 基本例子
 * **/
public class CuratorTools {
	static CuratorFramework zkclient=null;
	static String nameSpace="php";
	static {
		  String zkhost="192.168.46.22:2181";//zk的host
		  RetryPolicy rp=new ExponentialBackoffRetry(1000, 3);//重试机制
		  Builder builder = CuratorFrameworkFactory.builder().connectString(zkhost)
				  .connectionTimeoutMs(5000)
				  .sessionTimeoutMs(5000)
				  .retryPolicy(rp);
		  builder.namespace(nameSpace);
		  CuratorFramework zclient = builder.build();
		  zkclient=zclient;
		  zkclient.start();// 放在这前面执行
		  zkclient.newNamespaceAwareEnsurePath(nameSpace);
	}
	public static void main(String[] args)throws Exception {
		CuratorTools ct=new  CuratorTools();
		//ct.getListChildren("/zk/bb");
		//ct.upload("/jianli/123.txt", "D:\\123.txt");
		//ct.createrOrUpdate("/zk/cc334/zzz","c");
		//ct.delete("/qinb/bb");
		//ct.checkExist("/zk");
		ct.read("/jianli/123.txt");
		zkclient.close();
	}
	/**
	 * 创建或更新一个节点
	 * 
	 * @param path 路径
	 * @param content 内容
	 * **/
	public void createrOrUpdate(String path,String content)throws Exception{
		zkclient.newNamespaceAwareEnsurePath(path).ensure(zkclient.getZookeeperClient());
	    zkclient.setData().forPath(path,content.getBytes());	
	    System.out.println("添加成功！！！");
	}
	/**
	 * 删除zk节点
	 * @param path 删除节点的路径
	 * 
	 * **/
	public void delete(String path)throws Exception{
		zkclient.delete().guaranteed().deletingChildrenIfNeeded().forPath(path);
		System.out.println("删除成功!");
	}
	/**
	 * 判断路径是否存在
	 * @param path
	 * **/
	public void checkExist(String path)throws Exception{
		if(zkclient.checkExists().forPath(path)==null){
			System.out.println("路径不存在!");
		}else{
			System.out.println("路径已经存在!");
		}
	}
	/**
	 * 读取的路径
	 * @param path
	 * **/
	public void read(String path)throws Exception{
		String data=new String(zkclient.getData().forPath(path),"gbk");
		System.out.println("读取的数据:" data);
	}
	/**
	 * @param path 路径
	 * 获取某个节点下的所有子文件
	 * */
	public void getListChildren(String path)throws Exception{
		List<String> paths=zkclient.getChildren().forPath(path);
		for(String p:paths){
			System.out.println(p);
		}
	}
	/**
	 * @param zkPath zk上的路径
	 * @param localpath 本地上的文件路径
	 * 
	 * **/
	public void upload(String zkPath,String localpath)throws Exception{
		createrOrUpdate(zkPath, "");//创建路径
		byte[] bs=FileUtils.readFileToByteArray(new File(localpath));
		zkclient.setData().forPath(zkPath, bs);
		System.out.println("上传文件成功！");
	}
}
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import com.netflix.curator.RetryPolicy;
import com.netflix.curator.framework.CuratorFramework;
import com.netflix.curator.framework.CuratorFrameworkFactory;
import com.netflix.curator.framework.recipes.locks.InterProcessMutex;
import com.netflix.curator.retry.ExponentialBackoffRetry;
public class TestCuratorLock {
 /**
  * @param args
  * @throws InterruptedException
  */
 public static void main(String[] args) throws InterruptedException {
  // TODO Auto-generated method stub
  CountDownLatch latch = new CountDownLatch(5);
  String zookeeperConnectionString = "localhost:2181,localhost:2182,localhost:2183";
  RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3);
  CuratorFramework client = CuratorFrameworkFactory.newClient(
    zookeeperConnectionString, retryPolicy);
  client.start();
  System.out.println("客户端启动。。。。");
  ExecutorService exec = Executors.newCachedThreadPool();
  for (int i = 0; i < 5; i++) {
   exec.submit(new MyLock("client" + i, client, latch));
  }
  exec.shutdown();
  latch.await();
  System.out.println("所有任务执行完毕");
  client.close();
  System.out.println("客户端关闭。。。。");
 }
 static class MyLock implements Runnable {
  private String name;
  private CuratorFramework client;
  private CountDownLatch latch;
  public MyLock(String name, CuratorFramework client, CountDownLatch latch) {
   this.name = name;
   this.client = client;
   this.latch = latch;
  }
  public String getName() {
   return name;
  }
  public void setName(String name) {
   this.name = name;
  }
  @Override
  public void run() {
   // TODO Auto-generated method stub
   InterProcessMutex lock = new InterProcessMutex(client,
     "/test_group");
   try {
    if (lock.acquire(120, TimeUnit.SECONDS)) {
     try {
      // do some work inside of the critical section here
      System.out.println("----------" + this.name
        + "获得资源----------");
      System.out.println("----------" + this.name
        + "正在处理资源----------");
      Thread.sleep(10 * 1000);
      System.out.println("----------" + this.name
        + "资源使用完毕----------");
      latch.countDown();
     } finally {
      lock.release();
      System.out.println("----------" + this.name
        + "释放----------");
     }
    }
   } catch (Exception e) {
    // TODO Auto-generated catch block
    e.printStackTrace();
   }
   }
 }
 }
使用curator实现zookeeper锁服务的示例分享

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import com.netflix.curator.RetryPolicy;
import com.netflix.curator.framework.CuratorFramework;
import com.netflix.curator.framework.CuratorFrameworkFactory;
import com.netflix.curator.framework.recipes.locks.InterProcessMutex;
import com.netflix.curator.retry.ExponentialBackoffRetry;
public class TestCuratorLock {
 /**
  * @param args
  * @throws InterruptedException
  */
 public static void main(String[] args) throws InterruptedException {
  // TODO Auto-generated method stub
  CountDownLatch latch = new CountDownLatch(5);
  String zookeeperConnectionString = "localhost:2181,localhost:2182,localhost:2183";
  RetryPolicy retryPolicy = new ExponentialBackoffRetry(1000, 3);
  CuratorFramework client = CuratorFrameworkFactory.newClient(
    zookeeperConnectionString, retryPolicy);
  client.start();
  System.out.println("客户端启动。。。。");
  ExecutorService exec = Executors.newCachedThreadPool();
  for (int i = 0; i < 5; i++) {
   exec.submit(new MyLock("client" + i, client, latch));
  }
  exec.shutdown();
  latch.await();
  System.out.println("所有任务执行完毕");
  client.close();
  System.out.println("客户端关闭。。。。");
 }
 static class MyLock implements Runnable {
  private String name;
  private CuratorFramework client;
  private CountDownLatch latch;
  public MyLock(String name, CuratorFramework client, CountDownLatch latch) {
   this.name = name;
   this.client = client;
   this.latch = latch;
  }
  public String getName() {
   return name;
  }
  public void setName(String name) {
   this.name = name;
  }
  @Override
  public void run() {
   // TODO Auto-generated method stub
   InterProcessMutex lock = new InterProcessMutex(client,
     "/test_group");
   try {
    if (lock.acquire(120, TimeUnit.SECONDS)) {
     try {
      // do some work inside of the critical section here
      System.out.println("----------" + this.name
        + "获得资源----------");
      System.out.println("----------" + this.name
        + "正在处理资源----------");
      Thread.sleep(10 * 1000);
      System.out.println("----------" + this.name
        + "资源使用完毕----------");
      latch.countDown();
     } finally {
      lock.release();
      System.out.println("----------" + this.name
        + "释放----------");
     }
    }
   } catch (Exception e) {
    // TODO Auto-generated catch block
    e.printStackTrace();
   }
   }
 }
 }