risingwave_meta_model_migration/
m20250916_120000_add_refresh_fields.rs

1use sea_orm_migration::prelude::*;
2
3#[derive(DeriveMigrationName)]
4pub struct Migration;
5
6#[async_trait::async_trait]
7impl MigrationTrait for Migration {
8    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
9        // Add refresh_state column as string, default to "IDLE"
10        manager
11            .alter_table(
12                Table::alter()
13                    .table(TableEnum::Table)
14                    .add_column(ColumnDef::new(TableEnum::RefreshState).string())
15                    .to_owned(),
16            )
17            .await
18    }
19
20    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
21        manager
22            .alter_table(
23                Table::alter()
24                    .table(TableEnum::Table)
25                    .drop_column(TableEnum::RefreshState)
26                    .to_owned(),
27            )
28            .await
29    }
30}
31
32#[derive(DeriveIden)]
33enum TableEnum {
34    #[sea_orm(iden = "table")]
35    Table,
36    RefreshState,
37}