evento-sql-migrator 2.0.0-alpha.28

SQL migrations for evento event sourcing library.
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

//! Widens columns that were narrowed on databases migrated before the
//! in-place edits to `m0001`/`m0003`, and widens `subscriber.key` /
//! `event.routing_key` for long routing-key-prefixed subscription keys.
//!
//! SQLite ignores VARCHAR lengths entirely, so it needs no alteration.
//! `down` is a no-op: shrinking columns is destructive and the widened
//! definitions are the canonical schema.

use sea_query::{ColumnDef, Table, TableAlterStatement};

use evento_sql::{Event, Snapshot, Subscriber};

pub struct Operation;

fn up_statements() -> Vec<TableAlterStatement> {
    vec![
        Table::alter()
            .table(Event::Table)
            .modify_column(
                ColumnDef::new(Event::AggregatorId)
                    .string()
                    .string_len(64)
                    .not_null(),
            )
            .to_owned(),
        Table::alter()
            .table(Event::Table)
            .modify_column(
                ColumnDef::new(Event::Name)
                    .string()
                    .string_len(50)
                    .not_null(),
            )
            .to_owned(),
        Table::alter()
            .table(Event::Table)
            .modify_column(ColumnDef::new(Event::RoutingKey).string().string_len(255))
            .to_owned(),
        Table::alter()
            .table(Snapshot::Table)
            .modify_column(
                ColumnDef::new(Snapshot::Id)
                    .string()
                    .string_len(64)
                    .not_null(),
            )
            .to_owned(),
        Table::alter()
            .table(Subscriber::Table)
            .modify_column(
                ColumnDef::new(Subscriber::Key)
                    .string()
                    .string_len(255)
                    .not_null(),
            )
            .to_owned(),
    ]
}

#[cfg(feature = "sqlite")]
#[async_trait::async_trait]
impl sqlx_migrator::Operation<sqlx::Sqlite> for Operation {
    async fn up(
        &self,
        _connection: &mut sqlx::SqliteConnection,
    ) -> Result<(), sqlx_migrator::Error> {
        // SQLite ignores VARCHAR lengths; nothing to widen.
        Ok(())
    }

    async fn down(
        &self,
        _connection: &mut sqlx::SqliteConnection,
    ) -> Result<(), sqlx_migrator::Error> {
        Ok(())
    }
}

#[cfg(feature = "mysql")]
#[async_trait::async_trait]
impl sqlx_migrator::Operation<sqlx::MySql> for Operation {
    async fn up(&self, connection: &mut sqlx::MySqlConnection) -> Result<(), sqlx_migrator::Error> {
        for statement in up_statements() {
            let statement = statement.to_string(sea_query::MysqlQueryBuilder);
            sqlx::query(sqlx::AssertSqlSafe(statement.as_str()))
                .execute(&mut *connection)
                .await?;
        }

        Ok(())
    }

    async fn down(
        &self,
        _connection: &mut sqlx::MySqlConnection,
    ) -> Result<(), sqlx_migrator::Error> {
        Ok(())
    }
}

#[cfg(feature = "postgres")]
#[async_trait::async_trait]
impl sqlx_migrator::Operation<sqlx::Postgres> for Operation {
    async fn up(&self, connection: &mut sqlx::PgConnection) -> Result<(), sqlx_migrator::Error> {
        for statement in up_statements() {
            let statement = statement.to_string(sea_query::PostgresQueryBuilder);
            sqlx::query(sqlx::AssertSqlSafe(statement.as_str()))
                .execute(&mut *connection)
                .await?;
        }

        Ok(())
    }

    async fn down(&self, _connection: &mut sqlx::PgConnection) -> Result<(), sqlx_migrator::Error> {
        Ok(())
    }
}