builtin devices: Reorganize

2025-10-16 01:51:45 +00:00 · 2021-03-24 12:53:20 +01:00
parent 66a4df0667
commit fc49687879
15 changed files with 132 additions and 125 deletions
--- a/fairmq/devices/Proxy.h
+++ b/fairmq/devices/Proxy.h
@@ -0,0 +1,74 @@
+/********************************************************************************
+ * Copyright (C) 2014-2021 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH  *
+ *                                                                              *
+ *              This software is distributed under the terms of the             *
+ *              GNU Lesser General Public Licence (LGPL) version 3,             *
+ *                  copied verbatim in the file "LICENSE"                       *
+ ********************************************************************************/
+
+#ifndef FAIR_MQ_PROXY_H
+#define FAIR_MQ_PROXY_H
+
+#include <fairmq/Device.h>
+
+#include <string>
+
+namespace fair::mq
+{
+
+class Proxy : public Device
+{
+  public:
+    Proxy()
+        : fMultipart(true)
+        , fInChannelName()
+        , fOutChannelName()
+    {}
+
+  protected:
+    bool fMultipart;
+    std::string fInChannelName;
+    std::string fOutChannelName;
+
+    void InitTask() override
+    {
+        fMultipart = fConfig->GetProperty<bool>("multipart");
+        fInChannelName = fConfig->GetProperty<std::string>("in-channel");
+        fOutChannelName = fConfig->GetProperty<std::string>("out-channel");
+    }
+
+    void Run() override
+    {
+        if (fMultipart) {
+            while (!NewStatePending()) {
+                FairMQParts payload;
+                if (Receive(payload, fInChannelName) >= 0) {
+                    if (Send(payload, fOutChannelName) < 0) {
+                        LOG(debug) << "Transfer interrupted";
+                        break;
+                    }
+                } else {
+                    LOG(debug) << "Transfer interrupted";
+                    break;
+                }
+            }
+        } else {
+            while (!NewStatePending()) {
+                FairMQMessagePtr payload(fTransportFactory->CreateMessage());
+                if (Receive(payload, fInChannelName) >= 0) {
+                    if (Send(payload, fOutChannelName) < 0) {
+                        LOG(debug) << "Transfer interrupted";
+                        break;
+                    }
+                } else {
+                    LOG(debug) << "Transfer interrupted";
+                    break;
+                }
+            }
+        }
+    }
+};
+
+} // namespace fair::mq
+
+#endif /* FAIR_MQ_PROXY_H */