From ab23761e090b8ab6311a360eada7131f6663a3bf Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Thu, 1 Sep 2022 18:45:04 +0200 Subject: Fork from m6-ie project --- src/db.rs | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 src/db.rs (limited to 'src/db.rs') 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 = 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!"), + } +} -- cgit v1.2.3-71-g8e6c