Examples with CounterEntity org.apache.nifi.web.api.entity.CounterEntity used on opensource projects

Example 1 with CounterEntity

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

the class CountersResource method updateCounter.

/**
 * Update the specified counter. This will reset the counter value to 0.
 *
 * @param httpServletRequest request
 * @param id                 The id of the counter.
 * @return A counterEntity.
 */
@PUT
@Consumes(MediaType.WILDCARD)
@Produces(MediaType.APPLICATION_JSON)
@Path("{id}")
@ApiOperation(value = "Updates the specified counter. This will reset the counter value to 0", notes = NON_GUARANTEED_ENDPOINT, response = CounterEntity.class, authorizations = { @Authorization(value = "Write - /counters") })
@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 updateCounter(@Context final HttpServletRequest httpServletRequest, @ApiParam(value = "The id of the counter.") @PathParam("id") final String id) {
    if (isReplicateRequest()) {
        return replicate(HttpMethod.PUT);
    }
    final ComponentEntity requestComponentEntity = new ComponentEntity();
    requestComponentEntity.setId(id);
    return withWriteLock(serviceFacade, requestComponentEntity, lookup -> {
        authorizeCounters(RequestAction.WRITE);
    }, null, (componentEntity) -> {
        // reset the specified counter
        final CounterDTO counter = serviceFacade.updateCounter(componentEntity.getId());
        // create the response entity
        final CounterEntity entity = new CounterEntity();
        entity.setCounter(counter);
        // generate the response
        return generateOkResponse(entity).build();
    });
}