From 49a06ccbf087c4ed0db3d61f3ac4892f45858ce1 Mon Sep 17 00:00:00 2001
From: Harald Eilertsen <haraldei@anduin.net>
Date: Thu, 11 Jan 2018 23:17:19 +0100
Subject: Read .env file in main.

---
 src/main.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index be61d2d..4c05523 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -18,10 +18,17 @@ mod schema;
 mod controllers;
 use controllers::home_controller;
 use controllers::posts_controller;
+use dotenv::dotenv;
+use std::env;
 
 fn main() {
+    dotenv().expect("No .env file found.");
+
+    let dburl = env::var("DATABASE_URL")
+        .expect("DATABASE_URL environment variable must be set");
+
     rocket::ignite()
-        .manage(utils::init_db_pool())
+        .manage(utils::init_db_pool(&dburl))
         .mount("/",
             routes![
                 home_controller::index,
-- 
cgit v1.2.3