refactor db to init once, open connections otherwise

1 files changed, 6 insertions, 4 deletions

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)
             }))
         });