如何在Sea-ORM中创建PostgreSQL复合类型？

PostgreSQL中创建复合类型的原生SQL语法如下：

CREATE TYPE name AS
( attribute_name data_type [ COLLATE collation ] [, ... ] )

你已经成功用Sea-ORM的Schema Builder实现了枚举类型的创建：

DbBackend::Postgres => {
    manager
        .create_type(
            Type::create()
                .as_enum(SomeTable::Table)
                .values([SomeTable::Value, SomeTable::AnotherValue])
                .to_owned(),
        )
        .await?;
}

对于复合类型，由于Sea-ORM的Schema Builder目前没有提供直接的封装API，你可以通过以下两种方式实现：

方式一：直接执行原生SQL #

获取数据库连接后，直接执行原生的复合类型创建语句：

DbBackend::Postgres => {
    let conn = manager.get_connection().await?;
    conn.execute_unprepared(
        "CREATE TYPE user_info AS (username VARCHAR(50), age INT, email VARCHAR(100))"
    ).await?;
}

方式二：自定义Schema Builder扩展 #

如果想贴合Sea-ORM的链式调用风格，可以自定义扩展方法，通过custom方法拼接复合类型的定义：

use sea_orm::sea_query::{Type, TypeCreateStatement};

// 自定义复合类型构建扩展
trait CompositeTypeCreateExt {
    fn as_composite(self, type_name: &str, fields: &[(&str, &str)]) -> Self;
}

impl CompositeTypeCreateExt for TypeCreateStatement {
    fn as_composite(mut self, type_name: &str, fields: &[(&str, &str)]) -> Self {
        let fields_def = fields
            .iter()
            .map(|(name, typ)| format!("{} {}", name, typ))
            .collect::<Vec<_>>()
            .join(", ");
        
        self.custom(format!("CREATE TYPE {} AS ({})", type_name, fields_def));
        self
    }
}

// 使用示例
DbBackend::Postgres => {
    manager
        .create_type(
            Type::create()
                .as_composite("user_info", &[("username", "VARCHAR(50)"), ("age", "INT"), ("email", "VARCHAR(100)")])
                .to_owned(),
        )
        .await?;
}

如果需要支持COLLATE等额外参数，可以在自定义扩展中增加对应的处理逻辑，确保生成的SQL符合PostgreSQL语法要求。

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