cluster_config.h

// Copyright 2018 Google 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
//
//     https://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.
 
#ifndef GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGTABLE_CLUSTER_CONFIG_H
#define GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGTABLE_CLUSTER_CONFIG_H
 
#include "google/cloud/bigtable/version.h"
#include <google/bigtable/admin/v2/bigtable_instance_admin.pb.h>
#include <string>
 
namespace google {
namespace cloud {
namespace bigtable {
GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_BEGIN
 
/// Specify the initial configuration for a new cluster.
class ClusterConfig {
 public:
  using StorageType = ::google::bigtable::admin::v2::StorageType;
  // NOLINTNEXTLINE(readability-identifier-naming)
  constexpr static StorageType STORAGE_TYPE_UNSPECIFIED =
      google::bigtable::admin::v2::STORAGE_TYPE_UNSPECIFIED;
  // NOLINTNEXTLINE(readability-identifier-naming)
  constexpr static StorageType SSD = google::bigtable::admin::v2::SSD;
  // NOLINTNEXTLINE(readability-identifier-naming)
  constexpr static StorageType HDD = google::bigtable::admin::v2::HDD;
 
  // NOLINTNEXTLINE(google-explicit-constructor)
  ClusterConfig(google::bigtable::admin::v2::Cluster cluster)
      : proto_(std::move(cluster)) {}
 
  ClusterConfig(std::string location, std::int32_t serve_nodes,
                StorageType storage) {
    proto_.set_location(std::move(location));
    proto_.set_serve_nodes(serve_nodes);
    proto_.set_default_storage_type(storage);
  }
 
  ClusterConfig& SetEncryptionConfig(
      google::bigtable::admin::v2::Cluster::EncryptionConfig encryption) & {
    *proto_.mutable_encryption_config() = std::move(encryption);
    return *this;
  }
  ClusterConfig&& SetEncryptionConfig(
      google::bigtable::admin::v2::Cluster::EncryptionConfig encryption) && {
    return std::move(SetEncryptionConfig(std::move(encryption)));
  }
 
  std::string const& GetName() { return proto_.name(); }
 
  google::bigtable::admin::v2::Cluster const& as_proto() const& {
    return proto_;
  }
 
  google::bigtable::admin::v2::Cluster&& as_proto() && {
    return std::move(proto_);
  }
 
 private:
  google::bigtable::admin::v2::Cluster proto_;
};
 
GOOGLE_CLOUD_CPP_INLINE_NAMESPACE_END
}  // namespace bigtable
}  // namespace cloud
}  // namespace google
 
#endif  // GOOGLE_CLOUD_CPP_GOOGLE_CLOUD_BIGTABLE_CLUSTER_CONFIG_H