Search in sources :

Example 1 with Status

use of org.apache.sling.hc.api.Result.Status in project sling by apache.

the class HealthCheckExecutorServlet method doGet.

protected void doGet(final HttpServletRequest request, final HttpServletResponse response, final String format) throws ServletException, IOException {
    HealthCheckSelector selector = HealthCheckSelector.empty();
    String pathInfo = request.getPathInfo();
    String pathTokensStr = StringUtils.removeStart(splitFormat(pathInfo)[0], "/");
    List<String> tags = new ArrayList<String>();
    List<String> names = new ArrayList<String>();
    if (StringUtils.isNotBlank(pathTokensStr)) {
        String[] pathTokens = pathTokensStr.split(PARAM_SPLIT_REGEX);
        for (String pathToken : pathTokens) {
            if (pathToken.indexOf(' ') >= 0) {
                // token contains space. assume it is a name
                names.add(pathToken);
            } else {
                tags.add(pathToken);
            }
        }
    }
    if (tags.size() == 0) {
        // if not provided via path use parameter or default
        tags = Arrays.asList(StringUtils.defaultIfEmpty(request.getParameter(PARAM_TAGS.name), "").split(PARAM_SPLIT_REGEX));
    }
    selector.withTags(tags.toArray(new String[0]));
    if (names.size() == 0) {
        // if not provided via path use parameter or default
        names = Arrays.asList(StringUtils.defaultIfEmpty(request.getParameter(PARAM_NAMES.name), "").split(PARAM_SPLIT_REGEX));
    }
    selector.withNames(names.toArray(new String[0]));
    final Boolean includeDebug = Boolean.valueOf(request.getParameter(PARAM_INCLUDE_DEBUG.name));
    final Map<Result.Status, Integer> statusMapping = request.getParameter(PARAM_HTTP_STATUS.name) != null ? getStatusMapping(request.getParameter(PARAM_HTTP_STATUS.name)) : null;
    HealthCheckExecutionOptions executionOptions = new HealthCheckExecutionOptions();
    executionOptions.setCombineTagsWithOr(Boolean.valueOf(StringUtils.defaultString(request.getParameter(PARAM_COMBINE_TAGS_WITH_OR.name), "true")));
    executionOptions.setForceInstantExecution(Boolean.valueOf(request.getParameter(PARAM_FORCE_INSTANT_EXECUTION.name)));
    String overrideGlobalTimeoutVal = request.getParameter(PARAM_OVERRIDE_GLOBAL_TIMEOUT.name);
    if (StringUtils.isNumeric(overrideGlobalTimeoutVal)) {
        executionOptions.setOverrideGlobalTimeout(Integer.valueOf(overrideGlobalTimeoutVal));
    }
    List<HealthCheckExecutionResult> executionResults = this.healthCheckExecutor.execute(selector, executionOptions);
    Result.Status mostSevereStatus = Result.Status.DEBUG;
    for (HealthCheckExecutionResult executionResult : executionResults) {
        Status status = executionResult.getHealthCheckResult().getStatus();
        if (status.ordinal() > mostSevereStatus.ordinal()) {
            mostSevereStatus = status;
        }
    }
    Result overallResult = new Result(mostSevereStatus, "Overall status " + mostSevereStatus);
    sendNoCacheHeaders(response);
    if (statusMapping != null) {
        Integer httpStatus = statusMapping.get(overallResult.getStatus());
        response.setStatus(httpStatus);
    }
    if (FORMAT_HTML.equals(format)) {
        sendHtmlResponse(overallResult, executionResults, request, response, includeDebug);
    } else if (FORMAT_JSON.equals(format)) {
        sendJsonResponse(overallResult, executionResults, null, response, includeDebug);
    } else if (FORMAT_JSONP.equals(format)) {
        String jsonpCallback = StringUtils.defaultIfEmpty(request.getParameter(PARAM_JSONP_CALLBACK.name), JSONP_CALLBACK_DEFAULT);
        sendJsonResponse(overallResult, executionResults, jsonpCallback, response, includeDebug);
    } else if (StringUtils.endsWith(format, FORMAT_TXT)) {
        sendTxtResponse(overallResult, response, StringUtils.equals(format, FORMAT_VERBOSE_TXT), executionResults, includeDebug);
    } else {
        response.setContentType("text/plain");
        response.getWriter().println("Invalid format " + format + " - supported formats: html|json|jsonp|txt|verbose.txt");
    }
}
Also used : Status(org.apache.sling.hc.api.Result.Status) ArrayList(java.util.ArrayList) Result(org.apache.sling.hc.api.Result) HealthCheckExecutionResult(org.apache.sling.hc.api.execution.HealthCheckExecutionResult) Status(org.apache.sling.hc.api.Result.Status) HealthCheckSelector(org.apache.sling.hc.api.execution.HealthCheckSelector) HealthCheckExecutionOptions(org.apache.sling.hc.api.execution.HealthCheckExecutionOptions) HealthCheckExecutionResult(org.apache.sling.hc.api.execution.HealthCheckExecutionResult)

Aggregations

ArrayList (java.util.ArrayList)1 Result (org.apache.sling.hc.api.Result)1 Status (org.apache.sling.hc.api.Result.Status)1 HealthCheckExecutionOptions (org.apache.sling.hc.api.execution.HealthCheckExecutionOptions)1 HealthCheckExecutionResult (org.apache.sling.hc.api.execution.HealthCheckExecutionResult)1 HealthCheckSelector (org.apache.sling.hc.api.execution.HealthCheckSelector)1