risingwave_meta_model_migration/
m20250121_085800_change_wasm_udf_identifier.rs

1use sea_orm_migration::prelude::{Table as MigrationTable, *};
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        manager
10            .alter_table(
11                MigrationTable::alter()
12                    .table(Function::Table)
13                    .rename_column(Function::Identifier, Function::NameInRuntime)
14                    .to_owned(),
15            )
16            .await?;
17
18        let stmt = Query::update()
19            .table(Function::Table)
20            .value(Function::NameInRuntime, Expr::col(Function::Name))
21            .and_where(Expr::col(Function::Language).is_in(vec!["wasm", "rust"]))
22            .to_owned();
23        manager.exec_stmt(stmt).await
24    }
25
26    async fn down(&self, _manager: &SchemaManager) -> Result<(), DbErr> {
27        Err(DbErr::Migration(
28            "cannot rollback wasm udf identifier migration".to_owned(),
29        ))?
30    }
31}
32
33#[derive(DeriveIden)]
34enum Function {
35    Table,
36    Name,
37    Identifier,
38    NameInRuntime,
39    Language,
40}