Examples with InstanceStatus org.apache.nifi.minifi.commons.status.instance.InstanceStatus used on opensource projects

Example 1 with InstanceStatus

use of org.apache.nifi.minifi.commons.status.instance.InstanceStatus in project nifi-minifi by apache.

the class StatusConfigReporter method getStatus.

public static FlowStatusReport getStatus(FlowController flowController, String statusRequest, Logger logger) throws StatusRequestException {
    if (statusRequest == null) {
        logger.error("Received a status request which was null");
        throw new StatusRequestException("Cannot complete status request because the statusRequest is null");
    }
    if (flowController == null) {
        logger.error("Received a status but the Flow Controller is null");
        throw new StatusRequestException("Cannot complete status request because the Flow Controller is null");
    }
    FlowStatusReport flowStatusReport = new FlowStatusReport();
    List<String> errorsGeneratingReport = new LinkedList<>();
    flowStatusReport.setErrorsGeneratingReport(errorsGeneratingReport);
    String[] itemsToReport = statusRequest.split(";");
    ProcessGroupStatus rootGroupStatus = flowController.getControllerStatus();
    Map<String, ProcessorStatus> processorStatusMap = null;
    Map<String, ConnectionStatus> connectionStatusMap = null;
    Map<String, RemoteProcessGroupStatus> remoteProcessGroupStatusMap = null;
    for (String item : itemsToReport) {
        String[] sections = item.split(":");
        try {
            switch(sections[0].toLowerCase().trim()) {
                case "systemdiagnostics":
                    SystemDiagnosticsStatus systemDiagnosticsStatus = parseSystemDiagnosticsRequest(flowController.getSystemDiagnostics(), sections[1]);
                    flowStatusReport.setSystemDiagnosticsStatus(systemDiagnosticsStatus);
                    break;
                case "instance":
                    InstanceStatus instanceStatus = parseInstanceRequest(sections[1], flowController, rootGroupStatus);
                    flowStatusReport.setInstanceStatus(instanceStatus);
                    break;
                case "remoteprocessgroup":
                    if (flowStatusReport.getRemoteProcessGroupStatusList() == null) {
                        List<RemoteProcessGroupStatusBean> remoteProcessGroupStatusList = new LinkedList<>();
                        flowStatusReport.setRemoteProcessGroupStatusList(remoteProcessGroupStatusList);
                    }
                    handleRemoteProcessGroupRequest(sections, rootGroupStatus, flowController, flowStatusReport.getRemoteProcessGroupStatusList(), remoteProcessGroupStatusMap, logger);
                    break;
                case "processor":
                    if (flowStatusReport.getProcessorStatusList() == null) {
                        List<ProcessorStatusBean> processorStatusList = new LinkedList<>();
                        flowStatusReport.setProcessorStatusList(processorStatusList);
                    }
                    handleProcessorRequest(sections, rootGroupStatus, flowController, flowStatusReport.getProcessorStatusList(), processorStatusMap, logger);
                    break;
                case "connection":
                    if (flowStatusReport.getConnectionStatusList() == null) {
                        List<ConnectionStatusBean> connectionStatusList = new LinkedList<>();
                        flowStatusReport.setConnectionStatusList(connectionStatusList);
                    }
                    handleConnectionRequest(sections, rootGroupStatus, flowStatusReport.getConnectionStatusList(), connectionStatusMap, logger);
                    break;
                case "provenancereporting":
                    if (flowStatusReport.getRemoteProcessGroupStatusList() == null) {
                        List<ReportingTaskStatus> reportingTaskStatusList = new LinkedList<>();
                        flowStatusReport.setReportingTaskStatusList(reportingTaskStatusList);
                    }
                    handleReportingTaskRequest(sections, flowController, flowStatusReport.getReportingTaskStatusList(), logger);
                    break;
                case "controllerservices":
                    if (flowStatusReport.getControllerServiceStatusList() == null) {
                        List<ControllerServiceStatus> controllerServiceStatusList = new LinkedList<>();
                        flowStatusReport.setControllerServiceStatusList(controllerServiceStatusList);
                    }
                    handleControllerServices(sections, flowController, flowStatusReport.getControllerServiceStatusList(), logger);
                    break;
            }
        } catch (Exception e) {
            logger.error("Hit exception while requesting status for item '" + item + "'", e);
            errorsGeneratingReport.add("Unable to get status for request '" + item + "' due to:" + e);
        }
    }
    return flowStatusReport;
}

