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

Example 6 with StatusHistoryDTO

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

the class StatusHistoryEndpointMerger method merge.

@Override
public NodeResponse merge(URI uri, String method, Set<NodeResponse> successfulResponses, Set<NodeResponse> problematicResponses, NodeResponse clientResponse) {
    final Map<String, MetricDescriptor<?>> metricDescriptors = getStandardMetricDescriptors(uri);
    final StatusHistoryEntity responseEntity = clientResponse.getClientResponse().readEntity(StatusHistoryEntity.class);
    final Set<StatusDescriptorDTO> fieldDescriptors = new LinkedHashSet<>();
    boolean includeCounters = true;
    StatusHistoryDTO lastStatusHistory = null;
    final List<NodeStatusSnapshotsDTO> nodeStatusSnapshots = new ArrayList<>(successfulResponses.size());
    LinkedHashMap<String, String> noReadPermissionsComponentDetails = null;
    for (final NodeResponse nodeResponse : successfulResponses) {
        final StatusHistoryEntity nodeResponseEntity = nodeResponse == clientResponse ? responseEntity : nodeResponse.getClientResponse().readEntity(StatusHistoryEntity.class);
        final StatusHistoryDTO nodeStatus = nodeResponseEntity.getStatusHistory();
        lastStatusHistory = nodeStatus;
        if (noReadPermissionsComponentDetails == null && !nodeResponseEntity.getCanRead()) {
            // If component details from a history with no read permissions is encountered for the first time, hold on to them to be used in the merged response
            noReadPermissionsComponentDetails = nodeStatus.getComponentDetails();
        }
        if (!Boolean.TRUE.equals(nodeResponseEntity.getCanRead())) {
            includeCounters = false;
        }
        final NodeIdentifier nodeId = nodeResponse.getNodeId();
        final NodeStatusSnapshotsDTO nodeStatusSnapshot = new NodeStatusSnapshotsDTO();
        nodeStatusSnapshot.setNodeId(nodeId.getId());
        nodeStatusSnapshot.setAddress(nodeId.getApiAddress());
        nodeStatusSnapshot.setApiPort(nodeId.getApiPort());
        nodeStatusSnapshot.setStatusSnapshots(nodeStatus.getAggregateSnapshots());
        nodeStatusSnapshots.add(nodeStatusSnapshot);
        final List<StatusDescriptorDTO> descriptors = nodeStatus.getFieldDescriptors();
        if (descriptors != null) {
            fieldDescriptors.addAll(descriptors);
        }
    }
    // the user is not authorized, we want to assume that the user is, in fact, not authorized.
    if (includeCounters) {
        for (final StatusDescriptorDTO descriptorDto : fieldDescriptors) {
            final String fieldName = descriptorDto.getField();
            if (!metricDescriptors.containsKey(fieldName)) {
                final ValueMapper<ProcessorStatus> valueMapper = s -> {
                    final Map<String, Long> counters = s.getCounters();
                    if (counters == null) {
                        return 0L;
                    }
                    return counters.getOrDefault(descriptorDto.getField(), 0L);
                };
                final MetricDescriptor<ProcessorStatus> metricDescriptor = new StandardMetricDescriptor<>(descriptorDto.getField(), descriptorDto.getLabel(), descriptorDto.getDescription(), Formatter.COUNT, valueMapper);
                metricDescriptors.put(fieldName, metricDescriptor);
            }
        }
    }
    final StatusHistoryDTO clusterStatusHistory = new StatusHistoryDTO();
    clusterStatusHistory.setAggregateSnapshots(mergeStatusHistories(nodeStatusSnapshots, metricDescriptors));
    clusterStatusHistory.setGenerated(new Date());
    clusterStatusHistory.setNodeSnapshots(nodeStatusSnapshots);
    if (lastStatusHistory != null) {
        clusterStatusHistory.setComponentDetails(noReadPermissionsComponentDetails == null ? lastStatusHistory.getComponentDetails() : noReadPermissionsComponentDetails);
    }
    clusterStatusHistory.setFieldDescriptors(new ArrayList<>(fieldDescriptors));
    final StatusHistoryEntity clusterEntity = new StatusHistoryEntity();
    clusterEntity.setStatusHistory(clusterStatusHistory);
    clusterEntity.setCanRead(noReadPermissionsComponentDetails == null);
    return new NodeResponse(clientResponse, clusterEntity);
}

