TransformJobDefinition.h

 
#pragma once
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/BatchStrategy.h>
#include <aws/sagemaker/model/TransformInput.h>
#include <aws/sagemaker/model/TransformOutput.h>
#include <aws/sagemaker/model/TransformResources.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class TransformJobDefinition {
 public:
  AWS_SAGEMAKER_API TransformJobDefinition() = default;
  AWS_SAGEMAKER_API TransformJobDefinition(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API TransformJobDefinition& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  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 TransformJobDefinition& WithMaxConcurrentTransforms(int value) {
    SetMaxConcurrentTransforms(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 TransformJobDefinition& 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 TransformJobDefinition& 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>>
  TransformJobDefinition& WithEnvironment(EnvironmentT&& value) {
    SetEnvironment(std::forward<EnvironmentT>(value));
    return *this;
  }
  template <typename EnvironmentKeyT = Aws::String, typename EnvironmentValueT = Aws::String>
  TransformJobDefinition& 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>
  TransformJobDefinition& 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>
  TransformJobDefinition& WithTransformOutput(TransformOutputT&& value) {
    SetTransformOutput(std::forward<TransformOutputT>(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>
  TransformJobDefinition& WithTransformResources(TransformResourcesT&& value) {
    SetTransformResources(std::forward<TransformResourcesT>(value));
    return *this;
  }
 private:
  int m_maxConcurrentTransforms{0};
 
  int m_maxPayloadInMB{0};
 
  BatchStrategy m_batchStrategy{BatchStrategy::NOT_SET};
 
  Aws::Map<Aws::String, Aws::String> m_environment;
 
  TransformInput m_transformInput;
 
  TransformOutput m_transformOutput;
 
  TransformResources m_transformResources;
  bool m_maxConcurrentTransformsHasBeenSet = false;
  bool m_maxPayloadInMBHasBeenSet = false;
  bool m_batchStrategyHasBeenSet = false;
  bool m_environmentHasBeenSet = false;
  bool m_transformInputHasBeenSet = false;
  bool m_transformOutputHasBeenSet = false;
  bool m_transformResourcesHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws