使用curator實現zookeeper鎖服務的示例分享

更新時間：2014年02月24日 09:58:59 作者：

這篇文章主要介紹了使用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("所有任務執(zhí)行完畢");

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();
   }
   }
}
}