Java Code Examples for org.apache.nifi.processor.Relationship#getName()

The following examples show how to use org.apache.nifi.processor.Relationship#getName() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
Source File: MockProcessSession.java    From localization_nifi with Apache License 2.0 6 votes vote down vote up 
@Override
public void transfer(final FlowFile flowFile, final Relationship relationship) {
    if (relationship == Relationship.SELF) {
        transfer(flowFile);
        return;
    }
    if(!processor.getRelationships().contains(relationship)){
        throw new IllegalArgumentException("this relationship " + relationship.getName() + " is not known");
    }

    validateState(flowFile);
    List<MockFlowFile> list = transferMap.get(relationship);
    if (list == null) {
        list = new ArrayList<>();
        transferMap.put(relationship, list);
    }

    beingProcessed.remove(flowFile.getId());
    list.add((MockFlowFile) flowFile);
}
Example 2
Source File: MockProcessSession.java    From nifi with Apache License 2.0 6 votes vote down vote up 
@Override
public void transfer(FlowFile flowFile, final Relationship relationship) {
    if (relationship == Relationship.SELF) {
        transfer(flowFile);
        return;
    }
    if(!processor.getRelationships().contains(relationship)){
        throw new IllegalArgumentException("this relationship " + relationship.getName() + " is not known");
    }

    flowFile = validateState(flowFile);
    List<MockFlowFile> list = transferMap.computeIfAbsent(relationship, r -> new ArrayList<>());

    beingProcessed.remove(flowFile.getId());
    list.add((MockFlowFile) flowFile);
    updateLastQueuedDate((MockFlowFile) flowFile);
}
Example 3
Source File: StatelessProcessSession.java    From nifi with Apache License 2.0 6 votes vote down vote up 
@Override
public void transfer(FlowFile flowFile, final Relationship relationship) {
    if (relationship == Relationship.SELF) {
        transfer(flowFile);
        return;
    }
    if (!processor.getRelationships().contains(relationship)) {
        throw new IllegalArgumentException("this relationship " + relationship.getName() + " is not known");
    }

    flowFile = validateState(flowFile);

    if (outputMap.containsKey(relationship)) {
        Queue<StatelessFlowFile> queue = this.outputMap.get(relationship);
        queue.add((StatelessFlowFile) flowFile);

    }
    beingProcessed.remove(flowFile.getId());
    updateLastQueuedDate((StatelessFlowFile) flowFile);
}
Example 4
Source File: StandardProcessorNode.java    From nifi with Apache License 2.0 6 votes vote down vote up 
@Override
public void setAutoTerminatedRelationships(final Set<Relationship> terminate) {
    if (isRunning()) {
        throw new IllegalStateException("Cannot modify Processor configuration while the Processor is running");
    }

    for (final Relationship rel : terminate) {
        if (!getConnections(rel).isEmpty()) {
            throw new IllegalStateException("Cannot mark relationship '" + rel.getName()
                    + "' as auto-terminated because Connection already exists with this relationship");
        }
    }

    undefinedRelationshipsToTerminate.set(new HashSet<>(terminate));
    LOG.debug("Resetting Validation State of {} due to setting auto-terminated relationships", this);
    resetValidationState();
}
Example 5
Source File: AbstractPort.java    From localization_nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public Set<Connection> getConnections(final Relationship relationship) {
    readLock.lock();
    try {
        if (relationship.equals(PORT_RELATIONSHIP)) {
            return Collections.unmodifiableSet(outgoingConnections);
        }

        throw new IllegalArgumentException("No relationship with name " + relationship.getName() + " exists for Local Ports");
    } finally {
        readLock.unlock();
    }
}
Example 6
Source File: StandardFunnel.java    From localization_nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public Set<Connection> getConnections(final Relationship relationship) {
    readLock.lock();
    try {
        if (relationship.equals(Relationship.ANONYMOUS)) {
            return Collections.unmodifiableSet(outgoingConnections);
        }

        throw new IllegalArgumentException("No relationship with name " + relationship.getName() + " exists for Funnels");
    } finally {
        readLock.unlock();
    }
}
Example 7
Source File: StandardProcessorNode.java    From localization_nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public void setAutoTerminatedRelationships(final Set<Relationship> terminate) {
    if (isRunning()) {
        throw new IllegalStateException("Cannot modify Processor configuration while the Processor is running");
    }

    for (final Relationship rel : terminate) {
        if (!getConnections(rel).isEmpty()) {
            throw new IllegalStateException("Cannot mark relationship '" + rel.getName()
                    + "' as auto-terminated because Connection already exists with this relationship");
        }
    }
    undefinedRelationshipsToTerminate.set(new HashSet<>(terminate));
}
Example 8
Source File: StandardProcessSession.java    From localization_nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public void transfer(final FlowFile flowFile, final Relationship relationship) {
    validateRecordState(flowFile);
    final int numDestinations = context.getConnections(relationship).size();
    final int multiplier = Math.max(1, numDestinations);

    boolean autoTerminated = false;
    boolean selfRelationship = false;
    if (numDestinations == 0 && context.getConnectable().isAutoTerminated(relationship)) {
        // auto terminated.
        autoTerminated = true;
    } else if (numDestinations == 0 && relationship == Relationship.SELF) {
        selfRelationship = true;
    } else if (numDestinations == 0) {
        // the relationship specified is not known in this session/context
        throw new IllegalArgumentException("Relationship '" + relationship.getName() + "' is not known");
    }
    final StandardRepositoryRecord record = records.get(flowFile);
    record.setTransferRelationship(relationship);
    updateLastQueuedDate(record);

    if (autoTerminated) {
        removedCount += multiplier;
        removedBytes += flowFile.getSize();
    } else if (!selfRelationship) {
        flowFilesOut += multiplier;
        contentSizeOut += flowFile.getSize() * multiplier;
    }
}
Example 9
Source File: StandardProcessSession.java    From localization_nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public void transfer(final Collection<FlowFile> flowFiles, final Relationship relationship) {
    validateRecordState(flowFiles);

    boolean autoTerminated = false;
    boolean selfRelationship = false;
    final int numDestinations = context.getConnections(relationship).size();
    if (numDestinations == 0 && context.getConnectable().isAutoTerminated(relationship)) {
        // auto terminated.
        autoTerminated = true;
    } else if (numDestinations == 0 && relationship == Relationship.SELF) {
        selfRelationship = true;
    } else if (numDestinations == 0) {
        // the relationship specified is not known in this session/context
        throw new IllegalArgumentException("Relationship '" + relationship.getName() + "' is not known");
    }

    final int multiplier = Math.max(1, numDestinations);

    long contentSize = 0L;
    for (final FlowFile flowFile : flowFiles) {
        final StandardRepositoryRecord record = records.get(flowFile);
        record.setTransferRelationship(relationship);
        updateLastQueuedDate(record);

        contentSize += flowFile.getSize() * multiplier;
    }

    if (autoTerminated) {
        removedCount += multiplier * flowFiles.size();
        removedBytes += contentSize;
    } else if (!selfRelationship) {
        flowFilesOut += multiplier * flowFiles.size();
        contentSizeOut += multiplier * contentSize;
    }
}
Example 10
Source File: AbstractPort.java    From nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public Set<Connection> getConnections(final Relationship relationship) {
    readLock.lock();
    try {
        if (relationship.equals(PORT_RELATIONSHIP)) {
            return Collections.unmodifiableSet(outgoingConnections);
        }

        throw new IllegalArgumentException("No relationship with name " + relationship.getName() + " exists for Local Ports");
    } finally {
        readLock.unlock();
    }
}
Example 11
Source File: StandardFunnel.java    From nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public Set<Connection> getConnections(final Relationship relationship) {
    readLock.lock();
    try {
        if (relationship.equals(Relationship.ANONYMOUS)) {
            return Collections.unmodifiableSet(outgoingConnections);
        }

        throw new IllegalArgumentException("No relationship with name " + relationship.getName() + " exists for Funnels");
    } finally {
        readLock.unlock();
    }
}
Example 12
Source File: StandardProcessSession.java    From nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public void transfer(FlowFile flowFile, final Relationship relationship) {
    verifyTaskActive();
    flowFile = validateRecordState(flowFile);
    final int numDestinations = context.getConnections(relationship).size();
    final int multiplier = Math.max(1, numDestinations);

    boolean autoTerminated = false;
    boolean selfRelationship = false;
    if (numDestinations == 0 && context.getConnectable().isAutoTerminated(relationship)) {
        // auto terminated.
        autoTerminated = true;
    } else if (numDestinations == 0 && relationship == Relationship.SELF) {
        selfRelationship = true;
    } else if (numDestinations == 0) {
        // the relationship specified is not known in this session/context
        throw new IllegalArgumentException("Relationship '" + relationship.getName() + "' is not known");
    }
    final StandardRepositoryRecord record = getRecord(flowFile);
    record.setTransferRelationship(relationship);
    updateLastQueuedDate(record);

    if (autoTerminated) {
        removedCount += multiplier;
        removedBytes += flowFile.getSize();
    } else if (!selfRelationship) {
        flowFilesOut += multiplier;
        contentSizeOut += flowFile.getSize() * multiplier;
    }
}
Example 13
Source File: StandardProcessSession.java    From nifi with Apache License 2.0 5 votes vote down vote up 
@Override
public void transfer(Collection<FlowFile> flowFiles, final Relationship relationship) {
    verifyTaskActive();
    flowFiles = validateRecordState(flowFiles);

    boolean autoTerminated = false;
    boolean selfRelationship = false;
    final int numDestinations = context.getConnections(relationship).size();
    if (numDestinations == 0 && context.getConnectable().isAutoTerminated(relationship)) {
        // auto terminated.
        autoTerminated = true;
    } else if (numDestinations == 0 && relationship == Relationship.SELF) {
        selfRelationship = true;
    } else if (numDestinations == 0) {
        // the relationship specified is not known in this session/context
        throw new IllegalArgumentException("Relationship '" + relationship.getName() + "' is not known");
    }

    final int multiplier = Math.max(1, numDestinations);

    final long queuedTime = System.currentTimeMillis();
    long contentSize = 0L;
    for (final FlowFile flowFile : flowFiles) {
        final FlowFileRecord flowFileRecord = (FlowFileRecord) flowFile;
        final StandardRepositoryRecord record = getRecord(flowFileRecord);
        record.setTransferRelationship(relationship);
        updateLastQueuedDate(record, queuedTime);

        contentSize += flowFile.getSize();
    }

    if (autoTerminated) {
        removedCount += multiplier * flowFiles.size();
        removedBytes += multiplier * contentSize;
    } else if (!selfRelationship) {
        flowFilesOut += multiplier * flowFiles.size();
        contentSizeOut += multiplier * contentSize;
    }
}
Example 14
Source File: StandardProvenanceEventRecord.java    From localization_nifi with Apache License 2.0 4 votes vote down vote up 
@Override
public Builder setRelationship(Relationship relationship) {
    this.relationship = relationship.getName();
    return this;
}
Example 15
Source File: StandardProvenanceEventRecord.java    From nifi with Apache License 2.0 4 votes vote down vote up 
@Override
public Builder setRelationship(Relationship relationship) {
    this.relationship = relationship.getName();
    return this;
}