input/thread: implement size and seek

This commit is contained in:
Max Kellermann
2024-07-29 22:01:51 +02:00
parent dc51015c75
commit ec30716e01
2 changed files with 58 additions and 3 deletions

View File

@@ -46,6 +46,14 @@ ThreadInputStream::Start()
thread.Start();
}
void
ThreadInputStream::ThreadSeek([[maybe_unused]] offset_type new_offset)
{
assert(!IsSeekable());
throw std::runtime_error{"Not seekable"};
}
inline void
ThreadInputStream::ThreadFunc() noexcept
{
@@ -67,6 +75,24 @@ ThreadInputStream::ThreadFunc() noexcept
while (!close) {
assert(!postponed_exception);
if (IsSeeking()) {
const auto seek_offset_copy = offset = seek_offset;
seek_offset = UNKNOWN_SIZE;
eof = false;
buffer.Clear();
try {
const ScopeUnlock unlock(mutex);
ThreadSeek(seek_offset_copy);
} catch (...) {
postponed_exception = std::current_exception();
InvokeOnAvailable();
break;
}
offset = seek_offset_copy;
}
auto w = buffer.Write();
if (w.empty()) {
wake_cond.wait(lock);
@@ -115,6 +141,14 @@ ThreadInputStream::IsAvailable() const noexcept
return !IsEOF() || postponed_exception;
}
void
ThreadInputStream::Seek([[maybe_unused]] std::unique_lock<Mutex> &lock,
offset_type new_offset)
{
seek_offset = new_offset;
wake_cond.notify_one();
}
size_t
ThreadInputStream::Read(std::unique_lock<Mutex> &lock,
std::span<std::byte> dest)
@@ -125,6 +159,11 @@ ThreadInputStream::Read(std::unique_lock<Mutex> &lock,
if (postponed_exception)
std::rethrow_exception(postponed_exception);
if (IsSeeking()) {
caller_cond.wait(lock);
continue;
}
auto r = buffer.Read();
if (!r.empty()) {
size_t nbytes = std::min(dest.size(), r.size());
@@ -147,5 +186,5 @@ ThreadInputStream::IsEOF() const noexcept
{
assert(!thread.IsInside());
return eof && buffer.empty();
return eof && buffer.empty() && !IsSeeking();
}