main
use models::migrate::MIGRATION_LOCK_KEY;
use sea_orm::{ConnectionTrait, DatabaseConnection, DbErr, TransactionTrait};
/// 实体优先迁移：实体定义是表结构的唯一真相。
///
/// 每次 app 启动执行一次：先跑实体表达不了的兼容性 SQL，再按实体注册表同步
/// （只新增缺失的表/列/唯一键/外键，幂等），最后补实体表达不了的约束与索引。
/// 整个过程在单个事务内、同一数据库上以 advisory lock 串行化。
///
/// `sea_orm::SchemaBuilder::sync` 会把库里有、实体里没有的单列唯一键补上，也会把
/// 库里存在、实体表达不出来的唯一索引当过期索引删掉（Postgres 上走
/// `ALTER TABLE .. DROP CONSTRAINT`，普通索引会直接报错）。因此复合唯一索引、
/// 带谓词的部分唯一索引必须自己管：`before` 里 `DROP INDEX IF EXISTS`，`after` 里
/// `CREATE UNIQUE INDEX IF NOT EXISTS`。
pub struct EntityFirst<'a> {
before: Vec<&'a str>,
entities: Vec<&'a str>,
after: Vec<&'a str>,
}
impl<'a> EntityFirst<'a> {
pub fn new(entities: &[&'a str]) -> Self {
Self {
before: Vec::new(),
entities: entities.to_vec(),
after: Vec::new(),
}
}
/// 实体同步前必须完成的 SQL（老表补列等），保证实体同步能落地。
pub fn before<T: AsRef<[&'a str]>>(mut self, sql: T) -> Self {
self.before.extend_from_slice(sql.as_ref());
self
}
/// 实体表达不了的约束、复合索引等补充 SQL。
pub fn after<T: AsRef<[&'a str]>>(mut self, sql: T) -> Self {
self.after.extend_from_slice(sql.as_ref());
self
}
pub async fn run(&self, db: &DatabaseConnection) -> Result<(), DbErr> {
let txn = db.begin().await?;
txn.execute_unprepared(&format!(
"SELECT pg_advisory_xact_lock({MIGRATION_LOCK_KEY})"
))
.await?;
for statement in &self.before {
txn.execute_unprepared(statement).await?;
}
for prefix in &self.entities {
db.get_schema_registry(prefix).sync(&txn).await?;
}
for statement in &self.after {
txn.execute_unprepared(statement).await?;
}
txn.commit().await
}
}