rustio_core/
server.rs

1//! Hyper-backed HTTP/1 server.
2//!
3//! Bind an address with [`Server::bind`] and serve either a raw handler
4//! via [`Server::serve`] or a [`Router`] via [`Server::serve_router`].
5
6use std::convert::Infallible;
7use std::future::Future;
8use std::net::SocketAddr;
9use std::sync::Arc;
10
11use hyper::server::conn::http1;
12use hyper::service::service_fn;
13use hyper_util::rt::TokioIo;
14use tokio::net::TcpListener;
15
16use crate::http::{Request, Response};
17use crate::router::Router;
18
19pub struct Server {
20    addr: SocketAddr,
21}
22
23impl Server {
24    pub fn bind(addr: SocketAddr) -> Self {
25        Self { addr }
26    }
27
28    pub async fn serve<F, Fut>(self, handler: F) -> std::io::Result<()>
29    where
30        F: Fn(Request) -> Fut + Clone + Send + Sync + 'static,
31        Fut: Future<Output = Response> + Send + 'static,
32    {
33        let listener = TcpListener::bind(self.addr).await?;
34        eprintln!("rustio-core: listening on http://{}", self.addr);
35
36        loop {
37            let (stream, _peer) = listener.accept().await?;
38            let io = TokioIo::new(stream);
39            let handler = handler.clone();
40
41            tokio::spawn(async move {
42                let service = service_fn(move |raw: hyper::Request<hyper::body::Incoming>| {
43                    let handler = handler.clone();
44                    async move {
45                        let req = Request::new(raw);
46                        Ok::<Response, Infallible>(handler(req).await)
47                    }
48                });
49
50                if let Err(err) = http1::Builder::new().serve_connection(io, service).await {
51                    eprintln!("rustio-core: connection error: {err}");
52                }
53            });
54        }
55    }
56
57    pub async fn serve_router(self, router: Router) -> std::io::Result<()> {
58        let router = Arc::new(router);
59        self.serve(move |req| {
60            let router = router.clone();
61            async move { router.dispatch(req).await }
62        })
63        .await
64    }
65}
rustio_core/server.rs

rustio_core/
server.rs
