优化整体任务性能

src/services/processing_service.py

@@ -4,6 +4,7 @@
 import asyncio
 import time
 import traceback
+from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime
 from pathlib import Path
 from typing import Optional, Dict, Any
@@ -46,6 +47,7 @@ class ProcessingService:
         self.multi_scheme_planner = MultiSchemeMoldPlanner()
         self.cad_exporter = CADExporter()
         self._export_shapes_cache: Dict[str, Dict[str, Dict[str, Any]]] = {}
+        self._occ_executor = ThreadPoolExecutor(max_workers=2, thread_name_prefix="occ")
 
     # ─── 对外入口 ───
 
@@ -115,8 +117,13 @@ class ProcessingService:
             )
 
             stage_started = time.perf_counter()
-            shape = self.stp_parser.load_step_file(Path(file_path))
-            geometry_data = self.stp_parser.analyze_geometry(shape)
+            loop = asyncio.get_running_loop()
+            shape = await loop.run_in_executor(
+                self._occ_executor, self.stp_parser.load_step_file, Path(file_path)
+            )
+            geometry_data = await loop.run_in_executor(
+                self._occ_executor, self.stp_parser.analyze_geometry, shape
+            )
             stage_timings["parse_stp"] = round(time.perf_counter() - stage_started, 3)
 
             # 2. 生成网格数据并持久化
@@ -206,24 +213,21 @@ class ProcessingService:
             pointcloud_data = None
             lod_data = None
             if mesh_result:
+                lod0 = mesh_result.get("lods", {}).get("0", {})
                 pointcloud_data = {
                     "points": mesh_result.get("points", []),
                     "normals": mesh_result.get("normals", []),
-                    "vertices": mesh_result.get("vertices", []),
-                    "faces": mesh_result.get("faces", []),
+                    "vertices": lod0.get("vertices", []),
+                    "faces": lod0.get("faces", []),
                     "point_count": mesh_result.get("point_count", 0),
                     "vertex_count": mesh_result.get("vertex_count", 0),
                     "face_count": mesh_result.get("face_count", 0),
                 }
 
-            # 生成多级LOD数据（用于前端按距离切换精度）
-            try:
-                lod_result = self.mesh_generator.generate_multi_lod_mesh(shape)
-                if lod_result and lod_result.get("lods"):
-                    lod_data = lod_result
-                    logger.info(f"LOD数据生成成功: {len(lod_result['lods'])} 级 (面数: {[lod_result['lods'][k]['face_count'] for k in sorted(lod_result['lods'].keys())]})")
-            except Exception as lod_err:
-                logger.warning(f"LOD数据生成失败，使用单级精度: {lod_err}")
+            if mesh_result and mesh_result.get("lods"):
+                lods = mesh_result["lods"]
+                lod_data = mesh_result
+                logger.info(f"LOD数据复用成功: {len(lods)} 级 (面数: {[lods[k]['face_count'] for k in sorted(lods.keys())]})")
 
             detailed_cavity_json = await self._attach_scheme_previews(
                 detailed_cavity_json=detailed_cavity_json,
@@ -260,10 +264,14 @@ class ProcessingService:
 
             # 9. 分析模具设计
             stage_started = time.perf_counter()
-            analysis_result = self.geometry_analyzer.analyze_mold_design(
-                geometry_data,
-                product_material=requested_material,
-                shape=shape,
+            loop = asyncio.get_running_loop()
+            analysis_result = await loop.run_in_executor(
+                self._occ_executor,
+                lambda: self.geometry_analyzer.analyze_mold_design(
+                    geometry_data,
+                    product_material=requested_material,
+                    shape=shape,
+                ),
             )
 
             if analysis_result:
@@ -328,15 +336,14 @@ class ProcessingService:
                 },
             )
 
-            # 更新任务缓存状态
+            # 更新任务缓存状态（仅保留轻量摘要，完整数据由PG+RustFS持久化）
             await redis_task_manager.update_task(task_id, {
+                "status": ProcessingStatus.COMPLETED,
+                "completed_at": str(datetime.now()),
                 "geometry_data": geometry_data,
                 "analysis_result": analysis_result,
-                "plan_result": detailed_cavity_json,
-                "candidate_schemes": detailed_cavity_json.get("candidate_schemes", []),
-                "best_scheme_id": detailed_cavity_json.get("best_scheme_id"),
-                "cavity_data": best_cavity_data,
                 "key_info": best_key_info,
+                "best_scheme_id": detailed_cavity_json.get("best_scheme_id"),
                 "material": requested_material,
                 "parameters": process_params,
                 "stage_timings": stage_timings,
@@ -344,8 +351,6 @@ class ProcessingService:
                 "verification": verification_result,
                 "llm_report": llm_report,
                 "export_artifacts": export_artifacts,
-                "status": ProcessingStatus.COMPLETED,
-                "completed_at": str(datetime.now()),
             })
 
             logger.info(f"模具型腔生成完成: {task_id}")
@@ -375,18 +380,22 @@ class ProcessingService:
         self, shape, geometry_data: dict, file_path: str,
         db_session: AsyncSession, stp_file_id: int, task_id: str,
     ) -> Optional[Dict[str, Any]]:
-        """生成网格数据并持久化，失败不影响主流程"""
+        """生成多级LOD网格并持久化，一次OCC剖分+trimesh简化，失败不影响主流程"""
         mesh_result = None
         try:
-            mesh_result = self.mesh_generator.generate_mesh_from_shape(shape)
+            loop = asyncio.get_running_loop()
+            mesh_result = await loop.run_in_executor(
+                self._occ_executor, self.mesh_generator.generate_multi_lod_mesh, shape
+            )
 
-            vertices = mesh_result.get("vertices", [])
-            faces = mesh_result.get("faces", [])
+            lod0 = mesh_result.get("lods", {}).get("0", {})
+            vertices = lod0.get("vertices", [])
+            faces = lod0.get("faces", [])
             points = mesh_result.get("points", [])
             normals = mesh_result.get("normals", [])
             point_count = mesh_result.get("point_count", 0)
-            vertex_count = mesh_result.get("vertex_count", 0)
-            face_count = mesh_result.get("face_count", 0)
+            vertex_count = lod0.get("vertex_count", mesh_result.get("vertex_count", 0))
+            face_count = lod0.get("face_count", mesh_result.get("face_count", 0))
 
             if vertices and faces:
                 bbox = geometry_data.get("bounding_box", {})
@@ -438,11 +447,15 @@ class ProcessingService:
         plan_result = None
         try:
             if shape:
-                plan_result = self.multi_scheme_planner.generate_plan(
-                    shape=shape,
-                    material=selected_material,
-                    is_foam_material=is_foam_material,
-                    process_params=process_params,
+                loop = asyncio.get_running_loop()
+                plan_result = await loop.run_in_executor(
+                    self._occ_executor,
+                    lambda: self.multi_scheme_planner.generate_plan(
+                        shape=shape,
+                        material=selected_material,
+                        is_foam_material=is_foam_material,
+                        process_params=process_params,
+                    ),
                 )
                 logger.info(
                     f"多方案分模完成: 生成 {len(plan_result.get('candidate_schemes', []))} 套方案"