github-repos/blrec
1
0
Fork 0
mirror of https://github.com/acgnhiki/blrec.git synced 2025-03-26 12:04:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: refactor segment_remuxer

Browse Source
This commit is contained in:
acgnhik 2022-06-11 10:17:28 +08:00
parent f0fa65f269
commit 3519df4fe7
1 changed files with 12 additions and 4 deletions

16
src/blrec/core/operators/segment_remuxer.py
View File

@ -74,8 +74,8 @@ class SegmentRemuxer:
self._stream_remuxer.input.write(cached_data)
self._stream_remuxer.input.write(data.payload)
except Exception as exc:
logger.warning(f'Failed to write data to stream remuxer: {exc}')
except Exception as e:
logger.warning(f'Failed to write data to stream remuxer: {repr(e)}')
self._stream_remuxer.stop()
segment_data_cache.append(data.payload)
@ -104,14 +104,22 @@ class RemuxedStream(io.RawIOBase):
self._offset: int = 0
def read(self, size: int = -1) -> bytes:
if self._stream_remuxer.stopped:
ready = self._stream_remuxer.wait(timeout=self._read_timmeout)
if not ready:
logger.debug(
f'Stream remuxer not ready in {self._read_timmeout} seconds'
)
raise EOFError
try:
data = wait_for(
self._stream_remuxer.output.read,
args=(size,),
timeout=self._read_timmeout,
)
except Exception as exc:
logger.warning(f'Failed to read data from stream remuxer: {exc}')
except Exception as e:
logger.warning(f'Failed to read data from stream remuxer: {repr(e)}')
self._stream_remuxer.stop()
raise EOFError
else: