v1.0.0_rc.11

frame_serialize/ZMQWriterNotificationPuller.cpp

@@ -0,0 +1,34 @@
+// Copyright (2019-2024) Paul Scherrer Institute
+
+#include "ZMQWriterNotificationPuller.h"
+#include "nlohmann/json.hpp"
+
+ZMQWriterNotificationPuller::ZMQWriterNotificationPuller(const std::string &addr, std::chrono::milliseconds timeout)
+        : socket(ZMQSocketType::Pull) {
+    socket.ReceiveTimeout(timeout);
+    socket.Bind(addr);
+}
+
+std::optional<ZMQWriterNotificationOutput> ZMQWriterNotificationPuller::Receive(uint64_t run_number, const std::string &run_name) {
+    ZMQMessage msg;
+    // Loop to ensure that messages with wrong run_number or run_name are filtered
+    while (socket.Receive(msg)) {
+        try {
+            nlohmann::json j = nlohmann::json::parse(std::string((char *) msg.data(), msg.size()));
+            if ((j["run_number"] == run_number) && (j["run_name"] == run_name)) {
+                ZMQWriterNotificationOutput ret{};
+                ret.ok = j["ok"];
+                ret.processed_images = j["processed_images"];
+                ret.socket_number = j["socket_number"];
+                return ret;
+            }
+        } catch (const std::exception &e) {
+            // Cannot parse properly
+        }
+    }
+    return {};
+}
+
+std::string ZMQWriterNotificationPuller::GetEndpointName() {
+    return socket.GetEndpointName();
+}