SageMakerRuntimeClient.h

 
#pragma once
#include <aws/core/client/AWSClient.h>
#include <aws/core/client/AWSClientAsyncCRTP.h>
#include <aws/core/client/ClientConfiguration.h>
#include <aws/core/utils/json/JsonSerializer.h>
#include <aws/sagemaker-runtime/SageMakerRuntimeServiceClientModel.h>
#include <aws/sagemaker-runtime/SageMakerRuntime_EXPORTS.h>
 
namespace Aws {
namespace SageMakerRuntime {
class AWS_SAGEMAKERRUNTIME_API SageMakerRuntimeClient : public Aws::Client::AWSJsonClient,
                                                        public Aws::Client::ClientWithAsyncTemplateMethods<SageMakerRuntimeClient> {
 public:
  typedef Aws::Client::AWSJsonClient BASECLASS;
  static const char* GetServiceName();
  static const char* GetAllocationTag();
 
  typedef SageMakerRuntimeClientConfiguration ClientConfigurationType;
  typedef SageMakerRuntimeEndpointProvider EndpointProviderType;
 
  SageMakerRuntimeClient(const Aws::SageMakerRuntime::SageMakerRuntimeClientConfiguration& clientConfiguration =
                             Aws::SageMakerRuntime::SageMakerRuntimeClientConfiguration(),
                         std::shared_ptr<SageMakerRuntimeEndpointProviderBase> endpointProvider = nullptr);
 
  SageMakerRuntimeClient(const Aws::Auth::AWSCredentials& credentials,
                         std::shared_ptr<SageMakerRuntimeEndpointProviderBase> endpointProvider = nullptr,
                         const Aws::SageMakerRuntime::SageMakerRuntimeClientConfiguration& clientConfiguration =
                             Aws::SageMakerRuntime::SageMakerRuntimeClientConfiguration());
 
  SageMakerRuntimeClient(const std::shared_ptr<Aws::Auth::AWSCredentialsProvider>& credentialsProvider,
                         std::shared_ptr<SageMakerRuntimeEndpointProviderBase> endpointProvider = nullptr,
                         const Aws::SageMakerRuntime::SageMakerRuntimeClientConfiguration& clientConfiguration =
                             Aws::SageMakerRuntime::SageMakerRuntimeClientConfiguration());
 
  /* Legacy constructors due deprecation */
  SageMakerRuntimeClient(const Aws::Client::ClientConfiguration& clientConfiguration);
 
  SageMakerRuntimeClient(const Aws::Auth::AWSCredentials& credentials, const Aws::Client::ClientConfiguration& clientConfiguration);
 
  SageMakerRuntimeClient(const std::shared_ptr<Aws::Auth::AWSCredentialsProvider>& credentialsProvider,
                         const Aws::Client::ClientConfiguration& clientConfiguration);
 
  /* End of legacy constructors due deprecation */
  virtual ~SageMakerRuntimeClient();
 
  virtual Model::InvokeEndpointOutcome InvokeEndpoint(const Model::InvokeEndpointRequest& request) const;
 
  template <typename InvokeEndpointRequestT = Model::InvokeEndpointRequest>
  Model::InvokeEndpointOutcomeCallable InvokeEndpointCallable(const InvokeEndpointRequestT& request) const {
    return SubmitCallable(&SageMakerRuntimeClient::InvokeEndpoint, request);
  }
 
  template <typename InvokeEndpointRequestT = Model::InvokeEndpointRequest>
  void InvokeEndpointAsync(const InvokeEndpointRequestT& request, const InvokeEndpointResponseReceivedHandler& handler,
                           const std::shared_ptr<const Aws::Client::AsyncCallerContext>& context = nullptr) const {
    return SubmitAsync(&SageMakerRuntimeClient::InvokeEndpoint, request, handler, context);
  }
 
  virtual Model::InvokeEndpointAsyncOutcome InvokeEndpointAsync(const Model::InvokeEndpointAsyncRequest& request) const;
 
  template <typename InvokeEndpointAsyncRequestT = Model::InvokeEndpointAsyncRequest>
  Model::InvokeEndpointAsyncOutcomeCallable InvokeEndpointAsyncCallable(const InvokeEndpointAsyncRequestT& request) const {
    return SubmitCallable(&SageMakerRuntimeClient::InvokeEndpointAsync, request);
  }
 
  template <typename InvokeEndpointAsyncRequestT = Model::InvokeEndpointAsyncRequest>
  void InvokeEndpointAsyncAsync(const InvokeEndpointAsyncRequestT& request, const InvokeEndpointAsyncResponseReceivedHandler& handler,
                                const std::shared_ptr<const Aws::Client::AsyncCallerContext>& context = nullptr) const {
    return SubmitAsync(&SageMakerRuntimeClient::InvokeEndpointAsync, request, handler, context);
  }
 
  virtual Model::InvokeEndpointWithResponseStreamOutcome InvokeEndpointWithResponseStream(
      Model::InvokeEndpointWithResponseStreamRequest& request) const;
 
  template <typename InvokeEndpointWithResponseStreamRequestT = Model::InvokeEndpointWithResponseStreamRequest>
  Model::InvokeEndpointWithResponseStreamOutcomeCallable InvokeEndpointWithResponseStreamCallable(
      InvokeEndpointWithResponseStreamRequestT& request) const {
    return SubmitCallable(&SageMakerRuntimeClient::InvokeEndpointWithResponseStream, request);
  }
 
  template <typename InvokeEndpointWithResponseStreamRequestT = Model::InvokeEndpointWithResponseStreamRequest>
  void InvokeEndpointWithResponseStreamAsync(InvokeEndpointWithResponseStreamRequestT& request,
                                             const InvokeEndpointWithResponseStreamResponseReceivedHandler& handler,
                                             const std::shared_ptr<const Aws::Client::AsyncCallerContext>& context = nullptr) const {
    return SubmitAsync(&SageMakerRuntimeClient::InvokeEndpointWithResponseStream, request, handler, context);
  }
 
  void OverrideEndpoint(const Aws::String& endpoint);
  std::shared_ptr<SageMakerRuntimeEndpointProviderBase>& accessEndpointProvider();
 
 private:
  friend class Aws::Client::ClientWithAsyncTemplateMethods<SageMakerRuntimeClient>;
  void init(const SageMakerRuntimeClientConfiguration& clientConfiguration);
 
  SageMakerRuntimeClientConfiguration m_clientConfiguration;
  std::shared_ptr<SageMakerRuntimeEndpointProviderBase> m_endpointProvider;
};
 
}  // namespace SageMakerRuntime
}  // namespace Aws