x

src/services/storage_integration_rustfs.py

@@ -127,10 +127,14 @@ class StorageIntegrationService:
         logger.info(f"STP文件保存成功 RustFS: {stp_file.id}, 批次: {batch_id}")
         return stp_file
 
-    async def create_processing_task(self, session: AsyncSession,
-                                   task_id: str, 
-                                   stp_file_id: int, 
-                                   task_type: str = "stp_parsing") -> ProcessingTask:
+    async def create_processing_task(
+        self,
+        session: AsyncSession,
+        task_id: str,
+        stp_file_id: int,
+        task_type: str = "stp_parsing",
+        parameters: Optional[Dict[str, Any]] = None,
+    ) -> ProcessingTask:
         """创建处理任务记录"""
         try:
             task = ProcessingTask(
@@ -138,7 +142,8 @@ class StorageIntegrationService:
                 stp_file_id=stp_file_id,
                 task_type=task_type,
                 status="pending",
-                started_time=datetime.now()
+                started_time=datetime.now(),
+                parameters=parameters or {},
             )
             
             session.add(task)
@@ -189,6 +194,30 @@ class StorageIntegrationService:
             logger.error(f"更新任务状态失败: {e}")
             raise
 
+    async def update_task_parameters(
+        self,
+        session: AsyncSession,
+        task_id: str,
+        parameters: Dict[str, Any],
+    ):
+        """合并更新任务参数，便于保存阶段耗时等元数据。"""
+        try:
+            task = await session.execute(
+                select(ProcessingTask).where(ProcessingTask.task_id == task_id)
+            )
+            task = task.scalar_one_or_none()
+            if task is None:
+                return
+
+            merged = dict(task.parameters or {})
+            merged.update(parameters or {})
+            task.parameters = merged
+            await session.commit()
+        except Exception as e:
+            await session.rollback()
+            logger.error(f"更新任务参数失败: {e}")
+            raise
+
     async def update_stp_file_status(self, session: AsyncSession, stp_file_id: int, status: str):
         """更新STP文件状态"""
         try:
@@ -616,7 +645,11 @@ class StorageIntegrationService:
                 'volume_utilization': stp_file.analysis_metrics.volume_utilization,
                 'topology_complexity': stp_file.analysis_metrics.topology_complexity,
                 'wall_uniformity': stp_file.analysis_metrics.wall_uniformity,
-                'analysis_summary': stp_file.analysis_metrics.analysis_summary
+                'analysis_summary': stp_file.analysis_metrics.analysis_summary,
+                'verification_status': stp_file.analysis_metrics.verification_status,
+                'verification_volume_diff': stp_file.analysis_metrics.verification_volume_diff,
+                'verification_area_diff': stp_file.analysis_metrics.verification_area_diff,
+                'verification_details': stp_file.analysis_metrics.verification_details,
             }
 
         return result