Example usage for org.apache.zookeeper WatchedEvent getState

List of usage examples for org.apache.zookeeper WatchedEvent getState

Introduction

In this page you can find the example usage for org.apache.zookeeper WatchedEvent getState.

Prototype

public KeeperState getState() 

Source Link

Usage

From source file:org.apache.whirr.service.cdh.integration.CdhZooKeeperServiceTest.java

License:Apache License

@Test(timeout = TestConstants.ITEST_TIMEOUT)
public void test() throws Exception {
    class ConnectionWatcher implements Watcher {

        private ZooKeeper zk;
        private CountDownLatch latch = new CountDownLatch(1);

        public void connect(String hosts) throws IOException, InterruptedException {
            zk = new ZooKeeper(hosts, 5000, this);
            latch.await();/*from  www  .ja va 2s.  c o  m*/
        }

        public ZooKeeper getZooKeeper() {
            return zk;
        }

        @Override
        public void process(WatchedEvent event) {
            if (event.getState() == KeeperState.SyncConnected) {
                latch.countDown();
            }
        }

        public void close() throws InterruptedException {
            if (zk != null) {
                zk.close();
            }
        }

    }

    String path = "/data";
    String data = "Hello";
    ConnectionWatcher watcher = new ConnectionWatcher();
    watcher.connect(hosts);
    watcher.getZooKeeper().create(path, data.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
    watcher.close();

    watcher = new ConnectionWatcher();
    watcher.connect(hosts);
    byte[] actualData = watcher.getZooKeeper().getData(path, false, null);
    assertEquals(data, new String(actualData));
    watcher.close();
}

From source file:org.apache.whirr.service.zookeeper.ZooKeeperServiceTest.java

License:Apache License

@Test
public void test() throws Exception {
    class ConnectionWatcher implements Watcher {

        private ZooKeeper zk;
        private CountDownLatch latch = new CountDownLatch(1);

        public void connect(String hosts) throws IOException, InterruptedException {
            zk = new ZooKeeper(hosts, 5000, this);
            latch.await();/*  w w w  .  j a  va2  s .  c om*/
        }

        public ZooKeeper getZooKeeper() {
            return zk;
        }

        @Override
        public void process(WatchedEvent event) {
            if (event.getState() == KeeperState.SyncConnected) {
                latch.countDown();
            }
        }

        public void close() throws InterruptedException {
            if (zk != null) {
                zk.close();
            }
        }

    }

    String path = "/data";
    String data = "Hello";
    ConnectionWatcher watcher = new ConnectionWatcher();
    watcher.connect(cluster.getHosts());
    watcher.getZooKeeper().create(path, data.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
    watcher.close();

    watcher = new ConnectionWatcher();
    watcher.connect(cluster.getHosts());
    byte[] actualData = watcher.getZooKeeper().getData(path, false, null);
    assertEquals(data, new String(actualData));
    watcher.close();
}

From source file:org.bboxdb.distribution.zookeeper.ZookeeperClient.java

License:Apache License

/**
 * Connect to zookeeper//ww w .  j  ava 2s .co  m
 */
@Override
public void init() {
    if (zookeeperHosts == null || zookeeperHosts.isEmpty()) {
        logger.warn("No Zookeeper hosts are defined, not connecting to zookeeper");
        return;
    }

    try {
        shutdownPending = false;

        final CountDownLatch connectLatch = new CountDownLatch(1);

        zookeeper = new ZooKeeper(generateConnectString(), ZOOKEEPER_SESSION_TIMEOUT, new Watcher() {
            @Override
            public void process(final WatchedEvent event) {
                if (event.getState() == Watcher.Event.KeeperState.SyncConnected) {
                    connectLatch.countDown();
                }
            }
        });

        boolean waitResult = connectLatch.await(ZOOKEEPER_CONNCT_TIMEOUT, TimeUnit.SECONDS);

        if (waitResult == false) {
            logger.warn("Unable to connect in " + ZOOKEEPER_CONNCT_TIMEOUT + " seconds");
            closeZookeeperConnectionNE();
            return;
        }

        createDirectoryStructureIfNeeded();

        registerInstanceIfNameWasSet();
    } catch (Exception e) {
        logger.warn("Got exception while connecting to zookeeper", e);
    }
}

From source file:org.cane.rpc.servicectrl.ServiceRegistry.java

License:Open Source License

/**
 * create connection to zookeeper server
 * @return//  ww w  . j  av a 2  s .c  o m
 */
private ZooKeeper connectZookeeperServer() {
    ZooKeeper zk = null;
    try {
        zk = new ZooKeeper(zookeeperAddress, ServiceConstant.SESSION_TIMEOUT, new Watcher() {

            @Override
            public void process(WatchedEvent event) {
                if (event.getState() == KeeperState.SyncConnected) {
                    latch.countDown();
                }
            }
        });
        latch.await();
    } catch (Exception e) {
        LOG.error("Can not connect given zookeeper server!", e);
    }
    return zk;
}

From source file:org.cloudata.core.commitlog.CommitLogServer.java

License:Apache License

@Override
public void process(WatchedEvent event) {
    if (event.getType() == Event.EventType.None) {
        switch (event.getState()) {
        case SyncConnected:
            break;
        case Disconnected:
            LOG.warn("Disconnected:" + event);
            break;
        case Expired:
            LOG.info("Shutdown cause lock expired:" + event);
            stopCommitLogServer();//from  w w w  .j a  v  a2s.c  om
        }
    }
}

From source file:org.cloudata.core.master.CloudataMaster.java

License:Apache License

@Override
public void process(WatchedEvent event) {
    if (event.getType() == Event.EventType.None) {
        switch (event.getState()) {
        case SyncConnected:
            break;
        case Disconnected:
            LOG.warn("Disconnected:" + event);
            break;
        case Expired:
            LOG.info("Shutdown cause lock expired:" + event);
            shutdown();/*  w  w  w .  j  a  v  a  2s.co  m*/
            break;
        }
    }
}

From source file:org.cloudata.core.tabletserver.TabletServer.java

License:Apache License

@Override
public void process(WatchedEvent event) {
    if (event.getType() == Event.EventType.None) {
        switch (event.getState()) {
        case SyncConnected:
            break;
        case Disconnected:
            LOG.warn("Disconnected:" + event);
            break;
        case Expired:
            LOG.fatal("Shutdown cause lock expired:" + event);
            shutdown();/*www  . j  ava  2s .c  om*/
            break;
        }
    }
}

From source file:org.deeplearning4j.hadoop.util.HdfsLock.java

License:Apache License

@Override
public void process(WatchedEvent event) {
    if (event.getState() == KeeperState.Expired) {
        zk = new ZookeeperBuilder().setHost(host).setPort(port).setWatcher(this).build();

    }//ww w. ja  v  a  2  s .c o  m
}

From source file:org.deeplearning4j.scaleout.zookeeper.ZookeeperBuilder.java

License:Apache License

@Override
public void process(WatchedEvent event) {
    if (event.getState() == KeeperState.SyncConnected) {
        log.info("Synced");
    } else if (event.getState() == KeeperState.Disconnected) {
        keeper = build();// ww  w.j a v a  2 s .c  o m
    } else if (event.getState() == KeeperState.Expired) {
        keeper = build();
    }
    if (watcher != null)
        watcher.process(event);
    log.info("Processed event...");
}

From source file:org.deeplearning4j.scaleout.zookeeper.ZookeeperConfigurationRetriever.java

License:Apache License

@Override
public void process(WatchedEvent event) {
    if (event.getState() == KeeperState.Expired) {
        keeper = new ZookeeperBuilder().setHost(host).setPort(port).setWatcher(this).build();

    }/*from w w w  .j ava 2 s.  c o  m*/
}