com.hubspot.baragon.agent.healthcheck.ZooKeeperHealthcheck Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of BaragonAgentService Show documentation

There is a newer version: 0.6.2

package com.hubspot.baragon.agent.healthcheck;

import java.util.concurrent.atomic.AtomicReference;

import com.google.inject.Inject;
import com.google.inject.name.Named;
import com.hubspot.baragon.BaragonDataModule;
import com.hubspot.dropwizard.guice.InjectableHealthCheck;
import org.apache.curator.framework.state.ConnectionState;

public class ZooKeeperHealthcheck extends InjectableHealthCheck {
  private final AtomicReference connectionState;

  @Inject
  public ZooKeeperHealthcheck(@Named(BaragonDataModule.BARAGON_ZK_CONNECTION_STATE) AtomicReference connectionState) {
    this.connectionState = connectionState;
  }

  @Override
  public String getName() {
    return "zookeeper";
  }

  @Override
  protected Result check() throws Exception {
    final ConnectionState currentConnectionState = connectionState.get();

    if (currentConnectionState == null) {
      return Result.unhealthy("Connection state is null");
    }

    switch (currentConnectionState) {
      case CONNECTED:
      case RECONNECTED:
        return Result.healthy(currentConnectionState.name());
      default:
        return Result.unhealthy(currentConnectionState.name());
    }
  }
}