com.hubspot.blazar.zookeeper.BlazarCuratorProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of BlazarService Show documentation
The newest version!
package com.hubspot.blazar.zookeeper;

import com.google.common.base.Preconditions;
import com.hubspot.blazar.config.BlazarConfiguration;
import com.hubspot.blazar.config.ZooKeeperConfiguration;
import io.dropwizard.lifecycle.Managed;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.framework.state.ConnectionStateListener;
import org.apache.curator.retry.ExponentialBackoffRetry;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.inject.Inject;
import javax.inject.Provider;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;

public class BlazarCuratorProvider implements Provider, Managed {
  private static final Logger LOG = LoggerFactory.getLogger(BlazarCuratorProvider.class);

  private final CuratorFramework curatorFramework;
  private final AtomicBoolean started = new AtomicBoolean();
  private final AtomicBoolean stopped = new AtomicBoolean();

  @Inject
  public BlazarCuratorProvider(BlazarConfiguration configuration, Set listeners) {
    ZooKeeperConfiguration zooKeeperConfiguration = configuration.getZooKeeperConfiguration();

    this.curatorFramework = CuratorFrameworkFactory.builder()
        .connectString(zooKeeperConfiguration.getQuorum())
        .sessionTimeoutMs(zooKeeperConfiguration.getSessionTimeoutMillis())
        .connectionTimeoutMs(zooKeeperConfiguration.getConnectTimeoutMillis())
        .retryPolicy(new ExponentialBackoffRetry(zooKeeperConfiguration.getInitialRetryBackoffMillis(), zooKeeperConfiguration.getMaxRetries()))
        .namespace(zooKeeperConfiguration.getNamespace())
        .defaultData(null)
        .build();

    for (ConnectionStateListener listener : listeners) {
      curatorFramework.getConnectionStateListenable().addListener(listener);
    }
  }

  @Override
  public void start() throws Exception {
    if (started.compareAndSet(false, true)) {
      curatorFramework.start();

      long start = System.currentTimeMillis();

      try {
        Preconditions.checkState(curatorFramework.getZookeeperClient().blockUntilConnectedOrTimedOut(), "Could not connect to zookeeper");
      } catch (InterruptedException e) {
        Thread.currentThread().interrupt();
      }

      long end = System.currentTimeMillis();

      LOG.info("Connected to ZooKeeper in {}ms", end - start);
    }
  }

  @Override
  public CuratorFramework get() {
    return curatorFramework;
  }

  @Override
  public void stop() throws Exception {
    if (started.get() && stopped.compareAndSet(false, true)) {
      curatorFramework.close();
    }
  }
}