shockrah/freechat
shockrah/freechat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

child calls from main_responder (see route_dipatcher + route handlers) now borrow a connection for the sql db

Browse Source
This commit is contained in:
shockrah 2020-06-02 03:25:11 -07:00
parent 19ad0eee9f
commit d1cfc4284c
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server/src/main.rs
View File

@ -23,6 +23,7 @@ mod auth;
use auth::AuthReason;
mod routes;
mod invites;
fn map_qs(query_string_raw: Option<&str>) -> HashMap<&str, &str> {
/*
@ -42,9 +43,9 @@ fn map_qs(query_string_raw: Option<&str>) -> HashMap<&str, &str> {
map
}
async fn route_dispatcher(resp: &mut Response<Body>, meth: &Method, path: &str, params: &HashMap<&str, &str>) {
async fn route_dispatcher(conn: &Conn, resp: &mut Response<Body>, meth: &Method, path: &str, params: &HashMap<&str, &str>) {
match (meth, path) {
(&Method::GET, routes::INVITE_JOIN) => invites::join_invite_code(&mut resp, meth, path, params),
(&Method::GET, routes::INVITE_JOIN) => invites::join_invite_code(conn, &mut resp, params).await,
_ => {
*resp.status_mut() = StatusCode::NOT_FOUND;
}
@ -61,7 +62,7 @@ async fn main_responder(request: Request<Body>) -> Result<Response<Body>, hyper:
if let Ok(conn) = Conn::from_url(env::var("DATABASE_URL").unwrap()).await {
// some more information in the response would be great right about here
match auth::wall_entry(path, conn, &params).await {
OpenAuth | Good => route_dispatcher(&mut response, &method, path, &params).await,
OpenAuth | Good => route_dispatcher(&conn, &mut response, &method, path, &params).await,
LimitPassed => *response.status_mut() = StatusCode::UNAUTHORIZED,
NoKey => *response.status_mut() = StatusCode::UNAUTHORIZED,
InternalFailure => *response.status_mut() = StatusCode::INTERNAL_SERVER_ERROR