From 93d445dfdd4c51e4e56a516bae0b81036593434f Mon Sep 17 00:00:00 2001
From: yangdx <gzdaniel@me.com>
Date: Tue, 25 Nov 2025 18:24:39 +0800
Subject: [PATCH] Add pipeline status lock function for legacy compatibility

- Add get_pipeline_status_lock function
- Return NamespaceLock for consistency
- Support workspace parameter
- Enable logging option
- Legacy code compatibility
---
 lightrag/kg/shared_storage.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/lightrag/kg/shared_storage.py b/lightrag/kg/shared_storage.py
index 834cdc8f..ef0f61e2 100644
--- a/lightrag/kg/shared_storage.py
+++ b/lightrag/kg/shared_storage.py
@@ -1683,3 +1683,17 @@ def get_default_workspace() -> str:
     """
     global _default_workspace
     return _default_workspace
+
+
+def get_pipeline_status_lock(
+    enable_logging: bool = False, workspace: str = None
+) -> NamespaceLock:
+    """Return unified storage lock for pipeline status data consistency.
+
+    This function is for compatibility with legacy code only.
+    """
+    global _default_workspace
+    actual_workspace = workspace if workspace else _default_workspace
+    return get_namespace_lock(
+        "pipeline_status", workspace=actual_workspace, enable_logging=enable_logging
+    )