Add REQ-REP zmq/nn sockets to FairMQ together with a little example program.

Also includes example of using boost::program_options.
2025-10-15 09:31:45 +00:00 · 2014-10-10 18:36:44 +02:00
parent bd79420f93
commit 9317f06c10
11 changed files with 519 additions and 2 deletions
--- a/fairmq/examples/req-rep/runExampleClient.cxx
+++ b/fairmq/examples/req-rep/runExampleClient.cxx
@@ -0,0 +1,142 @@
+/********************************************************************************
+ *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
+ *                                                                              *
+ *              This software is distributed under the terms of the             * 
+ *         GNU Lesser General Public Licence version 3 (LGPL) version 3,        *  
+ *                  copied verbatim in the file "LICENSE"                       *
+ ********************************************************************************/
+/**
+ * runExampleClient.cxx
+ *
+ * @since 2013-04-23
+ * @author D. Klein, A. Rybalchenko
+ */
+
+#include <iostream>
+#include <csignal>
+
+#include "boost/program_options.hpp"
+
+#include "FairMQLogger.h"
+#include "FairMQExampleClient.h"
+
+#ifdef NANOMSG
+#include "FairMQTransportFactoryNN.h"
+#else
+#include "FairMQTransportFactoryZMQ.h"
+#endif
+
+using namespace std;
+
+FairMQExampleClient client;
+
+static void s_signal_handler(int signal)
+{
+    cout << endl << "Caught signal " << signal << endl;
+
+    client.ChangeState(FairMQExampleClient::STOP);
+    client.ChangeState(FairMQExampleClient::END);
+
+    cout << "Shutdown complete. Bye!" << endl;
+    exit(1);
+}
+
+static void s_catch_signals(void)
+{
+    struct sigaction action;
+    action.sa_handler = s_signal_handler;
+    action.sa_flags = 0;
+    sigemptyset(&action.sa_mask);
+    sigaction(SIGINT, &action, NULL);
+    sigaction(SIGTERM, &action, NULL);
+}
+
+typedef struct DeviceOptions
+{
+    string text;
+} DeviceOptions_t;
+
+inline bool parse_cmd_line(int _argc, char* _argv[], DeviceOptions* _options)
+{
+    if (_options == NULL)
+        throw std::runtime_error("Internal error: options' container is empty.");
+
+    namespace bpo = boost::program_options;
+    bpo::options_description desc("Options");
+    desc.add_options()
+        ("text,t", bpo::value<string>()->default_value("something"), "Text to send to server")
+        ("help", "Print help messages");
+
+    bpo::variables_map vm;
+    bpo::store(bpo::parse_command_line(_argc, _argv, desc), vm);
+
+    if ( vm.count("help") )
+    {
+        LOG(INFO) << "EPN" << endl << desc;
+        return false;
+    }
+
+    bpo::notify(vm);
+
+    if ( vm.count("text") )
+        _options->text = vm["text"].as<string>();
+
+    return true;
+}
+
+int main(int argc, char** argv)
+{
+    s_catch_signals();
+
+    DeviceOptions_t options;
+    try
+    {
+        if (!parse_cmd_line(argc, argv, &options))
+            return 0;
+    }
+    catch (exception& e)
+    {
+        LOG(ERROR) << e.what();
+        return 1;
+    }
+
+    LOG(INFO) << "PID: " << getpid();
+
+#ifdef NANOMSG
+    FairMQTransportFactory* transportFactory = new FairMQTransportFactoryNN();
+#else
+    FairMQTransportFactory* transportFactory = new FairMQTransportFactoryZMQ();
+#endif
+
+    client.SetTransport(transportFactory);
+
+    client.SetProperty(FairMQExampleClient::Id, "client");
+    client.SetProperty(FairMQExampleClient::NumIoThreads, 1);
+    client.SetProperty(FairMQExampleClient::NumInputs, 0);
+    client.SetProperty(FairMQExampleClient::NumOutputs, 1);
+
+    client.ChangeState(FairMQExampleClient::INIT);
+
+    client.SetProperty(FairMQExampleClient::OutputSocketType, "req", 0);
+    client.SetProperty(FairMQExampleClient::OutputSndBufSize, 10000, 0);
+    client.SetProperty(FairMQExampleClient::OutputMethod, "connect", 0);
+    client.SetProperty(FairMQExampleClient::OutputAddress, "tcp://localhost:5005", 0);
+
+    client.SetProperty(FairMQExampleClient::Text, options.text);
+
+    client.ChangeState(FairMQExampleClient::SETOUTPUT);
+    client.ChangeState(FairMQExampleClient::SETINPUT);
+    client.ChangeState(FairMQExampleClient::RUN);
+
+    // wait until the running thread has finished processing.
+    boost::unique_lock<boost::mutex> lock(client.fRunningMutex);
+    while (!client.fRunningFinished)
+    {
+        client.fRunningCondition.wait(lock);
+    }
+
+    client.ChangeState(FairMQExampleClient::STOP);
+    client.ChangeState(FairMQExampleClient::END);
+
+    return 0;
+}