cloud_sdk_reqwest/blocking/
raw.rs1use 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#[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}