Search in sources :

Example 1 with ConnectionsEntity

use of org.apache.nifi.web.api.entity.ConnectionsEntity in project nifi by apache.

the class ProcessGroupResource method getConnections.

/**
 * Gets all the connections.
 *
 * @return A connectionsEntity.
 */
@GET
@Consumes(MediaType.WILDCARD)
@Produces(MediaType.APPLICATION_JSON)
@Path("{id}/connections")
@ApiOperation(value = "Gets all connections", response = ConnectionsEntity.class, authorizations = { @Authorization(value = "Read - /process-groups/{uuid}") })
@ApiResponses(value = { @ApiResponse(code = 400, message = "NiFi was unable to complete the request because it was invalid. The request should not be retried without modification."), @ApiResponse(code = 401, message = "Client could not be authenticated."), @ApiResponse(code = 403, message = "Client is not authorized to make this request."), @ApiResponse(code = 404, message = "The specified resource could not be found."), @ApiResponse(code = 409, message = "The request was valid but NiFi was not in the appropriate state to process it. Retrying the same request later may be successful.") })
public Response getConnections(@ApiParam(value = "The process group id.", required = true) @PathParam("id") String groupId) {
    if (isReplicateRequest()) {
        return replicate(HttpMethod.GET);
    }
    // authorize access
    serviceFacade.authorizeAccess(lookup -> {
        final Authorizable processGroup = lookup.getProcessGroup(groupId).getAuthorizable();
        processGroup.authorize(authorizer, RequestAction.READ, NiFiUserUtils.getNiFiUser());
    });
    // all of the relationships for the specified source processor
    Set<ConnectionEntity> connections = serviceFacade.getConnections(groupId);
    // create the client response entity
    ConnectionsEntity entity = new ConnectionsEntity();
    entity.setConnections(connectionResource.populateRemainingConnectionEntitiesContent(connections));
    // generate the response
    return generateOkResponse(entity).build();
}
Also used : ConnectionsEntity(org.apache.nifi.web.api.entity.ConnectionsEntity) ComponentAuthorizable(org.apache.nifi.authorization.ComponentAuthorizable) Authorizable(org.apache.nifi.authorization.resource.Authorizable) SnippetAuthorizable(org.apache.nifi.authorization.SnippetAuthorizable) TemplateContentsAuthorizable(org.apache.nifi.authorization.TemplateContentsAuthorizable) ProcessGroupAuthorizable(org.apache.nifi.authorization.ProcessGroupAuthorizable) ConnectionEntity(org.apache.nifi.web.api.entity.ConnectionEntity) Path(javax.ws.rs.Path) Consumes(javax.ws.rs.Consumes) Produces(javax.ws.rs.Produces) GET(javax.ws.rs.GET) ApiOperation(io.swagger.annotations.ApiOperation) ApiResponses(io.swagger.annotations.ApiResponses)

Example 2 with ConnectionsEntity

use of org.apache.nifi.web.api.entity.ConnectionsEntity in project nifi by apache.

the class ConnectionsEndpointMerger method merge.

@Override
public final NodeResponse merge(final URI uri, final String method, final Set<NodeResponse> successfulResponses, final Set<NodeResponse> problematicResponses, final NodeResponse clientResponse) {
    if (!canHandle(uri, method)) {
        throw new IllegalArgumentException("Cannot use Endpoint Mapper of type " + getClass().getSimpleName() + " to map responses for URI " + uri + ", HTTP Method " + method);
    }
    final ConnectionsEntity responseEntity = clientResponse.getClientResponse().readEntity(ConnectionsEntity.class);
    final Set<ConnectionEntity> connectionEntities = responseEntity.getConnections();
    final Map<String, Map<NodeIdentifier, ConnectionEntity>> entityMap = new HashMap<>();
    for (final NodeResponse nodeResponse : successfulResponses) {
        final ConnectionsEntity nodeResponseEntity = nodeResponse == clientResponse ? responseEntity : nodeResponse.getClientResponse().readEntity(ConnectionsEntity.class);
        final Set<ConnectionEntity> nodeConnectionEntities = nodeResponseEntity.getConnections();
        for (final ConnectionEntity nodeConnectionEntity : nodeConnectionEntities) {
            final String nodeConnectionEntityId = nodeConnectionEntity.getId();
            Map<NodeIdentifier, ConnectionEntity> innerMap = entityMap.get(nodeConnectionEntityId);
            if (innerMap == null) {
                innerMap = new HashMap<>();
                entityMap.put(nodeConnectionEntityId, innerMap);
            }
            innerMap.put(nodeResponse.getNodeId(), nodeConnectionEntity);
        }
    }
    ConnectionsEntityMerger.mergeConnections(connectionEntities, entityMap);
    // create a new client response
    return new NodeResponse(clientResponse, responseEntity);
}
Also used : ConnectionsEntity(org.apache.nifi.web.api.entity.ConnectionsEntity) HashMap(java.util.HashMap) NodeIdentifier(org.apache.nifi.cluster.protocol.NodeIdentifier) NodeResponse(org.apache.nifi.cluster.manager.NodeResponse) ConnectionEntity(org.apache.nifi.web.api.entity.ConnectionEntity) Map(java.util.Map) HashMap(java.util.HashMap)

Aggregations

ConnectionEntity (org.apache.nifi.web.api.entity.ConnectionEntity)2 ConnectionsEntity (org.apache.nifi.web.api.entity.ConnectionsEntity)2 ApiOperation (io.swagger.annotations.ApiOperation)1 ApiResponses (io.swagger.annotations.ApiResponses)1 HashMap (java.util.HashMap)1 Map (java.util.Map)1 Consumes (javax.ws.rs.Consumes)1 GET (javax.ws.rs.GET)1 Path (javax.ws.rs.Path)1 Produces (javax.ws.rs.Produces)1 ComponentAuthorizable (org.apache.nifi.authorization.ComponentAuthorizable)1 ProcessGroupAuthorizable (org.apache.nifi.authorization.ProcessGroupAuthorizable)1 SnippetAuthorizable (org.apache.nifi.authorization.SnippetAuthorizable)1 TemplateContentsAuthorizable (org.apache.nifi.authorization.TemplateContentsAuthorizable)1 Authorizable (org.apache.nifi.authorization.resource.Authorizable)1 NodeResponse (org.apache.nifi.cluster.manager.NodeResponse)1 NodeIdentifier (org.apache.nifi.cluster.protocol.NodeIdentifier)1