feat: Basic HTTP listener
This commit is contained in:
parent
c77223048c
commit
452da73413
7 changed files with 350 additions and 7 deletions
57
src/http/mod.rs
Normal file
57
src/http/mod.rs
Normal file
|
|
@ -0,0 +1,57 @@
|
|||
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"
|
||||
}
|
||||
Loading…
Reference in a new issue