57 lines
2 KiB
Rust
57 lines
2 KiB
Rust
|
use axum::Router;
|
||
|
|
use axum::routing::get;
|
||
|
|
use axum::serve::Listener as AxumListener;
|
||
|
|
|
||
|
|
use crate::db::{Database, DatabaseInterface};
|
||
|
|
use crate::listener::{Listener, ListenerKind};
|
||
|
|
use crate::stream::{AbstractSocketAddr, AbstractStreamKind};
|
||
|
|
|
||
|
|
impl AxumListener for Listener {
|
||
|
|
type Io = AbstractStreamKind;
|
||
|
|
type Addr = AbstractSocketAddr;
|
||
|
|
|
||
|
|
async fn accept(&mut self) -> (Self::Io, Self::Addr) {
|
||
|
|
loop {
|
||
|
|
let res = match &self.kind {
|
||
|
|
ListenerKind::Tcp(l) => l
|
||
|
|
.accept()
|
||
|
|
.await
|
||
|
|
.map(|(stream, remote_addr)| (stream.into(), remote_addr.into())),
|
||
|
|
ListenerKind::Uds(l) => l
|
||
|
|
.accept()
|
||
|
|
.await
|
||
|
|
.map(|(stream, remote_addr)| (stream.into(), remote_addr.into())),
|
||
|
|
};
|
||
|
|
|
||
|
|
match res {
|
||
|
|
Ok((stream, remote_addr)) => return (stream, remote_addr),
|
||
|
|
Err(e) => {
|
||
|
|
// Here the error could be fatal, or could simply be that a client aborted the connected,
|
||
|
|
// in which case we don't want to crash the server, simply skip this client connection.
|
||
|
|
// https://doc.rust-lang.org/stable/std/net/struct.TcpListener.html#errors
|
||
|
|
match e.kind() {
|
||
|
|
std::io::ErrorKind::ConnectionAborted => {}
|
||
|
|
_ => panic!("Unrecoverable HTTP client connection error: {e}"),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn local_addr(&self) -> std::io::Result<Self::Addr> {
|
||
|
|
match &self.kind {
|
||
|
|
ListenerKind::Tcp(l) => l.local_addr().map(Into::into),
|
||
|
|
ListenerKind::Uds(l) => l.local_addr().map(Into::into),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn http_listen<D: DatabaseInterface>(listener: Listener, db: Database<D>) {
|
||
|
|
let app = Router::new().route("/", get(handler)).with_state(db);
|
||
|
|
|
||
|
|
axum::serve(listener, app).await.unwrap();
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn handler() -> &'static str {
|
||
|
|
"hello world"
|
||
|
|
}
|