org.zodiac.actuate.zookeeper.ZookeeperHealthIndicator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of zodiac-actuator Show documentation

Zodiac Actuator

There is a newer version: 1.6.8

package org.zodiac.actuate.zookeeper;

import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.imps.CuratorFrameworkState;

import org.springframework.boot.actuate.health.AbstractHealthIndicator;
import org.springframework.boot.actuate.health.Health;

public class ZookeeperHealthIndicator extends AbstractHealthIndicator {

    private final CuratorFramework curator;

    public ZookeeperHealthIndicator(CuratorFramework curator) {
        this.curator = curator;
    }

    @Override
    protected void doHealthCheck(Health.Builder builder) throws Exception {
        try {
            CuratorFrameworkState state = this.curator.getState();
            if (state != CuratorFrameworkState.STARTED) {
                builder.down().withDetail("error", "Client not started");
            } else if (this.curator.checkExists().forPath("/") == null) {
                builder.down().withDetail("error", "Root for namespace does not exist");
            } else {
                builder.up();
            }
            builder.withDetail("connectionString", this.curator.getZookeeperClient().getCurrentConnectionString())
                .withDetail("state", state);
        } catch (Exception e) {
            builder.down(e);
        }
    }

}