ResponseStream.h

 
#pragma once
#include <aws/sagemaker-runtime/SageMakerRuntime_EXPORTS.h>
#include <aws/sagemaker-runtime/model/ModelStreamError.h>
#include <aws/sagemaker-runtime/model/PayloadPart.h>
 
#include <utility>
 
namespace Aws {
namespace Utils {
namespace Json {
class JsonValue;
class JsonView;
}  // namespace Json
}  // namespace Utils
namespace SageMakerRuntime {
namespace Model {
 
class ResponseStream {
 public:
  AWS_SAGEMAKERRUNTIME_API ResponseStream() = default;
  AWS_SAGEMAKERRUNTIME_API ResponseStream(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKERRUNTIME_API ResponseStream& operator=(Aws::Utils::Json::JsonView jsonValue);
  AWS_SAGEMAKERRUNTIME_API Aws::Utils::Json::JsonValue Jsonize() const;
 
 
  inline const PayloadPart& GetPayloadPart() const { return m_payloadPart; }
  inline bool PayloadPartHasBeenSet() const { return m_payloadPartHasBeenSet; }
  template <typename PayloadPartT = PayloadPart>
  void SetPayloadPart(PayloadPartT&& value) {
    m_payloadPartHasBeenSet = true;
    m_payloadPart = std::forward<PayloadPartT>(value);
  }
  template <typename PayloadPartT = PayloadPart>
  ResponseStream& WithPayloadPart(PayloadPartT&& value) {
    SetPayloadPart(std::forward<PayloadPartT>(value));
    return *this;
  }
 
 
  inline const ModelStreamError& GetModelStreamError() const { return m_modelStreamError; }
  inline bool ModelStreamErrorHasBeenSet() const { return m_modelStreamErrorHasBeenSet; }
  template <typename ModelStreamErrorT = ModelStreamError>
  void SetModelStreamError(ModelStreamErrorT&& value) {
    m_modelStreamErrorHasBeenSet = true;
    m_modelStreamError = std::forward<ModelStreamErrorT>(value);
  }
  template <typename ModelStreamErrorT = ModelStreamError>
  ResponseStream& WithModelStreamError(ModelStreamErrorT&& value) {
    SetModelStreamError(std::forward<ModelStreamErrorT>(value));
    return *this;
  }
 
 
  inline const SageMakerRuntimeError& GetInternalStreamFailure() const { return m_internalStreamFailure; }
  inline bool InternalStreamFailureHasBeenSet() const { return m_internalStreamFailureHasBeenSet; }
  template <typename InternalStreamFailureT = SageMakerRuntimeError>
  void SetInternalStreamFailure(InternalStreamFailureT&& value) {
    m_internalStreamFailureHasBeenSet = true;
    m_internalStreamFailure = std::forward<InternalStreamFailureT>(value);
  }
  template <typename InternalStreamFailureT = SageMakerRuntimeError>
  ResponseStream& WithInternalStreamFailure(InternalStreamFailureT&& value) {
    SetInternalStreamFailure(std::forward<InternalStreamFailureT>(value));
    return *this;
  }
 private:
  PayloadPart m_payloadPart;
 
  ModelStreamError m_modelStreamError;
 
  SageMakerRuntimeError m_internalStreamFailure;
  bool m_payloadPartHasBeenSet = false;
  bool m_modelStreamErrorHasBeenSet = false;
  bool m_internalStreamFailureHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMakerRuntime
}  // namespace Aws