CreateOptimizationJobRequest.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/SageMakerRequest.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/OptimizationConfig.h>
#include <aws/sagemaker/model/OptimizationJobDeploymentInstanceType.h>
#include <aws/sagemaker/model/OptimizationJobModelSource.h>
#include <aws/sagemaker/model/OptimizationJobOutputConfig.h>
#include <aws/sagemaker/model/OptimizationVpcConfig.h>
#include <aws/sagemaker/model/StoppingCondition.h>
#include <aws/sagemaker/model/Tag.h>
 
#include <utility>
 
namespace Aws {
namespace SageMaker {
namespace Model {
 
class CreateOptimizationJobRequest : public SageMakerRequest {
 public:
  AWS_SAGEMAKER_API CreateOptimizationJobRequest() = default;
 
  // Service request name is the Operation name which will send this request out,
  // each operation should has unique request name, so that we can get operation's name from this request.
  // Note: this is not true for response, multiple operations may have the same response name,
  // so we can not get operation's name from response.
  inline virtual const char* GetServiceRequestName() const override { return "CreateOptimizationJob"; }
 
  AWS_SAGEMAKER_API Aws::String SerializePayload() const override;
 
  AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::String& GetOptimizationJobName() const { return m_optimizationJobName; }
  inline bool OptimizationJobNameHasBeenSet() const { return m_optimizationJobNameHasBeenSet; }
  template <typename OptimizationJobNameT = Aws::String>
  void SetOptimizationJobName(OptimizationJobNameT&& value) {
    m_optimizationJobNameHasBeenSet = true;
    m_optimizationJobName = std::forward<OptimizationJobNameT>(value);
  }
  template <typename OptimizationJobNameT = Aws::String>
  CreateOptimizationJobRequest& WithOptimizationJobName(OptimizationJobNameT&& value) {
    SetOptimizationJobName(std::forward<OptimizationJobNameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetRoleArn() const { return m_roleArn; }
  inline bool RoleArnHasBeenSet() const { return m_roleArnHasBeenSet; }
  template <typename RoleArnT = Aws::String>
  void SetRoleArn(RoleArnT&& value) {
    m_roleArnHasBeenSet = true;
    m_roleArn = std::forward<RoleArnT>(value);
  }
  template <typename RoleArnT = Aws::String>
  CreateOptimizationJobRequest& WithRoleArn(RoleArnT&& value) {
    SetRoleArn(std::forward<RoleArnT>(value));
    return *this;
  }
 
 
  inline const OptimizationJobModelSource& GetModelSource() const { return m_modelSource; }
  inline bool ModelSourceHasBeenSet() const { return m_modelSourceHasBeenSet; }
  template <typename ModelSourceT = OptimizationJobModelSource>
  void SetModelSource(ModelSourceT&& value) {
    m_modelSourceHasBeenSet = true;
    m_modelSource = std::forward<ModelSourceT>(value);
  }
  template <typename ModelSourceT = OptimizationJobModelSource>
  CreateOptimizationJobRequest& WithModelSource(ModelSourceT&& value) {
    SetModelSource(std::forward<ModelSourceT>(value));
    return *this;
  }
 
 
  inline OptimizationJobDeploymentInstanceType GetDeploymentInstanceType() const { return m_deploymentInstanceType; }
  inline bool DeploymentInstanceTypeHasBeenSet() const { return m_deploymentInstanceTypeHasBeenSet; }
  inline void SetDeploymentInstanceType(OptimizationJobDeploymentInstanceType value) {
    m_deploymentInstanceTypeHasBeenSet = true;
    m_deploymentInstanceType = value;
  }
  inline CreateOptimizationJobRequest& WithDeploymentInstanceType(OptimizationJobDeploymentInstanceType value) {
    SetDeploymentInstanceType(value);
    return *this;
  }
 
 
  inline int GetMaxInstanceCount() const { return m_maxInstanceCount; }
  inline bool MaxInstanceCountHasBeenSet() const { return m_maxInstanceCountHasBeenSet; }
  inline void SetMaxInstanceCount(int value) {
    m_maxInstanceCountHasBeenSet = true;
    m_maxInstanceCount = value;
  }
  inline CreateOptimizationJobRequest& WithMaxInstanceCount(int value) {
    SetMaxInstanceCount(value);
    return *this;
  }
 
 
  inline const Aws::Map<Aws::String, Aws::String>& GetOptimizationEnvironment() const { return m_optimizationEnvironment; }
  inline bool OptimizationEnvironmentHasBeenSet() const { return m_optimizationEnvironmentHasBeenSet; }
  template <typename OptimizationEnvironmentT = Aws::Map<Aws::String, Aws::String>>
  void SetOptimizationEnvironment(OptimizationEnvironmentT&& value) {
    m_optimizationEnvironmentHasBeenSet = true;
    m_optimizationEnvironment = std::forward<OptimizationEnvironmentT>(value);
  }
  template <typename OptimizationEnvironmentT = Aws::Map<Aws::String, Aws::String>>
  CreateOptimizationJobRequest& WithOptimizationEnvironment(OptimizationEnvironmentT&& value) {
    SetOptimizationEnvironment(std::forward<OptimizationEnvironmentT>(value));
    return *this;
  }
  template <typename OptimizationEnvironmentKeyT = Aws::String, typename OptimizationEnvironmentValueT = Aws::String>
  CreateOptimizationJobRequest& AddOptimizationEnvironment(OptimizationEnvironmentKeyT&& key, OptimizationEnvironmentValueT&& value) {
    m_optimizationEnvironmentHasBeenSet = true;
    m_optimizationEnvironment.emplace(std::forward<OptimizationEnvironmentKeyT>(key), std::forward<OptimizationEnvironmentValueT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<OptimizationConfig>& GetOptimizationConfigs() const { return m_optimizationConfigs; }
  inline bool OptimizationConfigsHasBeenSet() const { return m_optimizationConfigsHasBeenSet; }
  template <typename OptimizationConfigsT = Aws::Vector<OptimizationConfig>>
  void SetOptimizationConfigs(OptimizationConfigsT&& value) {
    m_optimizationConfigsHasBeenSet = true;
    m_optimizationConfigs = std::forward<OptimizationConfigsT>(value);
  }
  template <typename OptimizationConfigsT = Aws::Vector<OptimizationConfig>>
  CreateOptimizationJobRequest& WithOptimizationConfigs(OptimizationConfigsT&& value) {
    SetOptimizationConfigs(std::forward<OptimizationConfigsT>(value));
    return *this;
  }
  template <typename OptimizationConfigsT = OptimizationConfig>
  CreateOptimizationJobRequest& AddOptimizationConfigs(OptimizationConfigsT&& value) {
    m_optimizationConfigsHasBeenSet = true;
    m_optimizationConfigs.emplace_back(std::forward<OptimizationConfigsT>(value));
    return *this;
  }
 
 
  inline const OptimizationJobOutputConfig& GetOutputConfig() const { return m_outputConfig; }
  inline bool OutputConfigHasBeenSet() const { return m_outputConfigHasBeenSet; }
  template <typename OutputConfigT = OptimizationJobOutputConfig>
  void SetOutputConfig(OutputConfigT&& value) {
    m_outputConfigHasBeenSet = true;
    m_outputConfig = std::forward<OutputConfigT>(value);
  }
  template <typename OutputConfigT = OptimizationJobOutputConfig>
  CreateOptimizationJobRequest& WithOutputConfig(OutputConfigT&& value) {
    SetOutputConfig(std::forward<OutputConfigT>(value));
    return *this;
  }
 
 
  inline const StoppingCondition& GetStoppingCondition() const { return m_stoppingCondition; }
  inline bool StoppingConditionHasBeenSet() const { return m_stoppingConditionHasBeenSet; }
  template <typename StoppingConditionT = StoppingCondition>
  void SetStoppingCondition(StoppingConditionT&& value) {
    m_stoppingConditionHasBeenSet = true;
    m_stoppingCondition = std::forward<StoppingConditionT>(value);
  }
  template <typename StoppingConditionT = StoppingCondition>
  CreateOptimizationJobRequest& WithStoppingCondition(StoppingConditionT&& value) {
    SetStoppingCondition(std::forward<StoppingConditionT>(value));
    return *this;
  }
 
 
  inline const Aws::Vector<Tag>& GetTags() const { return m_tags; }
  inline bool TagsHasBeenSet() const { return m_tagsHasBeenSet; }
  template <typename TagsT = Aws::Vector<Tag>>
  void SetTags(TagsT&& value) {
    m_tagsHasBeenSet = true;
    m_tags = std::forward<TagsT>(value);
  }
  template <typename TagsT = Aws::Vector<Tag>>
  CreateOptimizationJobRequest& WithTags(TagsT&& value) {
    SetTags(std::forward<TagsT>(value));
    return *this;
  }
  template <typename TagsT = Tag>
  CreateOptimizationJobRequest& AddTags(TagsT&& value) {
    m_tagsHasBeenSet = true;
    m_tags.emplace_back(std::forward<TagsT>(value));
    return *this;
  }
 
 
  inline const OptimizationVpcConfig& GetVpcConfig() const { return m_vpcConfig; }
  inline bool VpcConfigHasBeenSet() const { return m_vpcConfigHasBeenSet; }
  template <typename VpcConfigT = OptimizationVpcConfig>
  void SetVpcConfig(VpcConfigT&& value) {
    m_vpcConfigHasBeenSet = true;
    m_vpcConfig = std::forward<VpcConfigT>(value);
  }
  template <typename VpcConfigT = OptimizationVpcConfig>
  CreateOptimizationJobRequest& WithVpcConfig(VpcConfigT&& value) {
    SetVpcConfig(std::forward<VpcConfigT>(value));
    return *this;
  }
 private:
  Aws::String m_optimizationJobName;
 
  Aws::String m_roleArn;
 
  OptimizationJobModelSource m_modelSource;
 
  OptimizationJobDeploymentInstanceType m_deploymentInstanceType{OptimizationJobDeploymentInstanceType::NOT_SET};
 
  int m_maxInstanceCount{0};
 
  Aws::Map<Aws::String, Aws::String> m_optimizationEnvironment;
 
  Aws::Vector<OptimizationConfig> m_optimizationConfigs;
 
  OptimizationJobOutputConfig m_outputConfig;
 
  StoppingCondition m_stoppingCondition;
 
  Aws::Vector<Tag> m_tags;
 
  OptimizationVpcConfig m_vpcConfig;
  bool m_optimizationJobNameHasBeenSet = false;
  bool m_roleArnHasBeenSet = false;
  bool m_modelSourceHasBeenSet = false;
  bool m_deploymentInstanceTypeHasBeenSet = false;
  bool m_maxInstanceCountHasBeenSet = false;
  bool m_optimizationEnvironmentHasBeenSet = false;
  bool m_optimizationConfigsHasBeenSet = false;
  bool m_outputConfigHasBeenSet = false;
  bool m_stoppingConditionHasBeenSet = false;
  bool m_tagsHasBeenSet = false;
  bool m_vpcConfigHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws