Java Code Examples for org.apache.curator.framework.CuratorFramework#getState()

The following examples show how to use org.apache.curator.framework.CuratorFramework#getState() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: ZKUtil.java From kylin-on-parquet-v2 with Apache License 2.0

6 votes

private static CuratorFramework getZookeeperClient(final String zkString, final RetryPolicy retryPolicy) {
    if (StringUtils.isEmpty(zkString)) {
        throw new IllegalArgumentException("ZOOKEEPER_QUORUM is empty!");
    }
    try {
        CuratorFramework instance = CACHE.get(zkString, new Callable<CuratorFramework>() {
            @Override
            public CuratorFramework call() throws Exception {
                return newZookeeperClient(zkString, retryPolicy);
            }
        });
        // during test, curator may be closed by others, remove it from CACHE and reinitialize a new one
        if (instance.getState() != CuratorFrameworkState.STARTED) {
            logger.warn("curator for {} is closed by others unexpectedly, reinitialize a new one", zkString);
            CACHE.invalidate(zkString);
            instance = getZookeeperClient(zkString, retryPolicy);
        }
        return instance;
    } catch (Throwable e) {
        throw new RuntimeException(e);
    }
}

Example 2

Source File: EventContainer.java From DBus with Apache License 2.0

6 votes

/**
 * 在拉完全量后将此schema的kafka consumer的offset设置为最新
 *
 * @param dbSchema
 */
/*public void setKafkaOffsetToLargest(String targetTopic){
	if(targetTopic==null)
		return;
	TopicPartition partition0 = new TopicPartition(targetTopic, 0);
	KafkaConsumerContainer.getInstances().getConsumer(targetTopic).seekToEnd(Arrays.asList(partition0));
}*/
protected <T> T deserialize(String path, Class<T> clazz) throws Exception {
    T packet = null;
    CuratorFramework curator = CuratorContainer.getInstance().getCurator();
    if (curator.getState() == CuratorFrameworkState.STOPPED) {
        LOG.info("[EventContainer] CuratorFrameworkState:{}", CuratorFrameworkState.STOPPED.name());
    } else {
        byte[] bytes = curator.getData().forPath(path);
        if (bytes != null && bytes.length != 0) {
            packet = JsonUtil.fromJson(new String(bytes, Charset.forName("UTF-8")), clazz);
        }
    }
    return packet;
}

Example 3

Source File: Zk.java From rdf-delta with Apache License 2.0

6 votes

/** Connect a curator client to a running ZooKepper server. */
public static void connect(CuratorFramework client) {
    switch(client.getState()) {
        case LATENT :
            client.start();
            try { client.blockUntilConnected(); }
            catch (InterruptedException ex) { throw new RuntimeException(ex); }
            return;
        case STARTED :
            //LOG.warn("CuratorFramework already started");
            return ;
        case STOPPED :
            throw new DeltaException("CuratorFramework closed");
        default :
            break;
    }
}

Example 4

Source File: ZKUtil.java From kylin with Apache License 2.0

6 votes

private static CuratorFramework getZookeeperClient(final String zkString, final RetryPolicy retryPolicy) {
    if (StringUtils.isEmpty(zkString)) {
        throw new IllegalArgumentException("ZOOKEEPER_QUORUM is empty!");
    }
    try {
        CuratorFramework instance = CACHE.get(zkString, new Callable<CuratorFramework>() {
            @Override
            public CuratorFramework call() throws Exception {
                return newZookeeperClient(zkString, retryPolicy);
            }
        });
        // during test, curator may be closed by others, remove it from CACHE and reinitialize a new one
        if (instance.getState() != CuratorFrameworkState.STARTED) {
            logger.warn("curator for {} is closed by others unexpectedly, reinitialize a new one", zkString);
            CACHE.invalidate(zkString);
            instance = getZookeeperClient(zkString, retryPolicy);
        }
        return instance;
    } catch (Throwable e) {
        throw new RuntimeException(e);
    }
}

Example 5

Source File: AbstractEvent.java From DBus with Apache License 2.0

5 votes

protected <T> T deserialize(String path, Class<T> clazz) throws Exception {
    T packet = null;
    CuratorFramework curator = CuratorContainer.getInstance().getCurator();
    if (curator.getState() == CuratorFrameworkState.STOPPED) {
        LOG.info("[control-event] CuratorFrameworkState:{}", CuratorFrameworkState.STOPPED.name());
    } else {
        byte[] bytes = curator.getData().forPath(path);
        if (bytes != null && bytes.length != 0) {
            packet = JsonUtil.fromJson(new String(bytes, Charset.forName("UTF-8")), clazz);
        }
    }
    return packet;
}

Example 6

Source File: AbstractEvent.java From DBus with Apache License 2.0

5 votes

protected void saveZk(String node, String packet) {
    try {
        CuratorFramework curator = CuratorContainer.getInstance().getCurator();
        if (curator.getState() == CuratorFrameworkState.STOPPED) {
            LOG.info("[control-event] CuratorFrameworkState:{}", CuratorFrameworkState.STOPPED.name());
        } else {
            curator.setData().forPath(node, packet.getBytes());
        }
    } catch (Exception e) {
        LOG.error("[control-event] 报错znode: " + node + ",数据包:" + packet + "失败!", e);
    }
}

Example 7

Source File: ProjectCheckHeartBeatEvent.java From DBus with Apache License 2.0

5 votes

/**
 * 根据根路径，获取所有末端叶子ZK节点
 *
 * @param basePath 根路径
 * @return
 * @throws Exception
 */
