mirror of
https://github.com/dragonflydb/dragonfly.git
synced 2025-05-10 18:05:44 +02:00
50f50c8380
* feat(server): write journal recorod with optional await based on flag issue #788 Signed-off-by: adi_holden <adi@dragonflydb.io>
41 lines
1.1 KiB
C++
41 lines
1.1 KiB
C++
// Copyright 2022, DragonflyDB authors. All rights reserved.
|
|
// See LICENSE for licensing terms.
|
|
//
|
|
|
|
#include "server/journal/streamer.h"
|
|
|
|
namespace dfly {
|
|
|
|
void JournalStreamer::Start(io::Sink* dest) {
|
|
write_fb_ = util::fibers_ext::Fiber(&JournalStreamer::WriterFb, this, dest);
|
|
journal_cb_id_ =
|
|
journal_->RegisterOnChange([this](const journal::Entry& entry, bool allow_await) {
|
|
if (entry.opcode == journal::Op::NOOP) {
|
|
// No recode to write, just await if data was written so consumer will read the data.
|
|
return AwaitIfWritten();
|
|
}
|
|
writer_.Write(entry);
|
|
record_cnt_.fetch_add(1, std::memory_order_relaxed);
|
|
NotifyWritten(allow_await);
|
|
});
|
|
}
|
|
|
|
uint64_t JournalStreamer::GetRecordCount() const {
|
|
return record_cnt_.load(std::memory_order_relaxed);
|
|
}
|
|
|
|
void JournalStreamer::Cancel() {
|
|
journal_->UnregisterOnChange(journal_cb_id_);
|
|
Finalize();
|
|
|
|
if (write_fb_.IsJoinable())
|
|
write_fb_.Join();
|
|
}
|
|
|
|
void JournalStreamer::WriterFb(io::Sink* dest) {
|
|
if (auto ec = ConsumeIntoSink(dest); ec) {
|
|
cntx_->ReportError(ec);
|
|
}
|
|
}
|
|
|
|
} // namespace dfly
|