From 3a490fca353fa6d955a3d44e125bd031aeaca70c Mon Sep 17 00:00:00 2001
From: Jef Roosens <roosensjef@gmail.com>
Date: Sat, 17 Apr 2021 10:52:55 +0200
Subject: [PATCH] [#24] Added db connection to server

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

diff --git a/src/server/main.rs b/src/server/main.rs
index 58b0300..9e01983 100644
--- a/src/server/main.rs
+++ b/src/server/main.rs
@@ -2,6 +2,8 @@
 
 #[macro_use]
 extern crate rocket;
+#[macro_use]
+extern crate rocket_contrib;
 
 #[cfg(test)]
 mod tests;
@@ -14,6 +16,7 @@ mod routes;
 use rocket::fairing::{Fairing, Info, Kind};
 use rocket::http::Header;
 use rocket::{Request, Response};
+use rocket_contrib::databases::diesel;
 
 pub struct CORS;
 
@@ -36,9 +39,14 @@ impl Fairing for CORS {
     }
 }
 
+// This defines a connection to the database
+#[database("postgres_fej")]
+struct FejDbConn(diesel::PgConnection);
+
 fn rocket() -> rocket::Rocket {
     rocket::ignite()
         .attach(CORS)
+        .attach(FejDbConn::fairing())
         .mount("/ivago", routes::ivago())
         .register(catchers![catchers::not_found])
 }