[fix] celery task

api/app/celery_worker.py

@@ -19,15 +19,35 @@ import app.tasks
 @worker_process_init.connect
 def _reinit_db_pool(**kwargs):
     """
-    prefork 子进程启动时重建 SQLAlchemy 连接池。
+    prefork 子进程启动时重建被 fork 污染的资源。
     
-    fork() 后子进程继承了父进程的连接池和底层 TCP socket，
-    多个子进程共享同一个 socket 会导致 PostgreSQL 连接损坏。
-    dispose() 会关闭继承来的连接，后续操作会自动创建新连接。
+    fork() 后子进程继承了父进程的：
+    1. SQLAlchemy 连接池 — 多进程共享 TCP socket 导致 DB 连接损坏
+    2. ThreadPoolExecutor — fork 后线程状态不确定，第二个任务会死锁
     """
+    # 重建 DB 连接池
     from app.db import engine
     engine.dispose()
     logger.info("DB connection pool disposed for forked worker process")
 
+    # 重建模块级 ThreadPoolExecutor（fork 后线程池不可用）
+    try:
+        from app.core.rag.deepdoc.parser import figure_parser
+        from concurrent.futures import ThreadPoolExecutor
+        figure_parser.shared_executor = ThreadPoolExecutor(max_workers=10)
+        logger.info("figure_parser.shared_executor recreated")
+    except Exception as e:
+        logger.warning(f"Failed to recreate figure_parser.shared_executor: {e}")
+
+    try:
+        from app.core.rag.utils import libre_office
+        from concurrent.futures import ThreadPoolExecutor
+        import os
+        max_workers = os.cpu_count() * 2 if os.cpu_count() else 4
+        libre_office.executor = ThreadPoolExecutor(max_workers=max_workers)
+        logger.info("libre_office.executor recreated")
+    except Exception as e:
+        logger.warning(f"Failed to recreate libre_office.executor: {e}")
+
 
 __all__ = ['celery_app']