Java Code Examples for org.apache.nifi.nar.ExtensionManager#getExtensions()

/**
 * Uses the {@link ExtensionManager} to discover any {@link ControllerService} implementations that implement a specific
 * ControllerService API.
 *
 * @param parent the controller service API
 * @return a list of controller services that implement the controller service API
 */
private List<Class<? extends ControllerService>> lookupControllerServiceImpls(
        final Class<? extends ControllerService> parent) {

    final List<Class<? extends ControllerService>> implementations = new ArrayList<>();

    // first get all ControllerService implementations
    final Set<Class> controllerServices = ExtensionManager.getExtensions(ControllerService.class);

    // then iterate over all controller services looking for any that is a child of the parent
    // ControllerService API that was passed in as a parameter
    for (final Class<? extends ControllerService> controllerServiceClass : controllerServices) {
        if (parent.isAssignableFrom(controllerServiceClass)) {
            implementations.add(controllerServiceClass);
        }
    }

    return implementations;
}
 

/**
 * Gets the ControllerService types that this controller supports.
 *
 * @param serviceType type
 * @return the ControllerService types that this controller supports
 */
public Set<DocumentedTypeDTO> getControllerServiceTypes(final String serviceType) {
    final Set<Class> serviceImplementations = ExtensionManager.getExtensions(ControllerService.class);

    // identify the controller services that implement the specified serviceType if applicable
    final Set<Class> matchingServiceImplementions;
    if (serviceType != null) {
        matchingServiceImplementions = new HashSet<>();

        // check each type and remove those that aren't in the specified ancestry
        for (final Class type : serviceImplementations) {
            if (implementsServiceType(serviceType, type)) {
                matchingServiceImplementions.add(type);
            }
        }
    } else {
        matchingServiceImplementions = serviceImplementations;
    }

    return dtoFactory.fromDocumentedTypes(matchingServiceImplementions);
}
 

@Override
@SuppressWarnings("rawtypes")
public List<String> getComponentTypes() {
    final Set<Class> procClasses = ExtensionManager.getExtensions(Processor.class);
    final List<String> componentTypes = new ArrayList<>(procClasses.size() + 2);
    componentTypes.add(ProvenanceEventRecord.REMOTE_INPUT_PORT_TYPE);
    componentTypes.add(ProvenanceEventRecord.REMOTE_OUTPUT_PORT_TYPE);
    procClasses.stream()
        .map(procClass -> procClass.getSimpleName())
        .forEach(componentType -> componentTypes.add(componentType));
    return componentTypes;
}
 

@SuppressWarnings("rawtypes")
public Set<Class> getFlowFileProcessorClasses() {
    return ExtensionManager.getExtensions(Processor.class);
}
 

@SuppressWarnings("rawtypes")
public Set<Class> getFlowFileComparatorClasses() {
    return ExtensionManager.getExtensions(FlowFilePrioritizer.class);
}