ThroughputConfig.h

 
#pragma once
#include <aws/sagemaker/SageMaker_EXPORTS.h>
#include <aws/sagemaker/model/ThroughputMode.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMaker {
namespace Model {
 
class ThroughputConfig {
 public:
  AWS_SAGEMAKER_API ThroughputConfig() = default;
  AWS_SAGEMAKER_API ThroughputConfig(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API ThroughputConfig& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKER_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline ThroughputMode GetThroughputMode() const { return m_throughputMode; }
  inline bool ThroughputModeHasBeenSet() const { return m_throughputModeHasBeenSet; }
  inline void SetThroughputMode(ThroughputMode value) {
    m_throughputModeHasBeenSet = true;
    m_throughputMode = value;
  }
  inline ThroughputConfig& WithThroughputMode(ThroughputMode value) {
    SetThroughputMode(value);
    return *this;
  }
 
 
  inline int GetProvisionedReadCapacityUnits() const { return m_provisionedReadCapacityUnits; }
  inline bool ProvisionedReadCapacityUnitsHasBeenSet() const { return m_provisionedReadCapacityUnitsHasBeenSet; }
  inline void SetProvisionedReadCapacityUnits(int value) {
    m_provisionedReadCapacityUnitsHasBeenSet = true;
    m_provisionedReadCapacityUnits = value;
  }
  inline ThroughputConfig& WithProvisionedReadCapacityUnits(int value) {
    SetProvisionedReadCapacityUnits(value);
    return *this;
  }
 
 
  inline int GetProvisionedWriteCapacityUnits() const { return m_provisionedWriteCapacityUnits; }
  inline bool ProvisionedWriteCapacityUnitsHasBeenSet() const { return m_provisionedWriteCapacityUnitsHasBeenSet; }
  inline void SetProvisionedWriteCapacityUnits(int value) {
    m_provisionedWriteCapacityUnitsHasBeenSet = true;
    m_provisionedWriteCapacityUnits = value;
  }
  inline ThroughputConfig& WithProvisionedWriteCapacityUnits(int value) {
    SetProvisionedWriteCapacityUnits(value);
    return *this;
  }
 private:
  ThroughputMode m_throughputMode{ThroughputMode::NOT_SET};
 
  int m_provisionedReadCapacityUnits{0};
 
  int m_provisionedWriteCapacityUnits{0};
  bool m_throughputModeHasBeenSet = false;
  bool m_provisionedReadCapacityUnitsHasBeenSet = false;
  bool m_provisionedWriteCapacityUnitsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMaker
}  // namespace Aws