From 8ae488341a66d27c91230b1c76d415d59c0e418b Mon Sep 17 00:00:00 2001 From: Akshay Date: Mon, 24 Aug 2020 10:09:53 +0530 Subject: refactor db to init once, open connections otherwise --- src/main.rs | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) (limited to 'src/main.rs') diff --git a/src/main.rs b/src/main.rs index c9819d0..35e686e 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,22 +1,24 @@ +// extern use anyhow::Result; use hyper::service::{make_service_fn, service_fn}; use hyper::Server; use log::trace; +// internal mod db; -use db::init_db; +use db::{init_db, open_connection}; + mod service; use service::shortner_service; -use std::sync::{Arc, Mutex}; - fn main() -> Result<()> { pretty_env_logger::init(); + init_db("./urls.db_3")?; smol::run(async { let addr = ([127, 0, 0, 1], 3000).into(); let service = make_service_fn(move |_| async { Ok::<_, hyper::Error>(service_fn(move |req| { - let db_conn = init_db("./urls.db_3").unwrap(); + let db_conn = open_connection("./urls.db_3").unwrap(); shortner_service(req, db_conn) })) }); -- cgit v1.2.3