mirror of
https://github.com/actix/examples
synced 2024-11-23 14:31:07 +01:00
Don't stream body but wait for it to complete and only send it then
This commit is contained in:
parent
fb5779fe8f
commit
7ff7accd79
@ -1,6 +1,7 @@
|
||||
use actix_web::client::Client;
|
||||
use actix_web::{middleware, web, App, Error, HttpRequest, HttpResponse, HttpServer};
|
||||
use clap::{value_t, Arg};
|
||||
use futures::stream::Stream;
|
||||
use futures::Future;
|
||||
use std::net::ToSocketAddrs;
|
||||
use url::Url;
|
||||
@ -15,6 +16,8 @@ fn forward(
|
||||
new_url.set_path(req.uri().path());
|
||||
new_url.set_query(req.uri().query());
|
||||
|
||||
// TODO: This forwarded implementation is incomplete as it only handles the inofficial
|
||||
// X-Forwarded-For header but not the official Forwarded one.
|
||||
let forwarded_req = client
|
||||
.request_from(new_url.as_str(), req.head())
|
||||
.no_decompress();
|
||||
@ -27,19 +30,22 @@ fn forward(
|
||||
forwarded_req
|
||||
.send_body(body)
|
||||
.map_err(Error::from)
|
||||
.map(|res| {
|
||||
.map(|mut res| {
|
||||
let mut client_resp = HttpResponse::build(res.status());
|
||||
// Remove `Connection` as per
|
||||
// https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Connection#Directives
|
||||
for (header_name, header_value) in res
|
||||
.headers()
|
||||
.iter()
|
||||
.filter(|(h, _)| *h != "connection")
|
||||
for (header_name, header_value) in
|
||||
res.headers().iter().filter(|(h, _)| *h != "connection")
|
||||
{
|
||||
client_resp.header(header_name.clone(), header_value.clone());
|
||||
}
|
||||
client_resp.streaming(res)
|
||||
res.body()
|
||||
.into_stream()
|
||||
.concat2()
|
||||
.map(move |b| client_resp.body(b))
|
||||
.map_err(|e| e.into())
|
||||
})
|
||||
.flatten()
|
||||
}
|
||||
|
||||
fn main() -> std::io::Result<()> {
|
||||
|
Loading…
Reference in New Issue
Block a user