diff options
Diffstat (limited to 'kvmd/plugins/msd/__init__.py')
| -rw-r--r-- | kvmd/plugins/msd/__init__.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/kvmd/plugins/msd/__init__.py b/kvmd/plugins/msd/__init__.py index 1bb11413..b830b0a9 100644 --- a/kvmd/plugins/msd/__init__.py +++ b/kvmd/plugins/msd/__init__.py @@ -22,6 +22,7 @@ import os import contextlib +import time from typing import Dict from typing import Type @@ -36,6 +37,7 @@ from ...logging import get_logger from ...errors import OperationError from ...errors import IsBusyError +from ... import aiotools from ... import aiofs from .. import BasePlugin @@ -187,8 +189,9 @@ class MsdImageReader: logger.exception("Can't close image reader") -class MsdImageWriter: - def __init__(self, path: str, size: int, sync: int) -> None: +class MsdImageWriter: # pylint: disable=too-many-instance-attributes + def __init__(self, notifier: aiotools.AioNotifier, path: str, size: int, sync: int) -> None: + self.__notifier = notifier self.__name = os.path.basename(path) self.__path = path self.__size = size @@ -197,6 +200,7 @@ class MsdImageWriter: self.__file: Optional[aiofiles.base.AiofilesContextManager] = None self.__written = 0 self.__unsynced = 0 + self.__tick = 0.0 def get_file(self) -> aiofiles.base.AiofilesContextManager: assert self.__file is not None @@ -226,6 +230,11 @@ class MsdImageWriter: await aiofs.afile_sync(self.__file) self.__unsynced = 0 + now = time.monotonic() + if self.__tick + 1 < now or self.__written == self.__size: + self.__tick = now + await self.__notifier.notify() + return self.__written async def close(self) -> None: |