src/db/mod.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use artists::Artists;
use artworks::Artworks;
use comments::Comments;
use sqlx::{postgres::PgPoolOptions, Pool, Postgres};

pub mod artists;
pub mod artworks;
pub mod comments;

#[derive(Clone)]
pub struct Database(Pool<Postgres>);

impl Database {
    pub async fn new(connection_string: &str) -> Self {
        let pool = PgPoolOptions::new()
            .max_connections(5)
            .connect(connection_string)
            .await
            .unwrap();

        sqlx::migrate!("./migrations").run(&pool).await.unwrap();

        Self(pool)
    }

    pub fn artists(&self) -> Artists {
        Artists::new(self.0.clone())
    }

    pub fn artworks(&self) -> Artworks {
        Artworks::new(self.0.clone())
    }

    pub fn comments(&self) -> Comments {
        Comments::new(self.0.clone())
    }
}