✨ Feature: Add support for completely disabling the database.

README.md

@@ -170,6 +170,7 @@ yym68686/uni-api:latest
 
 - CONFIG_URL: The download address of the configuration file, which can be a local file or a remote file, optional
 - TIMEOUT: Request timeout, default is 100 seconds. The timeout can control the time needed to switch to the next channel when one channel does not respond. Optional
+- DISABLE_DATABASE: Whether to disable the database, default is false, optional
 
 ## Get statistical data
 
@@ -189,7 +190,7 @@ There are other statistical data that you can query yourself by writing SQL in t
 
 ## Vercel Deployment
 
-[![Deploy with Vercel](https://vercel.com/button)](https://vercel.com/new/clone?repository-url=https%3A%2F%2Fgithub.com%2Fyym68686%2Funi-api%2Ftree%2Fmain&project-name=uni-api-vercel&repository-name=uni-api-vercel)
+[![Deploy with Vercel](https://vercel.com/button)](https://vercel.com/new/clone?repository-url=https%3A%2F%2Fgithub.com%2Fyym68686%2Funi-api%2Ftree%2Fmain&env=CONFIG_URL,DISABLE_DATABASE&project-name=uni-api-vercel&repository-name=uni-api-vercel)
 
 ## Docker local deployment
 

README_CN.md

@@ -170,6 +170,7 @@ yym68686/uni-api:latest
 
 - CONFIG_URL: 配置文件的下载地址，可以是本地文件，也可以是远程文件，选填
 - TIMEOUT: 请求超时时间，默认为 100 秒，超时时间可以控制当一个渠道没有响应时，切换下一个渠道需要的时间。选填
+- DISABLE_DATABASE: 是否禁用数据库，默认为 false，选填
 
 ## 获取统计数据
 
@@ -189,7 +190,9 @@ yym68686/uni-api:latest
 
 ## Vercel 部署
 
-[![Deploy with Vercel](https://vercel.com/button)](https://vercel.com/new/clone?repository-url=https%3A%2F%2Fgithub.com%2Fyym68686%2Funi-api%2Ftree%2Fmain&project-name=uni-api-vercel&repository-name=uni-api-vercel)
+[![Deploy with Vercel](https://vercel.com/button)](https://vercel.com/new/clone?repository-url=https%3A%2F%2Fgithub.com%2Fyym68686%2Funi-api%2Ftree%2Fmain&env=CONFIG_URL,DISABLE_DATABASE&project-name=uni-api-vercel&repository-name=uni-api-vercel)
+
+点击上面的一键部署按钮后，设置环境变量 `CONFIG_URL` 为配置文件的直链，然后点击 Create 创建项目。
 
 ## Docker 本地部署
 

main.py

@@ -34,8 +34,13 @@ is_debug = bool(os.getenv("DEBUG", False))
 from sqlalchemy import inspect, text
 from sqlalchemy.sql import sqltypes
 
+# 添加新的环境变量检查
+DISABLE_DATABASE = os.getenv("DISABLE_DATABASE", "false").lower() == "true"
+
 async def create_tables():
-    async with engine.begin() as conn:
+    if DISABLE_DATABASE:
+        return
+    async with db_engine.begin() as conn:
         await conn.run_sync(Base.metadata.create_all)
 
         # 检查并添加缺失的列
@@ -85,7 +90,8 @@ async def lifespan(app: FastAPI):
     # for route in frontend_router.routes:
     #     print(f"Route: {route.path}, methods: {route.methods}")
     # 启动时的代码
-    await create_tables()
+    if not DISABLE_DATABASE:
+        await create_tables()
 
     TIMEOUT = float(os.getenv("TIMEOUT", 100))
     timeout = httpx.Timeout(connect=15.0, read=TIMEOUT, write=30.0, pool=30.0)
@@ -178,17 +184,19 @@ class ChannelStat(Base):
     success = Column(Boolean, default=False)
     timestamp = Column(DateTime(timezone=True), server_default=func.now())
 
-# 获取数据库路径
-db_path = os.getenv('DB_PATH', './data/stats.db')
 
-# 确保 data 目录存在
-data_dir = os.path.dirname(db_path)
-os.makedirs(data_dir, exist_ok=True)
+if not DISABLE_DATABASE:
+    # 获取数据库路径
+    db_path = os.getenv('DB_PATH', './data/stats.db')
+
+    # 确保 data 目录存在
+    data_dir = os.path.dirname(db_path)
+    os.makedirs(data_dir, exist_ok=True)
 
-# 创建异步引擎和会话
-# engine = create_async_engine('sqlite+aiosqlite:///' + db_path, echo=False)
-engine = create_async_engine('sqlite+aiosqlite:///' + db_path, echo=is_debug)
-async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+    # 创建异步引擎和会话
+    # db_engine = create_async_engine('sqlite+aiosqlite:///' + db_path, echo=False)
+    db_engine = create_async_engine('sqlite+aiosqlite:///' + db_path, echo=is_debug)
+    async_session = sessionmaker(db_engine, class_=AsyncSession, expire_on_commit=False)
 
 from starlette.types import Scope, Receive, Send
 from starlette.responses import Response
@@ -260,6 +268,8 @@ class LoggingStreamingResponse(Response):
     async def update_stats(self):
         # 这里添加更新数据库的逻辑
         # print("current_info2")
+        if DISABLE_DATABASE:
+            return
         async with async_session() as session:
             async with session.begin():
                 try:
@@ -426,6 +436,8 @@ class StatsMiddleware(BaseHTTPMiddleware):
             request_info.reset(current_request_info)
 
     async def update_stats(self, current_info):
+        if DISABLE_DATABASE:
+            return
         # 这里添加更新数据库的逻辑
         async with async_session() as session:
             async with session.begin():
@@ -440,6 +452,8 @@ class StatsMiddleware(BaseHTTPMiddleware):
                     logger.error(f"Error updating stats: {str(e)}")
 
     async def update_channel_stats(self, request_id, provider, model, api_key, success):
+        if DISABLE_DATABASE:
+            return
         async with async_session() as session:
             async with session.begin():
                 try:
@@ -958,6 +972,8 @@ async def get_stats(
     token: str = Depends(verify_admin_api_key),
     hours: int = Query(default=24, ge=1, le=720, description="Number of hours to look back for stats (1-720)")
 ):
+    if DISABLE_DATABASE:
+        return JSONResponse(content={"stats": {}})
     async with async_session() as session:
         # 计算指定时间范围的开始时间
         start_time = datetime.now(timezone.utc) - timedelta(hours=hours)