Complete ring buffer and direct download support #26

This commit is contained in:
Scott E. Graves 2024-12-23 08:43:21 -06:00
parent 8ff90ae769
commit f344665ddc

View File

@ -376,24 +376,31 @@ void ring_buffer_open_file::background_reader_thread() {
unique_mutex_lock read_lock(read_mtx_);
read_lock.unlock();
auto next_chunk = ring_pos_;
while (not stop_requested_) {
read_lock.lock();
auto next_chunk =
ring_pos_ + 1U >= ring_state_.size() ? 0U : ring_pos_ + 1U;
const auto check_and_wait = [this, &next_chunk]() {
unique_mutex_lock chunk_lock(chunk_mtx_);
if (get_read_state().all()) {
chunk_notify_.wait(chunk_lock);
next_chunk = ring_pos_;
}
chunk_notify_.notify_all();
};
next_chunk = next_chunk + 1U > ring_end_ ? ring_begin_ : next_chunk + 1U;
if (not ring_state_[next_chunk % ring_state_.size()]) {
read_lock.unlock();
check_and_wait();
continue;
}
read_lock.unlock();
download_chunk(next_chunk, true);
unique_mutex_lock chunk_lock(chunk_mtx_);
if (get_read_state().all()) {
chunk_notify_.wait(chunk_lock);
}
chunk_notify_.notify_all();
check_and_wait();
}
}
} // namespace repertory