fixed drop

This commit is contained in:
Yannick Stephan
2025-02-18 10:21:14 +01:00
parent 49b1fc9bdf
commit fc0cf2934e
4 changed files with 2 additions and 24 deletions

View File

@@ -49,7 +49,4 @@ class JsonKVStorage(BaseKVStorage):
async def delete(self, ids: list[str]) -> None:
for doc_id in ids:
self._data.pop(doc_id, None)
await self.index_done_callback()
async def drop(self) -> None:
self._data = {}
await self.index_done_callback()

View File

@@ -117,11 +117,6 @@ class MongoKVStorage(BaseKVStorage):
# Mongo handles persistence automatically
pass
async def drop(self) -> None:
"""Drop the collection"""
await self._data.drop()
@final
@dataclass
class MongoDocStatusStorage(DocStatusStorage):
@@ -202,11 +197,6 @@ class MongoDocStatusStorage(DocStatusStorage):
# Mongo handles persistence automatically
pass
async def drop(self) -> None:
"""Drop the collection"""
await self._data.drop()
@final
@dataclass
class MongoGraphStorage(BaseGraphStorage):

View File

@@ -60,9 +60,4 @@ class RedisKVStorage(BaseKVStorage):
async def index_done_callback(self) -> None:
# Redis handles persistence automatically
pass
async def drop(self) -> None:
keys = await self._redis.keys(f"{self.namespace}:*")
if keys:
await self._redis.delete(*keys)
pass