private Set<String> getZKNodePathsFromPath(String basePath, Set<String> result) {
    Set<String> defaultSet = new HashSet<>();
    try {
        CuratorFramework curator = CuratorContainer.getInstance().getCurator();
        if (curator.getState() == CuratorFrameworkState.STOPPED) {
            LOG.info("[traverse zk node paths] CuratorFrameworkState:{}", CuratorFrameworkState.STOPPED.name());
            return defaultSet;
        } else {
            //获取下一级path
            List<String> paths = curator.getChildren().forPath(basePath);
            if (paths.size() > 0) {
                //遍历添加下一级节点
                for (String path : paths) {
                    path = basePath + "/" + path;
                    getZKNodePathsFromPath(path, result);
                }
            } else {
                //如果是末端叶子节点，那么将完整的路径加入到结果集
                if (!HeartBeatConfigContainer.getInstance().getHbConf().getProjectMonitorPath()
                        .equals(basePath)) {// 默认的base path不加入
                    LOG.info("add nodePath : " + basePath);
                    result.add(basePath);
                }

            }
            return result;
        }
    } catch (Exception e) {
        LoggerFactory.getLogger().error("[ traverse zk node paths] path :" + basePath, e);
        return defaultSet;
    }
}

Example 8

Source File: EventContainer.java From DBus with Apache License 2.0

5 votes

protected void saveZk(String node, String packet) {
    try {
        CuratorFramework curator = CuratorContainer.getInstance().getCurator();
        if (curator.getState() == CuratorFrameworkState.STOPPED) {
            LOG.info("[EventContainer] CuratorFrameworkState:{}", CuratorFrameworkState.STOPPED.name());
        } else {
            curator.setData().forPath(node, packet.getBytes());
        }
    } catch (Exception e) {
        LOG.error("[control-event] 报错znode: " + node + ",数据包:" + packet + "失败!", e);
    }
}

Example 9

Source File: ZookeeperWorkerRegister.java From idworker with Apache License 2.0

5 votes

/**
 * 关闭注册
 */
@Override
public synchronized void logout() {
    CuratorFramework client = (CuratorFramework) regCenter.getRawClient();
    if (client != null && client.getState() == CuratorFrameworkState.STARTED) {
        // 移除注册节点（最大程度的自动释放资源）
        regCenter.remove(nodePath.getWorkerIdPath());
        // 关闭连接
        regCenter.close();
    }
}

Example 10

Source File: MesosMasterHealthCheck.java From myriad with Apache License 2.0

5 votes

@Override
protected Result check() throws Exception {
	String mesosMaster = cfg.getMesosMaster();
	int zkIndex = mesosMaster.indexOf("zk://", 0);
	if (zkIndex >= 0) {
		String zkHostPorts = mesosMaster.substring(5,
				mesosMaster.indexOf("/", 5));

		String[] hostPorts = zkHostPorts.split(",");

		for (String hostPort : hostPorts) {
			CuratorFramework client = CuratorFrameworkFactory.newClient(
					hostPort, new ExponentialBackoffRetry(1000, 3));
			client.start();
			client.blockUntilConnected(5, TimeUnit.SECONDS);

			switch (client.getState()) {
			case STARTED:
				return Result.healthy();
			case STOPPED:
				LOGGER.fine("Unable to reach: " + hostPort);
			case LATENT:
				LOGGER.fine("Unable to reach: " + hostPort);
			default:
				LOGGER.fine("Unable to reach: " + hostPort);
			}
		}
	} else {
		if (HealthCheckUtils.checkHostPort(mesosMaster)) {
			return Result.healthy();
		}
	}

	return Result.unhealthy("Unable to connect to: " + mesosMaster);
}

Example 11

Source File: CuratorManager.java From snowizard-discovery with BSD 3-Clause "New" or "Revised" License

5 votes

/**
 * Constructor
 * 
 * @param framework
 *            {@link CuratorFramework}
 */
public CuratorManager(final CuratorFramework framework) {
    this.framework = checkNotNull(framework);
    // start framework directly to allow other bundles to interact with zookeeper
    // during their run() method.
    if (framework.getState() != CuratorFrameworkState.STARTED) {
        framework.start();
    }
}

Example 12

Source File: MesosMasterHealthCheck.java From incubator-myriad with Apache License 2.0

4 votes

@Override
protected Result check() throws Exception {
  String mesosMaster = cfg.getMesosMaster();
  int zkIndex = mesosMaster.indexOf("zk://", 0);
  Result result = Result.unhealthy("Unable to connect to: " + mesosMaster);
  if (zkIndex >= 0) {
    final int fromIndex = 5;
    String zkHostPorts = mesosMaster.substring(fromIndex, mesosMaster.indexOf("/", fromIndex));

    String[] hostPorts = zkHostPorts.split(",");

    for (String hostPort : hostPorts) {
      final int maxRetries = 3;
      final int baseSleepTimeMs = 1000;
      CuratorFramework client = CuratorFrameworkFactory.newClient(hostPort, new ExponentialBackoffRetry(baseSleepTimeMs,
          maxRetries));
      client.start();
      final int blockTime = 5;
      client.blockUntilConnected(blockTime, TimeUnit.SECONDS);

      switch (client.getState()) {
        case STARTED:
          result = Result.healthy();
          break;
        case STOPPED:
          LOGGER.info("Unable to reach: ", hostPort);
          break;
        case LATENT:
          LOGGER.info("Unable to reach: ", hostPort);
          break;
        default:
          LOGGER.info("Unable to reach: ", hostPort);
      }
    }
  } else {
    if (HealthCheckUtils.checkHostPort(mesosMaster)) {
      result = Result.healthy();
    }
  }

  return result;
}