-
-
Notifications
You must be signed in to change notification settings - Fork 1.7k
Closed
Description
How to copy Request for useing sender.send_request and hyper::upgrade::on for web soket proxy at the same time
and return BAD_GATEWAY if let client_stream = TcpStream::connect(addr).await.unwrap(); return err ?
error[E0596]: cannot borrow `req` as mutable, as it is not declared as mutable
--> src\main.rs:32:42
|
18 | async fn server_upgrade(req: Request<Incoming>) -> Result<Response<Incoming>, hyper::Error> {
| --- help: consider changing this to be mutable: `mut req`
...
32 | let reqs = hyper::upgrade::on(&mut req).await;
| ^^^^^^^^ cannot borrow as mutable
my code
use std::net::SocketAddr;
use tokio::net::{TcpListener, TcpStream};
use hyper::server::conn::http1;
use hyper::service::service_fn;
use hyper::{Request, Response, StatusCode};
use hyper::body::Incoming;
use hyper_util::rt::TokioIo;
/// Handle server-side I/O after HTTP upgraded.
/*async fn server_upgraded_io(upgraded: Upgraded, conn: Connection<TokioIo<TcpStream>, Incoming>) -> Result<()> {
let upgraded = TokioIo::new(upgraded);
let tcp = conn.into_parts().io;
tokio::io::copy_bidirectional(&mut upgraded, &mut tcp).await?;
Ok(())
} */
/// Our server HTTP handler to initiate HTTP upgrades.
async fn server_upgrade(req: Request<Incoming>) -> Result<Response<Incoming>, hyper::Error> {
let addr = "localhost:8080";
let client_stream = TcpStream::connect(addr).await.unwrap();
let io = TokioIo::new(client_stream);
let (mut sender, conn) = hyper::client::conn::http1::handshake(io).await.unwrap();
tokio::task::spawn(async move {
if let Err(err) = conn.await {
println!("Connection failed: {:?}", err);
}
});
let mut ret = sender.send_request(req).await.unwrap();
if ret.status() == hyper::StatusCode::SWITCHING_PROTOCOLS {
tokio::task::spawn(async move {
let _rets = hyper::upgrade::on(&mut ret).await;
let _reqs = hyper::upgrade::on(&mut req).await;
});
}
return Ok(ret);
}
#[tokio::main]
async fn main() {
let other_addr: SocketAddr = ([127, 0, 0, 1], 3001).into();
let listener = TcpListener::bind(other_addr).await.expect("failed to bind");
loop {
let (stream, _) = listener.accept().await.unwrap();
let io = TokioIo::new(stream);
tokio::spawn(async move {
let task = http1::Builder::new().serve_connection(io, service_fn(server_upgrade));
let _conn = task.with_upgrades().await.unwrap();
});
}
}Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels
