Filip Leonarski
64002f1e29
Some checks failed
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 11m14s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 10m43s
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 11m35s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 9m20s
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 10m23s
Build Packages / Generate python client (push) Successful in 39s
Build Packages / build:rpm (rocky9_sls9) (push) Successful in 11m24s
Build Packages / Create release (push) Has been skipped
Build Packages / Build documentation (push) Successful in 1m0s
Build Packages / build:rpm (rocky8) (push) Successful in 10m35s
Build Packages / build:rpm (ubuntu2204) (push) Successful in 10m35s
Build Packages / build:rpm (rocky9) (push) Successful in 11m17s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 9m9s
Build Packages / Unit tests (push) Failing after 1h18m57s
This is an UNSTABLE release. The release has significant modifications and bug fixes, if things go wrong, it is better to revert to 1.0.0-rc.124. * jfjoch_broker: Significant improvements in TCP image socket, as a viable alternative for ZeroMQ sockets (only a single port on broker side, dynamically change number of writers, acknowledgments for written files) * jfjoch_broker: Delta phi is calculated also for still data in Bragg prediction * jfjoch_broker: Image pusher statistics are accessible via the REST interface * jfjoch_writer: Supports TCP image socket and for these auto-forking option Reviewed-on: #36 Co-authored-by: Filip Leonarski <filip.leonarski@psi.ch> Co-committed-by: Filip Leonarski <filip.leonarski@psi.ch>
59 lines
2.1 KiB
C++
59 lines
2.1 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
#ifndef JUNGFRAUJOCH_ZMQSTREAM2PUSHER_H
|
|
#define JUNGFRAUJOCH_ZMQSTREAM2PUSHER_H
|
|
|
|
#include "ImagePusher.h"
|
|
#include "../common/ZMQWrappers.h"
|
|
#include "../preview/PreviewCounter.h"
|
|
#include "ZMQWriterNotificationPuller.h"
|
|
#include "ZMQStream2PusherSocket.h"
|
|
#include "../frame_serialize/CBORStream2Serializer.h"
|
|
|
|
class ZMQStream2Pusher : public ImagePusher {
|
|
std::vector<uint8_t> serialization_buffer;
|
|
CBORStream2Serializer serializer;
|
|
|
|
std::vector<std::unique_ptr<ZMQStream2PusherSocket>> socket;
|
|
|
|
std::unique_ptr<ZMQWriterNotificationPuller> writer_notification_socket;
|
|
|
|
int64_t images_per_file = 1;
|
|
uint64_t run_number = 0;
|
|
std::string run_name;
|
|
std::atomic<bool> transmission_error = false;
|
|
|
|
mutable std::mutex images_written_mutex;
|
|
std::optional<uint64_t> images_written;
|
|
public:
|
|
explicit ZMQStream2Pusher(const std::vector<std::string>& addr,
|
|
std::optional<int32_t> send_buffer_high_watermark = {},
|
|
std::optional<int32_t> send_buffer_size = {});
|
|
|
|
ZMQStream2Pusher& WriterNotificationSocket(const std::string& addr);
|
|
std::string GetWriterNotificationSocketAddress() const override;
|
|
|
|
std::vector<std::string> GetAddress();
|
|
|
|
// Strictly serial, as order of these is important
|
|
void StartDataCollection(StartMessage& message) override;
|
|
bool EndDataCollection(const EndMessage& message) override;
|
|
bool SendCalibration(const CompressedImage& message) override;
|
|
|
|
// Thread-safe
|
|
void SendImage(ZeroCopyReturnValue &z) override;
|
|
bool SendImage(const uint8_t *image_data, size_t image_size, int64_t image_number) override;
|
|
|
|
std::string Finalize() override;
|
|
|
|
std::string PrintSetup() const override;
|
|
std::optional<uint64_t> GetImagesWritten() const override;
|
|
|
|
size_t GetConnectedWriters() const override;
|
|
|
|
ImagePusherType GetType() const override { return ImagePusherType::ZMQ; }
|
|
};
|
|
|
|
#endif //JUNGFRAUJOCH_ZMQSTREAM2PUSHER_H
|