io.github.devlibx.easy.app.dropwizard.healthcheck.ApplicationHealthCheck Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dropwizard-app Show documentation

There is a newer version: 3.java-17.1

package io.github.devlibx.easy.app.dropwizard.healthcheck;

import com.codahale.metrics.health.HealthCheck;
import io.dropwizard.setup.Environment;
import io.gitbub.devlibx.easy.helper.healthcheck.IHealthCheckProvider;
import io.gitbub.devlibx.easy.helper.string.StringHelper;
import lombok.Builder;
import lombok.Data;
import lombok.Getter;
import lombok.extern.slf4j.Slf4j;

import javax.inject.Inject;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@Slf4j
public class ApplicationHealthCheck {
    private final Map summarizers;
    private final StringHelper stringHelper;

    @Inject
    public ApplicationHealthCheck(Map summarizers, StringHelper stringHelper) {
        this.summarizers = summarizers;
        this.stringHelper = stringHelper;
    }

    public void setupHealthChecks(Environment environment) {
        healthChecks().forEach(healthCheckHolder -> {
            environment.healthChecks().register(healthCheckHolder.name, healthCheckHolder.healthCheck);
        });
    }

    private List healthChecks() {
        List list = new ArrayList<>();
        summarizers.forEach((name, healthCheckProvider) -> {
            HealthCheck healthCheck = new HealthCheck() {
                @Override
                protected Result check() throws Exception {
                    IHealthCheckProvider.Result result = healthCheckProvider.check();

                    ResultBuilder rb = Result.builder();
                    if (result.getDetails() != null) {
                        rb.withDetail("details", result.getDetails());
                    }
                    if (result.isHealthy()) {
                        rb.healthy().withMessage(result.getMessage());
                    } else {
                        rb.unhealthy().withMessage(result.getMessage());
                    }
                    return rb.build();
                }
            };
            list.add(
                    HealthCheckHolder.builder()
                            .healthCheck(healthCheck)
                            .name(name)
                            .build()
            );
        });

        return list;
    }

    @Data
    @Getter
    @Builder
    public static class HealthCheckHolder {
        private String name;
        private HealthCheck healthCheck;
    }
}