/**
* Discovers children of the given type underneath the given parent.
*
* @param parent look under this resource to find its children (if null, this looks for root
* resources)
* @param childType only find children of this type
* @param session session used to query the managed endpoint
* @param resourceConsumer if not null, will be a listener that gets notified when resources are
* discovered
*/
public <N> void discoverChildren(
Resource<L> parent,
ResourceType<L> childType,
Session<L> session,
Consumer<Resource<L>> resourceConsumer) {
try {
L parentLocation = parent != null ? parent.getLocation() : null;
log.debugf("Discovering children of [%s] of type [%s]", parent, childType);
final L childQuery =
session.getLocationResolver().absolutize(parentLocation, childType.getLocation());
Map<L, N> nativeResources = session.getDriver().fetchNodes(childQuery);
for (Map.Entry<L, N> entry : nativeResources.entrySet()) {
L location = entry.getKey(); // this is the unique DMR address for this resource
String resourceName =
session
.getLocationResolver()
.applyTemplate(
childType.getResourceNameTemplate(), location, session.getEndpoint().getName());
ID id = InventoryIdUtil.generateResourceId(session.getEndpoint(), location.toString());
Builder<L> builder =
Resource.<L>builder()
.id(id)
.name(new Name(resourceName))
.location(location)
.type(childType);
if (parent != null) {
builder.parent(parent);
}
// get the configuration of the resource
discoverResourceConfiguration(id, childType, location, entry.getValue(), builder, session);
// populate the metrics/avails based on the resource's type
addMetricAndAvailInstances(id, childType, location, entry.getValue(), builder, session);
Resource<L> resource = builder.build();
log.debugf("Discovered resource [%s]", resource);
if (resourceConsumer != null) {
resourceConsumer.accept(resource);
}
// recursively discover children of child types
Set<ResourceType<L>> childTypes = session.getResourceTypeManager().getChildren(childType);
for (ResourceType<L> nextLevelChildType : childTypes) {
discoverChildren(resource, nextLevelChildType, session, resourceConsumer);
}
}
} catch (Exception e) {
log.errorFailedToDiscoverResources(e, session.getEndpoint());
resourceConsumer.report(e);
}
}
private <N> void addMetricAndAvailInstances(
ID resourceId,
ResourceType<L> type,
L parentLocation,
N baseNode,
Resource.Builder<L> builder,
Session<L> session) {
for (MetricType<L> metricType : type.getMetricTypes()) {
AttributeLocation<L> location = metricType.getAttributeLocation();
try {
final AttributeLocation<L> instanceLocation =
session.getLocationResolver().absolutize(parentLocation, location);
if (session.getDriver().attributeExists(instanceLocation)) {
ID id =
InventoryIdUtil.generateMetricInstanceId(session.getFeedId(), resourceId, metricType);
Name name = metricType.getName();
MeasurementInstance<L, MetricType<L>> metricInstance =
new MeasurementInstance<>(id, name, instanceLocation, metricType);
builder.metric(metricInstance);
}
} catch (ProtocolException e) {
log.warnFailedToLocate(
e,
metricType.getClass().getName(),
String.valueOf(location),
String.valueOf(parentLocation));
}
}
for (AvailType<L> availType : type.getAvailTypes()) {
AttributeLocation<L> location = availType.getAttributeLocation();
try {
final AttributeLocation<L> instanceLocation =
session.getLocationResolver().absolutize(parentLocation, location);
if (session.getDriver().attributeExists(instanceLocation)) {
ID id =
InventoryIdUtil.generateAvailInstanceId(session.getFeedId(), resourceId, availType);
Name name = availType.getName();
MeasurementInstance<L, AvailType<L>> availInstance =
new MeasurementInstance<>(id, name, instanceLocation, availType);
builder.avail(availInstance);
}
} catch (ProtocolException e) {
log.warnFailedToLocate(
e,
availType.getClass().getName(),
String.valueOf(location),
String.valueOf(parentLocation));
}
}
}
private <N> void discoverResourceConfiguration(
ID resourceId,
ResourceType<L> type,
L parentAddress,
N baseNode,
Resource.Builder<L> builder,
Session<L> session) {
Collection<ResourceConfigurationPropertyType<L>> confPropTypes =
type.getResourceConfigurationPropertyTypes();
for (ResourceConfigurationPropertyType<L> confPropType : confPropTypes) {
try {
final AttributeLocation<L> location = confPropType.getAttributeLocation();
final LocationResolver<L> locationResolver = session.getLocationResolver();
final AttributeLocation<L> instanceLocation =
locationResolver.absolutize(parentAddress, location);
final Driver<L> driver = session.getDriver();
String resConfPropValue;
if (!locationResolver.isMultiTarget(instanceLocation.getLocation())) {
Object o = driver.fetchAttribute(instanceLocation);
resConfPropValue = ((o == null) ? null : o.toString());
} else {
// This resource config is a conglomeration of attrib values across multiple locations. We
// need to
// aggregate them all into a map and store the map as the resource configuration property
// value.
Map<L, Object> attribMap = driver.fetchAttributeAsMap(instanceLocation);
if (attribMap == null || attribMap.isEmpty()) {
resConfPropValue = null;
} else {
Map<String, String> resConfigPropMap = new HashMap<>(attribMap.size());
L multiTargetLocation = instanceLocation.getLocation();
for (Map.Entry<L, Object> entry : attribMap.entrySet()) {
L singleLocation = entry.getKey();
String attribValue = String.valueOf(entry.getValue());
String attribKey =
locationResolver.findWildcardMatch(multiTargetLocation, singleLocation);
resConfigPropMap.put(attribKey, attribValue);
}
resConfPropValue = Util.toJson(resConfigPropMap);
}
}
ResourceConfigurationPropertyInstance<L> cpi =
new ResourceConfigurationPropertyInstance<>(
ID.NULL_ID,
confPropType.getName(),
instanceLocation,
confPropType,
resConfPropValue);
builder.resourceConfigurationProperty(cpi);
} catch (Exception e) {
log.warnf(
e, "Failed to discover config [%s] for resource [%s]", confPropType, parentAddress);
}
}
}