Test MySQL Connection

Source/Server.rs

@@ -1,18 +1,57 @@
 #![feature(proc_macro_hygiene, decl_macro)]
 
-#[macro_use]
-extern crate rocket;
-
-#[macro_use]
-extern crate rocket_contrib;
+use std::env;
+use dotenv::dotenv;
 
+use rocket::{get, routes, catch, catchers};
+use rocket_contrib::json;
 use rocket_contrib::json::{Json, JsonValue};
 
+use mysql::*;
+use mysql::prelude::*;
+
+use serde::{Serialize, Deserialize};
+
+#[derive(Serialize, Deserialize, Debug)]
+struct Email {
+    address: String,
+    user_id: String,
+    is_confirmed: bool,
+    is_primary: bool,
+}
+
 #[get("/")]
 fn hello() -> JsonValue {
     return json!({ "TribuFu": "Hello World" });
 }
 
+#[get("/emails")]
+fn emails() -> JsonValue {
+    dotenv().ok();
+
+    let MYSQL_HOST: String = env::var("MYSQL_HOST").expect("Missing MYSQL_HOST in Environment");
+    let MYSQL_PORT: String = env::var("MYSQL_PORT").expect("Missing MYSQL_PORT in Environmment");
+    let MYSQL_USERNAME: String = env::var("MYSQL_USERNAME").expect("Missing MYSQL_USERNAME in Environmment");
+    let MYSQL_PASSWORD: String = env::var("MYSQL_PASSWORD").expect("Missing MYSQL_PASSWORD in Environmment");
+    let MYSQL_DATABASE: String = env::var("MYSQL_DATABASE").expect("Missing MYSQL_DATABASE in Environmment");
+
+    let url = format!("mysql://{}:{}@{}:{}/{}", MYSQL_USERNAME, MYSQL_PASSWORD, MYSQL_HOST, MYSQL_PORT, MYSQL_DATABASE);
+
+    let pool = Pool::new(url).unwrap();
+
+    let mut connection = pool.get_conn().unwrap();
+
+    // Let's select emails from database. Type inference should do the trick here.
+    let emails = connection.query_map(
+            "SELECT Address, UserId, IsConfirmed, IsPrimary from Email",
+            |(address, user_id, is_confirmed, is_primary)| {
+                Email { address, user_id, is_confirmed, is_primary }
+            },
+        ).unwrap();
+
+    return json!(emails);
+}
+
 #[catch(404)]
 fn not_found() -> JsonValue {
     return json!({
@@ -20,12 +59,9 @@ fn not_found() -> JsonValue {
     });
 }
 
-fn rocket() -> rocket::Rocket {
-    return rocket::ignite()
-        .mount("/", routes![hello])
-        .register(catchers![not_found]);
-}
-
 fn main() {
-    rocket().launch();
+    rocket::ignite()
+        .mount("/", routes![hello, emails])
+        .register(catchers![not_found])
+        .launch();
 }