#[cfg(feature = "log")]
use super::log::{print_request_log, print_response_log};
use crate::{
config::{ClientConfig, RequestConfig},
error::{Error, ErrorKind, Result},
http::{Client, HttpRequest, HttpResponse},
map,
};
use async_trait::async_trait;
use http_0_2::{Request, Response, StatusCode, request::Builder};
use isahc::{AsyncBody, HttpClient, config::RedirectPolicy, prelude::*};
use std::collections::HashMap;
use std::time::Duration;
use url::Url;
#[derive(Clone)]
pub struct ClientWrapper {
isahc_client: HttpClient,
}
impl Client for ClientWrapper {
type Inner = HttpClient;
fn new() -> Result<ClientWrapper> {
let client = HttpClient::builder()
.default_header("user-agent", "Feign HTTP")
.redirect_policy(RedirectPolicy::Limit(10))
.build()
.map_err(Error::build)?;
Ok(ClientWrapper {
isahc_client: client,
})
}
fn with_config(config: ClientConfig) -> Result<ClientWrapper> {
let mut client_builder = HttpClient::builder().default_header("user-agent", "Feign HTTP");
if let Some(millisecond) = config.connect_timeout {
client_builder = client_builder.connect_timeout(Duration::from_millis(millisecond));
}
if let Some(millisecond) = config.timeout {
client_builder = client_builder.timeout(Duration::from_millis(millisecond));
}
if let Some(millisecond) = config.read_timeout {
client_builder =
client_builder.low_speed_timeout(1, Duration::from_millis(millisecond));
}
client_builder = client_builder.redirect_policy(RedirectPolicy::Limit(10));
let client = client_builder.build().map_err(Error::build)?;
Ok(ClientWrapper {
isahc_client: client,
})
}
fn with_client(client: HttpClient) -> Result<ClientWrapper> {
Ok(ClientWrapper {
isahc_client: client,
})
}
fn get_client(&self) -> &HttpClient {
&self.isahc_client
}
}
pub struct RequestWrapper {
client_wrapper: ClientWrapper,
url: Url,
headers: HashMap<String, String>,
request: Builder,
}
pub struct ResponseWrapper {
response: Response<AsyncBody>,
}
impl HttpRequest for RequestWrapper {
fn headers(mut self, headers: HashMap<&str, String>) -> Self {
for (k, v) in headers {
self.headers.insert(k.to_lowercase(), v);
}
self
}
fn query(mut self, query: Vec<(&str, String)>) -> Self {
if query.len() == 0 {
return self;
}
let request = self.request;
let uri_ref = request.uri_ref().unwrap();
let mut url = uri_ref.to_string();
match uri_ref.query() {
Some(_) => {
url.push_str("&");
}
None => {
url.push_str("?");
}
}
let query = serde_urlencoded::to_string(query).unwrap();
url.push_str(&query);
self.url = Url::parse(url.as_str()).unwrap();
self.request = request.uri(url);
self
}
}
impl RequestWrapper {
pub fn new(client_wrapper: ClientWrapper, url: &str, method: &str) -> Result<RequestWrapper> {
let url = Url::parse(url).map_err(Error::build)?;
let request = Request::builder()
.uri(url.to_string())
.method(method.to_uppercase().as_str());
Ok(RequestWrapper {
client_wrapper,
url,
headers: map!(),
request,
})
}
pub fn with_config(
client_wrapper: ClientWrapper,
url: &str,
method: &str,
config: RequestConfig,
) -> Result<RequestWrapper> {
let mut request = Request::builder();
if let Some(millisecond) = config.timeout {
request = request.timeout(Duration::from_millis(millisecond));
}
let url = Url::parse(url).map_err(Error::build)?;
request = request
.uri(url.to_string())
.method(method.to_uppercase().as_str());
Ok(RequestWrapper {
client_wrapper,
url,
headers: map!(),
request,
})
}
fn set_header(mut self) -> Self {
let mut request = self.request;
for (k, v) in &self.headers {
request = request.header(k.as_str(), v);
}
self.request = request;
self
}
fn set_header_if_absent(&mut self, k: &str, v: String) {
if let None = self.headers.get(k) {
self.headers.insert(k.to_string(), v);
}
}
async fn send_body(self, body: Option<Vec<u8>>) -> Result<ResponseWrapper> {
let url = self.url.clone();
let client_wrapper = self.client_wrapper.clone();
let mut async_body = AsyncBody::from(());
if let Some(body) = body.clone() {
async_body = AsyncBody::from(body);
}
let request = self
.set_header()
.request
.body(async_body)
.map_err(Error::build)?;
#[cfg(feature = "log")]
print_request_log(&request, body);
return match client_wrapper.get_client().send_async(request).await {
Ok(response) => {
#[cfg(feature = "log")]
print_response_log(&response);
let status = response.status();
if status.is_client_error() || status.is_server_error() {
return Err(Error::status(url, status));
}
Ok(ResponseWrapper { response })
}
Err(e) => Err(Error::new(ErrorKind::Request, Some(e)).with_url(url)),
};
}
pub async fn send(self) -> Result<ResponseWrapper> {
self.send_body(None).await
}
pub async fn send_text(mut self, text: String) -> Result<ResponseWrapper> {
self.set_header_if_absent("content-type", "text/plain".to_string());
self.send_body(Some(text.as_bytes().to_vec())).await
}
pub async fn send_form<T>(mut self, form: &T) -> Result<ResponseWrapper>
where
T: serde::ser::Serialize,
{
self.set_header_if_absent(
"content-type",
"application/x-www-form-urlencoded".to_string(),
);
let form = serde_urlencoded::to_string(form).map_err(Error::encode)?;
self.send_body(Some(form.as_bytes().to_vec())).await
}
#[cfg(feature = "json")]
pub async fn send_json<T>(mut self, json: &T) -> Result<ResponseWrapper>
where
T: serde::ser::Serialize,
{
self.set_header_if_absent("content-type", "application/json".to_string());
let json = serde_json::to_string(json).map_err(Error::encode)?;
self.send_body(Some(json.as_bytes().to_vec())).await
}
pub async fn send_vec(mut self, vec: Vec<u8>) -> Result<ResponseWrapper> {
self.set_header_if_absent("content-type", "application/octet-stream".to_string());
self.send_body(Some(vec)).await
}
}
#[async_trait]
impl HttpResponse for ResponseWrapper {
fn status(&self) -> StatusCode {
self.response.status()
}
async fn none(self) -> Result<()> {
Ok(())
}
async fn text(mut self) -> Result<String> {
self.response.text().await.map_err(Error::decode)
}
async fn vec(mut self) -> Result<Vec<u8>> {
self.response.bytes().await.map_err(Error::decode)
}
#[cfg(feature = "json")]
async fn json<T>(mut self) -> Result<T>
where
T: serde::de::DeserializeOwned,
{
let bytes = self.response.bytes().await.map_err(Error::decode)?;
serde_json::from_slice(&bytes).map_err(Error::decode)
}
}