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

Example 1 with ReportingTaskEntity

use of org.apache.nifi.web.api.entity.ReportingTaskEntity in project kylo by Teradata.

the class NiFiReportingTaskRestClientV1 method update.

/**
 * update a reporting task
 *
 * @param reportingTaskDTO the reporting task
 * @return the updated reporting task, null if there was an error
 */
@Override
public ReportingTaskDTO update(final ReportingTaskDTO reportingTaskDTO) {
    String id = reportingTaskDTO.getId();
    Optional<ReportingTaskDTO> updated = findReportingTaskEntityById(id).flatMap(current -> {
        try {
            ReportingTaskEntity entity = client.put(REPORTING_TASK_PATH + id, toReportingTaskEntity(reportingTaskDTO, current.getRevision().getVersion()), ReportingTaskEntity.class);
            if (entity != null) {
                return Optional.of(entity.getComponent());
            } else {
                return null;
            }
        } catch (Exception e) {
            return Optional.empty();
        }
    });
    if (updated.isPresent()) {
        return updated.get();
    } else {
        return null;
    }
}

Example 2 with ReportingTaskEntity

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

the class StandardNiFiServiceFacade method updateReportingTask.

@Override
public ReportingTaskEntity updateReportingTask(final Revision revision, final ReportingTaskDTO reportingTaskDTO) {
    // get the component, ensure we have access to it, and perform the update request
    final ReportingTaskNode reportingTask = reportingTaskDAO.getReportingTask(reportingTaskDTO.getId());
    final RevisionUpdate<ReportingTaskDTO> snapshot = updateComponent(revision, reportingTask, () -> reportingTaskDAO.updateReportingTask(reportingTaskDTO), rt -> dtoFactory.createReportingTaskDto(rt));
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(reportingTask);
    final List<BulletinDTO> bulletins = dtoFactory.createBulletinDtos(bulletinRepository.findBulletinsForSource(reportingTask.getIdentifier()));
    final List<BulletinEntity> bulletinEntities = bulletins.stream().map(bulletin -> entityFactory.createBulletinEntity(bulletin, permissions.getCanRead())).collect(Collectors.toList());
    return entityFactory.createReportingTaskEntity(snapshot.getComponent(), dtoFactory.createRevisionDTO(snapshot.getLastModification()), permissions, bulletinEntities);
}

Example 3 with ReportingTaskEntity

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

the class StandardNiFiServiceFacade method createReportingTaskEntity.

private ReportingTaskEntity createReportingTaskEntity(final ReportingTaskNode reportingTask) {
    final RevisionDTO revision = dtoFactory.createRevisionDTO(revisionManager.getRevision(reportingTask.getIdentifier()));
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(reportingTask);
    final List<BulletinDTO> bulletins = dtoFactory.createBulletinDtos(bulletinRepository.findBulletinsForSource(reportingTask.getIdentifier()));
    final List<BulletinEntity> bulletinEntities = bulletins.stream().map(bulletin -> entityFactory.createBulletinEntity(bulletin, permissions.getCanRead())).collect(Collectors.toList());
    return entityFactory.createReportingTaskEntity(dtoFactory.createReportingTaskDto(reportingTask), revision, permissions, bulletinEntities);
}

Example 4 with ReportingTaskEntity

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

the class ControllerResource method createReportingTask.

// ---------------
// reporting tasks
// ---------------
/**
 * Creates a new Reporting Task.
 *
 * @param httpServletRequest  request
 * @param requestReportingTaskEntity A reportingTaskEntity.
 * @return A reportingTaskEntity.
 */
@POST
@Consumes(MediaType.APPLICATION_JSON)
@Produces(MediaType.APPLICATION_JSON)
@Path("reporting-tasks")
@ApiOperation(value = "Creates a new reporting task", response = ReportingTaskEntity.class, authorizations = { @Authorization(value = "Write - /controller"), @Authorization(value = "Read - any referenced Controller Services - /controller-services/{uuid}"), @Authorization(value = "Write - if the Reporting Task is restricted - /restricted-components") })
@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 = 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 createReportingTask(@Context final HttpServletRequest httpServletRequest, @ApiParam(value = "The reporting task configuration details.", required = true) final ReportingTaskEntity requestReportingTaskEntity) {
    if (requestReportingTaskEntity == null || requestReportingTaskEntity.getComponent() == null) {
        throw new IllegalArgumentException("Reporting task details must be specified.");
    }
    if (requestReportingTaskEntity.getRevision() == null || (requestReportingTaskEntity.getRevision().getVersion() == null || requestReportingTaskEntity.getRevision().getVersion() != 0)) {
        throw new IllegalArgumentException("A revision of 0 must be specified when creating a new Reporting task.");
    }
    final ReportingTaskDTO requestReportingTask = requestReportingTaskEntity.getComponent();
    if (requestReportingTask.getId() != null) {
        throw new IllegalArgumentException("Reporting task ID cannot be specified.");
    }
    if (StringUtils.isBlank(requestReportingTask.getType())) {
        throw new IllegalArgumentException("The type of reporting task to create must be specified.");
    }
    if (isReplicateRequest()) {
        return replicate(HttpMethod.POST, requestReportingTaskEntity);
    }
    return withWriteLock(serviceFacade, requestReportingTaskEntity, lookup -> {
        authorizeController(RequestAction.WRITE);
        ComponentAuthorizable authorizable = null;
        try {
            authorizable = lookup.getConfigurableComponent(requestReportingTask.getType(), requestReportingTask.getBundle());
            if (authorizable.isRestricted()) {
                authorizeRestrictions(authorizer, authorizable);
            }
            if (requestReportingTask.getProperties() != null) {
                AuthorizeControllerServiceReference.authorizeControllerServiceReferences(requestReportingTask.getProperties(), authorizable, authorizer, lookup);
            }
        } finally {
            if (authorizable != null) {
                authorizable.cleanUpResources();
            }
        }
    }, () -> serviceFacade.verifyCreateReportingTask(requestReportingTask), (reportingTaskEntity) -> {
        final ReportingTaskDTO reportingTask = reportingTaskEntity.getComponent();
        // set the processor id as appropriate
        reportingTask.setId(generateUuid());
        // create the reporting task and generate the json
        final Revision revision = getRevision(reportingTaskEntity, reportingTask.getId());
        final ReportingTaskEntity entity = serviceFacade.createReportingTask(revision, reportingTask);
        reportingTaskResource.populateRemainingReportingTaskEntityContent(entity);
        // build the response
        return generateCreatedResponse(URI.create(entity.getUri()), entity).build();
    });
}

Example 5 with ReportingTaskEntity

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

the class EntityFactory method createReportingTaskEntity.

public ReportingTaskEntity createReportingTaskEntity(final ReportingTaskDTO dto, final RevisionDTO revision, final PermissionsDTO permissions, final List<BulletinEntity> bulletins) {
    final ReportingTaskEntity entity = new ReportingTaskEntity();
    entity.setRevision(revision);
    if (dto != null) {
        entity.setPermissions(permissions);
        entity.setId(dto.getId());
        if (permissions != null && permissions.getCanRead()) {
            entity.setComponent(dto);
            entity.setBulletins(bulletins);
        }
    }
    return entity;
}