From 14968097b28919f7e1f71ec5e49b30191826fa33 Mon Sep 17 00:00:00 2001
From: Santo Cariotti <santo@dcariotti.me>
Date: Mon, 22 Aug 2022 16:06:28 +0200
Subject: Sqlx connection

---
 server/src/db.rs     | 20 ++++++++++++++++++++
 server/src/errors.rs | 11 +++++++++++
 server/src/main.rs   |  8 ++++++++
 3 files changed, 39 insertions(+)
 create mode 100644 server/src/db.rs

(limited to 'server/src')

diff --git a/server/src/db.rs b/server/src/db.rs
new file mode 100644
index 0000000..8cc4d1f
--- /dev/null
+++ b/server/src/db.rs
@@ -0,0 +1,20 @@
+use crate::errors::AppError;
+
+use sqlx::postgres::PgPool;
+
+static mut CONNECTION: Option<PgPool> = None;
+
+pub async fn setup() -> Result<(), AppError> {
+    let database_url =
+        std::env::var("DATABASE_URL").expect("Define `DATABASE_URL` environment variable.");
+
+    unsafe {
+        CONNECTION = Some(PgPool::connect(&database_url).await?);
+    }
+
+    Ok(())
+}
+
+pub unsafe fn get_client() -> Option<&'static PgPool> {
+    CONNECTION.as_ref()
+}
diff --git a/server/src/errors.rs b/server/src/errors.rs
index cf59d0c..dc0468e 100644
--- a/server/src/errors.rs
+++ b/server/src/errors.rs
@@ -7,6 +7,7 @@ use serde_json::json;
 
 pub enum AppError {
     Generic,
+    Database,
 }
 
 impl IntoResponse for AppError {
@@ -16,6 +17,10 @@ impl IntoResponse for AppError {
                 StatusCode::INTERNAL_SERVER_ERROR,
                 "Generic error, can't find why",
             ),
+            AppError::Database => (
+                StatusCode::INTERNAL_SERVER_ERROR,
+                "Error with database connection",
+            ),
         };
 
         let body = Json(json!({
@@ -25,3 +30,9 @@ impl IntoResponse for AppError {
         (status, body).into_response()
     }
 }
+
+impl From<sqlx::Error> for AppError {
+    fn from(_error: sqlx::Error) -> AppError {
+        AppError::Database
+    }
+}
diff --git a/server/src/main.rs b/server/src/main.rs
index 2f4a727..e061b83 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -1,6 +1,7 @@
 mod db;
 mod errors;
 mod logger;
+use sqlx::postgres::PgPool;
 
 use axum::{
     http::{header, Request},
@@ -29,6 +30,13 @@ async fn main() {
 
 async fn create_app() -> Router {
     logger::setup();
+    let _ = db::setup().await;
+    let pool = unsafe {
+        match db::get_client() {
+            Some(client) => client,
+            None => panic!("Connection not established!"),
+        }
+    };
 
     Router::new()
         .route("/", get(hej))
-- 
cgit v1.2.3-18-g5258