tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Block messages from being delivered upon ShutDown

Browse Source
This commit is contained in:
Tyler Neely 2022-11-17 21:27:48 +00:00
parent 262df5c6a2
commit cf73ed529d
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/io/simulator/simulator_handle.cpp
View File

@ -23,6 +23,12 @@ namespace memgraph::io::simulator {
void SimulatorHandle::ShutDown() { void SimulatorHandle::ShutDown() {
std::unique_lock<std::mutex> lock(mu_); std::unique_lock<std::mutex> lock(mu_);
should_shut_down_ = true; should_shut_down_ = true;
for (auto it = promises_.begin(); it != promises_.end();) {
auto &[promise_key, dop] = *it;
std::move(dop).promise.TimeOut();
it = promises_.erase(it);
}
can_receive_.clear();
cv_.notify_all(); cv_.notify_all();
} }
@ -62,7 +68,7 @@ bool SimulatorHandle::MaybeTickSimulator() {
const size_t blocked_servers = blocked_on_receive_.size(); const size_t blocked_servers = blocked_on_receive_.size();
if (blocked_servers < server_addresses_.size()) { if (should_shut_down_ || blocked_servers < server_addresses_.size()) {
// we only need to advance the simulator when all // we only need to advance the simulator when all
// servers have reached a quiescent state, blocked // servers have reached a quiescent state, blocked
// on their own futures or receive methods. // on their own futures or receive methods.