Skip to content

Commit 32e94d8

Browse files
increases db timeout and allow context reentry
1 parent 941e4af commit 32e94d8

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

lib/repositories/repo.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ def __new__(cls, *args, **kwargs):
2020
return cls._instances[cls]
2121

2222
def __init__(self, collection_name: str):
23+
self._entered = False
2324
if not getattr(self, '_initialized', False):
2425
self._collection_name = collection_name
2526
self._initialized_event = asyncio.Event()
@@ -45,7 +46,8 @@ async def _async_init(self):
4546
self._initialized_event.set()
4647

4748
async def __aenter__(self):
48-
await self._initialized_event.wait()
49+
if not self._entered:
50+
await self._initialized_event.wait()
4951
return self
5052

5153
async def __aexit__(self, exc_type, exc_value, traceback):
@@ -63,7 +65,7 @@ def _initialize_connection(self):
6365
server_api=ServerApi("1"),
6466
maxIdleTimeMS=5000,
6567
connectTimeoutMS=5000,
66-
serverSelectionTimeoutMS=15000,
68+
serverSelectionTimeoutMS=30000,
6769
)
6870
self._collection = self._client.rocketpy[self._collection_name]
6971
logger.info("MongoDB client initialized for %s", self.__class__)

0 commit comments

Comments
 (0)