Jungfraujoch/image_pusher/ZMQStream2PusherSocket.h

// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#ifndef JFJOCH_ZMQSTREAM2PUSHERSOCKET_H
#define JFJOCH_ZMQSTREAM2PUSHERSOCKET_H

#include <mutex>
#include <future>
#include <string>
#include <optional>

#include "ImagePusher.h"
#include "../common/ZMQWrappers.h"
#include "../common/ThreadSafeFIFO.h"
#include "../common/ZeroCopyReturnValue.h"

class ZMQStream2PusherSocket {
    std::mutex m;
    std::atomic<bool> active = false;
    std::future<void> send_future;

    ThreadSafeFIFO<ImagePusherQueueElement> queue;
    ZMQSocket s;
    void WriterThread();
public:
    explicit ZMQStream2PusherSocket(const std::string& addr,
                              std::optional<int32_t> send_buffer_high_watermark,
                              std::optional<int32_t> send_buffer_size);

    std::string GetEndpointName();

    // Strictly serial, as order of these is important
    bool Send(const uint8_t *data, size_t size);

    void StartWriterThread();
    void StopWriterThread();

    // Thread-safe
    void SendImage(ZeroCopyReturnValue &z);
};

#endif //JFJOCH_ZMQSTREAM2PUSHERSOCKET_H