From 5b9803842596343c89102f74ed80c16dc6f1ade5 Mon Sep 17 00:00:00 2001
From: Eric-Terminal <121368508+Eric-Terminal@users.noreply.github.com>
Date: Sat, 6 Dec 2025 02:16:00 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=B0=86=20pympler=20=E5=92=8C=20objgra?=
 =?UTF-8?q?ph=20=E6=94=B9=E4=B8=BA=E5=8F=AF=E9=80=89=E4=BE=9D=E8=B5=96?=
 =?UTF-8?q?=EF=BC=8C=E4=BF=AE=E5=A4=8D=20Docker=20=E9=95=9C=E5=83=8F?=
 =?UTF-8?q?=E5=90=AF=E5=8A=A8=E5=A4=B1=E8=B4=A5=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/common/mem_monitor.py | 31 +++++++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/src/common/mem_monitor.py b/src/common/mem_monitor.py
index e9f60585c..a64d3eaeb 100644
--- a/src/common/mem_monitor.py
+++ b/src/common/mem_monitor.py
@@ -22,9 +22,24 @@ from logging.handlers import RotatingFileHandler
 from pathlib import Path
 from typing import TYPE_CHECKING
 
-import objgraph
 import psutil
-from pympler import muppy, summary
+
+# objgraph 是可选依赖，用于对象增长监控
+try:
+    import objgraph
+    OBJGRAPH_AVAILABLE = True
+except ImportError:
+    OBJGRAPH_AVAILABLE = False
+    objgraph = None  # type: ignore[assignment]
+
+# pympler 是可选依赖，用于类型内存分析
+try:
+    from pympler import muppy, summary
+    PYMPLER_AVAILABLE = True
+except ImportError:
+    PYMPLER_AVAILABLE = False
+    muppy = None
+    summary = None
 
 if TYPE_CHECKING:
     from psutil import Process
@@ -153,6 +168,10 @@ def log_object_growth(limit: int = 20) -> None:
     Args:
         limit: 显示的最大增长类型数
     """
+    if not OBJGRAPH_AVAILABLE or objgraph is None:
+        logger.warning("objgraph not available, skipping object growth analysis")
+        return
+    
     logger.info("==== Objgraph growth (top %s) ====", limit)
     try:
         # objgraph.show_growth 默认输出到 stdout，需要捕获输出
@@ -182,6 +201,10 @@ def log_type_memory_diff() -> None:
     """使用 Pympler 查看各类型对象占用的内存变化"""
     global _last_type_summary
     
+    if not PYMPLER_AVAILABLE or muppy is None or summary is None:
+        logger.warning("pympler not available, skipping type memory analysis")
+        return
+    
     import io
     import sys
     
@@ -338,6 +361,10 @@ def debug_leak_for_type(type_name: str, max_depth: int = 5, filename: str | None
     Returns:
         是否成功生成引用图
     """
+    if not OBJGRAPH_AVAILABLE or objgraph is None:
+        logger.warning("objgraph not available, cannot generate backrefs graph")
+        return False
+    
     if filename is None:
         filename = f"{type_name}_backrefs.png"