@Override
public void process(ClusterEvent event) throws Exception {
Cluster cluster = event.getAttribute("Cluster");
if (cluster == null) {
throw new StageException("Missing attributes in event:" + event + ". Requires Cluster");
}
Map<ResourceId, ResourceConfig> resourceConfigMap =
event.getAttribute(AttributeName.RESOURCES.toString());
if (resourceConfigMap == null) {
throw new StageException("Resources must be computed prior to validation!");
}
Map<ResourceId, Resource> resourceMap = cluster.getResourceMap();
Map<String, Map<String, String>> idealStateRuleMap =
event.getAttribute(AttributeName.IDEAL_STATE_RULES.toString());
for (ResourceId resourceId : resourceMap.keySet()) {
// check every ideal state against the ideal state rules
// the pipeline should not process any resources that have an unsupported ideal state
IdealState idealState = resourceMap.get(resourceId).getIdealState();
if (idealState == null) {
continue;
}
if (idealStateRuleMap != null && !idealStateRuleMap.isEmpty()) {
boolean hasMatchingRule = false;
for (String ruleName : idealStateRuleMap.keySet()) {
Map<String, String> rule = idealStateRuleMap.get(ruleName);
boolean matches = idealStateMatchesRule(idealState, rule);
hasMatchingRule = hasMatchingRule || matches;
if (matches) {
break;
}
}
if (!hasMatchingRule) {
LOG.warn("Resource " + resourceId + " does not have a valid ideal state!");
resourceConfigMap.remove(resourceId);
}
}
// check that every resource to process has a live state model definition
StateModelDefinitionId stateModelDefId = idealState.getStateModelDefId();
StateModelDefinition stateModelDef = cluster.getStateModelMap().get(stateModelDefId);
if (stateModelDef == null) {
LOG.warn(
"Resource "
+ resourceId
+ " uses state model "
+ stateModelDefId
+ ", but it is not on the cluster!");
resourceConfigMap.remove(resourceId);
}
}
}
