serve-net 0.7.1

The network service engine for Rust: listeners over net-io, ingress from direct clients, reverse proxies (PROXY protocol or trusted forwarded headers), Unix sockets, and FastCGI, a per-connection engine with a slot for any sans-IO protocol engine, admission, drain, and restarts without dropped connections.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

//! The server behind real front ends, for the interop CI job: forwarded
//! headers on 127.0.0.1:9001 (nginx, Apache) and the PROXY protocol on
//! 127.0.0.1:9002 (HAProxy). Each reply says what the server learned, and
//! `.github/interop/run.sh` checks it. Press Enter to stop.

use std::time::Duration;

use serve_net::{Ingress, ListenerSpec, ProxySource, ProxyVersions, Server, TrustedProxies};

fn main() -> Result<(), serve_net::ServeError> {
    let forwarded = Ingress::Proxy(ProxySource::forwarded_for(TrustedProxies::loopback()));
    let proxy = Ingress::Proxy(ProxySource::proxy_protocol(
        ProxyVersions::Both,
        TrustedProxies::loopback(),
    ));
    let server = Server::builder()
        .listener(ListenerSpec::tcp("forwarded", "127.0.0.1:9001").ingress(forwarded))
        .listener(ListenerSpec::tcp("proxy", "127.0.0.1:9002").ingress(proxy))
        .serve(serve_net::handler(|req, reply| {
            let c = req.client();
            let addr = c
                .addr()
                .map_or_else(|| "unknown".to_owned(), |a| a.to_string());
            reply.text(200, &format!("{:?} {addr}\n", c.ingress()));
        }))?;
    println!("listening");
    let drain = server.drain_handle();
    std::thread::spawn(move || {
        let mut line = String::new();
        let _ = std::io::stdin().read_line(&mut line);
        drain.drain(Duration::from_secs(5));
    });
    let report = server.wait();
    println!(
        "stopped: {} graceful, {} killed",
        report.graceful, report.killed
    );
    Ok(())
}