38 lines
748 B
Rust
38 lines
748 B
Rust
|
use std::{ops::Deref, sync::Arc};
|
||
|
|
|
||
|
|
use anyhow::Context;
|
||
|
|
use sqlx::{Pool, Postgres};
|
||
|
|
|
||
|
|
#[derive(Clone)]
|
||
|
|
pub struct SharedState(pub Arc<State>);
|
||
|
|
|
||
|
|
impl Deref for SharedState {
|
||
|
|
type Target = Arc<State>;
|
||
|
|
|
||
|
|
fn deref(&self) -> &Self::Target {
|
||
|
|
&self.0
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct State {
|
||
|
|
pub _db: Pool<Postgres>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl State {
|
||
|
|
pub async fn new() -> anyhow::Result<Self> {
|
||
|
|
let db = sqlx::PgPool::connect(
|
||
|
|
&std::env::var("DATABASE_URL").context("DATABASE_URL is not set")?,
|
||
|
|
)
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
sqlx::migrate!("migrations/crdb")
|
||
|
|
.set_locking(false)
|
||
|
|
.run(&db)
|
||
|
|
.await?;
|
||
|
|
|
||
|
|
let _ = sqlx::query("SELECT 1;").fetch_one(&db).await?;
|
||
|
|
|
||
|
|
Ok(Self { _db: db })
|
||
|
|
}
|
||
|
|
}
|