60 lines
1.7 KiB
Rust
60 lines
1.7 KiB
Rust
use super::m20250905_000001_init::Thought;
|
|
use sea_orm_migration::prelude::*;
|
|
|
|
#[derive(DeriveMigrationName)]
|
|
pub struct Migration;
|
|
|
|
#[async_trait::async_trait]
|
|
impl MigrationTrait for Migration {
|
|
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
manager
|
|
.get_connection()
|
|
.execute_unprepared(
|
|
"CREATE TYPE thought_visibility AS ENUM ('public', 'friends_only', 'private')",
|
|
)
|
|
.await?;
|
|
|
|
// 2. Add the new column to the thoughts table
|
|
manager
|
|
.alter_table(
|
|
Table::alter()
|
|
.table(Thought::Table)
|
|
.add_column(
|
|
ColumnDef::new(ThoughtExtension::Visibility)
|
|
.enumeration(
|
|
"thought_visibility",
|
|
["public", "friends_only", "private"],
|
|
)
|
|
.not_null()
|
|
.default("public"), // Default new thoughts to public
|
|
)
|
|
.to_owned(),
|
|
)
|
|
.await
|
|
}
|
|
|
|
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
|
manager
|
|
.alter_table(
|
|
Table::alter()
|
|
.table(Thought::Table)
|
|
.drop_column(ThoughtExtension::Visibility)
|
|
.to_owned(),
|
|
)
|
|
.await?;
|
|
|
|
// Drop the ENUM type
|
|
manager
|
|
.get_connection()
|
|
.execute_unprepared("DROP TYPE thought_visibility")
|
|
.await?;
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
#[derive(DeriveIden)]
|
|
enum ThoughtExtension {
|
|
Visibility,
|
|
}
|