forked/reqwest-impersonate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

improvements

Browse Source
This commit is contained in:
Sean McArthur
2016-10-16 09:40:28 -07:00
parent 9bfdcec052
commit d78eff6918
6 changed files with 104 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/client.rs
View File

@@ -3,7 +3,8 @@ use std::io::{self, Read};
use hyper::{Method, StatusCode, HttpVersion, Url};
use hyper::header::Headers;
use super::sync;
use ::body::Body;
use ::sync;
pub struct Client {
inner: sync::Client,
@@ -42,20 +43,31 @@ pub struct RequestBuilder<'a> {
version: HttpVersion,
headers: Headers,
body: Option<()>,
body: Option<Body>,
}
impl<'a> RequestBuilder<'a> {
pub fn body(mut self, body: ()) -> RequestBuilder<'a> {
self.body = Some(body);
pub fn header<H: ::header::Header>(mut self, header: H) -> RequestBuilder<'a> {
self.headers.set(header);
self
}
pub fn send(mut self) -> Result<Response, String> {
self.headers.set(::hyper::header::ContentLength(0));
pub fn headers(mut self, headers: ::header::Headers) -> RequestBuilder<'a> {
self.headers.extend(headers.iter());
self
}
pub fn body<T: Into<Body>>(mut self, body: T) -> RequestBuilder<'a> {
self.body = Some(body.into());
self
}
pub fn send(mut self) -> ::Result<Response> {
self.headers.set(::header::ContentLength(0));
let req = try!(self.client.inner.request(self.method, self.url, self.version, self.headers));
let res = try!(req.end().map_err(|e| format!("RequestError: end: {}", e)));
let res = try!(req.end());
Ok(Response {
inner: res
})