From c346af050b2d7e0ae2dbd9fd5857077edb9cd4ca Mon Sep 17 00:00:00 2001
From: fit2cloud-chenyw <yawen.chen@fit2cloud.com>
Date: Thu, 4 Sep 2025 13:19:53 +0800
Subject: [PATCH] perf: Cache Disabled for MCP Service

---
 backend/common/core/sqlbot_cache.py | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/backend/common/core/sqlbot_cache.py b/backend/common/core/sqlbot_cache.py
index 6c827207..15018019 100644
--- a/backend/common/core/sqlbot_cache.py
+++ b/backend/common/core/sqlbot_cache.py
@@ -68,7 +68,7 @@ def cache(
         
         @wraps(func)
         async def wrapper(*args, **kwargs):
-            if not settings.CACHE_TYPE or settings.CACHE_TYPE.lower() == "none":
+            if not settings.CACHE_TYPE or settings.CACHE_TYPE.lower() == "none" or not is_cache_initialized():
                 return await func(*args, **kwargs)
             # 生成缓存键
             cache_key = used_key_builder(
@@ -96,7 +96,7 @@ def clear_cache(
     def decorator(func):
         @wraps(func)
         async def wrapper(*args, **kwargs):
-            if not settings.CACHE_TYPE or settings.CACHE_TYPE.lower() == "none":
+            if not settings.CACHE_TYPE or settings.CACHE_TYPE.lower() == "none" or not is_cache_initialized():
                 return await func(*args, **kwargs)
             cache_key = custom_key_builder(
                 func=func,
@@ -138,4 +138,21 @@ def init_sqlbot_cache():
         SQLBotLogUtil.info(f"SQLBot 使用Redis缓存, 可使用多进程模式")
     else:
         SQLBotLogUtil.warning("SQLBot 未启用缓存, 可使用多进程模式")
-    
\ No newline at end of file
+    
+
+def is_cache_initialized() -> bool:
+    # 检查必要的属性是否存在
+    if not hasattr(FastAPICache, "_backend") or not hasattr(FastAPICache, "_prefix"):
+        return False
+    
+    # 检查属性值是否为 None
+    if FastAPICache._backend is None or FastAPICache._prefix is None:
+        return False
+    
+    # 尝试获取后端确认
+    try:
+        backend = FastAPICache.get_backend()
+        return backend is not None
+    except (AssertionError, AttributeError, Exception) as e:
+        SQLBotLogUtil.debug(f"缓存初始化检查失败: {str(e)}")
+        return False
\ No newline at end of file