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> { // 1. ALTER articles 表：新增字段 manager .alter_table( Table::alter() .table(Article::Table) .add_column( ColumnDef::new(Article::Status) .string_len(20) .not_null() .default("draft"), ) .add_column(ColumnDef::new(Article::Slug).string_len(200).null()) .add_column( ColumnDef::new(Article::ContentType) .string_len(20) .not_null() .default("rich_text"), ) .add_column(ColumnDef::new(Article::ReviewedBy).uuid().null()) .add_column( ColumnDef::new(Article::ReviewedAt) .timestamp_with_time_zone() .null(), ) .add_column(ColumnDef::new(Article::ReviewNote).text().null()) .add_column( ColumnDef::new(Article::ViewCount) .integer() .not_null() .default(0), ) .add_column( ColumnDef::new(Article::SortOrder) .integer() .not_null() .default(0), ) .add_column(ColumnDef::new(Article::CategoryId).uuid().null()) .to_owned(), ) .await?; // 将已有 published_at 不为 null 的记录状态设为 published manager .get_connection() .execute_unprepared( "UPDATE article SET status = 'published' WHERE published_at IS NOT NULL AND deleted_at IS NULL", ) .await?; // 2. 创建 article_category 表 manager .create_table( Table::create() .table(ArticleCategory::Table) .col( ColumnDef::new(ArticleCategory::Id) .uuid() .not_null() .primary_key(), ) .col(ColumnDef::new(ArticleCategory::TenantId).uuid().not_null()) .col( ColumnDef::new(ArticleCategory::Name) .string_len(100) .not_null(), ) .col(ColumnDef::new(ArticleCategory::Slug).string_len(100).null()) .col(ColumnDef::new(ArticleCategory::ParentId).uuid().null()) .col(ColumnDef::new(ArticleCategory::Description).text().null()) .col( ColumnDef::new(ArticleCategory::SortOrder) .integer() .not_null() .default(0), ) .col( ColumnDef::new(ArticleCategory::CreatedAt) .timestamp_with_time_zone() .not_null() .default(Expr::current_timestamp()), ) .col( ColumnDef::new(ArticleCategory::UpdatedAt) .timestamp_with_time_zone() .not_null() .default(Expr::current_timestamp()), ) .col(ColumnDef::new(ArticleCategory::CreatedBy).uuid().null()) .col(ColumnDef::new(ArticleCategory::UpdatedBy).uuid().null()) .col( ColumnDef::new(ArticleCategory::DeletedAt) .timestamp_with_time_zone() .null(), ) .col( ColumnDef::new(ArticleCategory::Version) .integer() .not_null() .default(1), ) .to_owned(), ) .await?; manager .create_index( Index::create() .name("idx_article_category_tenant") .table(ArticleCategory::Table) .col(ArticleCategory::TenantId) .to_owned(), ) .await?; // 3. 创建 article_tag 表 manager .create_table( Table::create() .table(ArticleTag::Table) .col( ColumnDef::new(ArticleTag::Id) .uuid() .not_null() .primary_key(), ) .col(ColumnDef::new(ArticleTag::TenantId).uuid().not_null()) .col(ColumnDef::new(ArticleTag::Name).string_len(50).not_null()) .col( ColumnDef::new(ArticleTag::CreatedAt) .timestamp_with_time_zone() .not_null() .default(Expr::current_timestamp()), ) .col( ColumnDef::new(ArticleTag::UpdatedAt) .timestamp_with_time_zone() .not_null() .default(Expr::current_timestamp()), ) .col(ColumnDef::new(ArticleTag::CreatedBy).uuid().null()) .col(ColumnDef::new(ArticleTag::UpdatedBy).uuid().null()) .col( ColumnDef::new(ArticleTag::DeletedAt) .timestamp_with_time_zone() .null(), ) .col( ColumnDef::new(ArticleTag::Version) .integer() .not_null() .default(1), ) .to_owned(), ) .await?; manager .create_index( Index::create() .name("idx_article_tag_tenant") .table(ArticleTag::Table) .col(ArticleTag::TenantId) .to_owned(), ) .await?; // 4. 创建 article_article_tag 关联表（复合主键） manager .create_table( Table::create() .table(ArticleArticleTag::Table) .col( ColumnDef::new(ArticleArticleTag::ArticleId) .uuid() .not_null(), ) .col(ColumnDef::new(ArticleArticleTag::TagId).uuid().not_null()) .primary_key( Index::create() .col(ArticleArticleTag::ArticleId) .col(ArticleArticleTag::TagId), ) .to_owned(), ) .await?; // 5. 创建 article_revision 版本历史表 manager .create_table( Table::create() .table(ArticleRevision::Table) .col( ColumnDef::new(ArticleRevision::Id) .uuid() .not_null() .primary_key(), ) .col(ColumnDef::new(ArticleRevision::TenantId).uuid().not_null()) .col(ColumnDef::new(ArticleRevision::ArticleId).uuid().not_null()) .col( ColumnDef::new(ArticleRevision::RevisionNumber) .integer() .not_null(), ) .col( ColumnDef::new(ArticleRevision::Title) .string_len(255) .not_null(), ) .col(ColumnDef::new(ArticleRevision::Content).text().not_null()) .col(ColumnDef::new(ArticleRevision::Summary).text().null()) .col(ColumnDef::new(ArticleRevision::CreatedBy).uuid().null()) .col( ColumnDef::new(ArticleRevision::CreatedAt) .timestamp_with_time_zone() .not_null() .default(Expr::current_timestamp()), ) .to_owned(), ) .await?; manager .create_index( Index::create() .name("idx_article_revision_article") .table(ArticleRevision::Table) .col(ArticleRevision::ArticleId) .to_owned(), ) .await?; Ok(()) } async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> { manager .drop_table(Table::drop().table(ArticleRevision::Table).to_owned()) .await?; manager .drop_table(Table::drop().table(ArticleArticleTag::Table).to_owned()) .await?; manager .drop_table(Table::drop().table(ArticleTag::Table).to_owned()) .await?; manager .drop_table(Table::drop().table(ArticleCategory::Table).to_owned()) .await?; manager .alter_table( Table::alter() .table(Article::Table) .drop_column(Article::Status) .drop_column(Article::Slug) .drop_column(Article::ContentType) .drop_column(Article::ReviewedBy) .drop_column(Article::ReviewedAt) .drop_column(Article::ReviewNote) .drop_column(Article::ViewCount) .drop_column(Article::SortOrder) .drop_column(Article::CategoryId) .to_owned(), ) .await?; Ok(()) } } #[derive(DeriveIden)] enum Article { Table, Status, Slug, ContentType, ReviewedBy, ReviewedAt, ReviewNote, ViewCount, SortOrder, CategoryId, } #[derive(DeriveIden)] enum ArticleCategory { Table, Id, TenantId, Name, Slug, ParentId, Description, SortOrder, CreatedAt, UpdatedAt, CreatedBy, UpdatedBy, DeletedAt, Version, } #[derive(DeriveIden)] enum ArticleTag { Table, Id, TenantId, Name, CreatedAt, UpdatedAt, CreatedBy, UpdatedBy, DeletedAt, Version, } #[derive(DeriveIden)] enum ArticleArticleTag { Table, ArticleId, TagId, } #[derive(DeriveIden)] enum ArticleRevision { Table, Id, TenantId, ArticleId, RevisionNumber, Title, Content, Summary, CreatedBy, CreatedAt, }