kz.greetgo.logging.zookeeper.core.ZookeeperConnectParams Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of greetgo-logging-zookeeper Show documentation
Logging mechanism using in greetgo!
There is a newer version: 0.2.0
Show newest version
package kz.greetgo.logging.zookeeper.core;

import org.apache.curator.framework.AuthInfo;
import org.apache.curator.framework.CuratorFramework;
import org.apache.curator.framework.CuratorFrameworkFactory;
import org.apache.curator.retry.RetryNTimes;
import org.apache.zookeeper.client.ZKClientConfig;

import java.util.List;
import java.util.Map;
import java.util.Objects;
import java.util.function.IntSupplier;
import java.util.function.Supplier;

@SuppressWarnings("unused")
public class ZookeeperConnectParams {

  private IntSupplier sessionTimeoutMillis = () -> 3000;
  private IntSupplier connectTimeoutMillis = () -> 25000;
  private IntSupplier maxRetries = () -> 3;
  private IntSupplier sleepBetweenRetriesMillis = () -> 100;
  private Supplier connectStr = null;

  private IntSupplier retriesOnSessionExpiredException = () -> 3;

  private Supplier> config = Map::of;

  private Supplier> authInfoList = List::of;

  private void check() {
    if (connectStr == null) {
      throw new RuntimeException("1H2c3x3vAb :: Not specified connectStr. Value example: `localhost:2181`");
    }
  }

  public ZookeeperConnectParams config(Supplier> config) {
    this.config = config;
    return this;
  }

  public ZookeeperConnectParams authInfoList(Supplier> authInfoList) {
    this.authInfoList = authInfoList;
    return this;
  }

  public ZookeeperConnectParams maxRetries(IntSupplier maxRetries) {
    this.maxRetries = Objects.requireNonNull(maxRetries);
    return this;
  }

  public ZookeeperConnectParams retriesOnSessionExpiredException(IntSupplier retriesOnSessionExpiredException) {
    this.retriesOnSessionExpiredException = Objects.requireNonNull(retriesOnSessionExpiredException);
    return this;
  }

  public int getRetriesOnSessionExpiredException() {
    return retriesOnSessionExpiredException.getAsInt();
  }

  public ZookeeperConnectParams maxRetries(int maxRetries) {
    this.maxRetries = () -> maxRetries;
    return this;
  }

  public ZookeeperConnectParams sleepBetweenRetriesMillis(IntSupplier sleepBetweenRetriesMillis) {
    this.sleepBetweenRetriesMillis = Objects.requireNonNull(sleepBetweenRetriesMillis);
    return this;
  }

  public ZookeeperConnectParams sleepBetweenRetriesMillis(int sleepBetweenRetriesMillis) {
    this.sleepBetweenRetriesMillis = () -> sleepBetweenRetriesMillis;
    return this;
  }

  public ZookeeperConnectParams connectTimeoutMillis(IntSupplier connectTimeoutMillis) {
    this.connectTimeoutMillis = Objects.requireNonNull(connectTimeoutMillis);
    return this;
  }

  public ZookeeperConnectParams connectTimeoutMillis(int connectTimeoutMillis) {
    this.connectTimeoutMillis = () -> connectTimeoutMillis;
    return this;
  }

  public ZookeeperConnectParams sessionTimeoutMillis(IntSupplier sessionTimeoutMillis) {
    this.sessionTimeoutMillis = Objects.requireNonNull(sessionTimeoutMillis);
    return this;
  }

  public ZookeeperConnectParams sessionTimeoutMillis(int sessionTimeoutMillis) {
    this.sessionTimeoutMillis = () -> sessionTimeoutMillis;
    return this;
  }

  public ZookeeperConnectParams connectStr(Supplier connectStr) {
    this.connectStr = Objects.requireNonNull(connectStr);
    return this;
  }

  public ZookeeperConnectParams connectStr(String connectStr) {
    Objects.requireNonNull(connectStr);
    this.connectStr = () -> connectStr;
    return this;
  }

  private ZKClientConfig createZkConfig() {
    ZKClientConfig ret = new ZKClientConfig();
    config.get().forEach(ret::setProperty);
    return ret;
  }

  public CuratorFramework createClient() {
    check();

    var retryPolicy = new RetryNTimes(maxRetries.getAsInt(), sleepBetweenRetriesMillis.getAsInt());

    return CuratorFrameworkFactory.builder()
                                  .connectString(connectStr.get())
                                  .sessionTimeoutMs(sessionTimeoutMillis.getAsInt())
                                  .connectionTimeoutMs(connectTimeoutMillis.getAsInt())
                                  .retryPolicy(retryPolicy)
                                  .authorization(authInfoList.get())
                                  .zkClientConfig(createZkConfig())
                                  .build();
  }
}