Example 2 with InstanceStatus

use of org.apache.nifi.minifi.commons.status.instance.InstanceStatus in project nifi-minifi by apache.

the class StatusReportPopulator method addInstanceStatus.

public static void addInstanceStatus(FlowStatusReport flowStatusReport, boolean addHealth, boolean addStats, boolean addBulletins, boolean populateBulletins) {
    InstanceStatus instanceStatus = new InstanceStatus();
    if (addHealth) {
        InstanceHealth instanceHealth = new InstanceHealth();
        instanceHealth.setQueuedContentSize(1L);
        instanceHealth.setQueuedCount(2);
        instanceHealth.setActiveThreads(3);
        instanceHealth.setHasBulletins(populateBulletins);
        instanceStatus.setInstanceHealth(instanceHealth);
    }
    if (addBulletins) {
        if (populateBulletins) {
            BulletinStatus bulletinStatus = new BulletinStatus();
            bulletinStatus.setMessage("Bulletin message");
            bulletinStatus.setTimestamp(new Date(1464019245000L));
            instanceStatus.setBulletinList(Collections.singletonList(bulletinStatus));
        } else {
            instanceStatus.setBulletinList(Collections.EMPTY_LIST);
        }
    }
    if (addStats) {
        InstanceStats instanceStats = new InstanceStats();
        instanceStats.setBytesRead(1L);
        instanceStats.setBytesWritten(2L);
        instanceStats.setBytesSent(3L);
        instanceStats.setFlowfilesSent(4);
        instanceStats.setBytesTransferred(5L);
        instanceStats.setFlowfilesTransferred(6);
        instanceStats.setBytesReceived(7L);
        instanceStats.setFlowfilesReceived(8);
        instanceStatus.setInstanceStats(instanceStats);
    }
    flowStatusReport.setInstanceStatus(instanceStatus);
}

Example 3 with InstanceStatus

use of org.apache.nifi.minifi.commons.status.instance.InstanceStatus in project nifi-minifi by apache.

the class StatusRequestParser method parseInstanceRequest.

static InstanceStatus parseInstanceRequest(String statusTypes, FlowController flowController, ProcessGroupStatus rootGroupStatus) {
    InstanceStatus instanceStatus = new InstanceStatus();
    flowController.getAllControllerServices();
    List<Bulletin> bulletinList = flowController.getBulletinRepository().findBulletinsForController();
    String[] statusSplits = statusTypes.split(",");
    for (String statusType : statusSplits) {
        switch(statusType.toLowerCase().trim()) {
            case "health":
                InstanceHealth instanceHealth = new InstanceHealth();
                instanceHealth.setQueuedCount(rootGroupStatus.getQueuedCount());
                instanceHealth.setQueuedContentSize(rootGroupStatus.getQueuedContentSize());
                instanceHealth.setHasBulletins(!bulletinList.isEmpty());
                instanceHealth.setActiveThreads(rootGroupStatus.getActiveThreadCount());
                instanceStatus.setInstanceHealth(instanceHealth);
                break;
            case "bulletins":
                instanceStatus.setBulletinList(transformBulletins(flowController.getBulletinRepository().findBulletinsForController()));
                break;
            case "stats":
                InstanceStats instanceStats = new InstanceStats();
                instanceStats.setBytesRead(rootGroupStatus.getBytesRead());
                instanceStats.setBytesWritten(rootGroupStatus.getBytesWritten());
                instanceStats.setBytesSent(rootGroupStatus.getBytesSent());
                instanceStats.setFlowfilesSent(rootGroupStatus.getFlowFilesSent());
                instanceStats.setBytesTransferred(rootGroupStatus.getBytesTransferred());
                instanceStats.setFlowfilesTransferred(rootGroupStatus.getFlowFilesTransferred());
                instanceStats.setBytesReceived(rootGroupStatus.getBytesReceived());
                instanceStats.setFlowfilesReceived(rootGroupStatus.getFlowFilesReceived());
                instanceStatus.setInstanceStats(instanceStats);
                break;
        }
    }
    return instanceStatus;
}