How to get the local or remote address in a handler?

[arniu]

Feature Request

Motivation

We can extract the remote address with ConnectInfo<SocketAddr> in a handler,
while there is no way to get the remote address in it.

Proposal

Add a new extractor AddrPair:

pub struct AddrPair {
    pub local: SocketAddr,
    pub remote: SocketAddr,
}

Alternatives

Just add LocalAddr to get the local address.

[davidpdrsn]

We have this already https://docs.rs/axum/0.6.0-rc.1/axum/extract/struct.ConnectInfo.html. Did not find it in the docs or does it not work for you?

[arniu]

And how to get the server in a handler?

[davidpdrsn]

Pass it either via State or Extension

[arniu]

Pass the server to the consumed router?

[davidpdrsn]

Sorry I forgot that you have to pass the router to the server. You can use AddrIncoming from hyper:

use axum::{
    extract::{ConnectInfo, State},
    routing::get,
    Router,
};
use hyper::server::conn::AddrIncoming;
use std::net::SocketAddr;

#[tokio::main]
async fn main() {
    let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
    let incoming = AddrIncoming::bind(&addr).unwrap();
    let local_addr = incoming.local_addr();

    let app = Router::with_state(AppState { local_addr }).route("/", get(handler));

    axum::Server::builder(incoming)
        .serve(app.into_make_service_with_connect_info::<SocketAddr>())
        .await
        .unwrap();
}

#[derive(Clone)]
struct AppState {
    local_addr: SocketAddr,
}

async fn handler(state: State<AppState>, ConnectInfo(remote_addr): ConnectInfo<SocketAddr>) {
    dbg!(state.local_addr);
    dbg!(remote_addr);
}

[arniu]

AddrIncoming::local_addr differs slightly from AddrStream::local_addr.