Java Code Examples for org.apache.curator.framework.recipes.cache.TreeCacheEvent#getData()

/**
 * Resolve the name of service.
 * @param event {@link TreeCacheEvent}
 * @return If the Zookeeper's {@link ChildData#getPath() node path} that was
 * notified comes from {@link ServiceInstance the service instance}, return it's
 * parent path as the service name, or return <code>null</code>
 */
private String resolveServiceName(TreeCacheEvent event) {
	ChildData childData = event.getData();
	String path = childData.getPath();
	if (logger.isDebugEnabled()) {
		logger.debug("ZK node[path : {}] event type : {}", path, event.getType());
	}

	String serviceName = null;

	if (pathMatcher.match(serviceInstancePathPattern, path)) {
		Map<String, String> variables = pathMatcher
				.extractUriTemplateVariables(serviceInstancePathPattern, path);
		serviceName = variables.get(SERVICE_NAME_PATH_VARIABLE_NAME);
	}

	return serviceName;
}
 

@Override
public void childEvent(CuratorFramework curator, TreeCacheEvent event) throws Exception {
    ChildData data = event.getData();
    if (data == null) return;
    String dataPath = data.getPath();
    if (Strings.isNullOrEmpty(dataPath)) return;
    if (dataPath.startsWith(watchPath)) {
        switch (event.getType()) {
            case NODE_ADDED:
                listener.onServiceAdded(dataPath, Jsons.fromJson(data.getData(), CommonServiceNode.class));
                break;
            case NODE_REMOVED:
                listener.onServiceRemoved(dataPath, Jsons.fromJson(data.getData(), CommonServiceNode.class));
                break;
            case NODE_UPDATED:
                listener.onServiceUpdated(dataPath, Jsons.fromJson(data.getData(), CommonServiceNode.class));
                break;
        }
        Logs.RSD.info("ZK node data change={}, nodePath={}, watchPath={}, ns={}");
    }
}
 

private void invalidatePropertyDefinitions(TreeCacheEvent event) {
    if (event == null || event.getData() == null) {
        return;
    }

    String path = event.getData().getPath();
    byte[] bytes = event.getData().getData();
    if (path == null || bytes == null) {
        return;
    }

    if (!path.startsWith(zkPath + "/" + ZK_DEFINE_PROPERTY)) {
        return;
    }

    String key = new String(bytes, StandardCharsets.UTF_8);
    if (key == null) {
        return;
    }

    String propertyName = key.substring(METADATA_DEFINE_PROPERTY_PREFIX.length());
    LOGGER.debug("invalidating property definition: %s", propertyName);
    invalidatePropertyDefinition(propertyName);
}
 

@Override
public void childEvent(CuratorFramework client, TreeCacheEvent event) throws Exception {
  String path = null;
  byte[] data = null;
  if(event != null && event.getData() != null) {
    path = event.getData().getPath();
    data = event.getData().getData();
  }
  LOG.debug("Type: {}, Path: {}, Data: {}", event.getType(), (path == null?"":path) , (data == null?"":new String(data,
      StandardCharsets.UTF_8)));
  List<Callback> callback = callbacks.get(event.getType());
  if(callback != null) {
    for(Callback cb : callback) {
      cb.apply(client, path, data);
    }
  }
}
 

/**
 * Update cached application instance according to zookeeper node event.
 *
 * @param event zookeeper node changed event
 */
private void doCreateOrUpdateApplications(TreeCacheEvent event) {
    ChildData chileData = event.getData();
    Application app = ZookeeperRegistryUtils.parseSessionNode(chileData.getPath());
    if (app != null) {
        applications.compute(app.getAppName(), (key, value) -> {
            Set<Application> group = value == null ? new ConcurrentSkipListSet<>() : value;
            group.remove(app); // remove if exists
            group.add(app);
            return group;
        });
    }
}
 

/**
 * Remove cached application instance according to zookeeper node event.
 *
 * @param event zookeeper node changed event
 */
private void doRemoveApplications(TreeCacheEvent event) {
    ChildData chileData = event.getData();
    if (chileData == null) {
        return; // Maybe null if session is timeout
    }

    Application app = ZookeeperRegistryUtils.parseSessionNode(chileData.getPath());
    if (app != null) {
        applications.computeIfPresent(app.getAppName(), (key, value) -> {
            value.remove(app); // Always remove whatever if it's exists
            return value;
        });
    }
}
 

@Override
public void childEvent(CuratorFramework curator, TreeCacheEvent event) throws Exception {
    ChildData data = event.getData();
    if (data != null) {
        if (event.getType() == NODE_ADDED) {

        }
        String path = data.getPath();
        //判断是否为topics节点
        if (path.contains(String.format("%s/",ZkUtils.BrokerTopicsPath())) && (!path.contains("partitions"))) {
            Topic topic = JSONObject.parseObject(data.getData(), Topic.class);
            String name = path.substring(path.lastIndexOf("/") + 1, path.length());
            topic.setName(name);

            int[] tPartiyions = topic.getPartitions().keySet().stream().mapToInt((t) -> Integer.valueOf(t)).sorted().toArray();
            for (Object key : tPartiyions
                    ) {
                String partitionPath = String.format("%s/partitions/%s/state", path, key);
                String state = new String(curator.getData().forPath(partitionPath));
                Partition partition = JSONObject.parseObject(state, Partition.class);
                JSONArray replicas = topic.getPartitions().getJSONArray(String.valueOf(key));
                int[] replicasArray = new int[replicas.size()];
                for (int i = 0; i <
                        replicas.size(); i++) {
                    replicasArray[i] = replicas.getInteger(i);
                }
                partition.setReplicasArray(replicasArray);

                topic.getPartitionMap().put((Integer) key, partition);
            }
            topicList.add(topic);
        }
    }
}
 

@Override
public final void childEvent(final CuratorFramework client, final TreeCacheEvent event) {
    ChildData childData = event.getData();
    if (null == childData) {
        return;
    }
    String path = childData.getPath();
    if (path.isEmpty()) {
        return;
    }
    dataChanged(path, event.getType(), null == childData.getData() ? "" : new String(childData.getData(), Charsets.UTF_8));
}