arche 2.4.0

An opinionated backend foundation for Axum applications, providing batteries-included integrations for cloud services, databases, authentication, middleware, and logging.
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128

use serde_json::Value;
use sqlx::postgres::PgPoolOptions;

use crate::config::{resolve_optional_string, resolve_required, resolve_required_string};
use crate::error::AppError;

pub use crate::config::pg::{PgConfig, PgConfigBuilder};

#[derive(sqlx::FromRow)]
pub struct PgHealth {
    pub id: String,
    pub value: String,
}

#[derive(serde::Deserialize, Debug)]
struct PgCredentials {
    username: String,
    password: String,
}

pub type PgPool = sqlx::PgPool;

fn get_credentials(config: &PgConfig) -> Result<PgCredentials, AppError> {
    let credentials_json =
        resolve_optional_string(config.credentials_json.clone(), "PG_CREDENTIALS");

    if let Some(credentials_str) = credentials_json {
        let credentials_json = serde_json::from_str::<Value>(&credentials_str).map_err(|e| {
            AppError::internal_error(
                format!("Config error [credentials_json/PG_CREDENTIALS]: Failed to parse credentials JSON: {}", e),
                None,
            )
        })?;

        let username = credentials_json
            .get("username")
            .and_then(|v| v.as_str())
            .ok_or_else(|| {
                AppError::internal_error(
                    "Config error [username/PG_CREDENTIALS]: username not found in credentials JSON".to_string(),
                    None,
                )
            })?
            .to_string();

        let password = credentials_json
            .get("password")
            .and_then(|v| v.as_str())
            .ok_or_else(|| {
                AppError::internal_error(
                    "Config error [password/PG_CREDENTIALS]: password not found in credentials JSON".to_string(),
                    None,
                )
            })?
            .to_string();

        return Ok(PgCredentials { username, password });
    }

    Ok(PgCredentials {
        username: resolve_required_string(config.username.clone(), "PG_USERNAME", "username")?,
        password: resolve_required_string(config.password.clone(), "PG_PASSWORD", "password")?,
    })
}

pub async fn get_pg_pool(config: impl Into<Option<PgConfig>>) -> Result<PgPool, AppError> {
    let config = config.into().unwrap_or_default();
    let credentials = get_credentials(&config)?;

    let host = resolve_required_string(config.host, "PG_HOST", "host")?;
    let port: u16 = resolve_required(config.port, "PG_PORT", "port")?;
    let database = resolve_required_string(config.database, "PG_DATABASE", "database")?;
    let max_conn: u32 = resolve_required(config.max_connections, "PG_MAX_CONN", "max_connections")?;

    let pg_url = format!(
        "postgres://{}:{}@{}:{}/{}",
        credentials.username, credentials.password, host, port, database
    );

    PgPoolOptions::new()
        .max_connections(max_conn)
        .connect(&pg_url)
        .await
        .map_err(|e| AppError::internal_error(format!("Failed to create PG Pool: {}", e), None))
}

pub async fn test_pg(pg_pool: sqlx::PgPool) -> Result<bool, AppError> {
    let insert_query = r#"
    INSERT INTO health (id, value) VALUES ($1, $2)"#;
    let select_query = r#"
    SELECT id, value FROM health WHERE id = $1"#;
    let delete_query = r#"
    DELETE FROM health WHERE id = $1"#;

    let id = nanoid::nanoid!();
    let value = "test-value";

    let insert_result = sqlx::query(insert_query)
        .bind(&id)
        .bind(value)
        .execute(&pg_pool)
        .await
        .map_err(|e| {
            AppError::internal_error(e.to_string(), Some("Failed to insert data".to_string()))
        })?;

    let select_result = sqlx::query_as::<_, PgHealth>(select_query)
        .bind(&id)
        .fetch_one(&pg_pool)
        .await
        .map_err(|e| {
            AppError::internal_error(e.to_string(), Some("Failed to select data".to_string()))
        })?;

    sqlx::query(delete_query)
        .bind(&id)
        .execute(&pg_pool)
        .await
        .map_err(|e| {
            AppError::internal_error(e.to_string(), Some("Failed to delete data".to_string()))
        })?;

    Ok(
        insert_result.rows_affected() == 1
            && select_result.id == id
            && select_result.value == value,
    )
}