优化

src/moldinsight/services/storage_integration_rustfs.py

@@ -121,7 +121,8 @@ class StorageIntegrationService:
         )
 
         session.add(stp_file)
-        await session.commit()
+        # D9：仅 flush，与 ProcessingTask 由路由层一并原子提交（避免孤儿文件记录）
+        await session.flush()
         await session.refresh(stp_file)
 
         logger.info(f"STP文件保存成功 RustFS: {stp_file.id}, 批次: {batch_id}")
@@ -134,8 +135,11 @@ class StorageIntegrationService:
         stp_file_id: int,
         task_type: str = "stp_parsing",
         parameters: Optional[Dict[str, Any]] = None,
+        batch_id: Optional[str] = None,
     ) -> ProcessingTask:
-        """创建处理任务记录"""
+        """创建处理任务记录（D9：仅 flush 不 commit，事务由调用方收口——
+        与 STPFile 记录同批提交，避免留下无任务的孤儿文件记录；batch_id 用于批量任务聚合查询）
+        """
         try:
             task = ProcessingTask(
                 task_id=task_id,
@@ -144,15 +148,15 @@ class StorageIntegrationService:
                 status="pending",
                 started_time=datetime.now(),
                 parameters=parameters or {},
+                batch_id=batch_id,
             )
-            
+
             session.add(task)
-            await session.commit()
-            await session.refresh(task)
-            
+            await session.flush()
+
             logger.info(f"处理任务创建成功: {task_id}")
             return task
-            
+
         except Exception as e:
             await session.rollback()
             logger.error(f"创建处理任务失败: {e}")
@@ -167,7 +171,8 @@ class StorageIntegrationService:
         current_step: Optional[str] = None,
         error_message: Optional[str] = None
     ):
-        """更新任务状态"""
+        """更新任务状态（保留即时 commit：进度/状态需跨事务对外可见，
+        处理链路中的各阶段进度依赖它落库——D9 收口仅针对数据本体写方法）"""
         try:
             update_data = {
                 "status": status,
@@ -200,7 +205,7 @@ class StorageIntegrationService:
         task_id: str,
         parameters: Dict[str, Any],
     ):
-        """合并更新任务参数，便于保存阶段耗时等元数据。"""
+        """合并更新任务参数，便于保存阶段耗时等元数据。（D9：flush 不 commit，事务由调用方收口）"""
         try:
             task = await session.execute(
                 select(ProcessingTask).where(ProcessingTask.task_id == task_id)
@@ -212,14 +217,14 @@ class StorageIntegrationService:
             merged = dict(task.parameters or {})
             merged.update(parameters or {})
             task.parameters = merged
-            await session.commit()
+            await session.flush()
         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文件状态"""
+        """更新STP文件状态（保留即时 commit，理由同 update_task_status）"""
         try:
             await session.execute(
                 update(STPFile)
@@ -280,7 +285,8 @@ class StorageIntegrationService:
         )
 
         session.add(geometry_data)
-        await session.commit()
+        # D9：数据本体仅 flush，与网格等同阶段数据由编排层统一 commit（原子落库）
+        await session.flush()
         await session.refresh(geometry_data)
 
         logger.info(f"几何数据保存成功 RustFS: {geometry_data.id}")
@@ -336,7 +342,8 @@ class StorageIntegrationService:
         )
 
         session.add(mesh_data)
-        await session.commit()
+        # D9：数据本体仅 flush，与几何数据同阶段由编排层统一 commit
+        await session.flush()
         await session.refresh(mesh_data)
 
         logger.info(f"网格数据保存成功 RustFS: {mesh_data.id}")
@@ -417,7 +424,8 @@ class StorageIntegrationService:
         )
 
         session.add(mold_cavity)
-        await session.commit()
+        # D9：数据本体仅 flush，型腔/HTML/特征同属结果包，由编排层统一 commit
+        await session.flush()
         await session.refresh(mold_cavity)
 
         logger.info(f"模具型腔数据保存成功 RustFS: {mold_cavity.id}")
@@ -466,7 +474,8 @@ class StorageIntegrationService:
         )
 
         session.add(html_file)
-        await session.commit()
+        # D9：数据本体仅 flush，型腔/HTML/特征同属结果包，由编排层统一 commit
+        await session.flush()
         await session.refresh(html_file)
 
         logger.info(f"HTML文件保存成功 RustFS: {html_file.id}")
@@ -504,7 +513,8 @@ class StorageIntegrationService:
             )
             session.add(rec_record)
 
-        await session.commit()
+        # D9：数据本体仅 flush，型腔/HTML/特征同属结果包，由编排层统一 commit
+        await session.flush()
         logger.info(f"保存了 {len(features)} 个特征和 {len(recommendations)} 个建议")
 
     async def log_user_activity(self, session: AsyncSession,
@@ -798,7 +808,8 @@ class StorageIntegrationService:
                     .where(STPFile.id == stp_file_id)
                     .values(**update_data)
                 )
-                await session.commit()
+                # D9：flush 不 commit，随结果包由编排层统一提交
+                await session.flush()
                 logger.info(f"STP文件分析摘要更新: ID {stp_file_id}")
                 
         except Exception as e: