com.datastax.oss.simulacron.common.cluster.NodePropertiesBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of simulacron-common Show documentation
There is a newer version: 0.12.0
/*
 * Copyright (C) 2017-2017 DataStax Inc.
 *
 * 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.datastax.oss.simulacron.common.cluster;

import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

/**
 * A base builder to use when defining a {@link NodeProperties} implementation that provides builder
 * methods for version, name, id, and peer info. Also provides a {@link #copy(NodeProperties)}
 * method for deriving {@link NodeProperties} values from an existing object.
 *
 * @param  The concrete type of the builder.
 * @param  The type of properties being created.
 */
public abstract class NodePropertiesBuilder<
    S extends NodePropertiesBuilder, N extends NodeProperties> {

  N parent;
  private final S myself;
  String cassandraVersion;
  String dseVersion;
  String name;
  Long id;
  UUID hostId;
  Map peerInfo = new HashMap<>();

  NodePropertiesBuilder(Class selfType) {
    this(selfType, null);
  }

  @SuppressWarnings("unchecked")
  NodePropertiesBuilder(Class selfType, N parent) {
    this.myself = (S) selfType.cast(this);
    this.parent = parent;
  }

  /**
   * Copies properties from a previously built {@link NodeProperties} into this. This is useful for
   * when we are creating a new copy that has new fields set like ID for example.
   *
   * Note that this doesn't copy values like children or environment specific properties (like
   * address).
   *
   * @param toCopy Existing {@link NodeProperties} toCopy.
   * @return Updated builder with copied properties.
   */
  public S copy(NodeProperties toCopy) {
    return this.withCassandraVersion(toCopy.getCassandraVersion())
        .withDSEVersion(toCopy.getDSEVersion())
        .withName(toCopy.getName())
        .withId(toCopy.getId())
        .withHostId(toCopy.getHostId())
        .withPeerInfo(toCopy.getPeerInfo());
  }

  public S withCassandraVersion(String cassandraVersion) {
    this.cassandraVersion = cassandraVersion;
    return myself;
  }

  public S withDSEVersion(String dseVersion) {
    this.dseVersion = dseVersion;
    return myself;
  }

  public S withName(String name) {
    this.name = name;
    return myself;
  }

  public S withId(Long id) {
    this.id = id;
    return myself;
  }

  public S withHostId(UUID hostId) {
    this.hostId = hostId;
    return myself;
  }

  public S withPeerInfo(String key, Object value) {
    peerInfo.put(key, value);
    return myself;
  }

  public S withPeerInfo(Map peerInfo) {
    this.peerInfo = new HashMap<>(peerInfo);
    return myself;
  }
}