PropertyKey idealStatePropertyKey = _keyBuilder.idealStates(tableName);
PropertyKey idealStatePropertyKey = _keyBuilder.idealStates(tableName);
@Override public Boolean call() { HelixDataAccessor dataAccessor = helixManager.getHelixDataAccessor(); PropertyKey idealStateKey = dataAccessor.keyBuilder().idealStates(resourceName); IdealState idealState = dataAccessor.getProperty(idealStateKey);
@Override public void dropResource(String clusterName, String resourceName) { logger.info("Drop resource {} from cluster {}", resourceName, clusterName); HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_zkClient)); Builder keyBuilder = accessor.keyBuilder(); accessor.removeProperty(keyBuilder.idealStates(resourceName)); accessor.removeProperty(keyBuilder.resourceConfig(resourceName)); }
@Deprecated @Override public void addIdealStateChangeListener(final org.apache.helix.IdealStateChangeListener listener) throws Exception { addListener(listener, new Builder(_clusterName).idealStates(), ChangeType.IDEAL_STATE, new EventType[] { EventType.NodeChildrenChanged }); }
@Override public void dropResource(String clusterName, String resourceName) { logger.info("Drop resource {} from cluster {}", resourceName, clusterName); HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_zkClient)); Builder keyBuilder = accessor.keyBuilder(); accessor.removeProperty(keyBuilder.idealStates(resourceName)); accessor.removeProperty(keyBuilder.resourceConfig(resourceName)); }
@Override public IdealState getResourceIdealState(String clusterName, String resourceName) { HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_zkClient)); Builder keyBuilder = accessor.keyBuilder(); return accessor.getProperty(keyBuilder.idealStates(resourceName)); }
@Override public void addIdealStateChangeListener(final IdealStateChangeListener listener) throws Exception { addListener(listener, new Builder(_clusterName).idealStates(), ChangeType.IDEAL_STATE, new EventType[] { EventType.NodeChildrenChanged }); }
StringRepresentation getIdealStateRepresentation(String clusterName, String resourceName) throws JsonGenerationException, JsonMappingException, IOException { Builder keyBuilder = new PropertyKey.Builder(clusterName); ZkClient zkclient = ResourceUtil.getAttributeFromCtx(getContext(), ResourceUtil.ContextKey.RAW_ZKCLIENT); String idealStateStr = ResourceUtil.readZkAsBytes(zkclient, keyBuilder.idealStates(resourceName)); StringRepresentation representation = new StringRepresentation(idealStateStr, MediaType.APPLICATION_JSON); return representation; }
private void checkForIdealStateAndContextRemoval(String workflow, String job) { // IdealState Assert.assertFalse( _baseAccessor.exists(_keyBuilder.idealStates(job).getPath(), AccessOption.PERSISTENT)); // JobContexts in old ZNode path String oldPath = String.format("/%s/PROPERTYSTORE/TaskRebalancer/%s/Context", CLUSTER_NAME, job); String newPath = _keyBuilder.jobContextZNode(workflow, job).getPath(); Assert.assertFalse(_baseAccessor.exists(oldPath, AccessOption.PERSISTENT)); Assert.assertFalse(_baseAccessor.exists(newPath, AccessOption.PERSISTENT)); } }
@Override public void addIdealStateChangeListener(final IdealStateChangeListener listener) throws Exception { addListener(listener, new Builder(_clusterName).idealStates(), ChangeType.IDEAL_STATE, new EventType[] { EventType.NodeChildrenChanged }); }
@Override public void setResourceIdealState(String clusterName, String resourceName, IdealState idealState) { logger .info("Set IdealState for resource {} in cluster {} with new IdealState {}.", resourceName, clusterName, idealState == null ? "NULL" : idealState.toString()); HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_zkClient)); Builder keyBuilder = accessor.keyBuilder(); accessor.setProperty(keyBuilder.idealStates(resourceName), idealState); }
private void verifyJobDeleted(String queueName, String jobName) throws Exception { HelixDataAccessor accessor = _manager.getHelixDataAccessor(); PropertyKey.Builder keyBuilder = accessor.keyBuilder(); Assert.assertNull(accessor.getProperty(keyBuilder.idealStates(jobName))); Assert.assertNull(accessor.getProperty(keyBuilder.resourceConfig(jobName))); TaskTestUtil.pollForEmptyJobState(_driver, queueName, jobName); }
@Override public IdealState getResourceIdealState(String clusterName, String resourceName) { HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_zkClient)); Builder keyBuilder = accessor.keyBuilder(); return accessor.getProperty(keyBuilder.idealStates(resourceName)); }
@Override public void setResourceIdealState(String clusterName, String resourceName, IdealState idealState) { logger .info("Set IdealState for resource {} in cluster {} with new IdealState {}.", resourceName, clusterName, idealState == null ? "NULL" : idealState.toString()); HelixDataAccessor accessor = new ZKHelixDataAccessor(clusterName, new ZkBaseDataAccessor<ZNRecord>(_zkClient)); Builder keyBuilder = accessor.keyBuilder(); accessor.setProperty(keyBuilder.idealStates(resourceName), idealState); }
private void verifyJobDeleted(String queueName, String jobName) throws Exception { HelixDataAccessor accessor = _manager.getHelixDataAccessor(); PropertyKey.Builder keyBuilder = accessor.keyBuilder(); Assert.assertNull(accessor.getProperty(keyBuilder.idealStates(jobName)), jobName + "'s idealstate has not been deleted!"); Assert.assertNull(accessor.getProperty(keyBuilder.resourceConfig(jobName)), jobName + "'s resourceConfig has not been deleted!"); TaskTestUtil.pollForEmptyJobState(_driver, queueName, jobName); }
private void createIS(HelixDataAccessor accessor, String resourceId, String stateModelDefRef, RebalanceMode rebalanceMode) { IdealState idealState = new IdealState(resourceId); idealState.setRebalanceMode(rebalanceMode); idealState.setStateModelDefRef(stateModelDefRef); idealState.setNumPartitions(1); idealState.setReplicas("1"); idealState.getRecord().setListField(resourceId + "_0", ImmutableList.of(PARTICIPANT)); idealState.getRecord().setMapField(resourceId + "_0", ImmutableMap.of(PARTICIPANT, STATE)); accessor.setProperty(accessor.keyBuilder().idealStates(resourceId), idealState); }
private void removeListeners(Listener listener) throws Exception { _manager.removeListener(new PropertyKey.Builder(_manager.getClusterName()).instanceConfigs(), listener); _manager .removeListener(new PropertyKey.Builder(_manager.getClusterName()).idealStates(), listener); }
ZkBaseDataAccessor zkBaseDataAccessor = (ZkBaseDataAccessor) dataAccessor.getBaseDataAccessor(); PropertyKey idealStateKey = dataAccessor.keyBuilder().idealStates(tableName); IdealState previousIdealState = dataAccessor.getProperty(idealStateKey);
public static IdealState getTableIdealState(HelixManager manager, String resourceName) { final HelixDataAccessor accessor = manager.getHelixDataAccessor(); final Builder builder = accessor.keyBuilder(); return accessor.getProperty(builder.idealStates(resourceName)); }