ContainerDefinition.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemaker/model/ImageConfig.h>
#include <aws/sagemaker/model/ContainerMode.h>
#include <aws/sagemaker/model/ModelDataSource.h>
#include <aws/core/utils/memory/stl/AWSVector.h>
#include <aws/core/utils/memory/stl/AWSMap.h>
#include <aws/sagemaker/model/MultiModelConfig.h>
#include <aws/sagemaker/model/AdditionalModelDataSource.h>
#include <utility>
 
namespace Aws
{
namespace Utils
{
namespace Json
{
  class JsonValue;
  class JsonView;
} // namespace Json
} // namespace Utils
namespace SageMaker
{
namespace Model
{
 
  class ContainerDefinition
  {
  public:
    AWS_SAGEMAKER_API ContainerDefinition() = default;
    AWS_SAGEMAKER_API ContainerDefinition(Aws::Utils::Json::JsonView jsonValue);
    AWS_SAGEMAKER_API ContainerDefinition& operator=(Aws::Utils::Json::JsonView jsonValue);
    AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
 
    inline const Aws::String& GetContainerHostname() const { return m_containerHostname; }
    inline bool ContainerHostnameHasBeenSet() const { return m_containerHostnameHasBeenSet; }
    template<typename ContainerHostnameT = Aws::String>
    void SetContainerHostname(ContainerHostnameT&& value) { m_containerHostnameHasBeenSet = true; m_containerHostname = std::forward<ContainerHostnameT>(value); }
    template<typename ContainerHostnameT = Aws::String>
    ContainerDefinition& WithContainerHostname(ContainerHostnameT&& value) { SetContainerHostname(std::forward<ContainerHostnameT>(value)); return *this;}
 
 
    inline const Aws::String& GetImage() const { return m_image; }
    inline bool ImageHasBeenSet() const { return m_imageHasBeenSet; }
    template<typename ImageT = Aws::String>
    void SetImage(ImageT&& value) { m_imageHasBeenSet = true; m_image = std::forward<ImageT>(value); }
    template<typename ImageT = Aws::String>
    ContainerDefinition& WithImage(ImageT&& value) { SetImage(std::forward<ImageT>(value)); return *this;}
 
 
    inline const ImageConfig& GetImageConfig() const { return m_imageConfig; }
    inline bool ImageConfigHasBeenSet() const { return m_imageConfigHasBeenSet; }
    template<typename ImageConfigT = ImageConfig>
    void SetImageConfig(ImageConfigT&& value) { m_imageConfigHasBeenSet = true; m_imageConfig = std::forward<ImageConfigT>(value); }
    template<typename ImageConfigT = ImageConfig>
    ContainerDefinition& WithImageConfig(ImageConfigT&& value) { SetImageConfig(std::forward<ImageConfigT>(value)); return *this;}
 
 
    inline ContainerMode GetMode() const { return m_mode; }
    inline bool ModeHasBeenSet() const { return m_modeHasBeenSet; }
    inline void SetMode(ContainerMode value) { m_modeHasBeenSet = true; m_mode = value; }
    inline ContainerDefinition& WithMode(ContainerMode value) { SetMode(value); return *this;}
 
 
    inline const Aws::String& GetModelDataUrl() const { return m_modelDataUrl; }
    inline bool ModelDataUrlHasBeenSet() const { return m_modelDataUrlHasBeenSet; }
    template<typename ModelDataUrlT = Aws::String>
    void SetModelDataUrl(ModelDataUrlT&& value) { m_modelDataUrlHasBeenSet = true; m_modelDataUrl = std::forward<ModelDataUrlT>(value); }
    template<typename ModelDataUrlT = Aws::String>
    ContainerDefinition& WithModelDataUrl(ModelDataUrlT&& value) { SetModelDataUrl(std::forward<ModelDataUrlT>(value)); return *this;}
 
 
    inline const ModelDataSource& GetModelDataSource() const { return m_modelDataSource; }
    inline bool ModelDataSourceHasBeenSet() const { return m_modelDataSourceHasBeenSet; }
    template<typename ModelDataSourceT = ModelDataSource>
    void SetModelDataSource(ModelDataSourceT&& value) { m_modelDataSourceHasBeenSet = true; m_modelDataSource = std::forward<ModelDataSourceT>(value); }
    template<typename ModelDataSourceT = ModelDataSource>
    ContainerDefinition& WithModelDataSource(ModelDataSourceT&& value) { SetModelDataSource(std::forward<ModelDataSourceT>(value)); return *this;}
 
 
    inline const Aws::Vector<AdditionalModelDataSource>& GetAdditionalModelDataSources() const { return m_additionalModelDataSources; }
    inline bool AdditionalModelDataSourcesHasBeenSet() const { return m_additionalModelDataSourcesHasBeenSet; }
    template<typename AdditionalModelDataSourcesT = Aws::Vector<AdditionalModelDataSource>>
    void SetAdditionalModelDataSources(AdditionalModelDataSourcesT&& value) { m_additionalModelDataSourcesHasBeenSet = true; m_additionalModelDataSources = std::forward<AdditionalModelDataSourcesT>(value); }
    template<typename AdditionalModelDataSourcesT = Aws::Vector<AdditionalModelDataSource>>
    ContainerDefinition& WithAdditionalModelDataSources(AdditionalModelDataSourcesT&& value) { SetAdditionalModelDataSources(std::forward<AdditionalModelDataSourcesT>(value)); return *this;}
    template<typename AdditionalModelDataSourcesT = AdditionalModelDataSource>
    ContainerDefinition& AddAdditionalModelDataSources(AdditionalModelDataSourcesT&& value) { m_additionalModelDataSourcesHasBeenSet = true; m_additionalModelDataSources.emplace_back(std::forward<AdditionalModelDataSourcesT>(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>>
    ContainerDefinition& WithEnvironment(EnvironmentT&& value) { SetEnvironment(std::forward<EnvironmentT>(value)); return *this;}
    template<typename EnvironmentKeyT = Aws::String, typename EnvironmentValueT = Aws::String>
    ContainerDefinition& AddEnvironment(EnvironmentKeyT&& key, EnvironmentValueT&& value) {
      m_environmentHasBeenSet = true; m_environment.emplace(std::forward<EnvironmentKeyT>(key), std::forward<EnvironmentValueT>(value)); return *this;
    }
 
 
    inline const Aws::String& GetModelPackageName() const { return m_modelPackageName; }
    inline bool ModelPackageNameHasBeenSet() const { return m_modelPackageNameHasBeenSet; }
    template<typename ModelPackageNameT = Aws::String>
    void SetModelPackageName(ModelPackageNameT&& value) { m_modelPackageNameHasBeenSet = true; m_modelPackageName = std::forward<ModelPackageNameT>(value); }
    template<typename ModelPackageNameT = Aws::String>
    ContainerDefinition& WithModelPackageName(ModelPackageNameT&& value) { SetModelPackageName(std::forward<ModelPackageNameT>(value)); return *this;}
 
 
    inline const Aws::String& GetInferenceSpecificationName() const { return m_inferenceSpecificationName; }
    inline bool InferenceSpecificationNameHasBeenSet() const { return m_inferenceSpecificationNameHasBeenSet; }
    template<typename InferenceSpecificationNameT = Aws::String>
    void SetInferenceSpecificationName(InferenceSpecificationNameT&& value) { m_inferenceSpecificationNameHasBeenSet = true; m_inferenceSpecificationName = std::forward<InferenceSpecificationNameT>(value); }
    template<typename InferenceSpecificationNameT = Aws::String>
    ContainerDefinition& WithInferenceSpecificationName(InferenceSpecificationNameT&& value) { SetInferenceSpecificationName(std::forward<InferenceSpecificationNameT>(value)); return *this;}
 
 
    inline const MultiModelConfig& GetMultiModelConfig() const { return m_multiModelConfig; }
    inline bool MultiModelConfigHasBeenSet() const { return m_multiModelConfigHasBeenSet; }
    template<typename MultiModelConfigT = MultiModelConfig>
    void SetMultiModelConfig(MultiModelConfigT&& value) { m_multiModelConfigHasBeenSet = true; m_multiModelConfig = std::forward<MultiModelConfigT>(value); }
    template<typename MultiModelConfigT = MultiModelConfig>
    ContainerDefinition& WithMultiModelConfig(MultiModelConfigT&& value) { SetMultiModelConfig(std::forward<MultiModelConfigT>(value)); return *this;}
  private:
 
    Aws::String m_containerHostname;
    bool m_containerHostnameHasBeenSet = false;
 
    Aws::String m_image;
    bool m_imageHasBeenSet = false;
 
    ImageConfig m_imageConfig;
    bool m_imageConfigHasBeenSet = false;
 
    ContainerMode m_mode{ContainerMode::NOT_SET};
    bool m_modeHasBeenSet = false;
 
    Aws::String m_modelDataUrl;
    bool m_modelDataUrlHasBeenSet = false;
 
    ModelDataSource m_modelDataSource;
    bool m_modelDataSourceHasBeenSet = false;
 
    Aws::Vector<AdditionalModelDataSource> m_additionalModelDataSources;
    bool m_additionalModelDataSourcesHasBeenSet = false;
 
    Aws::Map<Aws::String, Aws::String> m_environment;
    bool m_environmentHasBeenSet = false;
 
    Aws::String m_modelPackageName;
    bool m_modelPackageNameHasBeenSet = false;
 
    Aws::String m_inferenceSpecificationName;
    bool m_inferenceSpecificationNameHasBeenSet = false;
 
    MultiModelConfig m_multiModelConfig;
    bool m_multiModelConfigHasBeenSet = false;
  };
 
} // namespace Model
} // namespace SageMaker
} // namespace Aws