Unverified Commit 41628dc1 authored by Teng Ma's avatar Teng Ma Committed by GitHub
Browse files

[HiCache] fix: check clear() method for storage backend (#10096)


Co-authored-by: default avatarhzh0425 <hzh0425@apache.org>
parent a12061df
...@@ -134,11 +134,24 @@ class HiRadixCache(RadixCache): ...@@ -134,11 +134,24 @@ class HiRadixCache(RadixCache):
height += 1 height += 1
return height return height
def clear_storage_backend(self): def clear_storage_backend(self) -> bool:
if self.enable_storage: if self.enable_storage:
self.cache_controller.storage_backend.clear() try:
logger.info("Hierarchical cache storage backend cleared successfully!") # Check if the storage backend has a clear method (for nixl backends)
return True if hasattr(self.cache_controller.storage_backend, "clear"):
self.cache_controller.storage_backend.clear()
logger.info(
"Hierarchical cache storage backend cleared successfully!"
)
return True
else:
logger.warning(
f"Storage backend {type(self.cache_controller.storage_backend).__name__} does not support clear operation."
)
return False
except Exception as e:
logger.error(f"Failed to clear hierarchical cache storage backend: {e}")
return False
else: else:
logger.warning("Hierarchical cache storage backend is not enabled.") logger.warning("Hierarchical cache storage backend is not enabled.")
return False return False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment