tideway 0.1.0

A batteries-included Rust web framework built on Axum for building SaaS applications quickly
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

use crate::error::{Result, TidewayError};
use sea_orm_migration::MigratorTrait;

/// Run pending migrations
///
/// # Example
///
/// ```rust,ignore
/// use sea_orm_migration::MigratorTrait;
/// use tideway::database::run_migrations;
///
/// // Define your migrator
/// pub struct Migrator;
///
/// #[async_trait::async_trait]
/// impl MigratorTrait for Migrator {
///     fn migrations() -> Vec<Box<dyn MigrationTrait>> {
///         vec![
///             Box::new(m20250101_create_users::Migration),
///             Box::new(m20250102_create_organizations::Migration),
///         ]
///     }
/// }
///
/// // Run migrations
/// run_migrations::<Migrator>(&db).await?;
/// ```
pub async fn run_migrations<M: MigratorTrait>(db: &sea_orm::DatabaseConnection) -> Result<()> {
    M::up(db, None)
        .await
        .map_err(|e| TidewayError::internal(format!("Migration failed: {}", e)))?;

    tracing::info!("Database migrations completed successfully");
    Ok(())
}

/// Check migration status
pub async fn migration_status<M: MigratorTrait>(db: &sea_orm::DatabaseConnection) -> Result<()> {
    M::status(db)
        .await
        .map_err(|e| TidewayError::internal(format!("Failed to check migration status: {}", e)))
}

/// Rollback the last migration
pub async fn rollback_migration<M: MigratorTrait>(
    db: &sea_orm::DatabaseConnection,
    steps: Option<u32>,
) -> Result<()> {
    M::down(db, steps)
        .await
        .map_err(|e| TidewayError::internal(format!("Rollback failed: {}", e)))?;

    tracing::info!("Rolled back {} migration(s)", steps.unwrap_or(1));
    Ok(())
}

/// Reset the database (down all migrations then up)
pub async fn reset_database<M: MigratorTrait>(db: &sea_orm::DatabaseConnection) -> Result<()> {
    M::fresh(db)
        .await
        .map_err(|e| TidewayError::internal(format!("Database reset failed: {}", e)))?;

    tracing::info!("Database reset completed");
    Ok(())
}