feat: add delete method for mongo storage implement
All checks were successful
Linting and Formatting / lint-and-format (push) Successful in 3m47s

This commit is contained in:
2025-05-22 04:41:52 +08:00
parent 702e87492c
commit 8916f8a912

View File

@@ -311,6 +311,17 @@ class MongoDocStatusStorage(DocStatusStorage):
logger.error(f"Error dropping doc status {self._collection_name}: {e}") logger.error(f"Error dropping doc status {self._collection_name}: {e}")
return {"status": "error", "message": str(e)} return {"status": "error", "message": str(e)}
async def delete(self, ids: list[str]) -> None:
try:
result = await self._data.delete_many({"_id": {"$in": ids}})
deleted_count = result.deleted_count
logger.info(
f"Dropped {deleted_count} documents from doc status {self._collection_name}"
)
except PyMongoError as e:
logger.error(f"Error deleting doc status {self._collection_name}: {e}")
@final @final
@dataclass @dataclass