modo-rs 0.8.0

Rust web framework for small monolithic apps
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

use crate::error::Result;

use super::config::Config;
use super::database::Database;

/// Open a local libsql database, apply PRAGMAs from [`Config`], and
/// optionally run migrations.
///
/// Creates parent directories for the database path if they do not exist.
/// When [`Config::migrations`] is set, SQL migration files from that
/// directory are applied via [`migrate`](super::migrate).
///
/// # Errors
///
/// Returns an error if directory creation, database opening, PRAGMA
/// execution, or migration fails.
pub async fn connect(config: &Config) -> Result<Database> {
    // Create parent directories if needed
    if let Some(parent) = std::path::Path::new(&config.path).parent()
        && !parent.as_os_str().is_empty()
    {
        tokio::fs::create_dir_all(parent).await.map_err(|e| {
            crate::error::Error::internal(format!(
                "failed to create database directory: {parent:?}"
            ))
            .chain(e)
        })?;
    }

    let db = libsql::Builder::new_local(&config.path)
        .build()
        .await
        .map_err(crate::error::Error::from)?;

    let conn = db.connect().map_err(crate::error::Error::from)?;

    // Apply PRAGMAs (use query() because PRAGMAs return rows in libsql)
    conn.query(
        &format!("PRAGMA journal_mode={}", config.journal_mode.as_str()),
        (),
    )
    .await
    .map_err(crate::error::Error::from)?;

    conn.query(
        &format!("PRAGMA synchronous={}", config.synchronous.as_str()),
        (),
    )
    .await
    .map_err(crate::error::Error::from)?;

    conn.query(&format!("PRAGMA busy_timeout={}", config.busy_timeout), ())
        .await
        .map_err(crate::error::Error::from)?;

    conn.query(&format!("PRAGMA cache_size=-{}", config.cache_size), ())
        .await
        .map_err(crate::error::Error::from)?;

    conn.query(&format!("PRAGMA mmap_size={}", config.mmap_size), ())
        .await
        .map_err(crate::error::Error::from)?;

    conn.query(
        &format!(
            "PRAGMA foreign_keys={}",
            if config.foreign_keys { "ON" } else { "OFF" }
        ),
        (),
    )
    .await
    .map_err(crate::error::Error::from)?;

    conn.query(
        &format!("PRAGMA temp_store={}", config.temp_store.as_str()),
        (),
    )
    .await
    .map_err(crate::error::Error::from)?;

    // Run migrations if configured
    if let Some(ref migrations_dir) = config.migrations {
        super::migrate::migrate(&conn, migrations_dir).await?;
    }

    Ok(Database::new(db, conn))
}