InvokeEndpointAsyncRequest.h

 
#pragma once
#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 SageMakerRuntimeRequest {
 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::String SerializePayload() const override;
 
  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& GetContentType() const { return m_contentType; }
  inline bool ContentTypeHasBeenSet() const { return m_contentTypeHasBeenSet; }
  template <typename ContentTypeT = Aws::String>
  void SetContentType(ContentTypeT&& value) {
    m_contentTypeHasBeenSet = true;
    m_contentType = std::forward<ContentTypeT>(value);
  }
  template <typename ContentTypeT = Aws::String>
  InvokeEndpointAsyncRequest& WithContentType(ContentTypeT&& value) {
    SetContentType(std::forward<ContentTypeT>(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 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_contentType;
 
  Aws::String m_accept;
 
  Aws::String m_customAttributes;
 
  Aws::String m_inferenceId;
 
  Aws::String m_inputLocation;
 
  int m_requestTTLSeconds{0};
 
  int m_invocationTimeoutSeconds{0};
  bool m_endpointNameHasBeenSet = false;
  bool m_contentTypeHasBeenSet = false;
  bool m_acceptHasBeenSet = false;
  bool m_customAttributesHasBeenSet = false;
  bool m_inferenceIdHasBeenSet = false;
  bool m_inputLocationHasBeenSet = false;
  bool m_requestTTLSecondsHasBeenSet = false;
  bool m_invocationTimeoutSecondsHasBeenSet = false;
};
 
}  // namespace Model
}  // namespace SageMakerRuntime
}  // namespace Aws