使用SeaORM实现PostgreSQL枚举列时遭遇编译错误求助

1. 配置依赖 #

首先确保Cargo.toml中包含必要的依赖，包括SeaORM的active-enum特性，以及strum相关库（用于枚举迭代）：

[dependencies]
sea-orm = { version = "0.12", features = ["postgres", "active-enum"] }
strum = { version = "0.25", features = ["derive"] }
strum_macros = "0.25"
sea-orm-migration = { version = "0.12", features = ["postgres"] }

2. 正确定义枚举类型 #

创建PaymentOptions枚举，同时派生SeaORM的ActiveEnum和strum的EnumIter trait，明确数据库映射规则：

use sea_orm::entity::prelude::*;
use strum_macros::EnumIter;

#[derive(Debug, Clone, PartialEq, Eq, EnumIter, ActiveEnum)]
#[sea_orm(
    rs_type = "String", 
    db_type = "Enum", 
    enum_name = "payment_options" // 数据库中枚举类型的名称
)]
pub enum PaymentOptions {
    #[sea_orm(string_value = "Cash")]
    Cash,
    #[sea_orm(string_value = "Credit")]
    Credit,
    #[sea_orm(string_value = "Debit")]
    Debit,
    #[sea_orm(string_value = "Voucher")]
    Voucher,
}

3. 在迁移中创建枚举列 #

在迁移文件中，导入strum的IntoEnumIterator trait（解决iter方法找不到的问题），并正确调用enumeration方法：

use sea_orm_migration::prelude::*;
use strum::IntoEnumIterator; // 必须导入此trait才能使用iter()方法
use crate::entities::PaymentOptions; // 替换为你的枚举所在路径

#[derive(DeriveMigrationName)]
pub struct Migration;

#[async_trait::async_trait]
impl MigrationTrait for Migration {
    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        // 先创建数据库枚举类型
        manager.create_enum(
            Enum::create()
                .name(PaymentOptions::enum_name())
                .values(PaymentOptions::iter().map(|v| v.string_value()))
                .to_owned()
        ).await?;

        // 创建Orders表
        manager
            .create_table(
                Table::create()
                    .table(Orders::Table)
                    .col(
                        ColumnDef::new(Orders::Id)
                            .integer()
                            .not_null()
                            .auto_increment()
                            .primary_key(),
                    )
                    .col(
                        ColumnDef::new(Orders::PaymentType)
                            // 第一个参数是数据库枚举名称，第二个是变体值列表
                            .enumeration(PaymentOptions::enum_name(), PaymentOptions::iter().map(|v| v.string_value()))
                            .not_null(),
                    )
                    // 添加其他列...
                    .to_owned(),
            )
            .await
    }

    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        manager.drop_table(Table::drop().table(Orders::Table).to_owned()).await?;
        manager.drop_enum(Enum::drop().name(PaymentOptions::enum_name()).to_owned()).await
    }
}

#[derive(Iden)]
enum Orders {
    Table,
    Id,
    PaymentType,
}

错误原因说明 #

1. 传入枚举类型而非有效值：enumeration方法需要两个参数：数据库中枚举类型的名称，以及枚举所有变体的字符串值列表，而非直接传入枚举类型本身。
2. 找不到iter方法：PaymentOptions::iter()依赖strum库的IntoEnumIterator trait，必须显式导入该trait才能调用迭代方法。

内容的提问来源于stack exchange，提问作者Remi_Zacharias