gallant/uberbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Rewrote DB webserver to use warp

Browse source
This commit is contained in:
famfo 2022-01-15 14:50:47 +01:00
parent cf99f5a4a1
commit 5ab4e3a221
1 changed files with 16 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
src/web_service.rs
View file

@ -1,29 +1,26 @@
use crate::ExecutorConnection;
use hyper::service::{make_service_fn, service_fn};
use hyper::{Body, Request, Response, Server};
use std::convert::Infallible;
use std::net::SocketAddr;
use std::sync::Arc;
use warp::Filter;
pub async fn run(db: ExecutorConnection, listen: SocketAddr) -> anyhow::Result<()> {
let db = Arc::new(db);
let db_filter = warp::any().map(move || db.clone());
let filter = warp::any().and(db_filter).and_then(handle);
Server::bind(&listen)
.serve(make_service_fn(|_| {
let db = Arc::clone(&db);
async move { Ok::<_, Infallible>(service_fn(move |r| handle(r, Arc::clone(&db)))) }
}))
.await?;
warp::serve(filter).run(listen).await;
Ok(())
}
async fn handle(
_req: Request<Body>,
db: Arc<ExecutorConnection>,
) -> Result<Response<Body>, Infallible> {
Ok(Response::new(Body::from(format!(
"{:?}",
db.get_quote(None).await
))))
async fn handle(db: ExecutorConnection) -> Result<impl warp::Reply, warp::Rejection> {
if let Some((a, b)) = db.get_quote(None).await {
Ok(warp::reply::with_status(
format!("{} {}", a, b),
warp::http::StatusCode::OK,
))
} else {
Ok(warp::reply::with_status(
format!("None"),
warp::http::StatusCode::NO_CONTENT,
))
}
}