gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optimize log info

Browse Source
This commit is contained in:
yangdx
2025-04-28 23:17:09 +08:00
parent 40a2357c14
commit 3aef63cc65
1 changed files with 16 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
lightrag/kg/shared_storage.py
View File

@@ -65,17 +65,17 @@ class UnifiedLock(Generic[T]):
async def __aenter__(self) -> "UnifiedLock[T]": async def __aenter__(self) -> "UnifiedLock[T]":
try: try:
direct_log( # direct_log(
f"== Lock == Process {self._pid}: Acquiring lock '{self._name}' (async={self._is_async})", # f"== Lock == Process {self._pid}: Acquiring lock '{self._name}' (async={self._is_async})",
enable_output=self._enable_logging, # enable_output=self._enable_logging,
) # )
# If in multiprocess mode and async lock exists, acquire it first # If in multiprocess mode and async lock exists, acquire it first
if not self._is_async and self._async_lock is not None: if not self._is_async and self._async_lock is not None:
direct_log( # direct_log(
f"== Lock == Process {self._pid}: Acquiring async lock for '{self._name}'", # f"== Lock == Process {self._pid}: Acquiring async lock for '{self._name}'",
enable_output=self._enable_logging, # enable_output=self._enable_logging,
) # )
await self._async_lock.acquire() await self._async_lock.acquire()
direct_log( direct_log(
f"== Lock == Process {self._pid}: Async lock for '{self._name}' acquired", f"== Lock == Process {self._pid}: Async lock for '{self._name}' acquired",
@@ -112,10 +112,10 @@ class UnifiedLock(Generic[T]):
async def __aexit__(self, exc_type, exc_val, exc_tb): async def __aexit__(self, exc_type, exc_val, exc_tb):
main_lock_released = False main_lock_released = False
try: try:
direct_log( # direct_log(
f"== Lock == Process {self._pid}: Releasing lock '{self._name}' (async={self._is_async})", # f"== Lock == Process {self._pid}: Releasing lock '{self._name}' (async={self._is_async})",
enable_output=self._enable_logging, # enable_output=self._enable_logging,
) # )
# Release main lock first # Release main lock first
if self._is_async: if self._is_async:
@@ -127,10 +127,10 @@ class UnifiedLock(Generic[T]):
# Then release async lock if in multiprocess mode # Then release async lock if in multiprocess mode
if not self._is_async and self._async_lock is not None: if not self._is_async and self._async_lock is not None:
direct_log( # direct_log(
f"== Lock == Process {self._pid}: Releasing async lock for '{self._name}'", # f"== Lock == Process {self._pid}: Releasing async lock for '{self._name}'",
enable_output=self._enable_logging, # enable_output=self._enable_logging,
) # )
self._async_lock.release() self._async_lock.release()
direct_log( direct_log(