1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74

mod noop;
#[cfg(all(
    not(target_arch = "wasm32"),
    any(feature = "enable_reqwest", feature = "enable_reqwest_rustls")
))]
mod reqwest;
#[cfg(all(
    target_arch = "wasm32",
    any(feature = "enable_reqwest", feature = "enable_reqwest_rustls")
))]
compile_error!("The `enable_request` and `enable_reqwest_rustls` features are not allowed for `wasm32` targets");

#[cfg(all(
    not(target_arch = "wasm32"),
    any(feature = "enable_reqwest", feature = "enable_reqwest_rustls")
))]
pub use self::reqwest::*;
pub use noop::*;

use std::sync::Arc;

/// Construct a new `HttpClient`
pub fn new_http_client() -> Arc<dyn HttpClient> {
    #[allow(unused)]
    let http_client: Arc<dyn HttpClient> = Arc::new(NoopClient);
    #[cfg(all(
        not(target_arch = "wasm32"),
        any(feature = "enable_reqwest", feature = "enable_reqwest_rustls")
    ))]
    let http_client = new_reqwest_client();
    http_client
}

use crate::error::ErrorKind;
use async_trait::async_trait;
use bytes::Bytes;
use serde::Serialize;

/// An HTTP client which can send requests.
#[cfg_attr(target_arch = "wasm32", async_trait(?Send))]
#[cfg_attr(not(target_arch = "wasm32"), async_trait)]
pub trait HttpClient: Send + Sync + std::fmt::Debug {
    /// Send out a request using `azure_core`'s types.
    ///
    /// It does not consume the request. Implementors are expected to clone the necessary parts
    /// of the request and pass them to the underlying transport.
    async fn execute_request(&self, request: &crate::Request) -> crate::Result<crate::Response>;

    /// DEPRECATED: the status check will be responsibility of another policy (not the transport one).
    /// Send out the request and collect the response body.
    /// An error is returned if the status is not success.
    async fn execute_request_check_status(
        &self,
        request: &crate::Request,
    ) -> crate::Result<crate::CollectedResponse> {
        let rsp = self.execute_request(request).await?;
        let (status, headers, body) = rsp.deconstruct();
        let body = body.collect().await?;

        if status.is_success() {
            Ok(crate::CollectedResponse::new(status, headers, body))
        } else {
            Err(ErrorKind::http_response_from_body(status, &body).into_error())
        }
    }
}

/// Serialize a type to json.
pub fn to_json<T>(value: &T) -> crate::Result<Bytes>
where
    T: ?Sized + Serialize,
{
    Ok(Bytes::from(serde_json::to_vec(value)?))
}