Created
October 31, 2023 05:57
-
-
Save abersheeran/03e01a40c57b9db75f497dc3a16a32ae to your computer and use it in GitHub Desktop.
Write after timing or accumulation.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import abc | |
import threading | |
from typing import Any | |
class WriteCache(metaclass=abc.ABCMeta): | |
def __init__(self, max_size: int = 100, min_time: float = 1) -> None: | |
self.max_size = max_size | |
self.min_time = min_time | |
self.__size = max_size | |
self.__lock = threading.RLock() | |
self.__event = threading.Event() | |
# Start timing thread | |
self.set_writed() | |
def timing_handle(self) -> None: | |
if self.__event.wait(self.min_time): | |
return | |
with self.__lock: | |
try: | |
self.real_write() | |
except: | |
pass | |
finally: | |
self.set_writed() | |
def write(self, data: Any) -> None: | |
with self.__lock: | |
self.__size -= 1 | |
try: | |
return self._write(data) | |
finally: | |
if self.should_write: | |
try: | |
self.real_write() | |
finally: | |
self.set_writed() | |
@property | |
def should_write(self) -> bool: | |
return self.__size <= 0 | |
def set_writed(self) -> None: | |
self.__size = self.max_size | |
self.__event.set() | |
self.__event.clear() | |
threading.Thread(target=self.timing_handle, daemon=True).start() | |
@abc.abstractmethod | |
def _write(self, data: Any) -> None: | |
raise NotImplementedError | |
@abc.abstractmethod | |
def real_write(self) -> None: | |
raise NotImplementedError |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment