iven/hms
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ai): 添加 ai_analysis_queue 迁移 + Entity

Browse Source 
异步分析队列表，支持优先级/重试/状态机（pending→running→completed/failed）
索引覆盖租户状态查询和调度扫描，迁移号 000118
This commit is contained in:
iven
2026-05-05 15:35:59 +08:00
parent c268229311
commit 3186c5aee9
4 changed files with 184 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
crates/erp-ai/src/entity/ai_analysis_queue.rs Normal file
View File

@@ -0,0 +1,34 @@
use sea_orm::entity::prelude::*;
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Serialize, Deserialize)]
#[sea_orm(table_name = "ai_analysis_queue")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = false)]
pub id: Uuid,
pub tenant_id: Uuid,
pub patient_id: Uuid,
pub analysis_type: String,
pub priority: i32,
pub status: String,
pub source_event: Option<String>,
pub source_ref: String,
pub scheduled_at: DateTimeUtc,
pub started_at: Option<DateTimeUtc>,
pub completed_at: Option<DateTimeUtc>,
pub result_analysis_id: Option<Uuid>,
pub error_message: Option<String>,
pub retry_count: i32,
pub max_retries: i32,
pub created_at: DateTimeUtc,
pub updated_at: DateTimeUtc,
pub created_by: Option<Uuid>,
pub updated_by: Option<Uuid>,
pub deleted_at: Option<DateTimeUtc>,
pub version_lock: i32,
}
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
pub enum Relation {}
impl ActiveModelBehavior for ActiveModel {}

1
crates/erp-ai/src/entity/mod.rs
View File

@@ -1,4 +1,5 @@
pub mod ai_analysis;
pub mod ai_analysis_queue;
pub mod ai_prompt;
pub mod ai_risk_threshold;
pub mod ai_suggestion;

2
crates/erp-server/migration/src/lib.rs
View File

@@ -117,6 +117,7 @@ mod m20260505_000114_dialysis_record_add_workflow_instance;
mod m20260505_000115_family_member_health_proxy;
mod m20260505_000116_seed_missing_health_menus;
mod m20260505_000117_create_ai_tenant_configs;
mod m20260505_000118_create_ai_analysis_queue;
pub struct Migrator;
@@ -241,6 +242,7 @@ impl MigratorTrait for Migrator {
Box::new(m20260505_000115_family_member_health_proxy::Migration),
Box::new(m20260505_000116_seed_missing_health_menus::Migration),
Box::new(m20260505_000117_create_ai_tenant_configs::Migration),
Box::new(m20260505_000118_create_ai_analysis_queue::Migration),
]
}
}

147
crates/erp-server/migration/src/m20260505_000118_create_ai_analysis_queue.rs Normal file
View File

@@ -0,0 +1,147 @@
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> {
manager
.create_table(
Table::create()
.table(AiAnalysisQueue::Table)
.if_not_exists()
.col(ColumnDef::new(AiAnalysisQueue::Id).uuid().not_null().primary_key())
.col(ColumnDef::new(AiAnalysisQueue::TenantId).uuid().not_null())
.col(ColumnDef::new(AiAnalysisQueue::PatientId).uuid().not_null())
.col(
ColumnDef::new(AiAnalysisQueue::AnalysisType)
.string_len(50)
.not_null(),
)
.col(
ColumnDef::new(AiAnalysisQueue::Priority)
.integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(AiAnalysisQueue::Status)
.string_len(20)
.not_null()
.default("pending"),
)
.col(ColumnDef::new(AiAnalysisQueue::SourceEvent).string_len(100).null())
.col(
ColumnDef::new(AiAnalysisQueue::SourceRef)
.string_len(200)
.not_null()
.default(""),
)
.col(
ColumnDef::new(AiAnalysisQueue::ScheduledAt)
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.col(ColumnDef::new(AiAnalysisQueue::StartedAt).timestamp_with_time_zone().null())
.col(ColumnDef::new(AiAnalysisQueue::CompletedAt).timestamp_with_time_zone().null())
.col(ColumnDef::new(AiAnalysisQueue::ResultAnalysisId).uuid().null())
.col(ColumnDef::new(AiAnalysisQueue::ErrorMessage).text().null())
.col(
ColumnDef::new(AiAnalysisQueue::RetryCount)
.integer()
.not_null()
.default(0),
)
.col(
ColumnDef::new(AiAnalysisQueue::MaxRetries)
.integer()
.not_null()
.default(3),
)
.col(
ColumnDef::new(AiAnalysisQueue::CreatedAt)
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.col(
ColumnDef::new(AiAnalysisQueue::UpdatedAt)
.timestamp_with_time_zone()
.not_null()
.default(Expr::current_timestamp()),
)
.col(ColumnDef::new(AiAnalysisQueue::CreatedBy).uuid().null())
.col(ColumnDef::new(AiAnalysisQueue::UpdatedBy).uuid().null())
.col(ColumnDef::new(AiAnalysisQueue::DeletedAt).timestamp_with_time_zone().null())
.col(
ColumnDef::new(AiAnalysisQueue::VersionLock)
.integer()
.not_null()
.default(1),
)
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.if_not_exists()
.name("idx_ai_analysis_queue_tenant_status")
.table(AiAnalysisQueue::Table)
.col(AiAnalysisQueue::TenantId)
.col(AiAnalysisQueue::Status)
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.if_not_exists()
.name("idx_ai_analysis_queue_scheduled")
.table(AiAnalysisQueue::Table)
.col(AiAnalysisQueue::Status)
.col(AiAnalysisQueue::ScheduledAt)
.to_owned(),
)
.await?;
Ok(())
}
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
manager
.drop_table(Table::drop().table(AiAnalysisQueue::Table).to_owned())
.await?;
Ok(())
}
}
#[derive(DeriveIden)]
enum AiAnalysisQueue {
Table,
Id,
TenantId,
PatientId,
AnalysisType,
Priority,
Status,
SourceEvent,
SourceRef,
ScheduledAt,
StartedAt,
CompletedAt,
ResultAnalysisId,
ErrorMessage,
RetryCount,
MaxRetries,
CreatedAt,
UpdatedAt,
CreatedBy,
UpdatedBy,
DeletedAt,
VersionLock,
}