mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-12 16:21:13 +00:00
5aaf27bf02
- give transport Initialize() method with access to device config. - avoid using global context in the transport. - simplify shutdown procedure (no need for extra thread).
133 lines
4.0 KiB
C++
133 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" *
|
|
********************************************************************************/
|
|
/**
|
|
* FairMQTransportFactoryZMQ.cxx
|
|
*
|
|
* @since 2014-01-20
|
|
* @author: A. Rybalchenko
|
|
*/
|
|
|
|
#include "zmq.h"
|
|
|
|
#include "FairMQTransportFactoryZMQ.h"
|
|
#include "../options/FairMQProgOptions.h"
|
|
|
|
using namespace std;
|
|
|
|
FairMQ::Transport FairMQTransportFactoryZMQ::fTransportType = FairMQ::Transport::ZMQ;
|
|
|
|
FairMQTransportFactoryZMQ::FairMQTransportFactoryZMQ()
|
|
: fContext(zmq_ctx_new())
|
|
{
|
|
int major, minor, patch;
|
|
zmq_version(&major, &minor, &patch);
|
|
LOG(DEBUG) << "Transport: Using ZeroMQ library, version: " << major << "." << minor << "." << patch;
|
|
|
|
if (!fContext)
|
|
{
|
|
LOG(ERROR) << "failed creating context, reason: " << zmq_strerror(errno);
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
}
|
|
|
|
void FairMQTransportFactoryZMQ::Initialize(const FairMQProgOptions* config)
|
|
{
|
|
int numIoThreads = 1;
|
|
if (config)
|
|
{
|
|
numIoThreads = config->GetValue<int>("io-threads");
|
|
}
|
|
else
|
|
{
|
|
LOG(WARN) << "zeromq: FairMQProgOptions not available! Using defaults.";
|
|
}
|
|
|
|
if (zmq_ctx_set(fContext, ZMQ_IO_THREADS, numIoThreads) != 0)
|
|
{
|
|
LOG(ERROR) << "failed configuring context, reason: " << zmq_strerror(errno);
|
|
}
|
|
|
|
// Set the maximum number of allowed sockets on the context.
|
|
if (zmq_ctx_set(fContext, ZMQ_MAX_SOCKETS, 10000) != 0)
|
|
{
|
|
LOG(ERROR) << "failed configuring context, reason: " << zmq_strerror(errno);
|
|
}
|
|
}
|
|
|
|
FairMQMessagePtr FairMQTransportFactoryZMQ::CreateMessage() const
|
|
{
|
|
return unique_ptr<FairMQMessage>(new FairMQMessageZMQ());
|
|
}
|
|
|
|
FairMQMessagePtr FairMQTransportFactoryZMQ::CreateMessage(const size_t size) const
|
|
{
|
|
return unique_ptr<FairMQMessage>(new FairMQMessageZMQ(size));
|
|
}
|
|
|
|
FairMQMessagePtr FairMQTransportFactoryZMQ::CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint) const
|
|
{
|
|
return unique_ptr<FairMQMessage>(new FairMQMessageZMQ(data, size, ffn, hint));
|
|
}
|
|
|
|
FairMQSocketPtr FairMQTransportFactoryZMQ::CreateSocket(const string& type, const string& name, const string& id /*= ""*/) const
|
|
{
|
|
assert(fContext);
|
|
return unique_ptr<FairMQSocket>(new FairMQSocketZMQ(type, name, id, fContext));
|
|
}
|
|
|
|
FairMQPollerPtr FairMQTransportFactoryZMQ::CreatePoller(const vector<FairMQChannel>& channels) const
|
|
{
|
|
return unique_ptr<FairMQPoller>(new FairMQPollerZMQ(channels));
|
|
}
|
|
|
|
FairMQPollerPtr FairMQTransportFactoryZMQ::CreatePoller(const unordered_map<string, vector<FairMQChannel>>& channelsMap, const vector<string>& channelList) const
|
|
{
|
|
return unique_ptr<FairMQPoller>(new FairMQPollerZMQ(channelsMap, channelList));
|
|
}
|
|
|
|
FairMQPollerPtr FairMQTransportFactoryZMQ::CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const
|
|
{
|
|
return unique_ptr<FairMQPoller>(new FairMQPollerZMQ(cmdSocket, dataSocket));
|
|
}
|
|
|
|
FairMQ::Transport FairMQTransportFactoryZMQ::GetType() const
|
|
{
|
|
return fTransportType;
|
|
}
|
|
|
|
void FairMQTransportFactoryZMQ::Shutdown()
|
|
{
|
|
if (zmq_ctx_shutdown(fContext) != 0)
|
|
{
|
|
LOG(ERROR) << "zeromq: failed shutting down context, reason: " << zmq_strerror(errno);
|
|
}
|
|
}
|
|
|
|
void FairMQTransportFactoryZMQ::Terminate()
|
|
{
|
|
if (fContext)
|
|
{
|
|
if (zmq_ctx_term(fContext) != 0)
|
|
{
|
|
if (errno == EINTR)
|
|
{
|
|
LOG(ERROR) << " failed closing context, reason: " << zmq_strerror(errno);
|
|
}
|
|
else
|
|
{
|
|
fContext = nullptr;
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
LOG(ERROR) << "shmem: Terminate(): context now available for shutdown";
|
|
}
|
|
}
|