actix/examples
1
0
Fork 0
mirror of https://github.com/actix/examples synced 2024-11-27 16:02:57 +01:00
Code Issues Releases Wiki Activity

Fix proxy to handle both chunked and non-chunked responses (#59)

Browse Source
This commit is contained in:
Rotem Yaari 2018-11-17 20:27:29 +02:00 committed by Douman
parent 8a81a3bb58
commit 5871328be2
1 changed files with 22 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
http-full-proxy/src/main.rs
View File

@ -11,7 +11,7 @@ use actix_web::{
HttpResponse, HttpResponse,
}; };
use clap::{value_t, Arg}; use clap::{value_t, Arg};
use futures::Future; use futures::{future, Future};
use std::net::ToSocketAddrs; use std::net::ToSocketAddrs;
use url::Url; use url::Url;
@ -55,16 +55,28 @@ fn forward(
forwarded_req forwarded_req
.send() .send()
.map_err(Error::from) .map_err(Error::from)
.and_then(move |resp| { .and_then(construct_response)
let mut client_resp = HttpResponse::build(resp.status()); .responder()
for (header_name, header_value) in }
resp.headers().iter().filter(|(h, _)| *h != "connection")
{
client_resp.header(header_name.clone(), header_value.clone());
}
Ok(client_resp.streaming(resp.payload())) fn construct_response(
}).responder() resp: client::ClientResponse,
) -> Box<dyn Future<Item = HttpResponse, Error = Error>> {
let mut client_resp = HttpResponse::build(resp.status());
for (header_name, header_value) in
resp.headers().iter().filter(|(h, _)| *h != "connection")
{
client_resp.header(header_name.clone(), header_value.clone());
}
if resp.chunked().unwrap_or(false) {
Box::new(future::ok(client_resp.streaming(resp.payload())))
} else {
Box::new(
resp.body()
.from_err()
.and_then(move |body| Ok(client_resp.body(body))),
)
}
} }
fn main() { fn main() {