这篇文章主要介绍了使用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();
   }
   }
 }
 }


Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