mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +00:00
147 lines
4.0 KiB
C++
147 lines
4.0 KiB
C++
/********************************************************************************
|
|
* 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
|
|
{
|
|
DeviceOptions() :
|
|
text() {}
|
|
|
|
string text;
|
|
} DeviceOptions_t;
|
|
|
|
inline bool parse_cmd_line(int _argc, char* _argv[], DeviceOptions* _options)
|
|
{
|
|
if (_options == NULL)
|
|
throw 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::OutputRcvBufSize, 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;
|
|
}
|