leonarski_f
f3e0a15d26
All checks were successful
Build Packages / build:rpm (rocky8_nocuda) (push) Successful in 10m51s
Build Packages / build:rpm (ubuntu2404_nocuda) (push) Successful in 8m0s
Build Packages / build:rpm (ubuntu2204_nocuda) (push) Successful in 9m6s
Build Packages / build:rpm (rocky9_nocuda) (push) Successful in 10m7s
Build Packages / build:rpm (rocky8_sls9) (push) Successful in 9m47s
Build Packages / Generate python client (push) Successful in 29s
Build Packages / Build documentation (push) Successful in 43s
Build Packages / Create release (push) Has been skipped
Build Packages / build:rpm (rocky9_sls9) (push) Successful in 10m46s
Build Packages / build:rpm (rocky8) (push) Successful in 9m33s
Build Packages / Unit tests (push) Has been skipped
Build Packages / build:rpm (ubuntu2204) (push) Successful in 8m47s
Build Packages / build:rpm (rocky9) (push) Successful in 9m55s
Build Packages / build:rpm (ubuntu2404) (push) Successful in 9m4s
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: Default EIGER readout time is 20 microseconds * jfjoch_broker: Multiple improvements regarding performance * jfjoch_broker: Image buffer allows to track frames in preparation and sending * jfjoch_broker: Dedicated thread for ZeroMQ transmission to better utilize the image buffer * jfjoch_broker: Experimental implementation of transmission with raw TCP/IP sockets * jfjoch_writer: Fixes regarding properly closing files in long data collections * jfjoch_process: Scale & merge has been significantly improved, but it is not yet integrated into mainstream code Reviewed-on: #34
85 lines
2.5 KiB
C++
85 lines
2.5 KiB
C++
// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
|
|
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
|
|
#ifndef JUNGFRAUJOCH_ZMQWRAPPERS_H
|
|
#define JUNGFRAUJOCH_ZMQWRAPPERS_H
|
|
|
|
#include <vector>
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <zmq.h>
|
|
|
|
#include "JFJochException.h"
|
|
|
|
class ZMQContext {
|
|
void *context;
|
|
public:
|
|
ZMQContext();
|
|
ZMQContext& NumThreads(int32_t threads);
|
|
~ZMQContext();
|
|
void *GetContext() const;
|
|
};
|
|
|
|
enum class ZMQSocketType : int {Push = ZMQ_PUSH, Pull = ZMQ_PULL, Req = ZMQ_REQ, Rep = ZMQ_REP,
|
|
Pub = ZMQ_PUB, Sub = ZMQ_SUB};
|
|
|
|
class ZMQMessage {
|
|
zmq_msg_t msg;
|
|
public:
|
|
ZMQMessage();
|
|
~ZMQMessage();
|
|
ZMQMessage(ZMQMessage &other) = delete;
|
|
ZMQMessage& operator=(ZMQMessage &other) = delete;
|
|
zmq_msg_t *GetMsg();
|
|
size_t size();
|
|
const uint8_t *data();
|
|
};
|
|
|
|
class ZMQSocket {
|
|
std::mutex m;
|
|
ZMQContext context;
|
|
ZMQSocketType socket_type;
|
|
void *socket;
|
|
void SetSocketOption(int32_t option_name, int32_t value);
|
|
public:
|
|
ZMQSocket(ZMQSocket &socket) = delete;
|
|
const ZMQSocket& operator=(ZMQSocket &socket) = delete;
|
|
explicit ZMQSocket(ZMQSocketType socket_type);
|
|
~ZMQSocket();
|
|
void Connect(const std::string& addr);
|
|
void Disconnect(const std::string& addr);
|
|
void Bind(const std::string& addr);
|
|
ZMQSocket &NoReceiveTimeout();
|
|
ZMQSocket &ReceiveTimeout(std::chrono::milliseconds input);
|
|
|
|
ZMQSocket &NoSendTimeout();
|
|
ZMQSocket &SendTimeout(std::chrono::milliseconds input);
|
|
|
|
ZMQSocket &Subscribe(const std::string &topic);
|
|
ZMQSocket &SubscribeAll();
|
|
ZMQSocket &NoLinger();
|
|
ZMQSocket &Conflate(bool input);
|
|
ZMQSocket &SendBufferSize(int32_t bytes);
|
|
ZMQSocket &ReceiverBufferSize(int32_t bytes);
|
|
|
|
bool Receive(ZMQMessage& msg, bool blocking = true);
|
|
|
|
void Send();
|
|
bool Send(const void *buf, size_t buf_size, bool blocking = true, bool multipart = false);
|
|
bool SendZeroCopy(void *data, size_t size,void (*callback)(void *, void *), void *hint, bool blocking = true);
|
|
template <class T> bool Send(const std::vector<T> &buf) {
|
|
return Send(buf.data(), buf.size() * sizeof(T));
|
|
}
|
|
void Send(const int32_t &value);
|
|
bool Send(const std::string &s, bool blocking = true, bool multipart = false);
|
|
void Send(zmq_msg_t *msg);
|
|
ZMQSocket &SendWaterMark(int32_t msgs);
|
|
ZMQSocket &ReceiveWaterMark(int32_t msgs);
|
|
ZMQSocket &EnableKeepAlive();
|
|
std::string GetEndpointName();
|
|
};
|
|
|
|
|
|
#endif //JUNGFRAUJOCH_ZMQWRAPPERS_H
|