Fork from m6-ie project

1 files changed, 29 insertions, 0 deletions

diff --git a/src/db.rs b/src/db.rs
new file mode 100644
index 0000000..43c3bd9
--- /dev/null
+++ b/src/db.rs
@@ -0,0 +1,29 @@
+use crate::errors::AppError;
+
+use sqlx::postgres::PgPool;
+
+/// Static variable used to manage the database connection. Called with value = None raises a panic
+/// error.
+static mut CONNECTION: Option<PgPool> = None;
+
+/// Setup database connection. Get variable `DATABASE_URL` from the environment. Sqlx crate already
+/// defines an error for environments without DATABASE_URL.
+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(())
+}
+
+/// Get connection. Raises an error if `setup()` has not been called yet.
+/// Managing static `CONNECTION` is an unsafe operation.
+pub unsafe fn get_client() -> &'static PgPool {
+    match &CONNECTION {
+        Some(client) => client,
+        None => panic!("Connection not established!"),
+    }
+}