Examples with HealthResult io.micronaut.management.health.indicator.HealthResult used on opensource projects

Example 1 with HealthResult

use of io.micronaut.management.health.indicator.HealthResult in project micronaut-core by micronaut-projects.

the class HealthResultFilter method doFilterOnce.

@Override
protected Publisher<MutableHttpResponse<?>> doFilterOnce(HttpRequest<?> request, ServerFilterChain chain) {
    return Publishers.map(chain.proceed(request), response -> {
        Object body = response.body();
        if (body instanceof HealthResult) {
            HealthResult healthResult = (HealthResult) body;
            HealthStatus status = healthResult.getStatus();
            HttpStatus httpStatus = healthEndpoint.getStatusConfiguration().getHttpMapping().get(status.getName());
            if (httpStatus != null) {
                response.status(httpStatus);
            } else {
                boolean operational = status.getOperational().orElse(true);
                if (!operational) {
                    response.status(HttpStatus.SERVICE_UNAVAILABLE);
                }
            }
        }
        return response;
    });
}

Example 2 with HealthResult

use of io.micronaut.management.health.indicator.HealthResult in project micronaut-core by micronaut-projects.

the class JdbcIndicator method getResult.

private Publisher<HealthResult> getResult(DataSource dataSource) {
    if (executorService == null) {
        throw new IllegalStateException("I/O ExecutorService is null");
    }
    return new AsyncSingleResultPublisher<>(executorService, () -> {
        Optional<Throwable> throwable = Optional.empty();
        Map<String, Object> details = null;
        String key;
        try (Connection connection = dataSource.getConnection()) {
            if (connection.isValid(CONNECTION_TIMEOUT)) {
                DatabaseMetaData metaData = connection.getMetaData();
                key = metaData.getURL();
                details = new LinkedHashMap<>(1);
                details.put("database", metaData.getDatabaseProductName());
                details.put("version", metaData.getDatabaseProductVersion());
            } else {
                throw new SQLException("Connection was not valid");
            }
        } catch (SQLException e) {
            throwable = Optional.of(e);
            try {
                String url = dataSource.getClass().getMethod("getUrl").invoke(dataSource).toString();
                if (url.startsWith("jdbc:")) {
                    url = url.substring(5);
                }
                url = url.replaceFirst(";", "?");
                url = url.replaceAll(";", "&");
                URI uri = new URI(url);
                key = uri.getHost() + ":" + uri.getPort() + uri.getPath();
            } catch (Exception n) {
                key = dataSource.getClass().getName() + "@" + Integer.toHexString(dataSource.hashCode());
            }
        }
        HealthResult.Builder builder = HealthResult.builder(key);
        if (throwable.isPresent()) {
            builder.exception(throwable.get());
            builder.status(HealthStatus.DOWN);
        } else {
            builder.status(HealthStatus.UP);
            builder.details(details);
        }
        return builder.build();
    });
}

Example 3 with HealthResult

use of io.micronaut.management.health.indicator.HealthResult in project micronaut-elasticsearch by micronaut-projects.

the class ElasticsearchClientHealthIndicator method getResult.

/**
 * Tries to call the cluster info API on Elasticsearch to obtain information about the cluster. If the call succeeds, the Elasticsearch cluster
 * health status (GREEN / YELLOW / RED) will be included in the health indicator details.
 *
 * @return A positive health result UP if the cluster can be communicated with and is in either GREEN or YELLOW status. A negative health result
 * DOWN if the cluster cannot be communicated with or is in RED status.
 */
@Override
public Publisher<HealthResult> getResult() {
    return (subscriber -> {
        final HealthResult.Builder resultBuilder = HealthResult.builder(NAME);
        try {
            client.cluster().health().handle((health, exception) -> {
                if (exception != null) {
                    subscriber.onNext(resultBuilder.status(DOWN).exception(exception).build());
                    subscriber.onComplete();
                } else {
                    HealthStatus status = health.status() == co.elastic.clients.elasticsearch._types.HealthStatus.Red ? DOWN : UP;
                    subscriber.onNext(resultBuilder.status(status).details(health).build());
                    subscriber.onComplete();
                }
                return health;
            });
        } catch (IOException e) {
            subscriber.onNext(resultBuilder.status(DOWN).exception(e).build());
            subscriber.onComplete();
        }
    });
}

Example 4 with HealthResult

use of io.micronaut.management.health.indicator.HealthResult in project micronaut-neo4j by micronaut-projects.

the class Neo4jHealthIndicator method getResult.

@Override
public Publisher<HealthResult> getResult() {
    try {
        Mono<HealthResult> healthResultSingle = Mono.create(emitter -> {
            AsyncSession session = boltDriver.asyncSession();
            CompletionStage<ResultSummary> query = session.writeTransactionAsync(tx -> tx.runAsync("RETURN 1 AS result").thenCompose(ResultCursor::consumeAsync));
            query.handleAsync((resultSummaryStage, throwable) -> {
                if (throwable != null) {
                    return buildErrorResult(throwable);
                } else {
                    HealthResult.Builder status = HealthResult.builder(NAME, HealthStatus.UP);
                    ServerInfo serverInfo = resultSummaryStage.server();
                    status.details(Collections.singletonMap("server", serverInfo.version() + "@" + serverInfo.address()));
                    return status.build();
                }
            }).thenComposeAsync(status -> session.closeAsync().handle((signal, throwable) -> status)).thenAccept(emitter::success);
        });
        return healthResultSingle.subscribeOn(Schedulers.fromExecutorService(ioExecutor));
    } catch (Throwable e) {
        return Mono.just(buildErrorResult(e));
    }
}

Example 5 with HealthResult

use of io.micronaut.management.health.indicator.HealthResult in project micronaut-mongodb by micronaut-projects.

the class MongoHealthIndicator method checkRegisteredMongoClient.

private Publisher<HealthResult> checkRegisteredMongoClient(BeanRegistration<MongoClient> registration) {
    MongoClient mongoClient = registration.getBean();
    String databaseName = "mongodb (" + registration.getIdentifier().getName() + ")";
    Flux<Map<String, String>> databasePings = Flux.from(pingMongo(mongoClient)).map(this::getVersionDetails).timeout(Duration.of(10, ChronoUnit.SECONDS)).retry(3);
    return databasePings.map(detail -> buildStatusUp(databaseName, detail)).onErrorResume(throwable -> Flux.just(buildStatusDown(throwable, databaseName)));
}