CreateTransformJobRequest.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/BatchDataCaptureConfig.h>
#include <aws/sagemaker/model/BatchStrategy.h>
#include <aws/sagemaker/model/DataProcessing.h>
#include <aws/sagemaker/model/ExperimentConfig.h>
#include <aws/sagemaker/model/ModelClientConfig.h>
#include <aws/sagemaker/model/Tag.h>
#include <aws/sagemaker/model/TransformInput.h>
#include <aws/sagemaker/model/TransformOutput.h>
#include <aws/sagemaker/model/TransformResources.h>
 
#include <utility>
 
namespace Aws {
namespace SageMaker {
namespace Model {
 
class CreateTransformJobRequest : public SageMakerRequest {
 public:
  AWS_SAGEMAKER_API CreateTransformJobRequest() = 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 "CreateTransformJob"; }
 
  AWS_SAGEMAKER_API Aws::String SerializePayload() const override;
 
  AWS_SAGEMAKER_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::String& GetTransformJobName() const { return m_transformJobName; }
  inline bool TransformJobNameHasBeenSet() const { return m_transformJobNameHasBeenSet; }
  template <typename TransformJobNameT = Aws::String>
  void SetTransformJobName(TransformJobNameT&& value) {
    m_transformJobNameHasBeenSet = true;
    m_transformJobName = std::forward<TransformJobNameT>(value);
  }
  template <typename TransformJobNameT = Aws::String>
  CreateTransformJobRequest& WithTransformJobName(TransformJobNameT&& value) {
    SetTransformJobName(std::forward<TransformJobNameT>(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>
  CreateTransformJobRequest& WithModelName(ModelNameT&& value) {
    SetModelName(std::forward<ModelNameT>(value));
    return *this;
  }
 
 
  inline int GetMaxConcurrentTransforms() const { return m_maxConcurrentTransforms; }
  inline bool MaxConcurrentTransformsHasBeenSet() const { return m_maxConcurrentTransformsHasBeenSet; }
  inline void SetMaxConcurrentTransforms(int value) {
    m_maxConcurrentTransformsHasBeenSet = true;
    m_maxConcurrentTransforms = value;
  }
  inline CreateTransformJobRequest& WithMaxConcurrentTransforms(int value) {
    SetMaxConcurrentTransforms(value);
    return *this;
  }
 
 
  inline const ModelClientConfig& GetModelClientConfig() const { return m_modelClientConfig; }
  inline bool ModelClientConfigHasBeenSet() const { return m_modelClientConfigHasBeenSet; }
  template <typename ModelClientConfigT = ModelClientConfig>
  void SetModelClientConfig(ModelClientConfigT&& value) {
    m_modelClientConfigHasBeenSet = true;
    m_modelClientConfig = std::forward<ModelClientConfigT>(value);
  }
  template <typename ModelClientConfigT = ModelClientConfig>
  CreateTransformJobRequest& WithModelClientConfig(ModelClientConfigT&& value) {
    SetModelClientConfig(std::forward<ModelClientConfigT>(value));
    return *this;
  }
 
 
  inline int GetMaxPayloadInMB() const { return m_maxPayloadInMB; }
  inline bool MaxPayloadInMBHasBeenSet() const { return m_maxPayloadInMBHasBeenSet; }
  inline void SetMaxPayloadInMB(int value) {
    m_maxPayloadInMBHasBeenSet = true;
    m_maxPayloadInMB = value;
  }
  inline CreateTransformJobRequest& WithMaxPayloadInMB(int value) {
    SetMaxPayloadInMB(value);
    return *this;
  }
 
 
  inline BatchStrategy GetBatchStrategy() const { return m_batchStrategy; }
  inline bool BatchStrategyHasBeenSet() const { return m_batchStrategyHasBeenSet; }
  inline void SetBatchStrategy(BatchStrategy value) {
    m_batchStrategyHasBeenSet = true;
    m_batchStrategy = value;
  }
  inline CreateTransformJobRequest& WithBatchStrategy(BatchStrategy value) {
    SetBatchStrategy(value);
    return *this;
  }
 
 
  inline const Aws::Map<Aws::String, Aws::String>& GetEnvironment() const { return m_environment; }
  inline bool EnvironmentHasBeenSet() const { return m_environmentHasBeenSet; }
  template <typename EnvironmentT = Aws::Map<Aws::String, Aws::String>>
  void SetEnvironment(EnvironmentT&& value) {
    m_environmentHasBeenSet = true;
    m_environment = std::forward<EnvironmentT>(value);
  }
  template <typename EnvironmentT = Aws::Map<Aws::String, Aws::String>>
  CreateTransformJobRequest& WithEnvironment(EnvironmentT&& value) {
    SetEnvironment(std::forward<EnvironmentT>(value));
    return *this;
  }
  template <typename EnvironmentKeyT = Aws::String, typename EnvironmentValueT = Aws::String>
  CreateTransformJobRequest& AddEnvironment(EnvironmentKeyT&& key, EnvironmentValueT&& value) {
    m_environmentHasBeenSet = true;
    m_environment.emplace(std::forward<EnvironmentKeyT>(key), std::forward<EnvironmentValueT>(value));
    return *this;
  }
 
 
  inline const TransformInput& GetTransformInput() const { return m_transformInput; }
  inline bool TransformInputHasBeenSet() const { return m_transformInputHasBeenSet; }
  template <typename TransformInputT = TransformInput>
  void SetTransformInput(TransformInputT&& value) {
    m_transformInputHasBeenSet = true;
    m_transformInput = std::forward<TransformInputT>(value);
  }
  template <typename TransformInputT = TransformInput>
  CreateTransformJobRequest& WithTransformInput(TransformInputT&& value) {
    SetTransformInput(std::forward<TransformInputT>(value));
    return *this;
  }
 
 
  inline const TransformOutput& GetTransformOutput() const { return m_transformOutput; }
  inline bool TransformOutputHasBeenSet() const { return m_transformOutputHasBeenSet; }
  template <typename TransformOutputT = TransformOutput>
  void SetTransformOutput(TransformOutputT&& value) {
    m_transformOutputHasBeenSet = true;
    m_transformOutput = std::forward<TransformOutputT>(value);
  }
  template <typename TransformOutputT = TransformOutput>
  CreateTransformJobRequest& WithTransformOutput(TransformOutputT&& value) {
    SetTransformOutput(std::forward<TransformOutputT>(value));
    return *this;
  }
 
 
  inline const BatchDataCaptureConfig& GetDataCaptureConfig() const { return m_dataCaptureConfig; }
  inline bool DataCaptureConfigHasBeenSet() const { return m_dataCaptureConfigHasBeenSet; }
  template <typename DataCaptureConfigT = BatchDataCaptureConfig>
  void SetDataCaptureConfig(DataCaptureConfigT&& value) {
    m_dataCaptureConfigHasBeenSet = true;
    m_dataCaptureConfig = std::forward<DataCaptureConfigT>(value);
  }
  template <typename DataCaptureConfigT = BatchDataCaptureConfig>
  CreateTransformJobRequest& WithDataCaptureConfig(DataCaptureConfigT&& value) {
    SetDataCaptureConfig(std::forward<DataCaptureConfigT>(value));
    return *this;
  }
 
 
  inline const TransformResources& GetTransformResources() const { return m_transformResources; }
  inline bool TransformResourcesHasBeenSet() const { return m_transformResourcesHasBeenSet; }
  template <typename TransformResourcesT = TransformResources>
  void SetTransformResources(TransformResourcesT&& value) {
    m_transformResourcesHasBeenSet = true;
    m_transformResources = std::forward<TransformResourcesT>(value);
  }
  template <typename TransformResourcesT = TransformResources>
  CreateTransformJobRequest& WithTransformResources(TransformResourcesT&& value) {
    SetTransformResources(std::forward<TransformResourcesT>(value));
    return *this;
  }
 
 
  inline const DataProcessing& GetDataProcessing() const { return m_dataProcessing; }
  inline bool DataProcessingHasBeenSet() const { return m_dataProcessingHasBeenSet; }
  template <typename DataProcessingT = DataProcessing>
  void SetDataProcessing(DataProcessingT&& value) {
    m_dataProcessingHasBeenSet = true;
    m_dataProcessing = std::forward<DataProcessingT>(value);
  }
  template <typename DataProcessingT = DataProcessing>
  CreateTransformJobRequest& WithDataProcessing(DataProcessingT&& value) {
    SetDataProcessing(std::forward<DataProcessingT>(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>>
  CreateTransformJobRequest& WithTags(TagsT&& value) {
    SetTags(std::forward<TagsT>(value));
    return *this;
  }
  template <typename TagsT = Tag>
  CreateTransformJobRequest& AddTags(TagsT&& value) {
    m_tagsHasBeenSet = true;
    m_tags.emplace_back(std::forward<TagsT>(value));
    return *this;
  }
 
 
  inline const ExperimentConfig& GetExperimentConfig() const { return m_experimentConfig; }
  inline bool ExperimentConfigHasBeenSet() const { return m_experimentConfigHasBeenSet; }
  template <typename ExperimentConfigT = ExperimentConfig>
  void SetExperimentConfig(ExperimentConfigT&& value) {
    m_experimentConfigHasBeenSet = true;
    m_experimentConfig = std::forward<ExperimentConfigT>(value);
  }
  template <typename ExperimentConfigT = ExperimentConfig>
  CreateTransformJobRequest& WithExperimentConfig(ExperimentConfigT&& value) {
    SetExperimentConfig(std::forward<ExperimentConfigT>(value));
    return *this;
  }
 private:
  Aws::String m_transformJobName;
 
  Aws::String m_modelName;
 
  int m_maxConcurrentTransforms{0};
 
  ModelClientConfig m_modelClientConfig;
 
  int m_maxPayloadInMB{0};
 
  BatchStrategy m_batchStrategy{BatchStrategy::NOT_SET};
 
  Aws::Map<Aws::String, Aws::String> m_environment;
 
  TransformInput m_transformInput;
 
  TransformOutput m_transformOutput;
 
  BatchDataCaptureConfig m_dataCaptureConfig;
 
  TransformResources m_transformResources;
 
  DataProcessing m_dataProcessing;
 
  Aws::Vector<Tag> m_tags;
 
  ExperimentConfig m_experimentConfig;
  bool m_transformJobNameHasBeenSet = false;
  bool m_modelNameHasBeenSet = false;
  bool m_maxConcurrentTransformsHasBeenSet = false;
  bool m_modelClientConfigHasBeenSet = false;
  bool m_maxPayloadInMBHasBeenSet = false;
  bool m_batchStrategyHasBeenSet = false;
  bool m_environmentHasBeenSet = false;
  bool m_transformInputHasBeenSet = false;
  bool m_transformOutputHasBeenSet = false;
  bool m_dataCaptureConfigHasBeenSet = false;
  bool m_transformResourcesHasBeenSet = false;
  bool m_dataProcessingHasBeenSet = false;
  bool m_tagsHasBeenSet = false;
  bool m_experimentConfigHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws