lariv-rs 0.1.0

Compile-time plugin web application framework built on Axum, SeaORM, Maud, and HTMX
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

use sea_orm::{DbBackend, Statement};
use sea_orm_migration::prelude::*;

#[derive(DeriveMigrationName)]
pub struct Migration;

async fn exec(manager: &SchemaManager<'_>, sql: &str) -> Result<(), DbErr> {
    manager
        .get_connection()
        .execute(Statement::from_string(
            manager.get_connection().get_database_backend(),
            sql.to_string(),
        ))
        .await
        .map(|_| ())
}

#[async_trait::async_trait]
impl MigrationTrait for Migration {
    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        // Unique phones cannot all become ''; stamp a stable per-row default.
        match manager.get_database_backend() {
            DbBackend::Postgres => {
                exec(
                    manager,
                    "UPDATE users SET phone = 'user-' || id::text \
                     WHERE phone IS NULL OR btrim(phone) = ''",
                )
                .await?;
                exec(
                    manager,
                    "ALTER TABLE users ALTER COLUMN phone SET DEFAULT ''",
                )
                .await?;
                exec(manager, "ALTER TABLE users ALTER COLUMN phone SET NOT NULL").await?;
            }
            _ => {
                exec(
                    manager,
                    "UPDATE users SET phone = 'user-' || id \
                     WHERE phone IS NULL OR trim(phone) = ''",
                )
                .await?;
            }
        }
        Ok(())
    }

    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        if manager.get_database_backend() == DbBackend::Postgres {
            exec(
                manager,
                "ALTER TABLE users ALTER COLUMN phone DROP NOT NULL",
            )
            .await?;
            exec(manager, "ALTER TABLE users ALTER COLUMN phone DROP DEFAULT").await?;
        }
        Ok(())
    }
}