reactor_with_wait_until_done.hpp

// reactor_with_wait_until_done.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) ===
#include <mutex>
#include <atomic>
#include <condition_variable>
// === third-party includes (if any) ===
// === local includes (if any) ===
 
#pragma once
 
namespace presage::smartspectra::grpc_bindings {
template<typename TReactor>
class ReactorWithWaitUntilDone : public TReactor {
public:
    void WaitUntilDone();
    void OnDone() override;
    void OnCancel() override;
 
private:
    void HandleWaitingThreads();
    std::mutex done_mutex;
    std::condition_variable done_cv;
    bool done = false;
    std::atomic_int waiting_thread_count = 0;
};
 
 
 
}// namespace presage::smartspectra::grpc_bindings