From d5fa816ebbab56b76255614872f03ad02223b518 Mon Sep 17 00:00:00 2001
From: KeepingRunning <1599949878@qq.com>
Date: Sun, 30 Mar 2025 09:46:37 +0800
Subject: [PATCH] =?UTF-8?q?ft:=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

---
 requirements.txt       | Bin 672 -> 682 bytes
 src/common/database.py |   9 ++++++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 0dfd751484930ec11fed6da3b69ff72e6f5be121..76e16077faa4e80d647ddafcb2cf1bc4592ac629 100644
GIT binary patch
delta 24
gcmZ3$x{7teI?iZ@Vum7yGKSd6^^Bq$58Yz~0AY*>DF6Tf

delta 10
ScmZ3*x`1`Ux{W(NF#-S?l?3kq

diff --git a/src/common/database.py b/src/common/database.py
index a3e5b4e3..f038955b 100644
--- a/src/common/database.py
+++ b/src/common/database.py
@@ -15,9 +15,12 @@ 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:
+        # Support both standard mongodb:// and mongodb+srv:// connection strings
+        if uri.startswith(("mongodb://", "mongodb+srv://")):
+            return MongoClient(uri)
+        else:
+            raise ValueError("Invalid MongoDB URI. Must start with 'mongodb://' or 'mongodb+srv://'")
 
     if username and password:
         # 如果有用户名和密码，使用认证连接