RecommendationJobInputConfig.h

 
#pragma once
#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/EndpointInfo.h>
#include <aws/sagemaker/model/EndpointInputConfiguration.h>
#include <aws/sagemaker/model/RecommendationJobContainerConfig.h>
#include <aws/sagemaker/model/RecommendationJobResourceLimit.h>
#include <aws/sagemaker/model/RecommendationJobVpcConfig.h>
#include <aws/sagemaker/model/TrafficPattern.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class RecommendationJobInputConfig {
 public:
  AWS_SAGEMAKER_API RecommendationJobInputConfig() = default;
  AWS_SAGEMAKER_API RecommendationJobInputConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API RecommendationJobInputConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const Aws::String& GetModelPackageVersionArn() const { return m_modelPackageVersionArn; }
  inline bool ModelPackageVersionArnHasBeenSet() const { return m_modelPackageVersionArnHasBeenSet; }
  template <typename ModelPackageVersionArnT = Aws::String>
  void SetModelPackageVersionArn(ModelPackageVersionArnT&& value) {
    m_modelPackageVersionArnHasBeenSet = true;
    m_modelPackageVersionArn = std::forward<ModelPackageVersionArnT>(value);
  }
  template <typename ModelPackageVersionArnT = Aws::String>
  RecommendationJobInputConfig& WithModelPackageVersionArn(ModelPackageVersionArnT&& value) {
    SetModelPackageVersionArn(std::forward<ModelPackageVersionArnT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetModelName() const { return m_modelName; }
  inline bool ModelNameHasBeenSet() const { return m_modelNameHasBeenSet; }
  template <typename ModelNameT = Aws::String>
  void SetModelName(ModelNameT&& value) {
    m_modelNameHasBeenSet = true;
    m_modelName = std::forward<ModelNameT>(value);
  }
  template <typename ModelNameT = Aws::String>
  RecommendationJobInputConfig& WithModelName(ModelNameT&& value) {
    SetModelName(std::forward<ModelNameT>(value));
    return *this;
  }
 
 
  inline int GetJobDurationInSeconds() const { return m_jobDurationInSeconds; }
  inline bool JobDurationInSecondsHasBeenSet() const { return m_jobDurationInSecondsHasBeenSet; }
  inline void SetJobDurationInSeconds(int value) {
    m_jobDurationInSecondsHasBeenSet = true;
    m_jobDurationInSeconds = value;
  }
  inline RecommendationJobInputConfig& WithJobDurationInSeconds(int value) {
    SetJobDurationInSeconds(value);
    return *this;
  }
 
 
  inline const TrafficPattern& GetTrafficPattern() const { return m_trafficPattern; }
  inline bool TrafficPatternHasBeenSet() const { return m_trafficPatternHasBeenSet; }
  template <typename TrafficPatternT = TrafficPattern>
  void SetTrafficPattern(TrafficPatternT&& value) {
    m_trafficPatternHasBeenSet = true;
    m_trafficPattern = std::forward<TrafficPatternT>(value);
  }
  template <typename TrafficPatternT = TrafficPattern>
  RecommendationJobInputConfig& WithTrafficPattern(TrafficPatternT&& value) {
    SetTrafficPattern(std::forward<TrafficPatternT>(value));
    return *this;
  }
 
 
  inline const RecommendationJobResourceLimit& GetResourceLimit() const { return m_resourceLimit; }
  inline bool ResourceLimitHasBeenSet() const { return m_resourceLimitHasBeenSet; }
  template <typename ResourceLimitT = RecommendationJobResourceLimit>
  void SetResourceLimit(ResourceLimitT&& value) {
    m_resourceLimitHasBeenSet = true;
    m_resourceLimit = std::forward<ResourceLimitT>(value);
  }
  template <typename ResourceLimitT = RecommendationJobResourceLimit>
  RecommendationJobInputConfig& WithResourceLimit(ResourceLimitT&& value) {
    SetResourceLimit(std::forward<ResourceLimitT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<EndpointInputConfiguration>& GetEndpointConfigurations() const { return m_endpointConfigurations; }
  inline bool EndpointConfigurationsHasBeenSet() const { return m_endpointConfigurationsHasBeenSet; }
  template <typename EndpointConfigurationsT = Aws::Vector<EndpointInputConfiguration>>
  void SetEndpointConfigurations(EndpointConfigurationsT&& value) {
    m_endpointConfigurationsHasBeenSet = true;
    m_endpointConfigurations = std::forward<EndpointConfigurationsT>(value);
  }
  template <typename EndpointConfigurationsT = Aws::Vector<EndpointInputConfiguration>>
  RecommendationJobInputConfig& WithEndpointConfigurations(EndpointConfigurationsT&& value) {
    SetEndpointConfigurations(std::forward<EndpointConfigurationsT>(value));
    return *this;
  }
  template <typename EndpointConfigurationsT = EndpointInputConfiguration>
  RecommendationJobInputConfig& AddEndpointConfigurations(EndpointConfigurationsT&& value) {
    m_endpointConfigurationsHasBeenSet = true;
    m_endpointConfigurations.emplace_back(std::forward<EndpointConfigurationsT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetVolumeKmsKeyId() const { return m_volumeKmsKeyId; }
  inline bool VolumeKmsKeyIdHasBeenSet() const { return m_volumeKmsKeyIdHasBeenSet; }
  template <typename VolumeKmsKeyIdT = Aws::String>
  void SetVolumeKmsKeyId(VolumeKmsKeyIdT&& value) {
    m_volumeKmsKeyIdHasBeenSet = true;
    m_volumeKmsKeyId = std::forward<VolumeKmsKeyIdT>(value);
  }
  template <typename VolumeKmsKeyIdT = Aws::String>
  RecommendationJobInputConfig& WithVolumeKmsKeyId(VolumeKmsKeyIdT&& value) {
    SetVolumeKmsKeyId(std::forward<VolumeKmsKeyIdT>(value));
    return *this;
  }
 
 
  inline const RecommendationJobContainerConfig& GetContainerConfig() const { return m_containerConfig; }
  inline bool ContainerConfigHasBeenSet() const { return m_containerConfigHasBeenSet; }
  template <typename ContainerConfigT = RecommendationJobContainerConfig>
  void SetContainerConfig(ContainerConfigT&& value) {
    m_containerConfigHasBeenSet = true;
    m_containerConfig = std::forward<ContainerConfigT>(value);
  }
  template <typename ContainerConfigT = RecommendationJobContainerConfig>
  RecommendationJobInputConfig& WithContainerConfig(ContainerConfigT&& value) {
    SetContainerConfig(std::forward<ContainerConfigT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<EndpointInfo>& GetEndpoints() const { return m_endpoints; }
  inline bool EndpointsHasBeenSet() const { return m_endpointsHasBeenSet; }
  template <typename EndpointsT = Aws::Vector<EndpointInfo>>
  void SetEndpoints(EndpointsT&& value) {
    m_endpointsHasBeenSet = true;
    m_endpoints = std::forward<EndpointsT>(value);
  }
  template <typename EndpointsT = Aws::Vector<EndpointInfo>>
  RecommendationJobInputConfig& WithEndpoints(EndpointsT&& value) {
    SetEndpoints(std::forward<EndpointsT>(value));
    return *this;
  }
  template <typename EndpointsT = EndpointInfo>
  RecommendationJobInputConfig& AddEndpoints(EndpointsT&& value) {
    m_endpointsHasBeenSet = true;
    m_endpoints.emplace_back(std::forward<EndpointsT>(value));
    return *this;
  }
 
 
  inline const RecommendationJobVpcConfig& GetVpcConfig() const { return m_vpcConfig; }
  inline bool VpcConfigHasBeenSet() const { return m_vpcConfigHasBeenSet; }
  template <typename VpcConfigT = RecommendationJobVpcConfig>
  void SetVpcConfig(VpcConfigT&& value) {
    m_vpcConfigHasBeenSet = true;
    m_vpcConfig = std::forward<VpcConfigT>(value);
  }
  template <typename VpcConfigT = RecommendationJobVpcConfig>
  RecommendationJobInputConfig& WithVpcConfig(VpcConfigT&& value) {
    SetVpcConfig(std::forward<VpcConfigT>(value));
    return *this;
  }
 private:
  Aws::String m_modelPackageVersionArn;
 
  Aws::String m_modelName;
 
  int m_jobDurationInSeconds{0};
 
  TrafficPattern m_trafficPattern;
 
  RecommendationJobResourceLimit m_resourceLimit;
 
  Aws::Vector<EndpointInputConfiguration> m_endpointConfigurations;
 
  Aws::String m_volumeKmsKeyId;
 
  RecommendationJobContainerConfig m_containerConfig;
 
  Aws::Vector<EndpointInfo> m_endpoints;
 
  RecommendationJobVpcConfig m_vpcConfig;
  bool m_modelPackageVersionArnHasBeenSet = false;
  bool m_modelNameHasBeenSet = false;
  bool m_jobDurationInSecondsHasBeenSet = false;
  bool m_trafficPatternHasBeenSet = false;
  bool m_resourceLimitHasBeenSet = false;
  bool m_endpointConfigurationsHasBeenSet = false;
  bool m_volumeKmsKeyIdHasBeenSet = false;
  bool m_containerConfigHasBeenSet = false;
  bool m_endpointsHasBeenSet = false;
  bool m_vpcConfigHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws