DataSource.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/DatasetSource.h>
#include <aws/sagemaker/model/FileSystemDataSource.h>
#include <aws/sagemaker/model/S3DataSource.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class DataSource {
 public:
  AWS_SAGEMAKER_API DataSource() = default;
  AWS_SAGEMAKER_API DataSource(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API DataSource& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const S3DataSource& GetS3DataSource() const { return m_s3DataSource; }
  inline bool S3DataSourceHasBeenSet() const { return m_s3DataSourceHasBeenSet; }
  template <typename S3DataSourceT = S3DataSource>
  void SetS3DataSource(S3DataSourceT&& value) {
    m_s3DataSourceHasBeenSet = true;
    m_s3DataSource = std::forward<S3DataSourceT>(value);
  }
  template <typename S3DataSourceT = S3DataSource>
  DataSource& WithS3DataSource(S3DataSourceT&& value) {
    SetS3DataSource(std::forward<S3DataSourceT>(value));
    return *this;
  }
 
 
  inline const FileSystemDataSource& GetFileSystemDataSource() const { return m_fileSystemDataSource; }
  inline bool FileSystemDataSourceHasBeenSet() const { return m_fileSystemDataSourceHasBeenSet; }
  template <typename FileSystemDataSourceT = FileSystemDataSource>
  void SetFileSystemDataSource(FileSystemDataSourceT&& value) {
    m_fileSystemDataSourceHasBeenSet = true;
    m_fileSystemDataSource = std::forward<FileSystemDataSourceT>(value);
  }
  template <typename FileSystemDataSourceT = FileSystemDataSource>
  DataSource& WithFileSystemDataSource(FileSystemDataSourceT&& value) {
    SetFileSystemDataSource(std::forward<FileSystemDataSourceT>(value));
    return *this;
  }
 
 
  inline const DatasetSource& GetDatasetSource() const { return m_datasetSource; }
  inline bool DatasetSourceHasBeenSet() const { return m_datasetSourceHasBeenSet; }
  template <typename DatasetSourceT = DatasetSource>
  void SetDatasetSource(DatasetSourceT&& value) {
    m_datasetSourceHasBeenSet = true;
    m_datasetSource = std::forward<DatasetSourceT>(value);
  }
  template <typename DatasetSourceT = DatasetSource>
  DataSource& WithDatasetSource(DatasetSourceT&& value) {
    SetDatasetSource(std::forward<DatasetSourceT>(value));
    return *this;
  }
 private:
  S3DataSource m_s3DataSource;
 
  FileSystemDataSource m_fileSystemDataSource;
 
  DatasetSource m_datasetSource;
  bool m_s3DataSourceHasBeenSet = false;
  bool m_fileSystemDataSourceHasBeenSet = false;
  bool m_datasetSourceHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws