galvyn-core 0.4.0

Core concepts for galvyn like trait definitions
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67

//! [`Module`] implementation for [`rorm::Database`]

use rorm::Database;
use rorm::DatabaseConfiguration;
use rorm::DatabaseDriver;
use serde::Deserialize;
use serde::Serialize;

use crate::InitError;
use crate::Module;
use crate::PreInitError;

/// Enum declaring how the database should be configured
#[derive(Default, Debug)]
pub enum DatabaseSetup {
    #[default]
    Default,
    Custom(DatabaseConfiguration),
}

/// Config struct the [`DatabaseSetup::Default`] will deserialize from environment variables
#[derive(Serialize, Deserialize, Debug)]
struct DatabaseConfig {
    pub postgres_db: String,
    pub postgres_host: String,
    pub postgres_port: u16,
    pub postgres_user: String,
    pub postgres_password: String,
}

impl Module for Database {
    type Setup = DatabaseSetup;

    type PreInit = DatabaseConfiguration;

    async fn pre_init(setup: Self::Setup) -> Result<Self::PreInit, PreInitError> {
        match setup {
            DatabaseSetup::Default => {
                let DatabaseConfig {
                    postgres_db,
                    postgres_host,
                    postgres_port,
                    postgres_user,
                    postgres_password,
                } = envy::from_env()?;

                Ok(DatabaseConfiguration::new(DatabaseDriver::Postgres {
                    name: postgres_db,
                    host: postgres_host,
                    port: postgres_port,
                    user: postgres_user,
                    password: postgres_password,
                }))
            }
            DatabaseSetup::Custom(config) => Ok(config),
        }
    }

    type Dependencies = ();

    async fn init(
        config: Self::PreInit,
        _dependencies: &mut Self::Dependencies,
    ) -> Result<Self, InitError> {
        Ok(Database::connect(config).await?)
    }
}