mayhem-db 0.2.0-beta.1

A Discord-like chat application optimized for security, stability, speed, and ultimate customization. => Database Module
Documentation 
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use std::sync::Arc;

use migration::{DbErr, Migrator, MigratorTrait};
use sea_orm::DbConn;

use super::{
    connector::{connect, ConnectionOptions},
    insertion::InsertionHelper,
    mutation::MutationHelper,
    query::QueryHelper,
    removal::RemovalHelper,
};

#[derive(Clone)]
pub struct Client {
    pub opts: Option<ConnectionOptions>,
    pub client: Arc<DbConn>,
    pub inserter: InsertionHelper,
    pub mutator: MutationHelper,
    pub remover: RemovalHelper,
    pub query: QueryHelper,
}

unsafe impl Sync for Client {}
unsafe impl Send for Client {}

impl Client {
    pub async fn connect(opts: ConnectionOptions) -> Self {
        return Self::of(connect(opts.clone()).await.unwrap());
    }

    pub fn of(conn: DbConn) -> Self {
        let conn = Arc::new(conn);

        let inserter = InsertionHelper::create(conn.clone());
        let mutator = MutationHelper::create(conn.clone());
        let remover = RemovalHelper::create(conn.clone());
        let query = QueryHelper::create(conn.clone());

        return Self {
            opts: None,
            client: conn,
            inserter,
            mutator,
            remover,
            query,
        };
    }

    pub async fn run_migrations(&self) -> Result<(), DbErr> {
        return Migrator::up(&self.client, None).await;
    }
}