Skip to main content

cloud_sdk_reqwest/blocking/
raw.rs

1use core::fmt;
2
3use cloud_sdk::transport::{
4    BlockingRawHttpExecutor, BoundTransport, EndpointIdentity, EndpointIdentityError,
5    RawResponsePolicy, ResponseStorageSanitizer, ResponseWriter, TransportFailure,
6    TransportRequest,
7};
8use cloud_sdk_sanitization::sanitize_bytes;
9use http::header::HeaderValue;
10
11use crate::shared::{HttpsEndpoint, RawHttpError, RawHyperClient, RawTransportFailure};
12
13/// Raw blocking HTTP executor with no implicit authentication or provider policy.
14#[derive(Clone)]
15pub struct RawBlockingClient {
16    inner: RawHyperClient,
17    endpoint: HttpsEndpoint,
18}
19
20impl RawBlockingClient {
21    pub(super) const fn new(inner: RawHyperClient, endpoint: HttpsEndpoint) -> Self {
22        Self { inner, endpoint }
23    }
24
25    fn execute_inner(
26        &self,
27        request: TransportRequest<'_>,
28        policy: RawResponsePolicy<'_>,
29        response_writer: &mut ResponseWriter<'_>,
30    ) -> Result<(), RawTransportFailure> {
31        if tokio::runtime::Handle::try_current().is_ok() {
32            return Err(TransportFailure::not_sent(
33                RawHttpError::BlockingRuntimeContext,
34            ));
35        }
36        let runtime = tokio::runtime::Builder::new_current_thread()
37            .enable_all()
38            .build()
39            .map_err(|_| TransportFailure::not_sent(RawHttpError::RuntimeInitializationFailed))?;
40        let mut attempt = response_writer
41            .begin_attempt()
42            .map_err(|_| TransportFailure::not_sent(RawHttpError::ResponseAlreadyCommitted))?;
43        let completion = runtime.block_on(self.inner.execute(request, policy, &mut attempt))?;
44        attempt
45            .commit_completion(completion)
46            .map_err(|_| TransportFailure::response_started(RawHttpError::ResponseCommitFailed))
47    }
48
49    pub(crate) fn execute_authenticated(
50        &self,
51        request: TransportRequest<'_>,
52        policy: RawResponsePolicy<'_>,
53        authorization: HeaderValue,
54        response_writer: &mut ResponseWriter<'_>,
55    ) -> Result<(), RawTransportFailure> {
56        if tokio::runtime::Handle::try_current().is_ok() {
57            return Err(TransportFailure::not_sent(
58                RawHttpError::BlockingRuntimeContext,
59            ));
60        }
61        let runtime = tokio::runtime::Builder::new_current_thread()
62            .enable_all()
63            .build()
64            .map_err(|_| TransportFailure::not_sent(RawHttpError::RuntimeInitializationFailed))?;
65        let mut attempt = response_writer
66            .begin_attempt()
67            .map_err(|_| TransportFailure::not_sent(RawHttpError::ResponseAlreadyCommitted))?;
68        let completion = runtime.block_on(self.inner.execute_authenticated(
69            request,
70            policy,
71            authorization,
72            &mut attempt,
73        ))?;
74        attempt
75            .commit_completion(completion)
76            .map_err(|_| TransportFailure::response_started(RawHttpError::ResponseCommitFailed))
77    }
78}
79
80impl BlockingRawHttpExecutor for RawBlockingClient {
81    type Error = RawTransportFailure;
82
83    fn execute(
84        &self,
85        request: TransportRequest<'_>,
86        policy: RawResponsePolicy<'_>,
87        response: &mut ResponseWriter<'_>,
88    ) -> Result<(), Self::Error> {
89        self.execute_inner(request, policy, response)
90    }
91}
92
93impl ResponseStorageSanitizer for RawBlockingClient {
94    fn sanitize_response_storage(&self, response_storage: &mut [u8]) {
95        sanitize_bytes(response_storage);
96    }
97}
98
99impl BoundTransport for RawBlockingClient {
100    fn endpoint_identity(&self) -> Result<EndpointIdentity<'_>, EndpointIdentityError> {
101        self.endpoint.identity()
102    }
103}
104
105impl fmt::Debug for RawBlockingClient {
106    fn fmt(&self, formatter: &mut fmt::Formatter<'_>) -> fmt::Result {
107        formatter
108            .debug_struct("RawBlockingClient")
109            .field("endpoint", &"[redacted]")
110            .finish_non_exhaustive()
111    }
112}