From 94e1c89dcb1e0c566d2e6b38c288123d5b61485e Mon Sep 17 00:00:00 2001
From: SengokuCola <1026294844@qq.com>
Date: Sat, 5 Apr 2025 19:10:51 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E5=A2=9E=E5=8A=A0MongoDB=20SRV=E6=A0=BC?=
 =?UTF-8?q?=E5=BC=8F=E6=95=B0=E6=8D=AE=E5=BA=93URI=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/common/database.py | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/common/database.py b/src/common/database.py
index a3e5b4e3b..ee0ead0bd 100644
--- a/src/common/database.py
+++ b/src/common/database.py
@@ -15,9 +15,16 @@ def __create_database_instance():
     password = os.getenv("MONGODB_PASSWORD")
     auth_source = os.getenv("MONGODB_AUTH_SOURCE")
 
-    if uri and uri.startswith("mongodb://"):
-        # 优先使用URI连接
-        return MongoClient(uri)
+    if uri:
+        # 支持标准mongodb://和mongodb+srv://连接字符串
+        if uri.startswith(("mongodb://", "mongodb+srv://")):
+            return MongoClient(uri)
+        else:
+            raise ValueError(
+                "Invalid MongoDB URI format. URI must start with 'mongodb://' or 'mongodb+srv://'. "
+                "For MongoDB Atlas, use 'mongodb+srv://' format. "
+                "See: https://www.mongodb.com/docs/manual/reference/connection-string/"
+            )
 
     if username and password:
         # 如果有用户名和密码，使用认证连接