Skip to main content

risingwave_meta_model_migration/
m20260804_000000_worker_property_resource.rs

1use sea_orm_migration::prelude::*;
2
3use crate::utils::ColumnDefExt;
4
5#[derive(DeriveMigrationName)]
6pub struct Migration;
7
8#[async_trait::async_trait]
9impl MigrationTrait for Migration {
10    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
11        manager
12            .alter_table(
13                Table::alter()
14                    .table(WorkerProperty::Table)
15                    .add_column(ColumnDef::new(WorkerProperty::Resource).rw_binary(manager))
16                    .to_owned(),
17            )
18            .await?;
19        manager
20            .alter_table(
21                Table::alter()
22                    .table(WorkerProperty::Table)
23                    .add_column(ColumnDef::new(WorkerProperty::StartedAt).big_integer())
24                    .to_owned(),
25            )
26            .await?;
27        manager
28            .exec_stmt(Query::delete().from_table(Worker::Table).to_owned())
29            .await?;
30        Ok(())
31    }
32
33    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
34        manager
35            .alter_table(
36                Table::alter()
37                    .table(WorkerProperty::Table)
38                    .drop_column(WorkerProperty::Resource)
39                    .to_owned(),
40            )
41            .await?;
42        manager
43            .alter_table(
44                Table::alter()
45                    .table(WorkerProperty::Table)
46                    .drop_column(WorkerProperty::StartedAt)
47                    .to_owned(),
48            )
49            .await?;
50        Ok(())
51    }
52}
53
54#[derive(DeriveIden)]
55enum Worker {
56    Table,
57}
58
59#[derive(DeriveIden)]
60enum WorkerProperty {
61    Table,
62    Resource,
63    StartedAt,
64}