aws_smithy_runtime/client/http/test_util/
infallible.rsuse aws_smithy_runtime_api::client::connector_metadata::ConnectorMetadata;
use aws_smithy_runtime_api::client::http::{
HttpClient, HttpConnector, HttpConnectorFuture, HttpConnectorSettings, SharedHttpClient,
SharedHttpConnector,
};
use aws_smithy_runtime_api::client::orchestrator::{HttpRequest, HttpResponse};
use aws_smithy_runtime_api::client::result::ConnectorError;
use aws_smithy_runtime_api::client::runtime_components::RuntimeComponents;
use aws_smithy_runtime_api::shared::IntoShared;
use aws_smithy_types::body::SdkBody;
use std::fmt;
use std::sync::Arc;
pub fn infallible_client_fn<B>(
f: impl Fn(http_02x::Request<SdkBody>) -> http_02x::Response<B> + Send + Sync + 'static,
) -> SharedHttpClient
where
B: Into<SdkBody>,
{
InfallibleClientFn::new(f).into_shared()
}
#[derive(Clone)]
struct InfallibleClientFn {
#[allow(clippy::type_complexity)]
response: Arc<
dyn Fn(http_02x::Request<SdkBody>) -> Result<http_02x::Response<SdkBody>, ConnectorError>
+ Send
+ Sync,
>,
}
impl fmt::Debug for InfallibleClientFn {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("InfallibleClientFn").finish()
}
}
impl InfallibleClientFn {
fn new<B: Into<SdkBody>>(
f: impl Fn(http_02x::Request<SdkBody>) -> http_02x::Response<B> + Send + Sync + 'static,
) -> Self {
Self {
response: Arc::new(move |request| Ok(f(request).map(|b| b.into()))),
}
}
}
impl HttpConnector for InfallibleClientFn {
fn call(&self, request: HttpRequest) -> HttpConnectorFuture {
HttpConnectorFuture::ready(
(self.response)(request.try_into_http02x().unwrap())
.map(|res| HttpResponse::try_from(res).unwrap()),
)
}
}
impl HttpClient for InfallibleClientFn {
fn http_connector(
&self,
_: &HttpConnectorSettings,
_: &RuntimeComponents,
) -> SharedHttpConnector {
self.clone().into_shared()
}
fn connector_metadata(&self) -> Option<ConnectorMetadata> {
Some(ConnectorMetadata::new("infallible-client", None))
}
}