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

/**
 * @param relationshipName
 *            name
 * @return the relationship for this nodes processor for the given name or
 *         creates a new relationship for the given name
 */
@Override
public Relationship getRelationship(final String relationshipName) {
    final Relationship specRel = new Relationship.Builder().name(relationshipName).build();
    Relationship returnRel = specRel;

    final Set<Relationship> relationships;
    try (final NarCloseable narCloseable = NarCloseable.withComponentNarLoader(processor.getClass(), processor.getIdentifier())) {
        relationships = processor.getRelationships();
    }

    for (final Relationship rel : relationships) {
        if (rel.equals(specRel)) {
            returnRel = rel;
            break;
        }
    }
    return returnRel;
}
 

/**
 * @param relationshipName
 *            name
 * @return the relationship for this nodes processor for the given name or
 *         creates a new relationship for the given name
 */
@Override
public Relationship getRelationship(final String relationshipName) {
    final Relationship specRel = new Relationship.Builder().name(relationshipName).build();
    Relationship returnRel = specRel;

    final Set<Relationship> relationships;
    final Processor processor = processorRef.get().getProcessor();
    try (final NarCloseable narCloseable = NarCloseable.withComponentNarLoader(getExtensionManager(), processor.getClass(), processor.getIdentifier())) {
        relationships = processor.getRelationships();
    }

    for (final Relationship rel : relationships) {
        if (rel.equals(specRel)) {
            returnRel = rel;
            break;
        }
    }
    return returnRel;
}
 

/**
 * Asserts that all FlowFiles that were transferred were transferred to the
 * given relationship
 *
 * @param relationship to validate
 */
public void assertAllFlowFilesTransferred(final Relationship relationship) {
    for (final Map.Entry<Relationship, List<MockFlowFile>> entry : transferMap.entrySet()) {
        final Relationship rel = entry.getKey();
        final List<MockFlowFile> flowFiles = entry.getValue();

        if (!rel.equals(relationship) && flowFiles != null && !flowFiles.isEmpty()) {
            Assert.fail("Expected all Transferred FlowFiles to go to " + relationship + " but " + flowFiles.size() + " were routed to " + rel);
        }
    }
}
 

/**
 * Asserts that all FlowFiles that were transferred in the given relationship
 * are compliant with the given validator.
 *
 * @param validator validator to use
 */
public void assertAllFlowFiles(Relationship relationship, FlowFileValidator validator) {
    for (final Map.Entry<Relationship, List<MockFlowFile>> entry : transferMap.entrySet()) {
        final List<MockFlowFile> flowFiles = entry.getValue();
        final Relationship rel = entry.getKey();
        for (MockFlowFile mockFlowFile : flowFiles) {
            if(rel.equals(relationship)) {
                validator.assertFlowFile(mockFlowFile);
            }
        }
    }
}
 

@Override
public void addConnection(final Connection connection) throws IllegalArgumentException {
    writeLock.lock();
    try {
        if (!requireNonNull(connection).getSource().equals(this)) {
            if (connection.getDestination().equals(this)) {
                // don't add the connection twice. This may occur if we have a self-loop because we will be told
                // to add the connection once because we are the source and again because we are the destination.
                if (!incomingConnections.contains(connection)) {
                    incomingConnections.add(connection);
                }

                return;
            } else {
                throw new IllegalArgumentException("Cannot add a connection to a LocalPort for which the LocalPort is neither the Source nor the Destination");
            }
        }

        for (final Relationship relationship : connection.getRelationships()) {
            if (!relationship.equals(PORT_RELATIONSHIP)) {
                throw new IllegalArgumentException("No relationship with name " + relationship + " exists for Local Ports");
            }
        }

        // don't add the connection twice. This may occur if we have a self-loop because we will be told
        // to add the connection once because we are the source and again because we are the destination.
        if (!outgoingConnections.contains(connection)) {
            outgoingConnections.add(connection);
        }
    } finally {
        writeLock.unlock();
    }
}
 

@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();
    }
}
 

