ClusterInstanceGroupDetails.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/ActiveClusterOperationName.h>
#include <aws/sagemaker/model/ClusterCapacityRequirements.h>
#include <aws/sagemaker/model/ClusterInstanceStorageConfig.h>
#include <aws/sagemaker/model/ClusterInstanceType.h>
#include <aws/sagemaker/model/ClusterKubernetesConfigDetails.h>
#include <aws/sagemaker/model/ClusterLifeCycleConfig.h>
#include <aws/sagemaker/model/ClusterSlurmConfigDetails.h>
#include <aws/sagemaker/model/DeepHealthCheckType.h>
#include <aws/sagemaker/model/DeploymentConfiguration.h>
#include <aws/sagemaker/model/InstanceGroupStatus.h>
#include <aws/sagemaker/model/ScheduledUpdateConfig.h>
#include <aws/sagemaker/model/SoftwareUpdateStatus.h>
#include <aws/sagemaker/model/VpcConfig.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class ClusterInstanceGroupDetails {
 public:
  AWS_SAGEMAKER_API ClusterInstanceGroupDetails() = default;
  AWS_SAGEMAKER_API ClusterInstanceGroupDetails(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API ClusterInstanceGroupDetails& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline int GetCurrentCount() const { return m_currentCount; }
  inline bool CurrentCountHasBeenSet() const { return m_currentCountHasBeenSet; }
  inline void SetCurrentCount(int value) {
    m_currentCountHasBeenSet = true;
    m_currentCount = value;
  }
  inline ClusterInstanceGroupDetails& WithCurrentCount(int value) {
    SetCurrentCount(value);
    return *this;
  }
 
 
  inline int GetTargetCount() const { return m_targetCount; }
  inline bool TargetCountHasBeenSet() const { return m_targetCountHasBeenSet; }
  inline void SetTargetCount(int value) {
    m_targetCountHasBeenSet = true;
    m_targetCount = value;
  }
  inline ClusterInstanceGroupDetails& WithTargetCount(int value) {
    SetTargetCount(value);
    return *this;
  }
 
 
  inline int GetMinCount() const { return m_minCount; }
  inline bool MinCountHasBeenSet() const { return m_minCountHasBeenSet; }
  inline void SetMinCount(int value) {
    m_minCountHasBeenSet = true;
    m_minCount = value;
  }
  inline ClusterInstanceGroupDetails& WithMinCount(int value) {
    SetMinCount(value);
    return *this;
  }
 
 
  inline const Aws::String& GetInstanceGroupName() const { return m_instanceGroupName; }
  inline bool InstanceGroupNameHasBeenSet() const { return m_instanceGroupNameHasBeenSet; }
  template <typename InstanceGroupNameT = Aws::String>
  void SetInstanceGroupName(InstanceGroupNameT&& value) {
    m_instanceGroupNameHasBeenSet = true;
    m_instanceGroupName = std::forward<InstanceGroupNameT>(value);
  }
  template <typename InstanceGroupNameT = Aws::String>
  ClusterInstanceGroupDetails& WithInstanceGroupName(InstanceGroupNameT&& value) {
    SetInstanceGroupName(std::forward<InstanceGroupNameT>(value));
    return *this;
  }
 
 
  inline ClusterInstanceType GetInstanceType() const { return m_instanceType; }
  inline bool InstanceTypeHasBeenSet() const { return m_instanceTypeHasBeenSet; }
  inline void SetInstanceType(ClusterInstanceType value) {
    m_instanceTypeHasBeenSet = true;
    m_instanceType = value;
  }
  inline ClusterInstanceGroupDetails& WithInstanceType(ClusterInstanceType value) {
    SetInstanceType(value);
    return *this;
  }
 
 
  inline const ClusterLifeCycleConfig& GetLifeCycleConfig() const { return m_lifeCycleConfig; }
  inline bool LifeCycleConfigHasBeenSet() const { return m_lifeCycleConfigHasBeenSet; }
  template <typename LifeCycleConfigT = ClusterLifeCycleConfig>
  void SetLifeCycleConfig(LifeCycleConfigT&& value) {
    m_lifeCycleConfigHasBeenSet = true;
    m_lifeCycleConfig = std::forward<LifeCycleConfigT>(value);
  }
  template <typename LifeCycleConfigT = ClusterLifeCycleConfig>
  ClusterInstanceGroupDetails& WithLifeCycleConfig(LifeCycleConfigT&& value) {
    SetLifeCycleConfig(std::forward<LifeCycleConfigT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetExecutionRole() const { return m_executionRole; }
  inline bool ExecutionRoleHasBeenSet() const { return m_executionRoleHasBeenSet; }
  template <typename ExecutionRoleT = Aws::String>
  void SetExecutionRole(ExecutionRoleT&& value) {
    m_executionRoleHasBeenSet = true;
    m_executionRole = std::forward<ExecutionRoleT>(value);
  }
  template <typename ExecutionRoleT = Aws::String>
  ClusterInstanceGroupDetails& WithExecutionRole(ExecutionRoleT&& value) {
    SetExecutionRole(std::forward<ExecutionRoleT>(value));
    return *this;
  }
 
 
  inline int GetThreadsPerCore() const { return m_threadsPerCore; }
  inline bool ThreadsPerCoreHasBeenSet() const { return m_threadsPerCoreHasBeenSet; }
  inline void SetThreadsPerCore(int value) {
    m_threadsPerCoreHasBeenSet = true;
    m_threadsPerCore = value;
  }
  inline ClusterInstanceGroupDetails& WithThreadsPerCore(int value) {
    SetThreadsPerCore(value);
    return *this;
  }
 
 
  inline const Aws::Vector<ClusterInstanceStorageConfig>& GetInstanceStorageConfigs() const { return m_instanceStorageConfigs; }
  inline bool InstanceStorageConfigsHasBeenSet() const { return m_instanceStorageConfigsHasBeenSet; }
  template <typename InstanceStorageConfigsT = Aws::Vector<ClusterInstanceStorageConfig>>
  void SetInstanceStorageConfigs(InstanceStorageConfigsT&& value) {
    m_instanceStorageConfigsHasBeenSet = true;
    m_instanceStorageConfigs = std::forward<InstanceStorageConfigsT>(value);
  }
  template <typename InstanceStorageConfigsT = Aws::Vector<ClusterInstanceStorageConfig>>
  ClusterInstanceGroupDetails& WithInstanceStorageConfigs(InstanceStorageConfigsT&& value) {
    SetInstanceStorageConfigs(std::forward<InstanceStorageConfigsT>(value));
    return *this;
  }
  template <typename InstanceStorageConfigsT = ClusterInstanceStorageConfig>
  ClusterInstanceGroupDetails& AddInstanceStorageConfigs(InstanceStorageConfigsT&& value) {
    m_instanceStorageConfigsHasBeenSet = true;
    m_instanceStorageConfigs.emplace_back(std::forward<InstanceStorageConfigsT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<DeepHealthCheckType>& GetOnStartDeepHealthChecks() const { return m_onStartDeepHealthChecks; }
  inline bool OnStartDeepHealthChecksHasBeenSet() const { return m_onStartDeepHealthChecksHasBeenSet; }
  template <typename OnStartDeepHealthChecksT = Aws::Vector<DeepHealthCheckType>>
  void SetOnStartDeepHealthChecks(OnStartDeepHealthChecksT&& value) {
    m_onStartDeepHealthChecksHasBeenSet = true;
    m_onStartDeepHealthChecks = std::forward<OnStartDeepHealthChecksT>(value);
  }
  template <typename OnStartDeepHealthChecksT = Aws::Vector<DeepHealthCheckType>>
  ClusterInstanceGroupDetails& WithOnStartDeepHealthChecks(OnStartDeepHealthChecksT&& value) {
    SetOnStartDeepHealthChecks(std::forward<OnStartDeepHealthChecksT>(value));
    return *this;
  }
  inline ClusterInstanceGroupDetails& AddOnStartDeepHealthChecks(DeepHealthCheckType value) {
    m_onStartDeepHealthChecksHasBeenSet = true;
    m_onStartDeepHealthChecks.push_back(value);
    return *this;
  }
 
 
  inline InstanceGroupStatus GetStatus() const { return m_status; }
  inline bool StatusHasBeenSet() const { return m_statusHasBeenSet; }
  inline void SetStatus(InstanceGroupStatus value) {
    m_statusHasBeenSet = true;
    m_status = value;
  }
  inline ClusterInstanceGroupDetails& WithStatus(InstanceGroupStatus value) {
    SetStatus(value);
    return *this;
  }
 
 
  inline const Aws::String& GetTrainingPlanArn() const { return m_trainingPlanArn; }
  inline bool TrainingPlanArnHasBeenSet() const { return m_trainingPlanArnHasBeenSet; }
  template <typename TrainingPlanArnT = Aws::String>
  void SetTrainingPlanArn(TrainingPlanArnT&& value) {
    m_trainingPlanArnHasBeenSet = true;
    m_trainingPlanArn = std::forward<TrainingPlanArnT>(value);
  }
  template <typename TrainingPlanArnT = Aws::String>
  ClusterInstanceGroupDetails& WithTrainingPlanArn(TrainingPlanArnT&& value) {
    SetTrainingPlanArn(std::forward<TrainingPlanArnT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetTrainingPlanStatus() const { return m_trainingPlanStatus; }
  inline bool TrainingPlanStatusHasBeenSet() const { return m_trainingPlanStatusHasBeenSet; }
  template <typename TrainingPlanStatusT = Aws::String>
  void SetTrainingPlanStatus(TrainingPlanStatusT&& value) {
    m_trainingPlanStatusHasBeenSet = true;
    m_trainingPlanStatus = std::forward<TrainingPlanStatusT>(value);
  }
  template <typename TrainingPlanStatusT = Aws::String>
  ClusterInstanceGroupDetails& WithTrainingPlanStatus(TrainingPlanStatusT&& value) {
    SetTrainingPlanStatus(std::forward<TrainingPlanStatusT>(value));
    return *this;
  }
 
 
  inline const VpcConfig& GetOverrideVpcConfig() const { return m_overrideVpcConfig; }
  inline bool OverrideVpcConfigHasBeenSet() const { return m_overrideVpcConfigHasBeenSet; }
  template <typename OverrideVpcConfigT = VpcConfig>
  void SetOverrideVpcConfig(OverrideVpcConfigT&& value) {
    m_overrideVpcConfigHasBeenSet = true;
    m_overrideVpcConfig = std::forward<OverrideVpcConfigT>(value);
  }
  template <typename OverrideVpcConfigT = VpcConfig>
  ClusterInstanceGroupDetails& WithOverrideVpcConfig(OverrideVpcConfigT&& value) {
    SetOverrideVpcConfig(std::forward<OverrideVpcConfigT>(value));
    return *this;
  }
 
 
  inline const ScheduledUpdateConfig& GetScheduledUpdateConfig() const { return m_scheduledUpdateConfig; }
  inline bool ScheduledUpdateConfigHasBeenSet() const { return m_scheduledUpdateConfigHasBeenSet; }
  template <typename ScheduledUpdateConfigT = ScheduledUpdateConfig>
  void SetScheduledUpdateConfig(ScheduledUpdateConfigT&& value) {
    m_scheduledUpdateConfigHasBeenSet = true;
    m_scheduledUpdateConfig = std::forward<ScheduledUpdateConfigT>(value);
  }
  template <typename ScheduledUpdateConfigT = ScheduledUpdateConfig>
  ClusterInstanceGroupDetails& WithScheduledUpdateConfig(ScheduledUpdateConfigT&& value) {
    SetScheduledUpdateConfig(std::forward<ScheduledUpdateConfigT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetCurrentImageId() const { return m_currentImageId; }
  inline bool CurrentImageIdHasBeenSet() const { return m_currentImageIdHasBeenSet; }
  template <typename CurrentImageIdT = Aws::String>
  void SetCurrentImageId(CurrentImageIdT&& value) {
    m_currentImageIdHasBeenSet = true;
    m_currentImageId = std::forward<CurrentImageIdT>(value);
  }
  template <typename CurrentImageIdT = Aws::String>
  ClusterInstanceGroupDetails& WithCurrentImageId(CurrentImageIdT&& value) {
    SetCurrentImageId(std::forward<CurrentImageIdT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetDesiredImageId() const { return m_desiredImageId; }
  inline bool DesiredImageIdHasBeenSet() const { return m_desiredImageIdHasBeenSet; }
  template <typename DesiredImageIdT = Aws::String>
  void SetDesiredImageId(DesiredImageIdT&& value) {
    m_desiredImageIdHasBeenSet = true;
    m_desiredImageId = std::forward<DesiredImageIdT>(value);
  }
  template <typename DesiredImageIdT = Aws::String>
  ClusterInstanceGroupDetails& WithDesiredImageId(DesiredImageIdT&& value) {
    SetDesiredImageId(std::forward<DesiredImageIdT>(value));
    return *this;
  }
 
 
  inline const Aws::Map<ActiveClusterOperationName, int>& GetActiveOperations() const { return m_activeOperations; }
  inline bool ActiveOperationsHasBeenSet() const { return m_activeOperationsHasBeenSet; }
  template <typename ActiveOperationsT = Aws::Map<ActiveClusterOperationName, int>>
  void SetActiveOperations(ActiveOperationsT&& value) {
    m_activeOperationsHasBeenSet = true;
    m_activeOperations = std::forward<ActiveOperationsT>(value);
  }
  template <typename ActiveOperationsT = Aws::Map<ActiveClusterOperationName, int>>
  ClusterInstanceGroupDetails& WithActiveOperations(ActiveOperationsT&& value) {
    SetActiveOperations(std::forward<ActiveOperationsT>(value));
    return *this;
  }
  inline ClusterInstanceGroupDetails& AddActiveOperations(ActiveClusterOperationName key, int value) {
    m_activeOperationsHasBeenSet = true;
    m_activeOperations.emplace(key, value);
    return *this;
  }
 
 
  inline const ClusterKubernetesConfigDetails& GetKubernetesConfig() const { return m_kubernetesConfig; }
  inline bool KubernetesConfigHasBeenSet() const { return m_kubernetesConfigHasBeenSet; }
  template <typename KubernetesConfigT = ClusterKubernetesConfigDetails>
  void SetKubernetesConfig(KubernetesConfigT&& value) {
    m_kubernetesConfigHasBeenSet = true;
    m_kubernetesConfig = std::forward<KubernetesConfigT>(value);
  }
  template <typename KubernetesConfigT = ClusterKubernetesConfigDetails>
  ClusterInstanceGroupDetails& WithKubernetesConfig(KubernetesConfigT&& value) {
    SetKubernetesConfig(std::forward<KubernetesConfigT>(value));
    return *this;
  }
 
 
  inline const ClusterCapacityRequirements& GetCapacityRequirements() const { return m_capacityRequirements; }
  inline bool CapacityRequirementsHasBeenSet() const { return m_capacityRequirementsHasBeenSet; }
  template <typename CapacityRequirementsT = ClusterCapacityRequirements>
  void SetCapacityRequirements(CapacityRequirementsT&& value) {
    m_capacityRequirementsHasBeenSet = true;
    m_capacityRequirements = std::forward<CapacityRequirementsT>(value);
  }
  template <typename CapacityRequirementsT = ClusterCapacityRequirements>
  ClusterInstanceGroupDetails& WithCapacityRequirements(CapacityRequirementsT&& value) {
    SetCapacityRequirements(std::forward<CapacityRequirementsT>(value));
    return *this;
  }
 
 
  inline int GetTargetStateCount() const { return m_targetStateCount; }
  inline bool TargetStateCountHasBeenSet() const { return m_targetStateCountHasBeenSet; }
  inline void SetTargetStateCount(int value) {
    m_targetStateCountHasBeenSet = true;
    m_targetStateCount = value;
  }
  inline ClusterInstanceGroupDetails& WithTargetStateCount(int value) {
    SetTargetStateCount(value);
    return *this;
  }
 
 
  inline SoftwareUpdateStatus GetSoftwareUpdateStatus() const { return m_softwareUpdateStatus; }
  inline bool SoftwareUpdateStatusHasBeenSet() const { return m_softwareUpdateStatusHasBeenSet; }
  inline void SetSoftwareUpdateStatus(SoftwareUpdateStatus value) {
    m_softwareUpdateStatusHasBeenSet = true;
    m_softwareUpdateStatus = value;
  }
  inline ClusterInstanceGroupDetails& WithSoftwareUpdateStatus(SoftwareUpdateStatus value) {
    SetSoftwareUpdateStatus(value);
    return *this;
  }
 
 
  inline const DeploymentConfiguration& GetActiveSoftwareUpdateConfig() const { return m_activeSoftwareUpdateConfig; }
  inline bool ActiveSoftwareUpdateConfigHasBeenSet() const { return m_activeSoftwareUpdateConfigHasBeenSet; }
  template <typename ActiveSoftwareUpdateConfigT = DeploymentConfiguration>
  void SetActiveSoftwareUpdateConfig(ActiveSoftwareUpdateConfigT&& value) {
    m_activeSoftwareUpdateConfigHasBeenSet = true;
    m_activeSoftwareUpdateConfig = std::forward<ActiveSoftwareUpdateConfigT>(value);
  }
  template <typename ActiveSoftwareUpdateConfigT = DeploymentConfiguration>
  ClusterInstanceGroupDetails& WithActiveSoftwareUpdateConfig(ActiveSoftwareUpdateConfigT&& value) {
    SetActiveSoftwareUpdateConfig(std::forward<ActiveSoftwareUpdateConfigT>(value));
    return *this;
  }
 
 
  inline const ClusterSlurmConfigDetails& GetSlurmConfig() const { return m_slurmConfig; }
  inline bool SlurmConfigHasBeenSet() const { return m_slurmConfigHasBeenSet; }
  template <typename SlurmConfigT = ClusterSlurmConfigDetails>
  void SetSlurmConfig(SlurmConfigT&& value) {
    m_slurmConfigHasBeenSet = true;
    m_slurmConfig = std::forward<SlurmConfigT>(value);
  }
  template <typename SlurmConfigT = ClusterSlurmConfigDetails>
  ClusterInstanceGroupDetails& WithSlurmConfig(SlurmConfigT&& value) {
    SetSlurmConfig(std::forward<SlurmConfigT>(value));
    return *this;
  }
 private:
  int m_currentCount{0};
 
  int m_targetCount{0};
 
  int m_minCount{0};
 
  Aws::String m_instanceGroupName;
 
  ClusterInstanceType m_instanceType{ClusterInstanceType::NOT_SET};
 
  ClusterLifeCycleConfig m_lifeCycleConfig;
 
  Aws::String m_executionRole;
 
  int m_threadsPerCore{0};
 
  Aws::Vector<ClusterInstanceStorageConfig> m_instanceStorageConfigs;
 
  Aws::Vector<DeepHealthCheckType> m_onStartDeepHealthChecks;
 
  InstanceGroupStatus m_status{InstanceGroupStatus::NOT_SET};
 
  Aws::String m_trainingPlanArn;
 
  Aws::String m_trainingPlanStatus;
 
  VpcConfig m_overrideVpcConfig;
 
  ScheduledUpdateConfig m_scheduledUpdateConfig;
 
  Aws::String m_currentImageId;
 
  Aws::String m_desiredImageId;
 
  Aws::Map<ActiveClusterOperationName, int> m_activeOperations;
 
  ClusterKubernetesConfigDetails m_kubernetesConfig;
 
  ClusterCapacityRequirements m_capacityRequirements;
 
  int m_targetStateCount{0};
 
  SoftwareUpdateStatus m_softwareUpdateStatus{SoftwareUpdateStatus::NOT_SET};
 
  DeploymentConfiguration m_activeSoftwareUpdateConfig;
 
  ClusterSlurmConfigDetails m_slurmConfig;
  bool m_currentCountHasBeenSet = false;
  bool m_targetCountHasBeenSet = false;
  bool m_minCountHasBeenSet = false;
  bool m_instanceGroupNameHasBeenSet = false;
  bool m_instanceTypeHasBeenSet = false;
  bool m_lifeCycleConfigHasBeenSet = false;
  bool m_executionRoleHasBeenSet = false;
  bool m_threadsPerCoreHasBeenSet = false;
  bool m_instanceStorageConfigsHasBeenSet = false;
  bool m_onStartDeepHealthChecksHasBeenSet = false;
  bool m_statusHasBeenSet = false;
  bool m_trainingPlanArnHasBeenSet = false;
  bool m_trainingPlanStatusHasBeenSet = false;
  bool m_overrideVpcConfigHasBeenSet = false;
  bool m_scheduledUpdateConfigHasBeenSet = false;
  bool m_currentImageIdHasBeenSet = false;
  bool m_desiredImageIdHasBeenSet = false;
  bool m_activeOperationsHasBeenSet = false;
  bool m_kubernetesConfigHasBeenSet = false;
  bool m_capacityRequirementsHasBeenSet = false;
  bool m_targetStateCountHasBeenSet = false;
  bool m_softwareUpdateStatusHasBeenSet = false;
  bool m_activeSoftwareUpdateConfigHasBeenSet = false;
  bool m_slurmConfigHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws