com.smoketurner.notification.application.config.RiakConfiguration Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of notification-application Show documentation
There is a newer version: 2.0.0
Show newest version
/**
 * Copyright 2015 Smoke Turner, LLC.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
 * in compliance with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing permissions and limitations under
 * the License.
 */
package com.smoketurner.notification.application.config;

import io.dropwizard.setup.Environment;
import io.dropwizard.util.Duration;
import io.dropwizard.validation.MinDuration;
import java.net.UnknownHostException;
import java.security.KeyStore;
import java.security.KeyStoreException;
import java.util.List;
import java.util.concurrent.TimeUnit;
import javax.annotation.Nonnull;
import javax.validation.constraints.Min;
import javax.validation.constraints.NotNull;
import org.hibernate.validator.constraints.NotEmpty;
import com.basho.riak.client.core.RiakCluster;
import com.basho.riak.client.core.RiakNode;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.google.common.base.Preconditions;
import com.google.common.base.Strings;
import com.google.common.collect.Lists;
import com.google.common.net.HostAndPort;
import com.smoketurner.notification.application.managed.RiakClusterManager;

public class RiakConfiguration {

  private static final int DEFAULT_RIAK_PORT = 8087;

  @NotEmpty
  private List nodes = Lists.newArrayList();

  private String username;

  private String password;

  private String keyStore;

  @Min(1)
  @NotNull
  private Integer minConnections = 1;

  @Min(0)
  @NotNull
  private Integer maxConnections = 0;

  @Min(0)
  @NotNull
  private Integer executionAttempts = 3;

  @NotNull
  @MinDuration(value = 1, unit = TimeUnit.MILLISECONDS)
  private Duration idleTimeout = Duration.milliseconds(1000);

  @NotNull
  @MinDuration(value = 0, unit = TimeUnit.MILLISECONDS)
  private Duration connectionTimeout = Duration.milliseconds(0);

  @JsonProperty
  public List getNodes() {
    return nodes;
  }

  @JsonProperty
  public void setNodes(final List nodes) {
    this.nodes = nodes;
  }

  @JsonProperty
  public String getUsername() {
    return username;
  }

  @JsonProperty
  public void setUsername(final String username) {
    this.username = username;
  }

  @JsonProperty
  public String getPassword() {
    return password;
  }

  @JsonProperty
  public void setPassword(final String password) {
    this.password = password;
  }

  @JsonProperty
  public String getKeyStore() {
    return keyStore;
  }

  @JsonProperty
  public void setKeyStore(final String keyStore) {
    this.keyStore = keyStore;
  }

  @JsonProperty
  public Integer getMinConnections() {
    return minConnections;
  }

  @JsonProperty
  public void setMinConnections(final Integer connections) {
    this.minConnections = connections;
  }

  @JsonProperty
  public Integer getMaxConnections() {
    return maxConnections;
  }

  @JsonProperty
  public void setMaxConnections(final Integer connections) {
    this.maxConnections = connections;
  }

  @JsonProperty
  public Integer getExecutionAttempts() {
    return executionAttempts;
  }

  @JsonProperty
  public void setExecutionAttempts(final Integer attempts) {
    this.executionAttempts = attempts;
  }

  @JsonProperty
  public Duration getIdleTimeout() {
    return idleTimeout;
  }

  @JsonProperty
  public void setIdleTimeout(final Duration timeout) {
    this.idleTimeout = timeout;
  }

  @JsonProperty
  public Duration getConnectionTimeout() {
    return connectionTimeout;
  }

  @JsonProperty
  public void setConnectionTimeout(final Duration timeout) {
    this.connectionTimeout = timeout;
  }

  @JsonIgnore
  public RiakCluster build(@Nonnull final Environment environment) throws UnknownHostException,
      KeyStoreException {
    Preconditions.checkNotNull(environment);

    final RiakNode.Builder builder =
        new RiakNode.Builder().withMinConnections(minConnections)
            .withConnectionTimeout((int) connectionTimeout.toMilliseconds())
            .withIdleTimeout((int) idleTimeout.toMilliseconds()).withBlockOnMaxConnections(false);
    if (maxConnections > 0) {
      builder.withMaxConnections(maxConnections);
    }
    if (!Strings.isNullOrEmpty(username) && !Strings.isNullOrEmpty(password)
        && !Strings.isNullOrEmpty(keyStore)) {
      // TODO finish keyStore implementation
      final KeyStore keystore = KeyStore.getInstance("PKS");
      builder.withAuth(username, password, keystore);
    }

    final List nodes = Lists.newArrayList();
    for (HostAndPort address : this.nodes) {
      final RiakNode node =
          builder.withRemoteAddress(address.getHostText())
              .withRemotePort(address.getPortOrDefault(DEFAULT_RIAK_PORT)).build();
      nodes.add(node);
    }

    final RiakCluster cluster =
        RiakCluster.builder(nodes).withExecutionAttempts(executionAttempts).build();
    environment.lifecycle().manage(new RiakClusterManager(cluster));
    return cluster;
  }
}