From 4c544c296ca06715dedf422cfecda327ac26754e Mon Sep 17 00:00:00 2001
From: "mark.tian" <mark.tian@shiliulizi.cn>
Date: Wed, 3 Dec 2025 08:52:13 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=A4=96=E9=83=A8=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E8=B0=83=E7=94=A8=E5=BC=80=E5=85=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .env          | 27 +++++++++++++++++++++++++++
 .env.example  |  2 ++
 app/config.py |  2 ++
 app/routes.py | 11 +++++++++++
 main.py       | 12 ++++++++++++
 5 files changed, 54 insertions(+)
 create mode 100644 .env

diff --git a/.env b/.env
new file mode 100644
index 0000000..1355a51
--- /dev/null
+++ b/.env
@@ -0,0 +1,27 @@
+# 数据库配置
+DATABASE_URL=postgresql+asyncpg://postgres:12345@localhost:5432/ai_talk_callback_db
+
+# Redis配置
+REDIS_URL=redis://localhost:6379/0
+REDIS_LOCK_TIMEOUT=30
+REDIS_LOCK_RETRY_DELAY=0.1
+REDIS_LOCK_MAX_RETRIES=100
+
+# 业务配置
+COUNT_THRESHOLD=3
+EXTERNAL_API_ENABLED=false
+EXTERNAL_API_RETRY_MAX=3
+EXTERNAL_API_URL=https://127.0.0.1/api/openapi/customerApi/aiTaskResultFail
+
+# 日志配置
+LOG_LEVEL=INFO
+LOG_FILE=logs/app.log
+LOG_MAX_BYTES=10485760
+LOG_BACKUP_COUNT=5
+LOG_FORMAT=%(asctime)s - %(name)s - %(levelname)s - %(message)s
+LOG_DATE_FORMAT=%Y-%m-%d %H:%M:%S
+
+# 应用配置
+APP_NAME=AI Talk Callback API
+ENVIRONMENT=development
+DEBUG=false
\ No newline at end of file
diff --git a/.env.example b/.env.example
index c8c99aa..54c0969 100644
--- a/.env.example
+++ b/.env.example
@@ -9,6 +9,7 @@ REDIS_LOCK_MAX_RETRIES=100
 
 # 业务配置
 COUNT_THRESHOLD=3
+EXTERNAL_API_ENABLED=false
 EXTERNAL_API_RETRY_MAX=3
 EXTERNAL_API_URL=https://jeep-api.d2c.stlassac.com/api/openapi/customerApi/aiTaskResultFail
 
@@ -22,4 +23,5 @@ LOG_DATE_FORMAT=%Y-%m-%d %H:%M:%S
 
 # 应用配置
 APP_NAME=AI Talk Callback API
+ENVIRONMENT=production # 环境: production, test, development
 DEBUG=false
\ No newline at end of file
diff --git a/app/config.py b/app/config.py
index 3574d24..704f91c 100644
--- a/app/config.py
+++ b/app/config.py
@@ -13,6 +13,7 @@ class Settings(BaseSettings):
     
     # 业务配置
     count_threshold: int = 3  # count阈值，大于等于此值直接返回
+    external_api_enabled: bool = False  # 是否启用外部API调用
     external_api_retry_max: int = 3  # 外部API最大重试次数
     external_api_url: str = ""
     
@@ -26,6 +27,7 @@ class Settings(BaseSettings):
     
     # 应用配置
     app_name: str = "AI Talk Callback API"
+    environment: str = "production"  # 环境: production, test, development
     debug: bool = False
     
     class Config:
diff --git a/app/routes.py b/app/routes.py
index 000b056..2b40c57 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -256,6 +256,17 @@ async def ai_talk_callback(
                 site_id=siteId
             )
         
+        # 检查是否启用外部API调用
+        if not settings.external_api_enabled:
+            logger.info(f"🔌 外部API调用已禁用，直接返回成功")
+            return CallbackResponse(
+                success=True,
+                message="外部API调用已禁用，直接返回",
+                processed=False,
+                retry_count=0,
+                site_id=siteId
+            )
+        
         logger.info(f"📞 count={callback_data.count} < {settings.count_threshold}，调用外部API")
         
         # count < 3，调用外部API，使用分布式锁防止并发调用
diff --git a/main.py b/main.py
index 9aca2fd..a693e3d 100644
--- a/main.py
+++ b/main.py
@@ -69,12 +69,24 @@ app.include_router(router)
 
 @app.get("/")
 async def root():
+    # 检查环境，生产环境下禁用根接口
+    if settings.environment == "production":
+        logger.warning("🚫 生产环境下禁止访问根接口")
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="Not Found")
+    
     logger.info("📝 根接口被访问")
     return {"message": f"Welcome to {settings.app_name}"}
 
 
 @app.get("/health")
 async def health_check():
+    # 检查环境，生产环境下禁用健康检查接口
+    if settings.environment == "production":
+        logger.warning("🚫 生产环境下禁止访问健康检查接口")
+        from fastapi import HTTPException
+        raise HTTPException(status_code=404, detail="Not Found")
+    
     logger.debug("💓 健康检查接口被访问")
     return {"status": "healthy"}