mirror of
https://github.com/tokio-rs/tokio.git
synced 2025-09-25 12:00:35 +00:00
8a7e57786a
In an effort to reach API stability, the `tokio` crate is shedding its
_public_ dependencies on crates that are either a) do not provide a
stable (1.0+) release with longevity guarantees or b) match the `tokio`
release cadence. Of course, implementing `std` traits fits the
requirements.
The on exception, for now, is the `Stream` trait found in `futures_core`.
It is expected that this trait will not change much and be moved into `std.
Since Tokio is not yet going reaching 1.0, I feel that it is acceptable to maintain
a dependency on this trait given how foundational it is.
Since the `Stream` implementation is optional, types that are logically
streams provide `async fn next_*` functions to obtain the next value.
Avoiding the `next()` name prevents fn conflicts with `StreamExt::next()`.
Additionally, some misc cleanup is also done:
- `tokio::io::io` -> `tokio::io::util`.
- `delay` -> `delay_until`.
- `Timeout::new` -> `timeout(...)`.
- `signal::ctrl_c()` returns a future instead of a stream.
- `{tcp,unix}::Incoming` is removed (due to lack of `Stream` trait).
- `time::Throttle` is removed (due to lack of `Stream` trait).
- Fix: `mpsc::UnboundedSender::send(&self)` (no more conflict with `Sink` fns).
68 lines
1.9 KiB
Rust
68 lines
1.9 KiB
Rust
//! A proxy that forwards data to another server and forwards that server's
|
|
//! responses back to clients.
|
|
//!
|
|
//! Because the Tokio runtime uses a thread pool, each TCP connection is
|
|
//! processed concurrently with all other TCP connections across multiple
|
|
//! threads.
|
|
//!
|
|
//! You can showcase this by running this in one terminal:
|
|
//!
|
|
//! cargo run --example proxy
|
|
//!
|
|
//! This in another terminal
|
|
//!
|
|
//! cargo run --example echo
|
|
//!
|
|
//! And finally this in another terminal
|
|
//!
|
|
//! cargo run --example connect 127.0.0.1:8081
|
|
//!
|
|
//! This final terminal will connect to our proxy, which will in turn connect to
|
|
//! the echo server, and you'll be able to see data flowing between them.
|
|
|
|
#![warn(rust_2018_idioms)]
|
|
|
|
use futures::{future::try_join, FutureExt};
|
|
use std::{env, error::Error};
|
|
use tokio::{
|
|
io::AsyncReadExt,
|
|
net::{TcpListener, TcpStream},
|
|
};
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn Error>> {
|
|
let listen_addr = env::args().nth(1).unwrap_or("127.0.0.1:8081".to_string());
|
|
let server_addr = env::args().nth(2).unwrap_or("127.0.0.1:8080".to_string());
|
|
|
|
println!("Listening on: {}", listen_addr);
|
|
println!("Proxying to: {}", server_addr);
|
|
|
|
let mut listener = TcpListener::bind(listen_addr).await?;
|
|
|
|
while let Ok((inbound, _)) = listener.accept().await {
|
|
let transfer = transfer(inbound, server_addr.clone()).map(|r| {
|
|
if let Err(e) = r {
|
|
println!("Failed to transfer; error={}", e);
|
|
}
|
|
});
|
|
|
|
tokio::spawn(transfer);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
async fn transfer(mut inbound: TcpStream, proxy_addr: String) -> Result<(), Box<dyn Error>> {
|
|
let mut outbound = TcpStream::connect(proxy_addr).await?;
|
|
|
|
let (mut ri, mut wi) = inbound.split();
|
|
let (mut ro, mut wo) = outbound.split();
|
|
|
|
let client_to_server = ri.copy(&mut wo);
|
|
let server_to_client = ro.copy(&mut wi);
|
|
|
|
try_join(client_to_server, server_to_client).await?;
|
|
|
|
Ok(())
|
|
}
|