gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: adds redis locks

Browse source
This commit is contained in:
hajdul88 2025-10-07 19:45:48 +02:00
parent c72c4c56b9
commit 466707b003
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
cognee/infrastructure/databases/graph/kuzu/adapter.py
View file

@ -225,22 +225,24 @@ class KuzuAdapter(GraphDBInterface):
logger.error(f"Query execution failed: {str(e)}") logger.error(f"Query execution failed: {str(e)}")
raise raise
if not cache_config.caching: if cache_config.caching:
async with self._connection_change_lock: async with self._connection_change_lock:
self.open_connections += 1 self.open_connections += 1
if self._is_closed: if self._is_closed:
self.redis_lock.acquire()
self.reopen() self.reopen()
logger.info(f"Open connections after open: {self.open_connections}") logger.info(f"Open connections after open: {self.open_connections}")
result = await loop.run_in_executor(self.executor, blocking_query) result = await loop.run_in_executor(self.executor, blocking_query)
if not cache_config.caching: if cache_config.caching:
async with self._connection_change_lock: async with self._connection_change_lock:
self.open_connections -= 1 self.open_connections -= 1
logger.info(f"Opened connections after closing {self.open_connections}") logger.info(f"Opened connections after closing {self.open_connections}")
if self.open_connections == 0: if self.open_connections == 0:
self.connection.execute("CHECKPOINT;") self.connection.execute("CHECKPOINT;")
self.close() self.close()
self.redis_lock.release()
return result return result