From f9051a919fa9af44a5703f55afc11e91607d68c6 Mon Sep 17 00:00:00 2001
From: Akshay <nerdy@peppe.rs>
Date: Sun, 23 Aug 2020 22:17:50 +0530
Subject: attempt to use single db connection

---
 src/main.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 6b48957..c9819d0 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -4,15 +4,22 @@ use hyper::Server;
 use log::trace;
 
 mod db;
+use db::init_db;
 mod service;
 use service::shortner_service;
 
+use std::sync::{Arc, Mutex};
+
 fn main() -> Result<()> {
     pretty_env_logger::init();
     smol::run(async {
         let addr = ([127, 0, 0, 1], 3000).into();
-        let service =
-            make_service_fn(|_| async { Ok::<_, hyper::Error>(service_fn(shortner_service)) });
+        let service = make_service_fn(move |_| async {
+            Ok::<_, hyper::Error>(service_fn(move |req| {
+                let db_conn = init_db("./urls.db_3").unwrap();
+                shortner_service(req, db_conn)
+            }))
+        });
         let server = Server::bind(&addr).serve(service);
         trace!("Listening on http://{}", addr);
         server.await.unwrap();
-- 
cgit v1.2.3