@Override
public void addConnection(final Connection connection) throws IllegalArgumentException {
    writeLock.lock();
    try {
        if (!requireNonNull(connection).getSource().equals(this) && !connection.getDestination().equals(this)) {
            throw new IllegalArgumentException("Cannot add a connection to a Funnel for which the Funnel is neither the Source nor the Destination");
        }
        if (connection.getSource().equals(this) && connection.getDestination().equals(this)) {
            throw new IllegalArgumentException("Cannot add a connection from a Funnel back to itself");
        }

        if (connection.getDestination().equals(this)) {
            // don't add the connection twice. This may occur if we have a self-loop because we will be told
            // to add the connection once because we are the source and again because we are the destination.
            if (!incomingConnections.contains(connection)) {
                incomingConnections.add(connection);
            }
        }

        if (connection.getSource().equals(this)) {
            // don't add the connection twice. This may occur if we have a self-loop because we will be told
            // to add the connection once because we are the source and again because we are the destination.
            if (!outgoingConnections.contains(connection)) {
                for (final Relationship relationship : connection.getRelationships()) {
                    if (!relationship.equals(Relationship.ANONYMOUS)) {
                        throw new IllegalArgumentException("No relationship with name " + relationship + " exists for Funnels");
                    }
                }

                outgoingConnections.add(connection);
            }
        }
    } finally {
        writeLock.unlock();
    }
}
 

@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();
    }
}
 

/**
 * Asserts that all FlowFiles that were transferred were transferred to the
 * given relationship
 *
 * @param relationship to validate
 */
public void assertAllFlowFilesTransferred(final Relationship relationship) {
    for (final Map.Entry<Relationship, List<MockFlowFile>> entry : transferMap.entrySet()) {
        final Relationship rel = entry.getKey();
        final List<MockFlowFile> flowFiles = entry.getValue();

        if (!rel.equals(relationship) && flowFiles != null && !flowFiles.isEmpty()) {
            Assert.fail("Expected all Transferred FlowFiles to go to " + relationship + " but " + flowFiles.size() + " were routed to " + rel);
        }
    }
}
 

/**
 * Asserts that all FlowFiles that were transferred in the given relationship
 * are compliant with the given validator.
 *
 * @param validator validator to use
 */
public void assertAllFlowFiles(Relationship relationship, FlowFileValidator validator) {
    for (final Map.Entry<Relationship, List<MockFlowFile>> entry : transferMap.entrySet()) {
        final List<MockFlowFile> flowFiles = entry.getValue();
        final Relationship rel = entry.getKey();
        for (MockFlowFile mockFlowFile : flowFiles) {
            if(rel.equals(relationship)) {
                validator.assertFlowFile(mockFlowFile);
            }
        }
    }
}
 

@Override
public void addConnection(final Connection connection) throws IllegalArgumentException {
    writeLock.lock();
    try {
        if (!requireNonNull(connection).getSource().equals(this)) {
            if (connection.getDestination().equals(this)) {
                // don't add the connection twice. This may occur if we have a self-loop because we will be told
                // to add the connection once because we are the source and again because we are the destination.
                if (!incomingConnections.contains(connection)) {
                    incomingConnections.add(connection);
                }

                return;
            } else {
                throw new IllegalArgumentException("Cannot add a connection to a LocalPort for which the LocalPort is neither the Source nor the Destination");
            }
        }

        for (final Relationship relationship : connection.getRelationships()) {
            if (!relationship.equals(PORT_RELATIONSHIP)) {
                throw new IllegalArgumentException("No relationship with name " + relationship + " exists for Local Ports");
            }
        }

        // don't add the connection twice. This may occur if we have a self-loop because we will be told
        // to add the connection once because we are the source and again because we are the destination.
        if (!outgoingConnections.contains(connection)) {
            outgoingConnections.add(connection);
        }
    } finally {
        writeLock.unlock();
    }
}
 

@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();
    }
}
 

@Override
public void addConnection(final Connection connection) throws IllegalArgumentException {
    writeLock.lock();
    try {
        if (!requireNonNull(connection).getSource().equals(this) && !connection.getDestination().equals(this)) {
            throw new IllegalArgumentException("Cannot add a connection to a Funnel for which the Funnel is neither the Source nor the Destination");
        }
        if (connection.getSource().equals(this) && connection.getDestination().equals(this)) {
            throw new IllegalArgumentException("Cannot add a connection from a Funnel back to itself");
        }

        if (connection.getDestination().equals(this)) {
            // don't add the connection twice. This may occur if we have a self-loop because we will be told
            // to add the connection once because we are the source and again because we are the destination.
            if (!incomingConnections.contains(connection)) {
                incomingConnections.add(connection);
            }
        }

        if (connection.getSource().equals(this)) {
            // don't add the connection twice. This may occur if we have a self-loop because we will be told
            // to add the connection once because we are the source and again because we are the destination.
            if (!outgoingConnections.contains(connection)) {
                for (final Relationship relationship : connection.getRelationships()) {
                    if (!relationship.equals(Relationship.ANONYMOUS)) {
                        throw new IllegalArgumentException("No relationship with name " + relationship + " exists for Funnels");
                    }
                }

                outgoingConnections.add(connection);
            }
        }
    } finally {
        writeLock.unlock();
    }
}
 

@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();
    }
}