如何在SeaORM中使用PostgreSQL自定义类型citext并解决实体生成错误？

步骤1：自定义Citext类型并实现SeaORM所需Trait #

SeaORM默认不支持PostgreSQL的citext类型，需要手动定义类型并实现FromSql、IntoSql和ColumnType等核心Trait，让SeaORM能正确处理该类型的读写和数据库映射：

use sea_orm::{entity::prelude::*, sea_query::Value};
use std::fmt;

#[derive(Debug, Clone, PartialEq, Eq)]
pub struct Citext(String);

impl Citext {
    // 便捷构造方法
    pub fn new(s: impl Into<String>) -> Self {
        Self(s.into())
    }

    // 字符串引用访问
    pub fn as_str(&self) -> &str {
        &self.0
    }
}

// 实现Display trait方便格式化输出
impl fmt::Display for Citext {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.0)
    }
}

// 支持从String/&str转换为Citext
impl From<String> for Citext {
    fn from(s: String) -> Self {
        Self(s)
    }
}

impl From<&str> for Citext {
    fn from(s: &str) -> Self {
        Self(s.to_string())
    }
}

// 实现FromSql：从数据库读取citext值
impl FromSql for Citext {
    fn from_sql(value: Value) -> Result<Self, DbErr> {
        match value {
            Value::String(Some(s)) => Ok(Self(s)),
            _ => Err(DbErr::Type(format!(
                "Expected String value for Citext, got {:?}",
                value
            ))),
        }
    }

    fn from_sql_null() -> Result<Self, DbErr> {
        Err(DbErr::Type("Citext cannot be null".into()))
    }

    fn from_sql_nullable(value: Option<Value>) -> Result<Self, DbErr> {
        match value {
            Some(v) => Self::from_sql(v),
            None => Err(DbErr::Type("Citext cannot be null".into())),
        }
    }
}

// 实现IntoSql：将Citext写入数据库
impl IntoSql for Citext {
    fn into_sql(self) -> Value {
        Value::String(Some(self.0))
    }

    fn into_sql_nullable(self) -> Value {
        self.into_sql()
    }
}

// 实现ColumnType：指定数据库类型为citext
impl ColumnType for Citext {
    fn column_type() -> ColumnDef {
        ColumnDef::new(ColumnType::Custom("citext".to_string()))
    }
}

步骤2：修改实体定义，替换默认类型 #

将生成的实体中name字段的类型从String替换为自定义的Citext，确保实体与数据库类型匹配：

use sea_orm::entity::prelude::*;
// 导入自定义的Citext类型，路径根据实际情况调整
use crate::Citext;

#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
#[sea_orm(table_name = "retailers")]
pub struct Model {
    #[sea_orm(primary_key)]
    pub id: i32,
    pub name: Citext, // 替换为自定义Citext类型
}

#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {}

impl ActiveModelBehavior for ActiveModel {}

步骤3：处理生成实体时的枚举错误 #

你遇到的错误是因为SeaORM CLI误将citext识别为枚举类型，生成了错误的sea_orm_active_enums.rs文件。解决方法：

• 直接删除自动生成的src/entity/sea_orm_active_enums.rs文件
• 如果后续需要重新生成实体，可通过--type-map参数指定类型映射，避免误识别：

sea-orm-cli generate entity -u "postgres://你的数据库连接串" --type-map citext=Citext

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