mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-14 20:16:48 +00:00
refactor: Use MongoDB singleton for connection management
This commit is contained in:
1 parent
7bf79675c1
commit
83f2fb1e62
6 files changed
+35
-10
No files matched your search
@@ -0,0 +1,25 @@
|
||||
from application.core import settings
|
||||
from pymongo import MongoClient
|
||||
from flask import current_app, g
|
||||
|
||||
|
||||
class MongoDB:
|
||||
_client = None
|
||||
|
||||
@classmethod
|
||||
def get_client(cls):
|
||||
"""
|
||||
Get the MongoDB client instance, creating it if necessary.
|
||||
"""
|
||||
if cls._client is None:
|
||||
cls._client = MongoClient(settings.MONGO_URI)
|
||||
return cls._client
|
||||
|
||||
@classmethod
|
||||
def close_client(cls):
|
||||
"""
|
||||
Close the MongoDB client connection.
|
||||
"""
|
||||
if cls._client is not None:
|
||||
cls._client.close()
|
||||
cls._client = None
|
||||
Reference in new issue
Block a user