Search in sources :

Example 1 with NiFiAllowableValue

use of com.thinkbiganalytics.nifi.rest.model.NiFiAllowableValue in project kylo by Teradata.

the class NiFiPropertyDescriptorTransformV1 method toNiFiAllowableValue.

/**
 * Transforms the specified {@link AllowableValueDTO} to a {@link NiFiAllowableValue}.
 *
 * @param dto the allowable value DTO
 * @return the NiFi allowable value
 */
@Nonnull
public NiFiAllowableValue toNiFiAllowableValue(@Nonnull final AllowableValueDTO dto) {
    final NiFiAllowableValue nifi = new NiFiAllowableValue();
    nifi.setDisplayName(dto.getDisplayName());
    nifi.setValue(dto.getValue());
    nifi.setDescription(dto.getDescription());
    return nifi;
}
Also used : NiFiAllowableValue(com.thinkbiganalytics.nifi.rest.model.NiFiAllowableValue) Nonnull(javax.annotation.Nonnull)

Example 2 with NiFiAllowableValue

use of com.thinkbiganalytics.nifi.rest.model.NiFiAllowableValue in project kylo by Teradata.

the class TemplateCreationHelper method updateProcessorControllerServiceReference.

private boolean updateProcessorControllerServiceReference(ProcessorDTO processorDTO, PropertyDescriptorDTO propertyDescriptorDTO, NifiProperty matchingProperty, Set<ProcessorDTO> updatedProcessors, Map<String, String> updatedProcessorProperties) {
    NiFiPropertyDescriptor descriptor = matchingProperty.getPropertyDescriptor();
    String propertyKey = matchingProperty.getKey();
    if (matchingProperty != null && descriptor != null && descriptor.getAllowableValues() != null) {
        NiFiAllowableValue matchingValue = descriptor.getAllowableValues().stream().filter(niFiAllowableValue -> niFiAllowableValue.getValue().equalsIgnoreCase(matchingProperty.getValue())).findFirst().orElse(null);
        if (matchingValue != null) {
            String name = matchingValue.getDisplayName();
            String validControllerServiceId = hasMatchingService(enabledServiceNameMap, name) ? enabledServiceNameMap.get(name).get(0).getId() : hasMatchingService(serviceNameMap, name) ? serviceNameMap.get(name).get(0).getId() : null;
            if (StringUtils.isNotBlank(validControllerServiceId) && (propertyDescriptorDTO.isRequired() || !propertyDescriptorDTO.isRequired() && StringUtils.isNotBlank(processorDTO.getConfig().getProperties().get(propertyKey)))) {
                processorDTO.getConfig().getProperties().put(propertyKey, validControllerServiceId);
                updatedProcessorProperties.put(propertyKey, validControllerServiceId);
                if (!updatedProcessors.contains(processorDTO)) {
                    updatedProcessors.add(processorDTO);
                }
                return true;
            }
        }
    }
    return false;
}
Also used : NiFiAllowableValue(com.thinkbiganalytics.nifi.rest.model.NiFiAllowableValue) NiFiPropertyDescriptor(com.thinkbiganalytics.nifi.rest.model.NiFiPropertyDescriptor)

Aggregations

NiFiAllowableValue (com.thinkbiganalytics.nifi.rest.model.NiFiAllowableValue)2 NiFiPropertyDescriptor (com.thinkbiganalytics.nifi.rest.model.NiFiPropertyDescriptor)1 Nonnull (javax.annotation.Nonnull)1