42ae580034
- 添加ComfyUI API数据模型 (comfyui.rs) - 实现HTTP客户端服务层 (comfyui_service.rs) - 创建Tauri命令层 (comfyui_commands.rs) - 集成服务到应用状态管理 - 注册所有API命令到Tauri处理器 - 添加urlencoding依赖 - 支持8个API端点的完整功能 API端点包括: - 工作流管理 (获取/发布/删除) - 工作流执行和规范查询 - 服务器状态监控 - 文件列表管理 - 连接测试和配置管理 遵循四层架构模式: 数据模型 基础设施服务 表示命令 应用状态集成
297 lines
14 KiB
Rust
297 lines
14 KiB
Rust
use std::sync::{Arc, Mutex};
|
||
use crate::data::repositories::project_repository::ProjectRepository;
|
||
use crate::data::repositories::material_repository::MaterialRepository;
|
||
use crate::data::repositories::model_repository::ModelRepository;
|
||
use crate::data::repositories::model_dynamic_repository::ModelDynamicRepository;
|
||
use crate::data::repositories::video_generation_repository::VideoGenerationRepository;
|
||
use crate::data::repositories::conversation_repository::ConversationRepository;
|
||
use crate::data::repositories::outfit_photo_generation_repository::OutfitPhotoGenerationRepository;
|
||
use crate::data::repositories::hedra_lipsync_repository::HedraLipSyncRepository;
|
||
use crate::infrastructure::database::Database;
|
||
use crate::infrastructure::performance::PerformanceMonitor;
|
||
use crate::infrastructure::event_bus::EventBusManager;
|
||
use crate::infrastructure::bowong_text_video_agent_service::BowongTextVideoAgentService;
|
||
use crate::infrastructure::comfyui_service::ComfyuiService;
|
||
|
||
|
||
/// 应用全局状态管理
|
||
/// 遵循 Tauri 开发规范的状态管理模式
|
||
pub struct AppState {
|
||
pub database: Mutex<Option<Arc<Database>>>,
|
||
pub project_repository: Mutex<Option<ProjectRepository>>,
|
||
pub material_repository: Mutex<Option<MaterialRepository>>,
|
||
pub model_repository: Mutex<Option<ModelRepository>>,
|
||
pub model_dynamic_repository: Mutex<Option<ModelDynamicRepository>>,
|
||
pub video_generation_repository: Mutex<Option<VideoGenerationRepository>>,
|
||
pub conversation_repository: Mutex<Option<Arc<ConversationRepository>>>,
|
||
pub outfit_photo_generation_repository: Mutex<Option<OutfitPhotoGenerationRepository>>,
|
||
pub hedra_lipsync_repository: Mutex<Option<Arc<HedraLipSyncRepository>>>,
|
||
pub performance_monitor: Mutex<PerformanceMonitor>,
|
||
pub event_bus_manager: Arc<EventBusManager>,
|
||
pub bowong_text_video_agent_service: Mutex<Option<BowongTextVideoAgentService>>,
|
||
pub comfyui_service: Mutex<Option<ComfyuiService>>,
|
||
}
|
||
|
||
impl AppState {
|
||
pub fn new() -> Self {
|
||
Self {
|
||
database: Mutex::new(None),
|
||
project_repository: Mutex::new(None),
|
||
material_repository: Mutex::new(None),
|
||
model_repository: Mutex::new(None),
|
||
model_dynamic_repository: Mutex::new(None),
|
||
video_generation_repository: Mutex::new(None),
|
||
conversation_repository: Mutex::new(None),
|
||
outfit_photo_generation_repository: Mutex::new(None),
|
||
hedra_lipsync_repository: Mutex::new(None),
|
||
performance_monitor: Mutex::new(PerformanceMonitor::new()),
|
||
event_bus_manager: Arc::new(EventBusManager::new()),
|
||
bowong_text_video_agent_service: Mutex::new(None),
|
||
comfyui_service: Mutex::new(None),
|
||
}
|
||
}
|
||
|
||
/// 初始化数据库连接
|
||
/// 遵循安全第一原则，确保数据库初始化的安全性
|
||
/// 默认使用连接池模式以提高并发性能
|
||
pub fn initialize_database(&self) -> anyhow::Result<()> {
|
||
println!("开始初始化数据库连接...");
|
||
|
||
// 暂时使用单连接模式，避免锁竞争问题
|
||
// TODO: 在解决 SQLite 锁问题后重新启用连接池
|
||
let database = Arc::new(Database::new()?);
|
||
println!("使用单连接模式初始化数据库");
|
||
|
||
// 连接池模式代码（暂时注释）
|
||
/*
|
||
let database = match Database::new_with_pool() {
|
||
Ok(db) => {
|
||
println!("连接池模式初始化成功");
|
||
Arc::new(db)
|
||
},
|
||
Err(e) => {
|
||
eprintln!("连接池模式初始化失败: {}, 回退到单连接模式", e);
|
||
Arc::new(Database::new()?)
|
||
}
|
||
};
|
||
*/
|
||
|
||
let project_repository = ProjectRepository::new(database.clone())?;
|
||
let material_repository = MaterialRepository::new(database.clone())?;
|
||
let model_repository = ModelRepository::new(database.clone());
|
||
let model_dynamic_repository = ModelDynamicRepository::new(database.clone());
|
||
let video_generation_repository = VideoGenerationRepository::new(database.clone());
|
||
let conversation_repository = Arc::new(ConversationRepository::new(database.clone()));
|
||
|
||
// 初始化数据库表
|
||
model_dynamic_repository.init_tables()?;
|
||
video_generation_repository.init_tables()?;
|
||
conversation_repository.initialize_tables()?;
|
||
|
||
// 初始化穿搭图片相关表
|
||
let outfit_image_repository = crate::data::repositories::outfit_image_repository::OutfitImageRepository::new(database.clone());
|
||
outfit_image_repository.init_tables()?;
|
||
|
||
// 初始化穿搭照片生成相关表
|
||
let outfit_photo_generation_repository = OutfitPhotoGenerationRepository::new(database.clone())?;
|
||
|
||
// 初始化Hedra口型合成记录表
|
||
let hedra_lipsync_repository = crate::data::repositories::hedra_lipsync_repository::HedraLipSyncRepository::new(database.clone());
|
||
hedra_lipsync_repository.init_tables()?;
|
||
let hedra_lipsync_repository = Arc::new(hedra_lipsync_repository);
|
||
|
||
*self.database.lock().unwrap() = Some(database.clone());
|
||
*self.project_repository.lock().unwrap() = Some(project_repository);
|
||
*self.material_repository.lock().unwrap() = Some(material_repository);
|
||
*self.model_repository.lock().unwrap() = Some(model_repository);
|
||
*self.model_dynamic_repository.lock().unwrap() = Some(model_dynamic_repository);
|
||
*self.video_generation_repository.lock().unwrap() = Some(video_generation_repository);
|
||
*self.conversation_repository.lock().unwrap() = Some(conversation_repository);
|
||
*self.outfit_photo_generation_repository.lock().unwrap() = Some(outfit_photo_generation_repository);
|
||
*self.hedra_lipsync_repository.lock().unwrap() = Some(hedra_lipsync_repository);
|
||
|
||
println!("数据库初始化完成，连接池状态: {}",
|
||
if database.has_pool() { "已启用" } else { "未启用" });
|
||
Ok(())
|
||
}
|
||
|
||
/// 初始化数据库连接（单连接模式，用于测试或特殊场景）
|
||
pub fn initialize_database_single_mode(&self) -> anyhow::Result<()> {
|
||
let database = Arc::new(Database::new()?);
|
||
|
||
let project_repository = ProjectRepository::new(database.clone())?;
|
||
let material_repository = MaterialRepository::new(database.clone())?;
|
||
let model_repository = ModelRepository::new(database.clone());
|
||
let model_dynamic_repository = ModelDynamicRepository::new(database.clone());
|
||
let video_generation_repository = VideoGenerationRepository::new(database.clone());
|
||
let conversation_repository = Arc::new(ConversationRepository::new(database.clone()));
|
||
|
||
// 初始化数据库表
|
||
model_dynamic_repository.init_tables()?;
|
||
video_generation_repository.init_tables()?;
|
||
conversation_repository.initialize_tables()?;
|
||
|
||
// 初始化穿搭图片相关表
|
||
let outfit_image_repository = crate::data::repositories::outfit_image_repository::OutfitImageRepository::new(database.clone());
|
||
outfit_image_repository.init_tables()?;
|
||
|
||
// 初始化穿搭照片生成相关表
|
||
let outfit_photo_generation_repository = OutfitPhotoGenerationRepository::new(database.clone())?;
|
||
|
||
// 初始化Hedra口型合成记录表
|
||
let hedra_lipsync_repository = crate::data::repositories::hedra_lipsync_repository::HedraLipSyncRepository::new(database.clone());
|
||
hedra_lipsync_repository.init_tables()?;
|
||
let hedra_lipsync_repository = Arc::new(hedra_lipsync_repository);
|
||
|
||
*self.database.lock().unwrap() = Some(database.clone());
|
||
*self.project_repository.lock().unwrap() = Some(project_repository);
|
||
*self.material_repository.lock().unwrap() = Some(material_repository);
|
||
*self.model_repository.lock().unwrap() = Some(model_repository);
|
||
*self.model_dynamic_repository.lock().unwrap() = Some(model_dynamic_repository);
|
||
*self.video_generation_repository.lock().unwrap() = Some(video_generation_repository);
|
||
*self.conversation_repository.lock().unwrap() = Some(conversation_repository);
|
||
*self.outfit_photo_generation_repository.lock().unwrap() = Some(outfit_photo_generation_repository);
|
||
*self.hedra_lipsync_repository.lock().unwrap() = Some(hedra_lipsync_repository);
|
||
|
||
println!("数据库初始化完成，使用单连接模式");
|
||
Ok(())
|
||
}
|
||
|
||
/// 获取项目仓库实例
|
||
pub fn get_project_repository(&self) -> anyhow::Result<std::sync::MutexGuard<Option<ProjectRepository>>> {
|
||
Ok(self.project_repository.lock().unwrap())
|
||
}
|
||
|
||
/// 获取素材仓库实例
|
||
pub fn get_material_repository(&self) -> anyhow::Result<std::sync::MutexGuard<Option<MaterialRepository>>> {
|
||
Ok(self.material_repository.lock().unwrap())
|
||
}
|
||
|
||
/// 获取模特仓库实例
|
||
pub fn get_model_repository(&self) -> anyhow::Result<std::sync::MutexGuard<Option<ModelRepository>>> {
|
||
Ok(self.model_repository.lock().unwrap())
|
||
}
|
||
|
||
/// 获取模特动态仓库实例
|
||
pub fn get_model_dynamic_repository(&self) -> anyhow::Result<std::sync::MutexGuard<Option<ModelDynamicRepository>>> {
|
||
Ok(self.model_dynamic_repository.lock().unwrap())
|
||
}
|
||
|
||
/// 获取视频生成仓库实例
|
||
pub fn get_video_generation_repository(&self) -> anyhow::Result<std::sync::MutexGuard<Option<VideoGenerationRepository>>> {
|
||
Ok(self.video_generation_repository.lock().unwrap())
|
||
}
|
||
|
||
/// 获取会话仓库实例
|
||
pub fn get_conversation_repository(&self) -> anyhow::Result<Arc<ConversationRepository>> {
|
||
let repo_guard = self.conversation_repository.lock().unwrap();
|
||
repo_guard.as_ref()
|
||
.ok_or_else(|| anyhow::anyhow!("会话仓库未初始化"))
|
||
.map(|repo| repo.clone())
|
||
}
|
||
|
||
/// 获取数据库实例
|
||
pub fn get_database(&self) -> Arc<Database> {
|
||
// 使用全局静态数据库实例，确保整个应用只有一个数据库实例
|
||
use std::sync::OnceLock;
|
||
static DATABASE_INSTANCE: OnceLock<Arc<Database>> = OnceLock::new();
|
||
|
||
DATABASE_INSTANCE.get_or_init(|| {
|
||
// 只在第一次调用时初始化数据库
|
||
Arc::new(Database::new().unwrap())
|
||
}).clone()
|
||
}
|
||
|
||
/// 获取数据库连接
|
||
pub fn get_connection(&self) -> anyhow::Result<std::sync::Arc<std::sync::Mutex<rusqlite::Connection>>> {
|
||
let database = self.get_database();
|
||
Ok(database.get_connection())
|
||
}
|
||
|
||
/// 用于测试的构造函数
|
||
#[cfg(test)]
|
||
pub fn new_with_database(_database: Arc<Database>) -> Self {
|
||
let state = Self {
|
||
database: Mutex::new(None),
|
||
project_repository: Mutex::new(None),
|
||
material_repository: Mutex::new(None),
|
||
model_repository: Mutex::new(None),
|
||
model_dynamic_repository: Mutex::new(None),
|
||
video_generation_repository: Mutex::new(None),
|
||
conversation_repository: Mutex::new(None),
|
||
outfit_photo_generation_repository: Mutex::new(None),
|
||
hedra_lipsync_repository: Mutex::new(None),
|
||
performance_monitor: Mutex::new(PerformanceMonitor::new()),
|
||
event_bus_manager: Arc::new(EventBusManager::new()),
|
||
bowong_text_video_agent_service: Mutex::new(None),
|
||
comfyui_service: Mutex::new(None),
|
||
};
|
||
// 不直接存储database，而是在需要时返回传入的database
|
||
state
|
||
}
|
||
|
||
/// 初始化 BowongTextVideoAgent 服务
|
||
pub fn initialize_bowong_service(&self, config: crate::data::models::bowong_text_video_agent::BowongTextVideoAgentConfig) -> anyhow::Result<()> {
|
||
let service = BowongTextVideoAgentService::new(config)?;
|
||
|
||
let mut bowong_service = self.bowong_text_video_agent_service.lock()
|
||
.map_err(|e| anyhow::anyhow!("Failed to acquire lock for BowongTextVideoAgent service: {}", e))?;
|
||
|
||
*bowong_service = Some(service);
|
||
|
||
println!("✅ BowongTextVideoAgent 服务初始化成功");
|
||
Ok(())
|
||
}
|
||
|
||
/// 获取 BowongTextVideoAgent 服务的引用
|
||
pub fn get_bowong_service(&self) -> anyhow::Result<std::sync::MutexGuard<Option<BowongTextVideoAgentService>>> {
|
||
self.bowong_text_video_agent_service.lock()
|
||
.map_err(|e| anyhow::anyhow!("Failed to acquire lock for BowongTextVideoAgent service: {}", e))
|
||
}
|
||
|
||
/// 初始化 ComfyUI 服务
|
||
pub fn initialize_comfyui_service(&self, config: crate::data::models::comfyui::ComfyuiConfig) -> anyhow::Result<()> {
|
||
let service = ComfyuiService::new(config)?;
|
||
|
||
let mut comfyui_service = self.comfyui_service.lock()
|
||
.map_err(|e| anyhow::anyhow!("Failed to acquire lock for ComfyUI service: {}", e))?;
|
||
|
||
*comfyui_service = Some(service);
|
||
|
||
println!("✅ ComfyUI 服务初始化成功");
|
||
Ok(())
|
||
}
|
||
|
||
/// 获取 ComfyUI 服务的引用
|
||
pub fn get_comfyui_service(&self) -> Option<ComfyuiService> {
|
||
match self.comfyui_service.lock() {
|
||
Ok(guard) => guard.clone(),
|
||
Err(e) => {
|
||
eprintln!("Failed to acquire lock for ComfyUI service: {}", e);
|
||
None
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 更新 ComfyUI 服务配置
|
||
pub async fn update_comfyui_config(&self, config: crate::data::models::comfyui::ComfyuiConfig) -> anyhow::Result<()> {
|
||
// 创建新的服务实例
|
||
let new_service = ComfyuiService::new(config)?;
|
||
|
||
// 更新服务
|
||
let mut comfyui_service = self.comfyui_service.lock()
|
||
.map_err(|e| anyhow::anyhow!("Failed to acquire lock for ComfyUI service: {}", e))?;
|
||
|
||
*comfyui_service = Some(new_service);
|
||
|
||
println!("✅ ComfyUI 服务配置更新成功");
|
||
Ok(())
|
||
}
|
||
}
|
||
|
||
impl Default for AppState {
|
||
fn default() -> Self {
|
||
Self::new()
|
||
}
|
||
}
|