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

pub mod test_api;

use barrel::Migration;
use quaint::{prelude::Queryable, single::Quaint};
use test_setup::{BitFlags, Tags};

pub type Result<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync + 'static>>;
pub type TestResult = Result<()>;

pub struct BarrelMigrationExecutor {
    database: Quaint,
    sql_variant: barrel::backend::SqlVariant,
    schema_name: String,
    tags: BitFlags<Tags>,
}

impl BarrelMigrationExecutor {
    pub async fn execute<F>(&self, migration_fn: F) -> TestResult
    where
        F: FnOnce(&mut Migration),
    {
        self.execute_with_schema(migration_fn, &self.schema_name).await?;

        Ok(())
    }

    pub async fn execute_with_schema<F>(&self, migration_fn: F, schema_name: &str) -> TestResult
    where
        F: FnOnce(&mut Migration),
    {
        let mut migration = if self.tags.intersects(Tags::Vitess) {
            Migration::new()
        } else {
            Migration::new().schema(schema_name)
        };

        migration_fn(&mut migration);

        let full_sql = migration.make_from(self.sql_variant);

        if full_sql.is_empty() {
            return Ok(());
        }

        self.database.raw_cmd(&full_sql).await?;

        Ok(())
    }
}