Example 7 with StatusHistoryDTO

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

the class StatusHistoryUtil method createStatusHistoryDTO.

public static StatusHistoryDTO createStatusHistoryDTO(final StatusHistory statusHistory) {
    final List<StatusSnapshotDTO> snapshotDtos = new ArrayList<>();
    final Set<MetricDescriptor<?>> metricDescriptors = new LinkedHashSet<>();
    final LinkedHashMap<String, String> componentDetails = new LinkedHashMap<>(statusHistory.getComponentDetails());
    final Set<String> metricNames = new HashSet<>();
    for (final StatusSnapshot snapshot : statusHistory.getStatusSnapshots()) {
        final StatusSnapshotDTO snapshotDto = StatusHistoryUtil.createStatusSnapshotDto(snapshot);
        snapshotDtos.add(snapshotDto);
        metricNames.addAll(snapshotDto.getStatusMetrics().keySet());
        metricDescriptors.addAll(snapshot.getStatusMetrics().keySet());
    }
    // So for any metric that has is not in the aggregate snapshot, add it with a value of 0
    for (final StatusSnapshotDTO snapshotDto : snapshotDtos) {
        final Map<String, Long> metrics = snapshotDto.getStatusMetrics();
        for (final String metricName : metricNames) {
            if (!metrics.containsKey(metricName)) {
                metrics.put(metricName, 0L);
            }
        }
    }
    final StatusHistoryDTO dto = new StatusHistoryDTO();
    dto.setGenerated(new Date());
    dto.setComponentDetails(componentDetails);
    dto.setFieldDescriptors(StatusHistoryUtil.createFieldDescriptorDtos(metricDescriptors));
    dto.setAggregateSnapshots(snapshotDtos);
    return dto;
}

Example 8 with StatusHistoryDTO

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

the class StandardNiFiServiceFacade method getRemoteProcessGroupStatusHistory.

@Override
public StatusHistoryEntity getRemoteProcessGroupStatusHistory(final String id) {
    final RemoteProcessGroup remoteProcessGroup = remoteProcessGroupDAO.getRemoteProcessGroup(id);
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(remoteProcessGroup);
    final StatusHistoryDTO dto = controllerFacade.getRemoteProcessGroupStatusHistory(id);
    return entityFactory.createStatusHistoryEntity(dto, permissions);
}

Example 9 with StatusHistoryDTO

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

the class StandardNiFiServiceFacade method getConnectionStatusHistory.

@Override
public StatusHistoryEntity getConnectionStatusHistory(final String connectionId) {
    final Connection connection = connectionDAO.getConnection(connectionId);
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(connection);
    final StatusHistoryDTO dto = controllerFacade.getConnectionStatusHistory(connectionId);
    return entityFactory.createStatusHistoryEntity(dto, permissions);
}

Example 10 with StatusHistoryDTO

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

the class ControllerFacade method getConnectionStatusHistory.

/**
 * Returns the status history for the specified connection.
 *
 * @param connectionId connection id
 * @return status history
 */
public StatusHistoryDTO getConnectionStatusHistory(final String connectionId) {
    final ProcessGroup root = flowController.getGroup(flowController.getRootGroupId());
    final Connection connection = root.findConnection(connectionId);
    // ensure the connection was found
    if (connection == null) {
        throw new ResourceNotFoundException(String.format("Unable to locate connection with id '%s'.", connectionId));
    }
    final StatusHistoryDTO statusHistory = flowController.getConnectionStatusHistory(connectionId);
    // if not authorized
    if (!connection.isAuthorized(authorizer, RequestAction.READ, NiFiUserUtils.getNiFiUser())) {
        statusHistory.getComponentDetails().put(ComponentStatusRepository.COMPONENT_DETAIL_NAME, connectionId);
        statusHistory.getComponentDetails().put(ComponentStatusRepository.COMPONENT_DETAIL_SOURCE_NAME, connection.getSource().getIdentifier());
        statusHistory.getComponentDetails().put(ComponentStatusRepository.COMPONENT_DETAIL_DESTINATION_NAME, connection.getDestination().getIdentifier());
    }
    return statusHistory;
}