queued_server_write_reactor.hpp

// queued_server_write_reactor.hpp
// Created by Cascade AI / GPT4.1, 2025
// Copyright (C) 2025 Presage Security, Inc.
//
// SPDX-License-Identifier: LGPL-3.0-or-later
#pragma once
 
#include <mutex>
#include <queue>
#include <grpc/grpc.h>
#include <grpcpp/grpcpp.h>
#include "reactor_with_wait_until_done.hpp"
 
namespace presage::smartspectra::grpc_bindings {
 
template <typename TMessage>
class QueuedServerWriteReactor : public ReactorWithWaitUntilDone<grpc::ServerWriteReactor<TMessage>> {
public:
    using Base = ReactorWithWaitUntilDone<grpc::ServerWriteReactor<TMessage>>;
    QueuedServerWriteReactor();
    virtual ~QueuedServerWriteReactor();
protected:
    void EnqueueWrite(const TMessage& message);
    void OnWriteDone(bool ok);
private:
    std::mutex write_mutex_;
    std::queue<std::shared_ptr<TMessage>> pending_writes;
    bool write_in_progress = false;
    void ProcessWriteQueue();
};
 
} // namespace presage::smartspectra::grpc_bindings