mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-12 16:21:13 +00:00
FairMQRegion and examples/advanced/Region
This commit is contained in:
parent
65f1b96dc3
commit
221d2567f0
|
@ -62,6 +62,7 @@ set(FAIRMQ_HEADER_FILES
|
||||||
FairMQMessage.h
|
FairMQMessage.h
|
||||||
FairMQParts.h
|
FairMQParts.h
|
||||||
FairMQPoller.h
|
FairMQPoller.h
|
||||||
|
FairMQRegion.h
|
||||||
FairMQSocket.h
|
FairMQSocket.h
|
||||||
FairMQStateMachine.h
|
FairMQStateMachine.h
|
||||||
FairMQTransportFactory.h
|
FairMQTransportFactory.h
|
||||||
|
@ -87,10 +88,11 @@ set(FAIRMQ_HEADER_FILES
|
||||||
runFairMQDevice.h
|
runFairMQDevice.h
|
||||||
shmem/FairMQMessageSHM.h
|
shmem/FairMQMessageSHM.h
|
||||||
shmem/FairMQPollerSHM.h
|
shmem/FairMQPollerSHM.h
|
||||||
|
shmem/FairMQRegionSHM.h
|
||||||
shmem/FairMQSocketSHM.h
|
shmem/FairMQSocketSHM.h
|
||||||
shmem/FairMQTransportFactorySHM.h
|
shmem/FairMQTransportFactorySHM.h
|
||||||
shmem/FairMQShmMonitor.h
|
shmem/FairMQShmMonitor.h
|
||||||
shmem/FairMQShmDeviceCounter.h
|
shmem/FairMQShmCommon.h
|
||||||
tools/CppSTL.h
|
tools/CppSTL.h
|
||||||
tools/Network.h
|
tools/Network.h
|
||||||
tools/runSimpleMQStateMachine.h
|
tools/runSimpleMQStateMachine.h
|
||||||
|
@ -98,6 +100,7 @@ set(FAIRMQ_HEADER_FILES
|
||||||
tools/Version.h
|
tools/Version.h
|
||||||
zeromq/FairMQMessageZMQ.h
|
zeromq/FairMQMessageZMQ.h
|
||||||
zeromq/FairMQPollerZMQ.h
|
zeromq/FairMQPollerZMQ.h
|
||||||
|
zeromq/FairMQRegionZMQ.h
|
||||||
zeromq/FairMQSocketZMQ.h
|
zeromq/FairMQSocketZMQ.h
|
||||||
zeromq/FairMQTransportFactoryZMQ.h
|
zeromq/FairMQTransportFactoryZMQ.h
|
||||||
)
|
)
|
||||||
|
@ -106,6 +109,7 @@ if(NANOMSG_FOUND)
|
||||||
set(FAIRMQ_HEADER_FILES ${FAIRMQ_HEADER_FILES}
|
set(FAIRMQ_HEADER_FILES ${FAIRMQ_HEADER_FILES}
|
||||||
nanomsg/FairMQMessageNN.h
|
nanomsg/FairMQMessageNN.h
|
||||||
nanomsg/FairMQPollerNN.h
|
nanomsg/FairMQPollerNN.h
|
||||||
|
nanomsg/FairMQRegionNN.h
|
||||||
nanomsg/FairMQSocketNN.h
|
nanomsg/FairMQSocketNN.h
|
||||||
nanomsg/FairMQTransportFactoryNN.h
|
nanomsg/FairMQTransportFactoryNN.h
|
||||||
)
|
)
|
||||||
|
@ -144,11 +148,13 @@ set(FAIRMQ_SOURCE_FILES
|
||||||
PluginServices.cxx
|
PluginServices.cxx
|
||||||
shmem/FairMQMessageSHM.cxx
|
shmem/FairMQMessageSHM.cxx
|
||||||
shmem/FairMQPollerSHM.cxx
|
shmem/FairMQPollerSHM.cxx
|
||||||
|
shmem/FairMQRegionSHM.cxx
|
||||||
shmem/FairMQSocketSHM.cxx
|
shmem/FairMQSocketSHM.cxx
|
||||||
shmem/FairMQTransportFactorySHM.cxx
|
shmem/FairMQTransportFactorySHM.cxx
|
||||||
shmem/FairMQShmMonitor.cxx
|
shmem/FairMQShmMonitor.cxx
|
||||||
zeromq/FairMQMessageZMQ.cxx
|
zeromq/FairMQMessageZMQ.cxx
|
||||||
zeromq/FairMQPollerZMQ.cxx
|
zeromq/FairMQPollerZMQ.cxx
|
||||||
|
zeromq/FairMQRegionZMQ.cxx
|
||||||
zeromq/FairMQSocketZMQ.cxx
|
zeromq/FairMQSocketZMQ.cxx
|
||||||
zeromq/FairMQTransportFactoryZMQ.cxx
|
zeromq/FairMQTransportFactoryZMQ.cxx
|
||||||
)
|
)
|
||||||
|
@ -157,6 +163,7 @@ if(NANOMSG_FOUND)
|
||||||
set(FAIRMQ_SOURCE_FILES ${FAIRMQ_SOURCE_FILES}
|
set(FAIRMQ_SOURCE_FILES ${FAIRMQ_SOURCE_FILES}
|
||||||
nanomsg/FairMQMessageNN.cxx
|
nanomsg/FairMQMessageNN.cxx
|
||||||
nanomsg/FairMQPollerNN.cxx
|
nanomsg/FairMQPollerNN.cxx
|
||||||
|
nanomsg/FairMQRegionNN.cxx
|
||||||
nanomsg/FairMQSocketNN.cxx
|
nanomsg/FairMQSocketNN.cxx
|
||||||
nanomsg/FairMQTransportFactoryNN.cxx
|
nanomsg/FairMQTransportFactoryNN.cxx
|
||||||
)
|
)
|
||||||
|
|
|
@ -47,12 +47,12 @@ static void CallSignalHandler(int signal)
|
||||||
}
|
}
|
||||||
|
|
||||||
FairMQDevice::FairMQDevice()
|
FairMQDevice::FairMQDevice()
|
||||||
: fChannels()
|
: fTransportFactory(nullptr)
|
||||||
|
, fTransports()
|
||||||
|
, fChannels()
|
||||||
, fConfig(nullptr)
|
, fConfig(nullptr)
|
||||||
, fId()
|
, fId()
|
||||||
, fNumIoThreads(1)
|
, fNumIoThreads(1)
|
||||||
, fTransportFactory(nullptr)
|
|
||||||
, fTransports()
|
|
||||||
, fInitialValidationFinished(false)
|
, fInitialValidationFinished(false)
|
||||||
, fInitialValidationCondition()
|
, fInitialValidationCondition()
|
||||||
, fInitialValidationMutex()
|
, fInitialValidationMutex()
|
||||||
|
@ -69,6 +69,7 @@ FairMQDevice::FairMQDevice()
|
||||||
, fMsgInputs()
|
, fMsgInputs()
|
||||||
, fMultipartInputs()
|
, fMultipartInputs()
|
||||||
, fMultitransportInputs()
|
, fMultitransportInputs()
|
||||||
|
, fChannelRegistry()
|
||||||
, fInputChannelKeys()
|
, fInputChannelKeys()
|
||||||
, fMultitransportMutex()
|
, fMultitransportMutex()
|
||||||
, fMultitransportProceed(false)
|
, fMultitransportProceed(false)
|
||||||
|
@ -78,12 +79,12 @@ FairMQDevice::FairMQDevice()
|
||||||
}
|
}
|
||||||
|
|
||||||
FairMQDevice::FairMQDevice(const fair::mq::tools::Version version)
|
FairMQDevice::FairMQDevice(const fair::mq::tools::Version version)
|
||||||
: fChannels()
|
: fTransportFactory(nullptr)
|
||||||
|
, fTransports()
|
||||||
|
, fChannels()
|
||||||
, fConfig(nullptr)
|
, fConfig(nullptr)
|
||||||
, fId()
|
, fId()
|
||||||
, fNumIoThreads(1)
|
, fNumIoThreads(1)
|
||||||
, fTransportFactory(nullptr)
|
|
||||||
, fTransports()
|
|
||||||
, fInitialValidationFinished(false)
|
, fInitialValidationFinished(false)
|
||||||
, fInitialValidationCondition()
|
, fInitialValidationCondition()
|
||||||
, fInitialValidationMutex()
|
, fInitialValidationMutex()
|
||||||
|
@ -100,6 +101,7 @@ FairMQDevice::FairMQDevice(const fair::mq::tools::Version version)
|
||||||
, fMsgInputs()
|
, fMsgInputs()
|
||||||
, fMultipartInputs()
|
, fMultipartInputs()
|
||||||
, fMultitransportInputs()
|
, fMultitransportInputs()
|
||||||
|
, fChannelRegistry()
|
||||||
, fInputChannelKeys()
|
, fInputChannelKeys()
|
||||||
, fMultitransportMutex()
|
, fMultitransportMutex()
|
||||||
, fMultitransportProceed(false)
|
, fMultitransportProceed(false)
|
||||||
|
@ -1293,8 +1295,6 @@ const FairMQChannel& FairMQDevice::GetChannel(const std::string& channelName, co
|
||||||
|
|
||||||
void FairMQDevice::Exit()
|
void FairMQDevice::Exit()
|
||||||
{
|
{
|
||||||
LOG(DEBUG) << "All transports are shut down.";
|
|
||||||
|
|
||||||
if (!fExternalConfig && fConfig)
|
if (!fExternalConfig && fConfig)
|
||||||
{
|
{
|
||||||
delete fConfig;
|
delete fConfig;
|
||||||
|
@ -1303,5 +1303,5 @@ void FairMQDevice::Exit()
|
||||||
|
|
||||||
FairMQDevice::~FairMQDevice()
|
FairMQDevice::~FairMQDevice()
|
||||||
{
|
{
|
||||||
LOG(DEBUG) << "Device destroyed";
|
LOG(DEBUG) << "Destructing device " << fId;
|
||||||
}
|
}
|
||||||
|
|
|
@ -18,6 +18,7 @@
|
||||||
#include "FairMQChannel.h"
|
#include "FairMQChannel.h"
|
||||||
#include "FairMQMessage.h"
|
#include "FairMQMessage.h"
|
||||||
#include "FairMQParts.h"
|
#include "FairMQParts.h"
|
||||||
|
#include "FairMQRegion.h"
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <memory> // unique_ptr
|
#include <memory> // unique_ptr
|
||||||
|
@ -242,6 +243,16 @@ class FairMQDevice : public FairMQStateMachine, public FairMQConfigurable
|
||||||
return fChannels.at(channel).at(index).NewSimpleMessage(data);
|
return fChannels.at(channel).at(index).NewSimpleMessage(data);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQRegionPtr NewRegion(const size_t size)
|
||||||
|
{
|
||||||
|
return Transport()->CreateRegion(size);
|
||||||
|
}
|
||||||
|
|
||||||
|
FairMQRegionPtr NewRegionFor(const std::string& channel, int index, const size_t size)
|
||||||
|
{
|
||||||
|
return fChannels.at(channel).at(index).Transport()->CreateRegion(size);
|
||||||
|
}
|
||||||
|
|
||||||
template<typename ...Ts>
|
template<typename ...Ts>
|
||||||
FairMQPollerPtr NewPoller(const Ts&... inputs)
|
FairMQPollerPtr NewPoller(const Ts&... inputs)
|
||||||
{
|
{
|
||||||
|
@ -272,7 +283,7 @@ class FairMQDevice : public FairMQStateMachine, public FairMQConfigurable
|
||||||
{
|
{
|
||||||
FairMQ::Transport type = channels.at(0)->Transport()->GetType();
|
FairMQ::Transport type = channels.at(0)->Transport()->GetType();
|
||||||
|
|
||||||
for (int i = 1; i < channels.size(); ++i)
|
for (unsigned int i = 1; i < channels.size(); ++i)
|
||||||
{
|
{
|
||||||
if (type != channels.at(i)->Transport()->GetType())
|
if (type != channels.at(i)->Transport()->GetType())
|
||||||
{
|
{
|
||||||
|
|
|
@ -1,33 +1,26 @@
|
||||||
/********************************************************************************
|
/********************************************************************************
|
||||||
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
||||||
* *
|
* *
|
||||||
* This software is distributed under the terms of the *
|
* This software is distributed under the terms of the *
|
||||||
* GNU Lesser General Public Licence version 3 (LGPL) version 3, *
|
* GNU Lesser General Public Licence version 3 (LGPL) version 3, *
|
||||||
* copied verbatim in the file "LICENSE" *
|
* copied verbatim in the file "LICENSE" *
|
||||||
********************************************************************************/
|
********************************************************************************/
|
||||||
#ifndef FAIRMQSHMDEVICECOUNTER_H_
|
|
||||||
#define FAIRMQSHMDEVICECOUNTER_H_
|
|
||||||
|
|
||||||
#include <atomic>
|
#ifndef FAIRMQREGION_H_
|
||||||
|
#define FAIRMQREGION_H_
|
||||||
|
|
||||||
namespace fair
|
#include <cstddef> // size_t
|
||||||
{
|
#include <memory> // unique_ptr
|
||||||
namespace mq
|
|
||||||
{
|
|
||||||
namespace shmem
|
|
||||||
{
|
|
||||||
|
|
||||||
struct DeviceCounter
|
class FairMQRegion
|
||||||
{
|
{
|
||||||
DeviceCounter(unsigned int c)
|
public:
|
||||||
: count(c)
|
virtual void* GetData() const = 0;
|
||||||
{}
|
virtual size_t GetSize() const = 0;
|
||||||
|
|
||||||
std::atomic<unsigned int> count;
|
virtual ~FairMQRegion() {};
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace shmem
|
using FairMQRegionPtr = std::unique_ptr<FairMQRegion>;
|
||||||
} // namespace mq
|
|
||||||
} // namespace fair
|
|
||||||
|
|
||||||
#endif /* FAIRMQSHMDEVICECOUNTER_H_ */
|
#endif /* FAIRMQREGION_H_ */
|
|
@ -12,6 +12,7 @@
|
||||||
#include "FairMQMessage.h"
|
#include "FairMQMessage.h"
|
||||||
#include "FairMQSocket.h"
|
#include "FairMQSocket.h"
|
||||||
#include "FairMQPoller.h"
|
#include "FairMQPoller.h"
|
||||||
|
#include "FairMQRegion.h"
|
||||||
#include "FairMQLogger.h"
|
#include "FairMQLogger.h"
|
||||||
#include "FairMQTransports.h"
|
#include "FairMQTransports.h"
|
||||||
#include <string>
|
#include <string>
|
||||||
|
@ -50,6 +51,8 @@ class FairMQTransportFactory
|
||||||
/// @return pointer to FairMQMessage
|
/// @return pointer to FairMQMessage
|
||||||
virtual FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const = 0;
|
virtual FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const = 0;
|
||||||
|
|
||||||
|
virtual FairMQMessagePtr CreateMessage(FairMQRegionPtr& region, void* data, const size_t size) const = 0;
|
||||||
|
|
||||||
/// Create a socket
|
/// Create a socket
|
||||||
virtual FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const = 0;
|
virtual FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const = 0;
|
||||||
|
|
||||||
|
@ -62,6 +65,8 @@ class FairMQTransportFactory
|
||||||
/// Create a poller for two sockets
|
/// Create a poller for two sockets
|
||||||
virtual FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const = 0;
|
virtual FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const = 0;
|
||||||
|
|
||||||
|
virtual FairMQRegionPtr CreateRegion(const size_t size) const = 0;
|
||||||
|
|
||||||
/// Get transport type
|
/// Get transport type
|
||||||
virtual FairMQ::Transport GetType() const = 0;
|
virtual FairMQ::Transport GetType() const = 0;
|
||||||
|
|
||||||
|
|
|
@ -31,4 +31,11 @@ Possible further implementation would be to run the monitor with `--self-destruc
|
||||||
|
|
||||||
The FairMQShmMonitor class can also be used independently from the supplied executable (built from `runFairMQShmMonitor.cxx`), allowing integration on any level. For example invoking the monitor could be a functionality that a device offers.
|
The FairMQShmMonitor class can also be used independently from the supplied executable (built from `runFairMQShmMonitor.cxx`), allowing integration on any level. For example invoking the monitor could be a functionality that a device offers.
|
||||||
|
|
||||||
|
FairMQ Shared Memory currently uses following names to register shared memory on the system:
|
||||||
|
|
||||||
|
`fairmq_shmem_main` - main segment name, used for user data (this name can be overridden via `--shm-segment-name`).
|
||||||
|
`fairmq_shmem_management` - management segment name, used for storing management data.
|
||||||
|
`fairmq_shmem_control_queue` - message queue for communicating between shm transport and shm monitor (exists independent of above segments).
|
||||||
|
`fairmq_shmem_mutex` - boost::interprocess::named_mutex for management purposes (exists independent of above segments).
|
||||||
|
|
||||||
← [Back](../README.md)
|
← [Back](../README.md)
|
||||||
|
|
|
@ -29,6 +29,7 @@ FairMQMessageNN::FairMQMessageNN()
|
||||||
: fMessage(nullptr)
|
: fMessage(nullptr)
|
||||||
, fSize(0)
|
, fSize(0)
|
||||||
, fReceiving(false)
|
, fReceiving(false)
|
||||||
|
, fRegion(false)
|
||||||
{
|
{
|
||||||
fMessage = nn_allocmsg(0, 0);
|
fMessage = nn_allocmsg(0, 0);
|
||||||
if (!fMessage)
|
if (!fMessage)
|
||||||
|
@ -41,6 +42,7 @@ FairMQMessageNN::FairMQMessageNN(const size_t size)
|
||||||
: fMessage(nullptr)
|
: fMessage(nullptr)
|
||||||
, fSize(0)
|
, fSize(0)
|
||||||
, fReceiving(false)
|
, fReceiving(false)
|
||||||
|
, fRegion(false)
|
||||||
{
|
{
|
||||||
fMessage = nn_allocmsg(size, 0);
|
fMessage = nn_allocmsg(size, 0);
|
||||||
if (!fMessage)
|
if (!fMessage)
|
||||||
|
@ -60,6 +62,7 @@ FairMQMessageNN::FairMQMessageNN(void* data, const size_t size, fairmq_free_fn*
|
||||||
: fMessage(nullptr)
|
: fMessage(nullptr)
|
||||||
, fSize(0)
|
, fSize(0)
|
||||||
, fReceiving(false)
|
, fReceiving(false)
|
||||||
|
, fRegion(false)
|
||||||
{
|
{
|
||||||
fMessage = nn_allocmsg(size, 0);
|
fMessage = nn_allocmsg(size, 0);
|
||||||
if (!fMessage)
|
if (!fMessage)
|
||||||
|
@ -81,6 +84,15 @@ FairMQMessageNN::FairMQMessageNN(void* data, const size_t size, fairmq_free_fn*
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQMessageNN::FairMQMessageNN(FairMQRegionPtr& region, void* data, const size_t size)
|
||||||
|
: fMessage(data)
|
||||||
|
, fSize(size)
|
||||||
|
, fReceiving(false)
|
||||||
|
, fRegion(true)
|
||||||
|
{
|
||||||
|
// currently nanomsg will copy the buffer (data) inside nn_sendmsg()
|
||||||
|
}
|
||||||
|
|
||||||
void FairMQMessageNN::Rebuild()
|
void FairMQMessageNN::Rebuild()
|
||||||
{
|
{
|
||||||
Clear();
|
Clear();
|
||||||
|
|
|
@ -19,6 +19,7 @@
|
||||||
#include <string>
|
#include <string>
|
||||||
|
|
||||||
#include "FairMQMessage.h"
|
#include "FairMQMessage.h"
|
||||||
|
#include "FairMQRegion.h"
|
||||||
|
|
||||||
class FairMQMessageNN : public FairMQMessage
|
class FairMQMessageNN : public FairMQMessage
|
||||||
{
|
{
|
||||||
|
@ -26,6 +27,8 @@ class FairMQMessageNN : public FairMQMessage
|
||||||
FairMQMessageNN();
|
FairMQMessageNN();
|
||||||
FairMQMessageNN(const size_t size);
|
FairMQMessageNN(const size_t size);
|
||||||
FairMQMessageNN(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr);
|
FairMQMessageNN(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr);
|
||||||
|
FairMQMessageNN(FairMQRegionPtr& region, void* data, const size_t size);
|
||||||
|
|
||||||
FairMQMessageNN(const FairMQMessageNN&) = delete;
|
FairMQMessageNN(const FairMQMessageNN&) = delete;
|
||||||
FairMQMessageNN operator=(const FairMQMessageNN&) = delete;
|
FairMQMessageNN operator=(const FairMQMessageNN&) = delete;
|
||||||
|
|
||||||
|
@ -53,6 +56,7 @@ class FairMQMessageNN : public FairMQMessage
|
||||||
void* fMessage;
|
void* fMessage;
|
||||||
size_t fSize;
|
size_t fSize;
|
||||||
bool fReceiving;
|
bool fReceiving;
|
||||||
|
bool fRegion;
|
||||||
static std::string fDeviceID;
|
static std::string fDeviceID;
|
||||||
static FairMQ::Transport fTransportType;
|
static FairMQ::Transport fTransportType;
|
||||||
|
|
||||||
|
|
34
fairmq/nanomsg/FairMQRegionNN.cxx
Normal file
34
fairmq/nanomsg/FairMQRegionNN.cxx
Normal file
|
@ -0,0 +1,34 @@
|
||||||
|
/********************************************************************************
|
||||||
|
* 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" *
|
||||||
|
********************************************************************************/
|
||||||
|
|
||||||
|
#include "FairMQRegionNN.h"
|
||||||
|
#include "FairMQLogger.h"
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
FairMQRegionNN::FairMQRegionNN(const size_t size)
|
||||||
|
: fBuffer(malloc(size))
|
||||||
|
, fSize(size)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void* FairMQRegionNN::GetData() const
|
||||||
|
{
|
||||||
|
return fBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t FairMQRegionNN::GetSize() const
|
||||||
|
{
|
||||||
|
return fSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
FairMQRegionNN::~FairMQRegionNN()
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "destroying region";
|
||||||
|
free(fBuffer);
|
||||||
|
}
|
35
fairmq/nanomsg/FairMQRegionNN.h
Normal file
35
fairmq/nanomsg/FairMQRegionNN.h
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
/********************************************************************************
|
||||||
|
* 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" *
|
||||||
|
********************************************************************************/
|
||||||
|
|
||||||
|
#ifndef FAIRMQREGIONNN_H_
|
||||||
|
#define FAIRMQREGIONNN_H_
|
||||||
|
|
||||||
|
#include "FairMQRegion.h"
|
||||||
|
|
||||||
|
#include <cstddef> // size_t
|
||||||
|
|
||||||
|
class FairMQRegionNN : public FairMQRegion
|
||||||
|
{
|
||||||
|
friend class FairMQSocketNN;
|
||||||
|
|
||||||
|
public:
|
||||||
|
FairMQRegionNN(const size_t size);
|
||||||
|
FairMQRegionNN(const FairMQRegionNN&) = delete;
|
||||||
|
FairMQRegionNN operator=(const FairMQRegionNN&) = delete;
|
||||||
|
|
||||||
|
virtual void* GetData() const override;
|
||||||
|
virtual size_t GetSize() const override;
|
||||||
|
|
||||||
|
virtual ~FairMQRegionNN();
|
||||||
|
|
||||||
|
private:
|
||||||
|
void* fBuffer;
|
||||||
|
size_t fSize;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif /* FAIRMQREGIONNN_H_ */
|
|
@ -127,7 +127,14 @@ int FairMQSocketNN::Send(FairMQMessagePtr& msg, const int flags)
|
||||||
while (true)
|
while (true)
|
||||||
{
|
{
|
||||||
void* ptr = msg->GetMessage();
|
void* ptr = msg->GetMessage();
|
||||||
nbytes = nn_send(fSocket, &ptr, NN_MSG, flags);
|
if (static_cast<FairMQMessageNN*>(msg.get())->fRegion == false)
|
||||||
|
{
|
||||||
|
nbytes = nn_send(fSocket, &ptr, NN_MSG, flags);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
nbytes = nn_send(fSocket, ptr, msg->GetSize(), flags);
|
||||||
|
}
|
||||||
if (nbytes >= 0)
|
if (nbytes >= 0)
|
||||||
{
|
{
|
||||||
fBytesTx += nbytes;
|
fBytesTx += nbytes;
|
||||||
|
|
|
@ -35,6 +35,11 @@ FairMQMessagePtr FairMQTransportFactoryNN::CreateMessage(void* data, const size_
|
||||||
return unique_ptr<FairMQMessage>(new FairMQMessageNN(data, size, ffn, hint));
|
return unique_ptr<FairMQMessage>(new FairMQMessageNN(data, size, ffn, hint));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQMessagePtr FairMQTransportFactoryNN::CreateMessage(FairMQRegionPtr& region, void* data, const size_t size) const
|
||||||
|
{
|
||||||
|
return unique_ptr<FairMQMessage>(new FairMQMessageNN(region, data, size));
|
||||||
|
}
|
||||||
|
|
||||||
FairMQSocketPtr FairMQTransportFactoryNN::CreateSocket(const string& type, const string& name) const
|
FairMQSocketPtr FairMQTransportFactoryNN::CreateSocket(const string& type, const string& name) const
|
||||||
{
|
{
|
||||||
return unique_ptr<FairMQSocket>(new FairMQSocketNN(type, name, GetId()));
|
return unique_ptr<FairMQSocket>(new FairMQSocketNN(type, name, GetId()));
|
||||||
|
@ -60,6 +65,11 @@ FairMQPollerPtr FairMQTransportFactoryNN::CreatePoller(const FairMQSocket& cmdSo
|
||||||
return unique_ptr<FairMQPoller>(new FairMQPollerNN(cmdSocket, dataSocket));
|
return unique_ptr<FairMQPoller>(new FairMQPollerNN(cmdSocket, dataSocket));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQRegionPtr FairMQTransportFactoryNN::CreateRegion(const size_t size) const
|
||||||
|
{
|
||||||
|
return unique_ptr<FairMQRegion>(new FairMQRegionNN(size));
|
||||||
|
}
|
||||||
|
|
||||||
FairMQ::Transport FairMQTransportFactoryNN::GetType() const
|
FairMQ::Transport FairMQTransportFactoryNN::GetType() const
|
||||||
{
|
{
|
||||||
return fTransportType;
|
return fTransportType;
|
||||||
|
|
|
@ -13,6 +13,7 @@
|
||||||
#include "FairMQMessageNN.h"
|
#include "FairMQMessageNN.h"
|
||||||
#include "FairMQSocketNN.h"
|
#include "FairMQSocketNN.h"
|
||||||
#include "FairMQPollerNN.h"
|
#include "FairMQPollerNN.h"
|
||||||
|
#include "FairMQRegionNN.h"
|
||||||
#include <options/FairMQProgOptions.h>
|
#include <options/FairMQProgOptions.h>
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
@ -27,6 +28,7 @@ class FairMQTransportFactoryNN : public FairMQTransportFactory
|
||||||
FairMQMessagePtr CreateMessage() const override;
|
FairMQMessagePtr CreateMessage() const override;
|
||||||
FairMQMessagePtr CreateMessage(const size_t size) const override;
|
FairMQMessagePtr CreateMessage(const size_t size) const override;
|
||||||
FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const override;
|
FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const override;
|
||||||
|
FairMQMessagePtr CreateMessage(FairMQRegionPtr& region, void* data, const size_t size) const override;
|
||||||
|
|
||||||
FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const override;
|
FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const override;
|
||||||
|
|
||||||
|
@ -35,6 +37,8 @@ class FairMQTransportFactoryNN : public FairMQTransportFactory
|
||||||
FairMQPollerPtr CreatePoller(const std::unordered_map<std::string, std::vector<FairMQChannel>>& channelsMap, const std::vector<std::string>& channelList) const override;
|
FairMQPollerPtr CreatePoller(const std::unordered_map<std::string, std::vector<FairMQChannel>>& channelsMap, const std::vector<std::string>& channelList) const override;
|
||||||
FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const override;
|
FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const override;
|
||||||
|
|
||||||
|
FairMQRegionPtr CreateRegion(const size_t size) const override;
|
||||||
|
|
||||||
FairMQ::Transport GetType() const override;
|
FairMQ::Transport GetType() const override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
|
|
@ -9,11 +9,15 @@
|
||||||
#include <cstdlib>
|
#include <cstdlib>
|
||||||
|
|
||||||
#include "FairMQMessageSHM.h"
|
#include "FairMQMessageSHM.h"
|
||||||
|
#include "FairMQRegionSHM.h"
|
||||||
#include "FairMQLogger.h"
|
#include "FairMQLogger.h"
|
||||||
|
#include "FairMQShmCommon.h"
|
||||||
|
|
||||||
using namespace std;
|
using namespace std;
|
||||||
using namespace fair::mq::shmem;
|
using namespace fair::mq::shmem;
|
||||||
|
|
||||||
|
namespace bipc = boost::interprocess;
|
||||||
|
|
||||||
// uint64_t FairMQMessageSHM::fMessageID = 0;
|
// uint64_t FairMQMessageSHM::fMessageID = 0;
|
||||||
// string FairMQMessageSHM::fDeviceID = string();
|
// string FairMQMessageSHM::fDeviceID = string();
|
||||||
atomic<bool> FairMQMessageSHM::fInterrupted(false);
|
atomic<bool> FairMQMessageSHM::fInterrupted(false);
|
||||||
|
@ -25,9 +29,11 @@ FairMQMessageSHM::FairMQMessageSHM()
|
||||||
// , fReceiving(false)
|
// , fReceiving(false)
|
||||||
, fQueued(false)
|
, fQueued(false)
|
||||||
, fMetaCreated(false)
|
, fMetaCreated(false)
|
||||||
|
, fRegionId(0)
|
||||||
, fHandle()
|
, fHandle()
|
||||||
, fChunkSize(0)
|
, fSize(0)
|
||||||
, fLocalPtr(nullptr)
|
, fLocalPtr(nullptr)
|
||||||
|
, fRemoteRegion(nullptr)
|
||||||
{
|
{
|
||||||
if (zmq_msg_init(&fMessage) != 0)
|
if (zmq_msg_init(&fMessage) != 0)
|
||||||
{
|
{
|
||||||
|
@ -47,9 +53,11 @@ FairMQMessageSHM::FairMQMessageSHM(const size_t size)
|
||||||
// , fReceiving(false)
|
// , fReceiving(false)
|
||||||
, fQueued(false)
|
, fQueued(false)
|
||||||
, fMetaCreated(false)
|
, fMetaCreated(false)
|
||||||
|
, fRegionId(0)
|
||||||
, fHandle()
|
, fHandle()
|
||||||
, fChunkSize(0)
|
, fSize(0)
|
||||||
, fLocalPtr(nullptr)
|
, fLocalPtr(nullptr)
|
||||||
|
, fRemoteRegion(nullptr)
|
||||||
{
|
{
|
||||||
InitializeChunk(size);
|
InitializeChunk(size);
|
||||||
}
|
}
|
||||||
|
@ -60,9 +68,11 @@ FairMQMessageSHM::FairMQMessageSHM(void* data, const size_t size, fairmq_free_fn
|
||||||
// , fReceiving(false)
|
// , fReceiving(false)
|
||||||
, fQueued(false)
|
, fQueued(false)
|
||||||
, fMetaCreated(false)
|
, fMetaCreated(false)
|
||||||
|
, fRegionId(0)
|
||||||
, fHandle()
|
, fHandle()
|
||||||
, fChunkSize(0)
|
, fSize(0)
|
||||||
, fLocalPtr(nullptr)
|
, fLocalPtr(nullptr)
|
||||||
|
, fRemoteRegion(nullptr)
|
||||||
{
|
{
|
||||||
if (InitializeChunk(size))
|
if (InitializeChunk(size))
|
||||||
{
|
{
|
||||||
|
@ -78,6 +88,35 @@ FairMQMessageSHM::FairMQMessageSHM(void* data, const size_t size, fairmq_free_fn
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQMessageSHM::FairMQMessageSHM(FairMQRegionPtr& region, void* data, const size_t size)
|
||||||
|
: fMessage()
|
||||||
|
// , fOwner(nullptr)
|
||||||
|
// , fReceiving(false)
|
||||||
|
, fQueued(false)
|
||||||
|
, fMetaCreated(false)
|
||||||
|
, fRegionId(static_cast<FairMQRegionSHM*>(region.get())->fRegionId)
|
||||||
|
, fHandle()
|
||||||
|
, fSize(size)
|
||||||
|
, fLocalPtr(data)
|
||||||
|
, fRemoteRegion(nullptr)
|
||||||
|
{
|
||||||
|
fHandle = (bipc::managed_shared_memory::handle_t)(reinterpret_cast<const char*>(data) - reinterpret_cast<const char*>(region->GetData()));
|
||||||
|
|
||||||
|
if (zmq_msg_init_size(&fMessage, sizeof(MetaHeader)) != 0)
|
||||||
|
{
|
||||||
|
LOG(ERROR) << "failed initializing meta message, reason: " << zmq_strerror(errno);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
MetaHeader* metaPtr = new(zmq_msg_data(&fMessage)) MetaHeader();
|
||||||
|
metaPtr->fSize = size;
|
||||||
|
metaPtr->fHandle = fHandle;
|
||||||
|
metaPtr->fRegionId = fRegionId;
|
||||||
|
|
||||||
|
fMetaCreated = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
bool FairMQMessageSHM::InitializeChunk(const size_t size)
|
bool FairMQMessageSHM::InitializeChunk(const size_t size)
|
||||||
{
|
{
|
||||||
// string chunkID = fDeviceID + "c" + to_string(fMessageID);
|
// string chunkID = fDeviceID + "c" + to_string(fMessageID);
|
||||||
|
@ -109,7 +148,7 @@ bool FairMQMessageSHM::InitializeChunk(const size_t size)
|
||||||
fHandle = Manager::Instance().Segment()->get_handle_from_address(fLocalPtr);
|
fHandle = Manager::Instance().Segment()->get_handle_from_address(fLocalPtr);
|
||||||
}
|
}
|
||||||
|
|
||||||
fChunkSize = size;
|
fSize = size;
|
||||||
|
|
||||||
if (zmq_msg_init_size(&fMessage, sizeof(MetaHeader)) != 0)
|
if (zmq_msg_init_size(&fMessage, sizeof(MetaHeader)) != 0)
|
||||||
{
|
{
|
||||||
|
@ -119,6 +158,7 @@ bool FairMQMessageSHM::InitializeChunk(const size_t size)
|
||||||
MetaHeader* metaPtr = new(zmq_msg_data(&fMessage)) MetaHeader();
|
MetaHeader* metaPtr = new(zmq_msg_data(&fMessage)) MetaHeader();
|
||||||
metaPtr->fSize = size;
|
metaPtr->fSize = size;
|
||||||
metaPtr->fHandle = fHandle;
|
metaPtr->fHandle = fHandle;
|
||||||
|
metaPtr->fRegionId = fRegionId;
|
||||||
|
|
||||||
// if (zmq_msg_init_data(&fMessage, const_cast<char*>(ownerID->c_str()), ownerID->length(), StringDeleter, ownerID) != 0)
|
// if (zmq_msg_init_data(&fMessage, const_cast<char*>(ownerID->c_str()), ownerID->length(), StringDeleter, ownerID) != 0)
|
||||||
// {
|
// {
|
||||||
|
@ -187,14 +227,21 @@ void* FairMQMessageSHM::GetData()
|
||||||
{
|
{
|
||||||
return fLocalPtr;
|
return fLocalPtr;
|
||||||
}
|
}
|
||||||
else if (fHandle)
|
|
||||||
{
|
|
||||||
return Manager::Instance().Segment()->get_address_from_handle(fHandle);
|
|
||||||
}
|
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
// LOG(ERROR) << "Trying to get data of an empty shared memory message";
|
if (fRegionId == 0)
|
||||||
return nullptr;
|
{
|
||||||
|
return Manager::Instance().Segment()->get_address_from_handle(fHandle);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (!fRemoteRegion)
|
||||||
|
{
|
||||||
|
fRemoteRegion = FairMQRegionPtr(new FairMQRegionSHM(fRegionId, true));
|
||||||
|
}
|
||||||
|
fLocalPtr = reinterpret_cast<char*>(fRemoteRegion->GetData()) + fHandle;
|
||||||
|
return fLocalPtr;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// if (fOwner)
|
// if (fOwner)
|
||||||
|
@ -210,7 +257,7 @@ void* FairMQMessageSHM::GetData()
|
||||||
|
|
||||||
size_t FairMQMessageSHM::GetSize()
|
size_t FairMQMessageSHM::GetSize()
|
||||||
{
|
{
|
||||||
return fChunkSize;
|
return fSize;
|
||||||
// if (fOwner)
|
// if (fOwner)
|
||||||
// {
|
// {
|
||||||
// return fOwner->fPtr->GetSize();
|
// return fOwner->fPtr->GetSize();
|
||||||
|
@ -324,7 +371,7 @@ void FairMQMessageSHM::CloseMessage()
|
||||||
// }
|
// }
|
||||||
// else
|
// else
|
||||||
// {
|
// {
|
||||||
if (fHandle && !fQueued)
|
if (fHandle && !fQueued && fRegionId == 0)
|
||||||
{
|
{
|
||||||
// LOG(WARN) << "Destroying unsent message";
|
// LOG(WARN) << "Destroying unsent message";
|
||||||
// Manager::Instance().Segment()->destroy_ptr(fHandle);
|
// Manager::Instance().Segment()->destroy_ptr(fHandle);
|
||||||
|
|
|
@ -8,13 +8,17 @@
|
||||||
#ifndef FAIRMQMESSAGESHM_H_
|
#ifndef FAIRMQMESSAGESHM_H_
|
||||||
#define FAIRMQMESSAGESHM_H_
|
#define FAIRMQMESSAGESHM_H_
|
||||||
|
|
||||||
#include <cstddef>
|
#include <cstddef> // size_t
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
|
||||||
#include <zmq.h>
|
#include <zmq.h>
|
||||||
|
|
||||||
|
#include <boost/interprocess/shared_memory_object.hpp>
|
||||||
|
#include <boost/interprocess/mapped_region.hpp>
|
||||||
|
|
||||||
#include "FairMQMessage.h"
|
#include "FairMQMessage.h"
|
||||||
|
#include "FairMQRegion.h"
|
||||||
#include "FairMQShmManager.h"
|
#include "FairMQShmManager.h"
|
||||||
|
|
||||||
class FairMQMessageSHM : public FairMQMessage
|
class FairMQMessageSHM : public FairMQMessage
|
||||||
|
@ -25,6 +29,8 @@ class FairMQMessageSHM : public FairMQMessage
|
||||||
FairMQMessageSHM();
|
FairMQMessageSHM();
|
||||||
FairMQMessageSHM(const size_t size);
|
FairMQMessageSHM(const size_t size);
|
||||||
FairMQMessageSHM(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr);
|
FairMQMessageSHM(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr);
|
||||||
|
FairMQMessageSHM(FairMQRegionPtr& region, void* data, const size_t size);
|
||||||
|
|
||||||
FairMQMessageSHM(const FairMQMessageSHM&) = delete;
|
FairMQMessageSHM(const FairMQMessageSHM&) = delete;
|
||||||
FairMQMessageSHM operator=(const FairMQMessageSHM&) = delete;
|
FairMQMessageSHM operator=(const FairMQMessageSHM&) = delete;
|
||||||
|
|
||||||
|
@ -62,9 +68,11 @@ class FairMQMessageSHM : public FairMQMessage
|
||||||
bool fMetaCreated;
|
bool fMetaCreated;
|
||||||
static std::atomic<bool> fInterrupted;
|
static std::atomic<bool> fInterrupted;
|
||||||
static FairMQ::Transport fTransportType;
|
static FairMQ::Transport fTransportType;
|
||||||
|
uint64_t fRegionId;
|
||||||
bipc::managed_shared_memory::handle_t fHandle;
|
bipc::managed_shared_memory::handle_t fHandle;
|
||||||
size_t fChunkSize;
|
size_t fSize;
|
||||||
void* fLocalPtr;
|
void* fLocalPtr;
|
||||||
|
FairMQRegionPtr fRemoteRegion;
|
||||||
};
|
};
|
||||||
|
|
||||||
#endif /* FAIRMQMESSAGESHM_H_ */
|
#endif /* FAIRMQMESSAGESHM_H_ */
|
||||||
|
|
97
fairmq/shmem/FairMQRegionSHM.cxx
Normal file
97
fairmq/shmem/FairMQRegionSHM.cxx
Normal file
|
@ -0,0 +1,97 @@
|
||||||
|
/********************************************************************************
|
||||||
|
* 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" *
|
||||||
|
********************************************************************************/
|
||||||
|
|
||||||
|
#include "FairMQRegionSHM.h"
|
||||||
|
#include "FairMQShmManager.h"
|
||||||
|
#include "FairMQShmCommon.h"
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
using namespace fair::mq::shmem;
|
||||||
|
|
||||||
|
namespace bipc = boost::interprocess;
|
||||||
|
|
||||||
|
atomic<bool> FairMQRegionSHM::fInterrupted(false);
|
||||||
|
|
||||||
|
FairMQRegionSHM::FairMQRegionSHM(const size_t size)
|
||||||
|
: fShmemObject()
|
||||||
|
, fRegion()
|
||||||
|
, fRegionId(0)
|
||||||
|
, fRegionIdStr()
|
||||||
|
, fRemote(false)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
RegionCounter* rc = Manager::Instance().ManagementSegment().find<RegionCounter>(bipc::unique_instance).first;
|
||||||
|
if (rc)
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "shmem: region counter found, with value of " << rc->fCount << ". incrementing.";
|
||||||
|
(rc->fCount)++;
|
||||||
|
LOG(DEBUG) << "shmem: incremented region counter, now: " << rc->fCount;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "shmem: no region counter found, creating one and initializing with 1";
|
||||||
|
rc = Manager::Instance().ManagementSegment().construct<RegionCounter>(bipc::unique_instance)(1);
|
||||||
|
LOG(DEBUG) << "shmem: initialized region counter with: " << rc->fCount;
|
||||||
|
}
|
||||||
|
|
||||||
|
fRegionId = rc->fCount;
|
||||||
|
fRegionIdStr = "fairmq_shmem_region_" + std::to_string(fRegionId);
|
||||||
|
|
||||||
|
fShmemObject = unique_ptr<bipc::shared_memory_object>(new bipc::shared_memory_object(bipc::create_only, fRegionIdStr.c_str(), bipc::read_write));
|
||||||
|
fShmemObject->truncate(size);
|
||||||
|
fRegion = unique_ptr<bipc::mapped_region>(new bipc::mapped_region(*fShmemObject, bipc::read_write)); // TODO: add HUGEPAGES flag here
|
||||||
|
}
|
||||||
|
catch (bipc::interprocess_exception& e)
|
||||||
|
{
|
||||||
|
LOG(ERROR) << "shmem: cannot create region. Already created/not cleaned up?";
|
||||||
|
LOG(ERROR) << e.what();
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
FairMQRegionSHM::FairMQRegionSHM(const uint64_t id, bool remote)
|
||||||
|
: fShmemObject()
|
||||||
|
, fRegion()
|
||||||
|
, fRegionId(id)
|
||||||
|
, fRegionIdStr()
|
||||||
|
, fRemote(remote)
|
||||||
|
{
|
||||||
|
try
|
||||||
|
{
|
||||||
|
fRegionIdStr = "fairmq_shmem_region_" + std::to_string(fRegionId);
|
||||||
|
|
||||||
|
fShmemObject = unique_ptr<bipc::shared_memory_object>(new bipc::shared_memory_object(bipc::open_only, fRegionIdStr.c_str(), bipc::read_write));
|
||||||
|
fRegion = unique_ptr<bipc::mapped_region>(new bipc::mapped_region(*fShmemObject, bipc::read_write)); // TODO: add HUGEPAGES flag here
|
||||||
|
}
|
||||||
|
catch (bipc::interprocess_exception& e)
|
||||||
|
{
|
||||||
|
LOG(ERROR) << "shmem: cannot open region. Already closed?";
|
||||||
|
LOG(ERROR) << e.what();
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void* FairMQRegionSHM::GetData() const
|
||||||
|
{
|
||||||
|
return fRegion->get_address();
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t FairMQRegionSHM::GetSize() const
|
||||||
|
{
|
||||||
|
return fRegion->get_size();
|
||||||
|
}
|
||||||
|
|
||||||
|
FairMQRegionSHM::~FairMQRegionSHM()
|
||||||
|
{
|
||||||
|
if (!fRemote)
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "destroying region";
|
||||||
|
bipc::shared_memory_object::remove(fRegionIdStr.c_str());
|
||||||
|
}
|
||||||
|
}
|
46
fairmq/shmem/FairMQRegionSHM.h
Normal file
46
fairmq/shmem/FairMQRegionSHM.h
Normal file
|
@ -0,0 +1,46 @@
|
||||||
|
/********************************************************************************
|
||||||
|
* 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" *
|
||||||
|
********************************************************************************/
|
||||||
|
|
||||||
|
#ifndef FAIRMQREGIONSHM_H_
|
||||||
|
#define FAIRMQREGIONSHM_H_
|
||||||
|
|
||||||
|
#include "FairMQRegion.h"
|
||||||
|
|
||||||
|
#include <boost/interprocess/shared_memory_object.hpp>
|
||||||
|
#include <boost/interprocess/mapped_region.hpp>
|
||||||
|
|
||||||
|
#include <cstddef> // size_t
|
||||||
|
#include <atomic>
|
||||||
|
#include <memory>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
class FairMQRegionSHM : public FairMQRegion
|
||||||
|
{
|
||||||
|
friend class FairMQSocketSHM;
|
||||||
|
friend class FairMQMessageSHM;
|
||||||
|
|
||||||
|
public:
|
||||||
|
FairMQRegionSHM(const size_t size);
|
||||||
|
|
||||||
|
virtual void* GetData() const override;
|
||||||
|
virtual size_t GetSize() const override;
|
||||||
|
|
||||||
|
virtual ~FairMQRegionSHM();
|
||||||
|
|
||||||
|
private:
|
||||||
|
FairMQRegionSHM(const uint64_t id, bool remote);
|
||||||
|
|
||||||
|
static std::atomic<bool> fInterrupted;
|
||||||
|
std::unique_ptr<boost::interprocess::shared_memory_object> fShmemObject;
|
||||||
|
std::unique_ptr<boost::interprocess::mapped_region> fRegion;
|
||||||
|
uint64_t fRegionId;
|
||||||
|
std::string fRegionIdStr;
|
||||||
|
bool fRemote;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif /* FAIRMQREGIONSHM_H_ */
|
60
fairmq/shmem/FairMQShmCommon.h
Normal file
60
fairmq/shmem/FairMQShmCommon.h
Normal file
|
@ -0,0 +1,60 @@
|
||||||
|
/********************************************************************************
|
||||||
|
* 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" *
|
||||||
|
********************************************************************************/
|
||||||
|
#ifndef FAIR_MQ_SHMEM_COMMON_H_
|
||||||
|
#define FAIR_MQ_SHMEM_COMMON_H_
|
||||||
|
|
||||||
|
#include <atomic>
|
||||||
|
|
||||||
|
#include <boost/interprocess/managed_shared_memory.hpp>
|
||||||
|
|
||||||
|
namespace fair
|
||||||
|
{
|
||||||
|
namespace mq
|
||||||
|
{
|
||||||
|
namespace shmem
|
||||||
|
{
|
||||||
|
|
||||||
|
struct DeviceCounter
|
||||||
|
{
|
||||||
|
DeviceCounter(unsigned int c)
|
||||||
|
: fCount(c)
|
||||||
|
{}
|
||||||
|
|
||||||
|
std::atomic<unsigned int> fCount;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct RegionCounter
|
||||||
|
{
|
||||||
|
RegionCounter(unsigned int c)
|
||||||
|
: fCount(c)
|
||||||
|
{}
|
||||||
|
|
||||||
|
std::atomic<unsigned int> fCount;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct MonitorStatus
|
||||||
|
{
|
||||||
|
MonitorStatus()
|
||||||
|
: fActive(true)
|
||||||
|
{}
|
||||||
|
|
||||||
|
bool fActive;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct alignas(32) MetaHeader
|
||||||
|
{
|
||||||
|
uint64_t fSize;
|
||||||
|
uint64_t fRegionId;
|
||||||
|
boost::interprocess::managed_shared_memory::handle_t fHandle;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace shmem
|
||||||
|
} // namespace mq
|
||||||
|
} // namespace fair
|
||||||
|
|
||||||
|
#endif /* FAIR_MQ_SHMEM_COMMON_H_ */
|
|
@ -115,18 +115,42 @@ class Manager
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Remove()
|
||||||
|
{
|
||||||
|
if (bipc::shared_memory_object::remove("fairmq_shmem_main"))
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "shmem: successfully removed \"fairmq_shmem_main\" segment after the device has stopped.";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "shmem: did not remove \"fairmq_shmem_main\" segment after the device stopped. Already removed?";
|
||||||
|
}
|
||||||
|
|
||||||
|
if (bipc::shared_memory_object::remove("fairmq_shmem_management"))
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "shmem: successfully removed \"fairmq_shmem_management\" segment after the device has stopped.";
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "shmem: did not remove \"fairmq_shmem_management\" segment after the device stopped. Already removed?";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bipc::managed_shared_memory& ManagementSegment()
|
||||||
|
{
|
||||||
|
return fManagementSegment;
|
||||||
|
}
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Manager()
|
Manager()
|
||||||
: fSegment(nullptr)
|
: fSegment(nullptr)
|
||||||
|
, fManagementSegment(bipc::open_or_create, "fairmq_shmem_management", 65536)
|
||||||
{}
|
{}
|
||||||
|
Manager(const Manager&) = delete;
|
||||||
|
Manager operator=(const Manager&) = delete;
|
||||||
|
|
||||||
bipc::managed_shared_memory* fSegment;
|
bipc::managed_shared_memory* fSegment;
|
||||||
};
|
bipc::managed_shared_memory fManagementSegment;
|
||||||
|
|
||||||
struct alignas(16) MetaHeader
|
|
||||||
{
|
|
||||||
uint64_t fSize;
|
|
||||||
bipc::managed_shared_memory::handle_t fHandle;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// class Chunk
|
// class Chunk
|
||||||
|
|
|
@ -7,16 +7,18 @@
|
||||||
********************************************************************************/
|
********************************************************************************/
|
||||||
|
|
||||||
#include "FairMQShmMonitor.h"
|
#include "FairMQShmMonitor.h"
|
||||||
#include "FairMQShmDeviceCounter.h"
|
#include "FairMQShmCommon.h"
|
||||||
|
|
||||||
#include <boost/interprocess/managed_shared_memory.hpp>
|
#include <boost/interprocess/managed_shared_memory.hpp>
|
||||||
#include <boost/interprocess/containers/vector.hpp>
|
#include <boost/interprocess/containers/vector.hpp>
|
||||||
#include <boost/interprocess/containers/string.hpp>
|
#include <boost/interprocess/containers/string.hpp>
|
||||||
#include <boost/interprocess/allocators/allocator.hpp>
|
#include <boost/interprocess/allocators/allocator.hpp>
|
||||||
|
|
||||||
|
#include <boost/interprocess/sync/named_mutex.hpp>
|
||||||
#include <boost/interprocess/ipc/message_queue.hpp>
|
#include <boost/interprocess/ipc/message_queue.hpp>
|
||||||
#include <boost/date_time/posix_time/posix_time.hpp>
|
#include <boost/date_time/posix_time/posix_time.hpp>
|
||||||
|
|
||||||
|
#include <csignal>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <iomanip>
|
#include <iomanip>
|
||||||
|
|
||||||
|
@ -32,6 +34,11 @@ using String = bipc::basic_string<char, std::char_traits<char>, CharAll
|
||||||
using StringAllocator = bipc::allocator<String, bipc::managed_shared_memory::segment_manager>;
|
using StringAllocator = bipc::allocator<String, bipc::managed_shared_memory::segment_manager>;
|
||||||
using StringVector = bipc::vector<String, StringAllocator>;
|
using StringVector = bipc::vector<String, StringAllocator>;
|
||||||
|
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
volatile std::sig_atomic_t gSignalStatus;
|
||||||
|
}
|
||||||
|
|
||||||
namespace fair
|
namespace fair
|
||||||
{
|
{
|
||||||
namespace mq
|
namespace mq
|
||||||
|
@ -39,6 +46,11 @@ namespace mq
|
||||||
namespace shmem
|
namespace shmem
|
||||||
{
|
{
|
||||||
|
|
||||||
|
void signalHandler(int signal)
|
||||||
|
{
|
||||||
|
gSignalStatus = signal;
|
||||||
|
}
|
||||||
|
|
||||||
Monitor::Monitor(const string& segmentName, bool selfDestruct, bool interactive, unsigned int timeoutInMS)
|
Monitor::Monitor(const string& segmentName, bool selfDestruct, bool interactive, unsigned int timeoutInMS)
|
||||||
: fSelfDestruct(selfDestruct)
|
: fSelfDestruct(selfDestruct)
|
||||||
, fInteractive(interactive)
|
, fInteractive(interactive)
|
||||||
|
@ -48,15 +60,43 @@ Monitor::Monitor(const string& segmentName, bool selfDestruct, bool interactive,
|
||||||
, fTerminating(false)
|
, fTerminating(false)
|
||||||
, fHeartbeatTriggered(false)
|
, fHeartbeatTriggered(false)
|
||||||
, fLastHeartbeat()
|
, fLastHeartbeat()
|
||||||
, fHeartbeatThread()
|
, fSignalThread()
|
||||||
|
, fManagementSegment(bipc::open_or_create, "fairmq_shmem_management", 65536)
|
||||||
{
|
{
|
||||||
if (bipc::message_queue::remove("fairmq_shmem_control_queue"))
|
MonitorStatus* monitorStatus = fManagementSegment.find<MonitorStatus>(bipc::unique_instance).first;
|
||||||
|
if (monitorStatus != nullptr)
|
||||||
{
|
{
|
||||||
// cout << "successfully removed control queue" << endl;
|
cout << "shmmonitor already started or not properly exited. Try `shmmonitor --cleanup`" << endl;
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
else
|
fManagementSegment.construct<MonitorStatus>(bipc::unique_instance)();
|
||||||
|
|
||||||
|
CleanupControlQueues();
|
||||||
|
}
|
||||||
|
|
||||||
|
void Monitor::CatchSignals()
|
||||||
|
{
|
||||||
|
signal(SIGINT, signalHandler);
|
||||||
|
signal(SIGTERM, signalHandler);
|
||||||
|
fSignalThread = thread(&Monitor::SignalMonitor, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
void Monitor::SignalMonitor()
|
||||||
|
{
|
||||||
|
while (true)
|
||||||
{
|
{
|
||||||
// cout << "could not remove control queue" << endl;
|
if (gSignalStatus != 0)
|
||||||
|
{
|
||||||
|
fTerminating = true;
|
||||||
|
cout << "signal: " << gSignalStatus << endl;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else if (fTerminating)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
this_thread::sleep_for(chrono::milliseconds(100));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -109,22 +149,15 @@ void Monitor::MonitorHeartbeats()
|
||||||
cout << ie.what() << endl;
|
cout << ie.what() << endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (bipc::message_queue::remove("fairmq_shmem_control_queue"))
|
CleanupControlQueues();
|
||||||
{
|
|
||||||
// cout << "successfully removed control queue" << endl;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
cout << "could not remove control queue" << endl;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void Monitor::Interactive()
|
void Monitor::Interactive()
|
||||||
{
|
{
|
||||||
char input;
|
char c;
|
||||||
pollfd inputFd[1];
|
pollfd cinfd[1];
|
||||||
inputFd[0].fd = fileno(stdin);
|
cinfd[0].fd = fileno(stdin);
|
||||||
inputFd[0].events = POLLIN;
|
cinfd[0].events = POLLIN;
|
||||||
|
|
||||||
struct termios t;
|
struct termios t;
|
||||||
tcgetattr(STDIN_FILENO, &t); // get the current terminal I/O structure
|
tcgetattr(STDIN_FILENO, &t); // get the current terminal I/O structure
|
||||||
|
@ -138,11 +171,16 @@ void Monitor::Interactive()
|
||||||
|
|
||||||
while (!fTerminating)
|
while (!fTerminating)
|
||||||
{
|
{
|
||||||
if (poll(inputFd, 1, 100))
|
if (poll(cinfd, 1, 100))
|
||||||
{
|
{
|
||||||
input = getchar();
|
if (fTerminating || gSignalStatus != 0)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
switch (input)
|
c = getchar();
|
||||||
|
|
||||||
|
switch (c)
|
||||||
{
|
{
|
||||||
case 'q':
|
case 'q':
|
||||||
cout << "[q] --> quitting." << endl;
|
cout << "[q] --> quitting." << endl;
|
||||||
|
@ -165,7 +203,7 @@ void Monitor::Interactive()
|
||||||
cout << "[\\n] --> invalid input." << endl;
|
cout << "[\\n] --> invalid input." << endl;
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
cout << "[" << input << "] --> invalid input." << endl;
|
cout << "[" << c << "] --> invalid input." << endl;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -177,6 +215,11 @@ void Monitor::Interactive()
|
||||||
PrintHeader();
|
PrintHeader();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (fTerminating)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
CheckSegment();
|
CheckSegment();
|
||||||
|
|
||||||
if (!fTerminating)
|
if (!fTerminating)
|
||||||
|
@ -228,10 +271,10 @@ void Monitor::CheckSegment()
|
||||||
|
|
||||||
unsigned int numDevices = 0;
|
unsigned int numDevices = 0;
|
||||||
|
|
||||||
pair<fair::mq::shmem::DeviceCounter*, size_t> result = segment.find<fair::mq::shmem::DeviceCounter>(bipc::unique_instance);
|
fair::mq::shmem::DeviceCounter* dc = segment.find<fair::mq::shmem::DeviceCounter>(bipc::unique_instance).first;
|
||||||
if (result.first != nullptr)
|
if (dc)
|
||||||
{
|
{
|
||||||
numDevices = result.first->count;
|
numDevices = dc->fCount;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto now = chrono::high_resolution_clock::now();
|
auto now = chrono::high_resolution_clock::now();
|
||||||
|
@ -294,13 +337,57 @@ void Monitor::CheckSegment()
|
||||||
|
|
||||||
void Monitor::Cleanup(const string& segmentName)
|
void Monitor::Cleanup(const string& segmentName)
|
||||||
{
|
{
|
||||||
if (bipc::shared_memory_object::remove(segmentName.c_str()))
|
try
|
||||||
{
|
{
|
||||||
cout << "Successfully removed shared memory \"" << segmentName.c_str() << "\"." << endl;
|
bipc::managed_shared_memory managementSegment(bipc::open_only, "fairmq_shmem_management");
|
||||||
|
RegionCounter* rc = managementSegment.find<RegionCounter>(bipc::unique_instance).first;
|
||||||
|
if (rc)
|
||||||
|
{
|
||||||
|
cout << "Region counter found: " << rc->fCount << endl;
|
||||||
|
unsigned int regionCount = rc->fCount;
|
||||||
|
for (int i = 1; i <= regionCount; ++i)
|
||||||
|
{
|
||||||
|
RemoveObject("fairmq_shmem_region_" + to_string(regionCount));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
cout << "shmem: no region counter found. no regions to cleanup." << endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
RemoveObject("fairmq_shmem_management");
|
||||||
|
}
|
||||||
|
catch (bipc::interprocess_exception& ie)
|
||||||
|
{
|
||||||
|
cout << "Did not find \"fairmq_shmem_management\" shared memory segment. No regions to cleanup." << endl;
|
||||||
|
}
|
||||||
|
|
||||||
|
RemoveObject(segmentName);
|
||||||
|
|
||||||
|
boost::interprocess::named_mutex::remove("fairmq_shmem_mutex");
|
||||||
|
}
|
||||||
|
|
||||||
|
void Monitor::RemoveObject(const std::string& name)
|
||||||
|
{
|
||||||
|
if (bipc::shared_memory_object::remove(name.c_str()))
|
||||||
|
{
|
||||||
|
cout << "Successfully removed \"" << name << "\" shared memory segment." << endl;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
cout << "Did not remove shared memory. Already removed?" << endl;
|
cout << "Did not remove \"" << name << "\" shared memory segment. Already removed?" << endl;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void Monitor::CleanupControlQueues()
|
||||||
|
{
|
||||||
|
if (bipc::message_queue::remove("fairmq_shmem_control_queue"))
|
||||||
|
{
|
||||||
|
// cout << "successfully removed control queue" << endl;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// cout << "could not remove control queue" << endl;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -311,19 +398,19 @@ void Monitor::PrintQueues()
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
bipc::managed_shared_memory segment(bipc::open_only, fSegmentName.c_str());
|
bipc::managed_shared_memory segment(bipc::open_only, fSegmentName.c_str());
|
||||||
pair<StringVector*, size_t> queues = segment.find<StringVector>("fairmq_shmem_queues");
|
StringVector* queues = segment.find<StringVector>("fairmq_shmem_queues").first;
|
||||||
if (queues.first != nullptr)
|
if (queues)
|
||||||
{
|
{
|
||||||
cout << "found " << queues.first->size() << " queue(s):" << endl;
|
cout << "found " << queues->size() << " queue(s):" << endl;
|
||||||
|
|
||||||
for (int i = 0; i < queues.first->size(); ++i)
|
for (int i = 0; i < queues->size(); ++i)
|
||||||
{
|
{
|
||||||
string name(queues.first->at(i).c_str());
|
string name(queues->at(i).c_str());
|
||||||
cout << '\t' << name << " : ";
|
cout << '\t' << name << " : ";
|
||||||
pair<atomic<int>*, size_t> queueSize = segment.find<atomic<int>>(name.c_str());
|
atomic<int>* queueSize = segment.find<atomic<int>>(name.c_str()).first;
|
||||||
if (queueSize.first != nullptr)
|
if (queueSize)
|
||||||
{
|
{
|
||||||
cout << *(queueSize.first) << " messages" << endl;
|
cout << *queueSize << " messages" << endl;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
|
@ -368,6 +455,15 @@ void Monitor::PrintHelp()
|
||||||
cout << "controls: [x] close memory, [p] print queues, [h] help, [q] quit." << endl;
|
cout << "controls: [x] close memory, [p] print queues, [h] help, [q] quit." << endl;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Monitor::~Monitor()
|
||||||
|
{
|
||||||
|
fManagementSegment.destroy<MonitorStatus>(bipc::unique_instance);
|
||||||
|
if (fSignalThread.joinable())
|
||||||
|
{
|
||||||
|
fSignalThread.join();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
} // namespace shmem
|
} // namespace shmem
|
||||||
} // namespace mq
|
} // namespace mq
|
||||||
} // namespace fair
|
} // namespace fair
|
||||||
|
|
|
@ -5,8 +5,10 @@
|
||||||
* GNU Lesser General Public Licence version 3 (LGPL) version 3, *
|
* GNU Lesser General Public Licence version 3 (LGPL) version 3, *
|
||||||
* copied verbatim in the file "LICENSE" *
|
* copied verbatim in the file "LICENSE" *
|
||||||
********************************************************************************/
|
********************************************************************************/
|
||||||
#ifndef FAIRMQSHMMONITOR_H_
|
#ifndef FAIR_MQ_SHMEM_MONITOR_H_
|
||||||
#define FAIRMQSHMMONITOR_H_
|
#define FAIR_MQ_SHMEM_MONITOR_H_
|
||||||
|
|
||||||
|
#include <boost/interprocess/managed_shared_memory.hpp>
|
||||||
|
|
||||||
#include <thread>
|
#include <thread>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
@ -28,11 +30,13 @@ class Monitor
|
||||||
Monitor(const Monitor&) = delete;
|
Monitor(const Monitor&) = delete;
|
||||||
Monitor operator=(const Monitor&) = delete;
|
Monitor operator=(const Monitor&) = delete;
|
||||||
|
|
||||||
|
void CatchSignals();
|
||||||
void Run();
|
void Run();
|
||||||
|
|
||||||
virtual ~Monitor() {}
|
virtual ~Monitor();
|
||||||
|
|
||||||
static void Cleanup(const std::string& segmentName);
|
static void Cleanup(const std::string& segmentName);
|
||||||
|
static void CleanupControlQueues();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void PrintHeader();
|
void PrintHeader();
|
||||||
|
@ -41,6 +45,8 @@ class Monitor
|
||||||
void MonitorHeartbeats();
|
void MonitorHeartbeats();
|
||||||
void CheckSegment();
|
void CheckSegment();
|
||||||
void Interactive();
|
void Interactive();
|
||||||
|
void SignalMonitor();
|
||||||
|
static void RemoveObject(const std::string&);
|
||||||
|
|
||||||
bool fSelfDestruct; // will self-destruct after the memory has been closed
|
bool fSelfDestruct; // will self-destruct after the memory has been closed
|
||||||
bool fInteractive; // running in interactive mode
|
bool fInteractive; // running in interactive mode
|
||||||
|
@ -50,11 +56,12 @@ class Monitor
|
||||||
std::atomic<bool> fTerminating;
|
std::atomic<bool> fTerminating;
|
||||||
std::atomic<bool> fHeartbeatTriggered;
|
std::atomic<bool> fHeartbeatTriggered;
|
||||||
std::chrono::high_resolution_clock::time_point fLastHeartbeat;
|
std::chrono::high_resolution_clock::time_point fLastHeartbeat;
|
||||||
std::thread fHeartbeatThread;
|
std::thread fSignalThread;
|
||||||
|
boost::interprocess::managed_shared_memory fManagementSegment;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace shmem
|
} // namespace shmem
|
||||||
} // namespace mq
|
} // namespace mq
|
||||||
} // namespace fair
|
} // namespace fair
|
||||||
|
|
||||||
#endif /* FAIRMQSHMMONITOR_H_ */
|
#endif /* FAIR_MQ_SHMEM_MONITOR_H_ */
|
|
@ -11,7 +11,9 @@
|
||||||
|
|
||||||
#include "FairMQSocketSHM.h"
|
#include "FairMQSocketSHM.h"
|
||||||
#include "FairMQMessageSHM.h"
|
#include "FairMQMessageSHM.h"
|
||||||
|
#include "FairMQRegionSHM.h"
|
||||||
#include "FairMQLogger.h"
|
#include "FairMQLogger.h"
|
||||||
|
#include "FairMQShmCommon.h"
|
||||||
|
|
||||||
using namespace std;
|
using namespace std;
|
||||||
using namespace fair::mq::shmem;
|
using namespace fair::mq::shmem;
|
||||||
|
@ -173,24 +175,22 @@ int FairMQSocketSHM::Receive(FairMQMessagePtr& msg, const int flags)
|
||||||
// ShPtrOwner* owner = Manager::Instance().Segment()->find<ShPtrOwner>(ownerID.c_str()).first;
|
// ShPtrOwner* owner = Manager::Instance().Segment()->find<ShPtrOwner>(ownerID.c_str()).first;
|
||||||
MetaHeader* hdr = static_cast<MetaHeader*>(zmq_msg_data(msgPtr));
|
MetaHeader* hdr = static_cast<MetaHeader*>(zmq_msg_data(msgPtr));
|
||||||
size_t size = 0;
|
size_t size = 0;
|
||||||
if (hdr->fHandle)
|
static_cast<FairMQMessageSHM*>(msg.get())->fHandle = hdr->fHandle;
|
||||||
{
|
static_cast<FairMQMessageSHM*>(msg.get())->fSize = hdr->fSize;
|
||||||
static_cast<FairMQMessageSHM*>(msg.get())->fHandle = hdr->fHandle;
|
static_cast<FairMQMessageSHM*>(msg.get())->fRegionId = hdr->fRegionId;
|
||||||
static_cast<FairMQMessageSHM*>(msg.get())->fChunkSize = hdr->fSize;
|
// static_cast<FairMQMessageSHM*>(msg.get())->fOwner = owner;
|
||||||
// static_cast<FairMQMessageSHM*>(msg.get())->fOwner = owner;
|
// static_cast<FairMQMessageSHM*>(msg.get())->fReceiving = true;
|
||||||
// static_cast<FairMQMessageSHM*>(msg.get())->fReceiving = true;
|
size = msg->GetSize();
|
||||||
size = msg->GetSize();
|
|
||||||
|
|
||||||
fBytesRx += size;
|
fBytesRx += size;
|
||||||
++fMessagesRx;
|
++fMessagesRx;
|
||||||
|
|
||||||
return size;
|
return size;
|
||||||
}
|
// else
|
||||||
else
|
// {
|
||||||
{
|
// LOG(ERROR) << "Received meta data, but could not find corresponding chunk";
|
||||||
LOG(ERROR) << "Received meta data, but could not find corresponding chunk";
|
// return -1;
|
||||||
return -1;
|
// }
|
||||||
}
|
|
||||||
}
|
}
|
||||||
else if (zmq_errno() == EAGAIN)
|
else if (zmq_errno() == EAGAIN)
|
||||||
{
|
{
|
||||||
|
@ -238,7 +238,6 @@ int64_t FairMQSocketSHM::Send(vector<FairMQMessagePtr>& msgVec, const int flags)
|
||||||
{
|
{
|
||||||
static_cast<FairMQMessageSHM*>(msgVec[i].get())->fQueued = true;
|
static_cast<FairMQMessageSHM*>(msgVec[i].get())->fQueued = true;
|
||||||
// static_cast<FairMQMessageSHM*>(msgVec[i].get())->fReceiving = false;
|
// static_cast<FairMQMessageSHM*>(msgVec[i].get())->fReceiving = false;
|
||||||
// static_cast<FairMQMessageSHM*>(msgVec[i].get())->fQueued = true;
|
|
||||||
size_t size = msgVec[i]->GetSize();
|
size_t size = msgVec[i]->GetSize();
|
||||||
|
|
||||||
totalSize += size;
|
totalSize += size;
|
||||||
|
@ -327,23 +326,21 @@ int64_t FairMQSocketSHM::Receive(vector<FairMQMessagePtr>& msgVec, const int fla
|
||||||
// ShPtrOwner* owner = Manager::Instance().Segment()->find<ShPtrOwner>(ownerID.c_str()).first;
|
// ShPtrOwner* owner = Manager::Instance().Segment()->find<ShPtrOwner>(ownerID.c_str()).first;
|
||||||
MetaHeader* hdr = static_cast<MetaHeader*>(zmq_msg_data(msgPtr));
|
MetaHeader* hdr = static_cast<MetaHeader*>(zmq_msg_data(msgPtr));
|
||||||
size_t size = 0;
|
size_t size = 0;
|
||||||
if (hdr->fHandle)
|
static_cast<FairMQMessageSHM*>(part.get())->fHandle = hdr->fHandle;
|
||||||
{
|
static_cast<FairMQMessageSHM*>(part.get())->fSize = hdr->fSize;
|
||||||
static_cast<FairMQMessageSHM*>(part.get())->fHandle = hdr->fHandle;
|
static_cast<FairMQMessageSHM*>(part.get())->fRegionId = hdr->fRegionId;
|
||||||
static_cast<FairMQMessageSHM*>(part.get())->fChunkSize = hdr->fSize;
|
// static_cast<FairMQMessageSHM*>(part.get())->fOwner = owner;
|
||||||
// static_cast<FairMQMessageSHM*>(msg.get())->fOwner = owner;
|
// static_cast<FairMQMessageSHM*>(part.get())->fReceiving = true;
|
||||||
// static_cast<FairMQMessageSHM*>(msg.get())->fReceiving = true;
|
size = part->GetSize();
|
||||||
size = part->GetSize();
|
|
||||||
|
|
||||||
msgVec.push_back(move(part));
|
msgVec.push_back(move(part));
|
||||||
|
|
||||||
totalSize += size;
|
totalSize += size;
|
||||||
}
|
// else
|
||||||
else
|
// {
|
||||||
{
|
// LOG(ERROR) << "Received meta data, but could not find corresponding chunk";
|
||||||
LOG(ERROR) << "Received meta data, but could not find corresponding chunk";
|
// return -1;
|
||||||
return -1;
|
// }
|
||||||
}
|
|
||||||
}
|
}
|
||||||
else if (zmq_errno() == EAGAIN)
|
else if (zmq_errno() == EAGAIN)
|
||||||
{
|
{
|
||||||
|
@ -399,12 +396,14 @@ void FairMQSocketSHM::Close()
|
||||||
void FairMQSocketSHM::Interrupt()
|
void FairMQSocketSHM::Interrupt()
|
||||||
{
|
{
|
||||||
FairMQMessageSHM::fInterrupted = true;
|
FairMQMessageSHM::fInterrupted = true;
|
||||||
|
FairMQRegionSHM::fInterrupted = true;
|
||||||
fInterrupted = true;
|
fInterrupted = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
void FairMQSocketSHM::Resume()
|
void FairMQSocketSHM::Resume()
|
||||||
{
|
{
|
||||||
FairMQMessageSHM::fInterrupted = false;
|
FairMQMessageSHM::fInterrupted = false;
|
||||||
|
FairMQRegionSHM::fInterrupted = true;
|
||||||
fInterrupted = false;
|
fInterrupted = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -13,6 +13,7 @@
|
||||||
#include <zmq.h>
|
#include <zmq.h>
|
||||||
|
|
||||||
#include <boost/version.hpp>
|
#include <boost/version.hpp>
|
||||||
|
#include <boost/filesystem.hpp>
|
||||||
#include <boost/interprocess/managed_shared_memory.hpp>
|
#include <boost/interprocess/managed_shared_memory.hpp>
|
||||||
|
|
||||||
#include <boost/interprocess/ipc/message_queue.hpp>
|
#include <boost/interprocess/ipc/message_queue.hpp>
|
||||||
|
@ -21,10 +22,13 @@
|
||||||
#include <boost/interprocess/sync/scoped_lock.hpp>
|
#include <boost/interprocess/sync/scoped_lock.hpp>
|
||||||
|
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
#include <thread>
|
||||||
|
#include <cstdlib> // std::system
|
||||||
|
|
||||||
using namespace std;
|
using namespace std;
|
||||||
using namespace fair::mq::shmem;
|
using namespace fair::mq::shmem;
|
||||||
namespace bipc = boost::interprocess;
|
namespace bipc = boost::interprocess;
|
||||||
|
namespace bfs = boost::filesystem;
|
||||||
namespace bpt = boost::posix_time;
|
namespace bpt = boost::posix_time;
|
||||||
|
|
||||||
FairMQ::Transport FairMQTransportFactorySHM::fTransportType = FairMQ::Transport::SHM;
|
FairMQ::Transport FairMQTransportFactorySHM::fTransportType = FairMQ::Transport::SHM;
|
||||||
|
@ -75,30 +79,84 @@ FairMQTransportFactorySHM::FairMQTransportFactorySHM(const string& id, const Fai
|
||||||
LOG(ERROR) << "shmem: failed configuring context, reason: " << zmq_strerror(errno);
|
LOG(ERROR) << "shmem: failed configuring context, reason: " << zmq_strerror(errno);
|
||||||
}
|
}
|
||||||
|
|
||||||
fSendHeartbeats = true;
|
|
||||||
fHeartbeatThread = thread(&FairMQTransportFactorySHM::SendHeartbeats, this);
|
|
||||||
|
|
||||||
Manager::Instance().InitializeSegment("open_or_create", segmentName, segmentSize);
|
Manager::Instance().InitializeSegment("open_or_create", segmentName, segmentSize);
|
||||||
LOG(DEBUG) << "shmem: created/opened shared memory segment of " << segmentSize << " bytes. Available are " << Manager::Instance().Segment()->get_free_memory() << " bytes.";
|
LOG(DEBUG) << "shmem: created/opened shared memory segment of " << segmentSize << " bytes. Available are " << Manager::Instance().Segment()->get_free_memory() << " bytes.";
|
||||||
|
|
||||||
{ // mutex scope
|
{
|
||||||
bipc::scoped_lock<bipc::named_mutex> lock(fShMutex);
|
bipc::scoped_lock<bipc::named_mutex> lock(fShMutex);
|
||||||
|
|
||||||
pair<DeviceCounter*, size_t> result = Manager::Instance().Segment()->find<DeviceCounter>(bipc::unique_instance);
|
fDeviceCounter = Manager::Instance().Segment()->find<DeviceCounter>(bipc::unique_instance).first;
|
||||||
if (result.first != nullptr)
|
if (fDeviceCounter)
|
||||||
{
|
{
|
||||||
fDeviceCounter = result.first;
|
LOG(DEBUG) << "shmem: device counter found, with value of " << fDeviceCounter->fCount << ". incrementing.";
|
||||||
LOG(DEBUG) << "shmem: device counter found, with value of " << fDeviceCounter->count << ". incrementing.";
|
(fDeviceCounter->fCount)++;
|
||||||
(fDeviceCounter->count)++;
|
LOG(DEBUG) << "shmem: incremented device counter, now: " << fDeviceCounter->fCount;
|
||||||
LOG(DEBUG) << "shmem: incremented device counter, now: " << fDeviceCounter->count;
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
LOG(DEBUG) << "shmem: no device counter found, creating one and initializing with 1";
|
LOG(DEBUG) << "shmem: no device counter found, creating one and initializing with 1";
|
||||||
fDeviceCounter = Manager::Instance().Segment()->construct<DeviceCounter>(bipc::unique_instance)(1);
|
fDeviceCounter = Manager::Instance().Segment()->construct<DeviceCounter>(bipc::unique_instance)(1);
|
||||||
LOG(DEBUG) << "shmem: initialized device counter with: " << fDeviceCounter->count;
|
LOG(DEBUG) << "shmem: initialized device counter with: " << fDeviceCounter->fCount;
|
||||||
|
}
|
||||||
|
|
||||||
|
// start shm monitor
|
||||||
|
// try
|
||||||
|
// {
|
||||||
|
// MonitorStatus* monitorStatus = fManagementSegment.find<MonitorStatus>(bipc::unique_instance).first;
|
||||||
|
// if (monitorStatus == nullptr)
|
||||||
|
// {
|
||||||
|
// LOG(DEBUG) << "shmem: no shmmonitor found, starting...";
|
||||||
|
// StartMonitor();
|
||||||
|
// }
|
||||||
|
// else
|
||||||
|
// {
|
||||||
|
// LOG(DEBUG) << "shmem: found shmmonitor in fairmq_shmem_management.";
|
||||||
|
// }
|
||||||
|
// }
|
||||||
|
// catch (std::exception& e)
|
||||||
|
// {
|
||||||
|
// LOG(ERROR) << "shmem: Exception during shmmonitor initialization: " << e.what() << ", application will now exit";
|
||||||
|
// exit(EXIT_FAILURE);
|
||||||
|
// }
|
||||||
|
}
|
||||||
|
|
||||||
|
fSendHeartbeats = true;
|
||||||
|
fHeartbeatThread = thread(&FairMQTransportFactorySHM::SendHeartbeats, this);
|
||||||
|
}
|
||||||
|
|
||||||
|
void FairMQTransportFactorySHM::StartMonitor()
|
||||||
|
{
|
||||||
|
int numTries = 0;
|
||||||
|
|
||||||
|
if (!bfs::exists(bfs::path("shmmonitor")))
|
||||||
|
{
|
||||||
|
LOG(ERROR) << "Could not find shmmonitor. Is it in the PATH? Monitor not started";
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: replace with Boost.Process once boost 1.64 is available
|
||||||
|
int r = system("shmmonitor --self-destruct &");
|
||||||
|
LOG(DEBUG) << r;
|
||||||
|
|
||||||
|
do
|
||||||
|
{
|
||||||
|
MonitorStatus* monitorStatus = Manager::Instance().ManagementSegment().find<MonitorStatus>(bipc::unique_instance).first;
|
||||||
|
if (monitorStatus)
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "shmem: shmmonitor started";
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
this_thread::sleep_for(std::chrono::milliseconds(10));
|
||||||
|
if (++numTries > 100)
|
||||||
|
{
|
||||||
|
LOG(ERROR) << "Did not get response from shmmonitor after " << 10 * 100 << " milliseconds. Exiting.";
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
while (true);
|
||||||
}
|
}
|
||||||
|
|
||||||
void FairMQTransportFactorySHM::SendHeartbeats()
|
void FairMQTransportFactorySHM::SendHeartbeats()
|
||||||
|
@ -142,6 +200,11 @@ FairMQMessagePtr FairMQTransportFactorySHM::CreateMessage(void* data, const size
|
||||||
return unique_ptr<FairMQMessage>(new FairMQMessageSHM(data, size, ffn, hint));
|
return unique_ptr<FairMQMessage>(new FairMQMessageSHM(data, size, ffn, hint));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQMessagePtr FairMQTransportFactorySHM::CreateMessage(FairMQRegionPtr& region, void* data, const size_t size) const
|
||||||
|
{
|
||||||
|
return unique_ptr<FairMQMessage>(new FairMQMessageSHM(region, data, size));
|
||||||
|
}
|
||||||
|
|
||||||
FairMQSocketPtr FairMQTransportFactorySHM::CreateSocket(const string& type, const string& name) const
|
FairMQSocketPtr FairMQTransportFactorySHM::CreateSocket(const string& type, const string& name) const
|
||||||
{
|
{
|
||||||
assert(fContext);
|
assert(fContext);
|
||||||
|
@ -168,6 +231,11 @@ FairMQPollerPtr FairMQTransportFactorySHM::CreatePoller(const FairMQSocket& cmdS
|
||||||
return unique_ptr<FairMQPoller>(new FairMQPollerSHM(cmdSocket, dataSocket));
|
return unique_ptr<FairMQPoller>(new FairMQPollerSHM(cmdSocket, dataSocket));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQRegionPtr FairMQTransportFactorySHM::CreateRegion(const size_t size) const
|
||||||
|
{
|
||||||
|
return unique_ptr<FairMQRegion>(new FairMQRegionSHM(size));
|
||||||
|
}
|
||||||
|
|
||||||
FairMQTransportFactorySHM::~FairMQTransportFactorySHM()
|
FairMQTransportFactorySHM::~FairMQTransportFactorySHM()
|
||||||
{
|
{
|
||||||
fSendHeartbeats = false;
|
fSendHeartbeats = false;
|
||||||
|
@ -196,24 +264,17 @@ FairMQTransportFactorySHM::~FairMQTransportFactorySHM()
|
||||||
{ // mutex scope
|
{ // mutex scope
|
||||||
bipc::scoped_lock<bipc::named_mutex> lock(fShMutex);
|
bipc::scoped_lock<bipc::named_mutex> lock(fShMutex);
|
||||||
|
|
||||||
(fDeviceCounter->count)--;
|
(fDeviceCounter->fCount)--;
|
||||||
|
|
||||||
if (fDeviceCounter->count == 0)
|
if (fDeviceCounter->fCount == 0)
|
||||||
{
|
{
|
||||||
LOG(DEBUG) << "shmem: last 'fairmq_shmem_main' user, removing segment.";
|
LOG(DEBUG) << "shmem: last 'fairmq_shmem_main' user, removing segment.";
|
||||||
|
|
||||||
if (bipc::shared_memory_object::remove("fairmq_shmem_main"))
|
Manager::Instance().Remove();
|
||||||
{
|
|
||||||
LOG(DEBUG) << "shmem: successfully removed shared memory segment after the device has stopped.";
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
LOG(DEBUG) << "shmem: did not remove shared memory segment after the device stopped. Already removed?";
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
LOG(DEBUG) << "shmem: other 'fairmq_shmem_main' users present (" << fDeviceCounter->count << "), not removing it.";
|
LOG(DEBUG) << "shmem: other 'fairmq_shmem_main' users present (" << fDeviceCounter->fCount << "), not removing it.";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -13,8 +13,9 @@
|
||||||
#include "FairMQMessageSHM.h"
|
#include "FairMQMessageSHM.h"
|
||||||
#include "FairMQSocketSHM.h"
|
#include "FairMQSocketSHM.h"
|
||||||
#include "FairMQPollerSHM.h"
|
#include "FairMQPollerSHM.h"
|
||||||
#include "FairMQShmDeviceCounter.h"
|
#include "FairMQShmCommon.h"
|
||||||
#include <options/FairMQProgOptions.h>
|
#include <options/FairMQProgOptions.h>
|
||||||
|
#include "FairMQRegionSHM.h"
|
||||||
|
|
||||||
#include <vector>
|
#include <vector>
|
||||||
#include <string>
|
#include <string>
|
||||||
|
@ -23,15 +24,17 @@
|
||||||
|
|
||||||
#include <boost/interprocess/sync/named_mutex.hpp>
|
#include <boost/interprocess/sync/named_mutex.hpp>
|
||||||
|
|
||||||
|
|
||||||
class FairMQTransportFactorySHM : public FairMQTransportFactory
|
class FairMQTransportFactorySHM : public FairMQTransportFactory
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
FairMQTransportFactorySHM(const std::string& id = "", const FairMQProgOptions* config = nullptr);
|
FairMQTransportFactorySHM(const std::string& id = "", const FairMQProgOptions* config = nullptr);
|
||||||
|
FairMQTransportFactorySHM(const FairMQTransportFactorySHM&) = delete;
|
||||||
|
FairMQTransportFactorySHM operator=(const FairMQTransportFactorySHM&) = delete;
|
||||||
|
|
||||||
FairMQMessagePtr CreateMessage() const override;
|
FairMQMessagePtr CreateMessage() const override;
|
||||||
FairMQMessagePtr CreateMessage(const size_t size) const override;
|
FairMQMessagePtr CreateMessage(const size_t size) const override;
|
||||||
FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const override;
|
FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const override;
|
||||||
|
FairMQMessagePtr CreateMessage(FairMQRegionPtr& region, void* data, const size_t size) const override;
|
||||||
|
|
||||||
FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const override;
|
FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const override;
|
||||||
|
|
||||||
|
@ -40,13 +43,16 @@ class FairMQTransportFactorySHM : public FairMQTransportFactory
|
||||||
FairMQPollerPtr CreatePoller(const std::unordered_map<std::string, std::vector<FairMQChannel>>& channelsMap, const std::vector<std::string>& channelList) const override;
|
FairMQPollerPtr CreatePoller(const std::unordered_map<std::string, std::vector<FairMQChannel>>& channelsMap, const std::vector<std::string>& channelList) const override;
|
||||||
FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const override;
|
FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const override;
|
||||||
|
|
||||||
FairMQ::Transport GetType() const override;
|
FairMQRegionPtr CreateRegion(const size_t size) const override;
|
||||||
|
|
||||||
void SendHeartbeats();
|
FairMQ::Transport GetType() const override;
|
||||||
|
|
||||||
~FairMQTransportFactorySHM() override;
|
~FairMQTransportFactorySHM() override;
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
void SendHeartbeats();
|
||||||
|
void StartMonitor();
|
||||||
|
|
||||||
static FairMQ::Transport fTransportType;
|
static FairMQ::Transport fTransportType;
|
||||||
void* fContext;
|
void* fContext;
|
||||||
void* fHeartbeatSocket;
|
void* fHeartbeatSocket;
|
||||||
|
|
|
@ -6,5 +6,3 @@ The transport manages shared memory via boost::interprocess library. The transfe
|
||||||
|
|
||||||
Under development:
|
Under development:
|
||||||
- Cleanup of the shared memory segment in case all devices crash. Currently at least one device has to stop properly for a cleanup.
|
- Cleanup of the shared memory segment in case all devices crash. Currently at least one device has to stop properly for a cleanup.
|
||||||
- Implement more than one transport per device.
|
|
||||||
- Configuration of the shared memory size (currently hard-coded).
|
|
||||||
|
|
|
@ -49,6 +49,7 @@ int main(int argc, char** argv)
|
||||||
{
|
{
|
||||||
cout << "Cleaning up \"" << segmentName << "\"..." << endl;
|
cout << "Cleaning up \"" << segmentName << "\"..." << endl;
|
||||||
fair::mq::shmem::Monitor::Cleanup(segmentName);
|
fair::mq::shmem::Monitor::Cleanup(segmentName);
|
||||||
|
fair::mq::shmem::Monitor::CleanupControlQueues();
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -56,6 +57,7 @@ int main(int argc, char** argv)
|
||||||
|
|
||||||
fair::mq::shmem::Monitor monitor{segmentName, selfDestruct, interactive, timeoutInMS};
|
fair::mq::shmem::Monitor monitor{segmentName, selfDestruct, interactive, timeoutInMS};
|
||||||
|
|
||||||
|
monitor.CatchSignals();
|
||||||
monitor.Run();
|
monitor.Run();
|
||||||
}
|
}
|
||||||
catch (exception& e)
|
catch (exception& e)
|
||||||
|
|
|
@ -50,6 +50,25 @@ FairMQMessageZMQ::FairMQMessageZMQ(void* data, const size_t size, fairmq_free_fn
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQMessageZMQ::FairMQMessageZMQ(FairMQRegionPtr& region, void* data, const size_t size)
|
||||||
|
: fMessage()
|
||||||
|
{
|
||||||
|
// FIXME: make this zero-copy:
|
||||||
|
// simply taking over the provided buffer can casue premature delete, since region could be destroyed before the message is sent out.
|
||||||
|
// Needs lifetime extension for the ZMQ region.
|
||||||
|
if (zmq_msg_init_size(&fMessage, size) != 0)
|
||||||
|
{
|
||||||
|
LOG(ERROR) << "failed initializing message with size, reason: " << zmq_strerror(errno);
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy(zmq_msg_data(&fMessage), data, size);
|
||||||
|
|
||||||
|
// if (zmq_msg_init_data(&fMessage, data, size, [](void*, void*){}, nullptr) != 0)
|
||||||
|
// {
|
||||||
|
// LOG(ERROR) << "failed initializing message with data, reason: " << zmq_strerror(errno);
|
||||||
|
// }
|
||||||
|
}
|
||||||
|
|
||||||
void FairMQMessageZMQ::Rebuild()
|
void FairMQMessageZMQ::Rebuild()
|
||||||
{
|
{
|
||||||
CloseMessage();
|
CloseMessage();
|
||||||
|
|
|
@ -21,6 +21,7 @@
|
||||||
#include <zmq.h>
|
#include <zmq.h>
|
||||||
|
|
||||||
#include "FairMQMessage.h"
|
#include "FairMQMessage.h"
|
||||||
|
#include "FairMQRegion.h"
|
||||||
|
|
||||||
class FairMQMessageZMQ : public FairMQMessage
|
class FairMQMessageZMQ : public FairMQMessage
|
||||||
{
|
{
|
||||||
|
@ -28,6 +29,7 @@ class FairMQMessageZMQ : public FairMQMessage
|
||||||
FairMQMessageZMQ();
|
FairMQMessageZMQ();
|
||||||
FairMQMessageZMQ(const size_t size);
|
FairMQMessageZMQ(const size_t size);
|
||||||
FairMQMessageZMQ(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr);
|
FairMQMessageZMQ(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr);
|
||||||
|
FairMQMessageZMQ(FairMQRegionPtr& region, void* data, const size_t size);
|
||||||
|
|
||||||
virtual void Rebuild();
|
virtual void Rebuild();
|
||||||
virtual void Rebuild(const size_t size);
|
virtual void Rebuild(const size_t size);
|
||||||
|
|
34
fairmq/zeromq/FairMQRegionZMQ.cxx
Normal file
34
fairmq/zeromq/FairMQRegionZMQ.cxx
Normal file
|
@ -0,0 +1,34 @@
|
||||||
|
/********************************************************************************
|
||||||
|
* 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" *
|
||||||
|
********************************************************************************/
|
||||||
|
|
||||||
|
#include "FairMQRegionZMQ.h"
|
||||||
|
#include "FairMQLogger.h"
|
||||||
|
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
FairMQRegionZMQ::FairMQRegionZMQ(const size_t size)
|
||||||
|
: fBuffer(malloc(size))
|
||||||
|
, fSize(size)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
void* FairMQRegionZMQ::GetData() const
|
||||||
|
{
|
||||||
|
return fBuffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
size_t FairMQRegionZMQ::GetSize() const
|
||||||
|
{
|
||||||
|
return fSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
FairMQRegionZMQ::~FairMQRegionZMQ()
|
||||||
|
{
|
||||||
|
LOG(DEBUG) << "destroying region";
|
||||||
|
free(fBuffer);
|
||||||
|
}
|
35
fairmq/zeromq/FairMQRegionZMQ.h
Normal file
35
fairmq/zeromq/FairMQRegionZMQ.h
Normal file
|
@ -0,0 +1,35 @@
|
||||||
|
/********************************************************************************
|
||||||
|
* 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" *
|
||||||
|
********************************************************************************/
|
||||||
|
|
||||||
|
#ifndef FAIRMQREGIONZMQ_H_
|
||||||
|
#define FAIRMQREGIONZMQ_H_
|
||||||
|
|
||||||
|
#include "FairMQRegion.h"
|
||||||
|
|
||||||
|
#include <cstddef> // size_t
|
||||||
|
|
||||||
|
class FairMQRegionZMQ : public FairMQRegion
|
||||||
|
{
|
||||||
|
friend class FairMQSocketSHM;
|
||||||
|
|
||||||
|
public:
|
||||||
|
FairMQRegionZMQ(const size_t size);
|
||||||
|
FairMQRegionZMQ(const FairMQRegionZMQ&) = delete;
|
||||||
|
FairMQRegionZMQ operator=(const FairMQRegionZMQ&) = delete;
|
||||||
|
|
||||||
|
virtual void* GetData() const override;
|
||||||
|
virtual size_t GetSize() const override;
|
||||||
|
|
||||||
|
virtual ~FairMQRegionZMQ();
|
||||||
|
|
||||||
|
private:
|
||||||
|
void* fBuffer;
|
||||||
|
size_t fSize;
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif /* FAIRMQREGIONZMQ_H_ */
|
|
@ -14,8 +14,8 @@ using namespace std;
|
||||||
FairMQ::Transport FairMQTransportFactoryZMQ::fTransportType = FairMQ::Transport::ZMQ;
|
FairMQ::Transport FairMQTransportFactoryZMQ::fTransportType = FairMQ::Transport::ZMQ;
|
||||||
|
|
||||||
FairMQTransportFactoryZMQ::FairMQTransportFactoryZMQ(const string& id, const FairMQProgOptions* config)
|
FairMQTransportFactoryZMQ::FairMQTransportFactoryZMQ(const string& id, const FairMQProgOptions* config)
|
||||||
: FairMQTransportFactory(id)
|
: FairMQTransportFactory(id)
|
||||||
, fContext(zmq_ctx_new())
|
, fContext(zmq_ctx_new())
|
||||||
{
|
{
|
||||||
int major, minor, patch;
|
int major, minor, patch;
|
||||||
zmq_version(&major, &minor, &patch);
|
zmq_version(&major, &minor, &patch);
|
||||||
|
@ -64,6 +64,11 @@ FairMQMessagePtr FairMQTransportFactoryZMQ::CreateMessage(void* data, const size
|
||||||
return unique_ptr<FairMQMessage>(new FairMQMessageZMQ(data, size, ffn, hint));
|
return unique_ptr<FairMQMessage>(new FairMQMessageZMQ(data, size, ffn, hint));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQMessagePtr FairMQTransportFactoryZMQ::CreateMessage(FairMQRegionPtr& region, void* data, const size_t size) const
|
||||||
|
{
|
||||||
|
return unique_ptr<FairMQMessage>(new FairMQMessageZMQ(region, data, size));
|
||||||
|
}
|
||||||
|
|
||||||
FairMQSocketPtr FairMQTransportFactoryZMQ::CreateSocket(const string& type, const string& name) const
|
FairMQSocketPtr FairMQTransportFactoryZMQ::CreateSocket(const string& type, const string& name) const
|
||||||
{
|
{
|
||||||
assert(fContext);
|
assert(fContext);
|
||||||
|
@ -90,6 +95,11 @@ FairMQPollerPtr FairMQTransportFactoryZMQ::CreatePoller(const FairMQSocket& cmdS
|
||||||
return unique_ptr<FairMQPoller>(new FairMQPollerZMQ(cmdSocket, dataSocket));
|
return unique_ptr<FairMQPoller>(new FairMQPollerZMQ(cmdSocket, dataSocket));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FairMQRegionPtr FairMQTransportFactoryZMQ::CreateRegion(const size_t size) const
|
||||||
|
{
|
||||||
|
return unique_ptr<FairMQRegion>(new FairMQRegionZMQ(size));
|
||||||
|
}
|
||||||
|
|
||||||
FairMQ::Transport FairMQTransportFactoryZMQ::GetType() const
|
FairMQ::Transport FairMQTransportFactoryZMQ::GetType() const
|
||||||
{
|
{
|
||||||
return fTransportType;
|
return fTransportType;
|
||||||
|
@ -114,6 +124,6 @@ FairMQTransportFactoryZMQ::~FairMQTransportFactoryZMQ()
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
LOG(ERROR) << "shmem: Terminate(): context now available for shutdown";
|
LOG(ERROR) << "Terminate(): context not available for shutdown";
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -22,17 +22,22 @@
|
||||||
#include "FairMQMessageZMQ.h"
|
#include "FairMQMessageZMQ.h"
|
||||||
#include "FairMQSocketZMQ.h"
|
#include "FairMQSocketZMQ.h"
|
||||||
#include "FairMQPollerZMQ.h"
|
#include "FairMQPollerZMQ.h"
|
||||||
|
#include "FairMQRegionZMQ.h"
|
||||||
#include <options/FairMQProgOptions.h>
|
#include <options/FairMQProgOptions.h>
|
||||||
|
|
||||||
class FairMQTransportFactoryZMQ : public FairMQTransportFactory
|
class FairMQTransportFactoryZMQ : public FairMQTransportFactory
|
||||||
{
|
{
|
||||||
public:
|
public:
|
||||||
FairMQTransportFactoryZMQ(const std::string& id = "", const FairMQProgOptions* config = nullptr);
|
FairMQTransportFactoryZMQ(const std::string& id = "", const FairMQProgOptions* config = nullptr);
|
||||||
|
FairMQTransportFactoryZMQ(const FairMQTransportFactoryZMQ&) = delete;
|
||||||
|
FairMQTransportFactoryZMQ operator=(const FairMQTransportFactoryZMQ&) = delete;
|
||||||
|
|
||||||
~FairMQTransportFactoryZMQ() override;
|
~FairMQTransportFactoryZMQ() override;
|
||||||
|
|
||||||
FairMQMessagePtr CreateMessage() const override;
|
FairMQMessagePtr CreateMessage() const override;
|
||||||
FairMQMessagePtr CreateMessage(const size_t size) const override;
|
FairMQMessagePtr CreateMessage(const size_t size) const override;
|
||||||
FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const override;
|
FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) const override;
|
||||||
|
FairMQMessagePtr CreateMessage(FairMQRegionPtr& region, void* data, const size_t size) const override;
|
||||||
|
|
||||||
FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const override;
|
FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const override;
|
||||||
|
|
||||||
|
@ -41,6 +46,8 @@ class FairMQTransportFactoryZMQ : public FairMQTransportFactory
|
||||||
FairMQPollerPtr CreatePoller(const std::unordered_map<std::string, std::vector<FairMQChannel>>& channelsMap, const std::vector<std::string>& channelList) const override;
|
FairMQPollerPtr CreatePoller(const std::unordered_map<std::string, std::vector<FairMQChannel>>& channelsMap, const std::vector<std::string>& channelList) const override;
|
||||||
FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const override;
|
FairMQPollerPtr CreatePoller(const FairMQSocket& cmdSocket, const FairMQSocket& dataSocket) const override;
|
||||||
|
|
||||||
|
FairMQRegionPtr CreateRegion(const size_t size) const override;
|
||||||
|
|
||||||
FairMQ::Transport GetType() const override;
|
FairMQ::Transport GetType() const override;
|
||||||
private:
|
private:
|
||||||
static FairMQ::Transport fTransportType;
|
static FairMQ::Transport fTransportType;
|
||||||
|
|
Loading…
Reference in New Issue
Block a user