reactor_with_wait_until_done_impl.hpp

// reactor_with_wait_until_done_impl.hpp
// Created by greg on 2/20/25.
// Copyright (C) 2025 Presage Security, Inc.
//
// This program is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 3 of the License, or (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program; if not, write to the Free Software Foundation,
// Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 
// === standard library includes (if any) ===
// === third-party includes (if any) ===
#include <glog/logging.h>
// === local includes (if any) ===
#include "reactor_with_wait_until_done.hpp"
 
#pragma once
 
namespace presage::smartspectra::grpc_bindings {
 
 
template<typename TReactor>
void ReactorWithWaitUntilDone<TReactor>::HandleWaitingThreads() {
    {
        std::unique_lock<std::mutex> lock(done_mutex);
        this->done = true;
        done_cv.notify_all();
    }
    if (this->waiting_thread_count > 0) {
        std::unique_lock<std::mutex> lock(done_mutex);
        done_cv.wait(lock, [this] { return waiting_thread_count == 0; });
    }
}
 
template<typename TReactor>
void ReactorWithWaitUntilDone<TReactor>::WaitUntilDone() {
    std::unique_lock<std::mutex> lock(done_mutex);
    this->waiting_thread_count++;
    done_cv.wait(lock, [this] { return this->done; });
    this->waiting_thread_count--;
    done_cv.notify_one();
}
 
template<typename TReactor>
void ReactorWithWaitUntilDone<TReactor>::OnDone() {
    HandleWaitingThreads();
}
 
template<typename TReactor>
void ReactorWithWaitUntilDone<TReactor>::OnCancel() {
    HandleWaitingThreads();
    LOG(ERROR) << "RPC Cancelled";
}
 
 
} // namespace presage::smartspectra::grpc_bindings