pipe-io-server/pipe_io_server/pipe_io_thread.py

46 lines
1.0 KiB
Python

from threading import Thread, Event
from .terminate_process_on_failure import terminate_process_on_failure
class PipeIOThread(Thread):
def __init__(self):
super().__init__(daemon=True)
self._stop_event = Event()
self._io_threads = []
@terminate_process_on_failure
def run(self):
self._init_io_thread()
for thread in self._io_threads:
thread.start()
self._stop_event.wait()
self._stop_threads()
def _init_io_thread(self):
raise NotImplementedError()
def _stop_threads(self):
for thread in self._io_threads:
if thread.is_alive():
thread.stop()
self.on_stop()
def on_stop(self):
pass
def stop(self):
self._stop_event.set()
if self.is_alive():
self.join()
def wait(self):
self._stop_event.wait()
def __enter__(self):
self.start()
return self
def __exit__(self, type_, value, tb):
self.stop()