Examples with NodeDTO org.apache.nifi.web.api.dto.NodeDTO used on opensource projects

Example 1 with NodeDTO

use of org.apache.nifi.web.api.dto.NodeDTO in project nifi by apache.

the class StandardNiFiServiceFacade method getCluster.

@Override
public ClusterDTO getCluster() {
    // create cluster summary dto
    final ClusterDTO clusterDto = new ClusterDTO();
    // set current time
    clusterDto.setGenerated(new Date());
    // create node dtos
    final List<NodeDTO> nodeDtos = clusterCoordinator.getNodeIdentifiers().stream().map(nodeId -> getNode(nodeId)).collect(Collectors.toList());
    clusterDto.setNodes(nodeDtos);
    return clusterDto;
}

Example 2 with NodeDTO

use of org.apache.nifi.web.api.dto.NodeDTO in project nifi by apache.

the class ControllerResource method getNode.

/**
 * Gets the contents of the specified node in this NiFi cluster.
 *
 * @param id The node id.
 * @return A nodeEntity.
 */
@GET
@Consumes(MediaType.WILDCARD)
@Produces(MediaType.APPLICATION_JSON)
@Path("cluster/nodes/{id}")
@ApiOperation(value = "Gets a node in the cluster", response = NodeEntity.class, authorizations = { @Authorization(value = "Read - /controller") })
@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 getNode(@ApiParam(value = "The node id.", required = true) @PathParam("id") String id) {
    authorizeController(RequestAction.READ);
    // ensure connected to the cluster
    if (!isConnectedToCluster()) {
        throw new IllegalClusterResourceRequestException("Only a node connected to a cluster can process the request.");
    }
    if (isReplicateRequest()) {
        return replicate(HttpMethod.GET, getClusterCoordinatorNode());
    }
    // get the specified relationship
    final NodeDTO dto = serviceFacade.getNode(id);
    // create the response entity
    final NodeEntity entity = new NodeEntity();
    entity.setNode(dto);
    // generate the response
    return generateOkResponse(entity).build();
}

Example 3 with NodeDTO

use of org.apache.nifi.web.api.dto.NodeDTO in project nifi by apache.

the class ControllerResource method updateNode.

/**
 * Updates the contents of the specified node in this NiFi cluster.
 *
 * @param id         The id of the node
 * @param nodeEntity A nodeEntity
 * @return A nodeEntity
 */
@PUT
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
@Path("cluster/nodes/{id}")
@ApiOperation(value = "Updates a node in the cluster", response = NodeEntity.class, authorizations = { @Authorization(value = "Write - /controller") })
@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 updateNode(@ApiParam(value = "The node id.", required = true) @PathParam("id") String id, @ApiParam(value = "The node configuration. The only configuration that will be honored at this endpoint is the status.", required = true) NodeEntity nodeEntity) {
    authorizeController(RequestAction.WRITE);
    // ensure connected to the cluster
    if (!isConnectedToCluster()) {
        throw new IllegalClusterResourceRequestException("Only a node connected to a cluster can process the request.");
    }
    if (nodeEntity == null || nodeEntity.getNode() == null) {
        throw new IllegalArgumentException("Node details must be specified.");
    }
    // get the request node
    final NodeDTO requestNodeDTO = nodeEntity.getNode();
    if (!id.equals(requestNodeDTO.getNodeId())) {
        throw new IllegalArgumentException(String.format("The node id (%s) in the request body does " + "not equal the node id of the requested resource (%s).", requestNodeDTO.getNodeId(), id));
    }
    if (isReplicateRequest()) {
        return replicateToCoordinator(HttpMethod.PUT, nodeEntity);
    }
    // update the node
    final NodeDTO node = serviceFacade.updateNode(requestNodeDTO);
    // create the response entity
    NodeEntity entity = new NodeEntity();
    entity.setNode(node);
    // generate the response
    return generateOkResponse(entity).build();
}

Example 4 with NodeDTO

use of org.apache.nifi.web.api.dto.NodeDTO in project nifi by apache.

the class FlowResource method searchCluster.

// --------------------
// search cluster nodes
// --------------------
/**
 * Searches the cluster for a node with a given address.
 *
 * @param value Search value that will be matched against a node's address
 * @return Nodes that match the specified criteria
 */
@GET
@Consumes(MediaType.WILDCARD)
@Produces(MediaType.APPLICATION_JSON)
@Path("cluster/search-results")
@ApiOperation(value = "Searches the cluster for a node with the specified address", notes = NON_GUARANTEED_ENDPOINT, response = ClusterSearchResultsEntity.class, authorizations = { @Authorization(value = "Read - /flow") })
@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 searchCluster(@ApiParam(value = "Node address to search for.", required = true) @QueryParam("q") @DefaultValue(StringUtils.EMPTY) String value) {
    authorizeFlow();
    // ensure connected to the cluster
    if (!isConnectedToCluster()) {
        throw new IllegalClusterResourceRequestException("Only a node connected to a cluster can process the request.");
    }
    final List<NodeSearchResultDTO> nodeMatches = new ArrayList<>();
    // get the nodes in the cluster
    final ClusterDTO cluster = serviceFacade.getCluster();
    // check each to see if it matches the search term
    for (NodeDTO node : cluster.getNodes()) {
        // ensure the node is connected
        if (!NodeConnectionState.CONNECTED.name().equals(node.getStatus())) {
            continue;
        }
        // determine the current nodes address
        final String address = node.getAddress() + ":" + node.getApiPort();
        // count the node if there is no search or it matches the address
        if (StringUtils.isBlank(value) || StringUtils.containsIgnoreCase(address, value)) {
            final NodeSearchResultDTO nodeMatch = new NodeSearchResultDTO();
            nodeMatch.setId(node.getNodeId());
            nodeMatch.setAddress(address);
            nodeMatches.add(nodeMatch);
        }
    }
    // build the response
    ClusterSearchResultsEntity results = new ClusterSearchResultsEntity();
    results.setNodeResults(nodeMatches);
    // generate an 200 - OK response
    return noCache(Response.ok(results)).build();
}