InvokeEndpointAsyncRequest.h

 
#pragma once
#include <aws/core/utils/Array.h>
#include <aws/core/utils/memory/stl/AWSString.h>
#include <aws/sagemaker-runtime/SageMakerRuntimeRequest.h>
#include <aws/sagemaker-runtime/SageMakerRuntime_EXPORTS.h>
 
#include <utility>
 
namespace Aws {
namespace SageMakerRuntime {
namespace Model {
 
class InvokeEndpointAsyncRequest : public StreamingSageMakerRuntimeRequest {
 public:
  AWS_SAGEMAKERRUNTIME_API InvokeEndpointAsyncRequest() = default;
 
  // Service request name is the Operation name which will send this request out,
  // each operation should has unique request name, so that we can get operation's name from this request.
  // Note: this is not true for response, multiple operations may have the same response name,
  // so we can not get operation's name from response.
  inline virtual const char* GetServiceRequestName() const override { return "InvokeEndpointAsync"; }
 
  AWS_SAGEMAKERRUNTIME_API Aws::Http::HeaderValueCollection GetRequestSpecificHeaders() const override;
 
 
  inline const Aws::String& GetEndpointName() const { return m_endpointName; }
  inline bool EndpointNameHasBeenSet() const { return m_endpointNameHasBeenSet; }
  template <typename EndpointNameT = Aws::String>
  void SetEndpointName(EndpointNameT&& value) {
    m_endpointNameHasBeenSet = true;
    m_endpointName = std::forward<EndpointNameT>(value);
  }
  template <typename EndpointNameT = Aws::String>
  InvokeEndpointAsyncRequest& WithEndpointName(EndpointNameT&& value) {
    SetEndpointName(std::forward<EndpointNameT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetAccept() const { return m_accept; }
  inline bool AcceptHasBeenSet() const { return m_acceptHasBeenSet; }
  template <typename AcceptT = Aws::String>
  void SetAccept(AcceptT&& value) {
    m_acceptHasBeenSet = true;
    m_accept = std::forward<AcceptT>(value);
  }
  template <typename AcceptT = Aws::String>
  InvokeEndpointAsyncRequest& WithAccept(AcceptT&& value) {
    SetAccept(std::forward<AcceptT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetCustomAttributes() const { return m_customAttributes; }
  inline bool CustomAttributesHasBeenSet() const { return m_customAttributesHasBeenSet; }
  template <typename CustomAttributesT = Aws::String>
  void SetCustomAttributes(CustomAttributesT&& value) {
    m_customAttributesHasBeenSet = true;
    m_customAttributes = std::forward<CustomAttributesT>(value);
  }
  template <typename CustomAttributesT = Aws::String>
  InvokeEndpointAsyncRequest& WithCustomAttributes(CustomAttributesT&& value) {
    SetCustomAttributes(std::forward<CustomAttributesT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetInferenceId() const { return m_inferenceId; }
  inline bool InferenceIdHasBeenSet() const { return m_inferenceIdHasBeenSet; }
  template <typename InferenceIdT = Aws::String>
  void SetInferenceId(InferenceIdT&& value) {
    m_inferenceIdHasBeenSet = true;
    m_inferenceId = std::forward<InferenceIdT>(value);
  }
  template <typename InferenceIdT = Aws::String>
  InvokeEndpointAsyncRequest& WithInferenceId(InferenceIdT&& value) {
    SetInferenceId(std::forward<InferenceIdT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetInputLocation() const { return m_inputLocation; }
  inline bool InputLocationHasBeenSet() const { return m_inputLocationHasBeenSet; }
  template <typename InputLocationT = Aws::String>
  void SetInputLocation(InputLocationT&& value) {
    m_inputLocationHasBeenSet = true;
    m_inputLocation = std::forward<InputLocationT>(value);
  }
  template <typename InputLocationT = Aws::String>
  InvokeEndpointAsyncRequest& WithInputLocation(InputLocationT&& value) {
    SetInputLocation(std::forward<InputLocationT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetS3OutputPathExtension() const { return m_s3OutputPathExtension; }
  inline bool S3OutputPathExtensionHasBeenSet() const { return m_s3OutputPathExtensionHasBeenSet; }
  template <typename S3OutputPathExtensionT = Aws::String>
  void SetS3OutputPathExtension(S3OutputPathExtensionT&& value) {
    m_s3OutputPathExtensionHasBeenSet = true;
    m_s3OutputPathExtension = std::forward<S3OutputPathExtensionT>(value);
  }
  template <typename S3OutputPathExtensionT = Aws::String>
  InvokeEndpointAsyncRequest& WithS3OutputPathExtension(S3OutputPathExtensionT&& value) {
    SetS3OutputPathExtension(std::forward<S3OutputPathExtensionT>(value));
    return *this;
  }
 
 
  inline const Aws::String& GetFilename() const { return m_filename; }
  inline bool FilenameHasBeenSet() const { return m_filenameHasBeenSet; }
  template <typename FilenameT = Aws::String>
  void SetFilename(FilenameT&& value) {
    m_filenameHasBeenSet = true;
    m_filename = std::forward<FilenameT>(value);
  }
  template <typename FilenameT = Aws::String>
  InvokeEndpointAsyncRequest& WithFilename(FilenameT&& value) {
    SetFilename(std::forward<FilenameT>(value));
    return *this;
  }
 
 
  inline int GetRequestTTLSeconds() const { return m_requestTTLSeconds; }
  inline bool RequestTTLSecondsHasBeenSet() const { return m_requestTTLSecondsHasBeenSet; }
  inline void SetRequestTTLSeconds(int value) {
    m_requestTTLSecondsHasBeenSet = true;
    m_requestTTLSeconds = value;
  }
  inline InvokeEndpointAsyncRequest& WithRequestTTLSeconds(int value) {
    SetRequestTTLSeconds(value);
    return *this;
  }
 
 
  inline int GetInvocationTimeoutSeconds() const { return m_invocationTimeoutSeconds; }
  inline bool InvocationTimeoutSecondsHasBeenSet() const { return m_invocationTimeoutSecondsHasBeenSet; }
  inline void SetInvocationTimeoutSeconds(int value) {
    m_invocationTimeoutSecondsHasBeenSet = true;
    m_invocationTimeoutSeconds = value;
  }
  inline InvokeEndpointAsyncRequest& WithInvocationTimeoutSeconds(int value) {
    SetInvocationTimeoutSeconds(value);
    return *this;
  }
 private:
  Aws::String m_endpointName;
 
  Aws::String m_accept;
 
  Aws::String m_customAttributes;
 
  Aws::String m_inferenceId;
 
  Aws::String m_inputLocation;
 
  Aws::String m_s3OutputPathExtension;
 
  Aws::String m_filename;
 
  int m_requestTTLSeconds{0};
 
  int m_invocationTimeoutSeconds{0};
 
  bool m_endpointNameHasBeenSet = false;
  bool m_acceptHasBeenSet = false;
  bool m_customAttributesHasBeenSet = false;
  bool m_inferenceIdHasBeenSet = false;
  bool m_inputLocationHasBeenSet = false;
  bool m_s3OutputPathExtensionHasBeenSet = false;
  bool m_filenameHasBeenSet = false;
  bool m_requestTTLSecondsHasBeenSet = false;
  bool m_invocationTimeoutSecondsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMakerRuntime
}  // namespace Aws