Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions examples/proxy.rs
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,13 @@ async fn main() {
let client = client.clone();
async move {
let uri = req.uri().clone();
let remote_addr = req.extensions().get::<std::net::SocketAddr>().unwrap();
println!(
"Received request from {}: {} {}",
remote_addr,
req.method(),
uri
);

// You can modify request here
// or You can just return response anywhere
Expand Down
8 changes: 6 additions & 2 deletions src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ where
{
/// Bind to a socket address and return a future that runs the proxy server.
/// URL for requests that passed to service are full URL including scheme.
/// remote address of client is stored in request extensions as std::net::SocketAddr.
pub async fn bind<A: ToSocketAddrs, S>(
self,
addr: A,
Expand All @@ -77,7 +78,7 @@ where

Ok(async move {
loop {
let (stream, _) = match listener.accept().await {
let (stream, remote_addr) = match listener.accept().await {
Ok(conn) => conn,
Err(err) => {
tracing::warn!("Failed to accept connection: {}", err);
Expand All @@ -94,7 +95,10 @@ where
.title_case_headers(true)
.serve_connection(
TokioIo::new(stream),
Self::wrap_service(proxy.clone(), service.clone()),
service_fn(move |mut req| {
req.extensions_mut().insert(remote_addr);
Self::wrap_service(proxy.clone(), service.clone()).call(req)
}),
)
.with_upgrades()
.await
Expand Down