mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 08:41:16 +00:00
d40bbfe208
the patch seems big but most of it is just propagating the new notion of constness of the factory - since it keeps track of created messages with the internal allocator it no longer is const
73 lines
3.0 KiB
C++
73 lines
3.0 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2016-2017 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
|
|
#ifndef FAIRMQTRANSPORTFACTORYSHM_H_
|
|
#define FAIRMQTRANSPORTFACTORYSHM_H_
|
|
|
|
#include <fairmq/shmem/Manager.h>
|
|
#include <fairmq/shmem/Common.h>
|
|
|
|
#include "FairMQTransportFactory.h"
|
|
#include "FairMQMessageSHM.h"
|
|
#include "FairMQSocketSHM.h"
|
|
#include "FairMQPollerSHM.h"
|
|
#include "FairMQUnmanagedRegionSHM.h"
|
|
#include <options/FairMQProgOptions.h>
|
|
|
|
#include <boost/interprocess/sync/named_mutex.hpp>
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <atomic>
|
|
|
|
class FairMQTransportFactorySHM final : public FairMQTransportFactory
|
|
{
|
|
public:
|
|
FairMQTransportFactorySHM(const std::string& id = "", const FairMQProgOptions* config = nullptr);
|
|
FairMQTransportFactorySHM(const FairMQTransportFactorySHM&) = delete;
|
|
FairMQTransportFactorySHM operator=(const FairMQTransportFactorySHM&) = delete;
|
|
|
|
FairMQMessagePtr CreateMessage() override;
|
|
FairMQMessagePtr CreateMessage(const size_t size) override;
|
|
FairMQMessagePtr CreateMessage(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) override;
|
|
FairMQMessagePtr CreateMessage(FairMQUnmanagedRegionPtr& region, void* data, const size_t size, void* hint = 0) override;
|
|
|
|
FairMQSocketPtr CreateSocket(const std::string& type, const std::string& name) const override;
|
|
|
|
FairMQPollerPtr CreatePoller(const std::vector<FairMQChannel>& channels) const override;
|
|
FairMQPollerPtr CreatePoller(const std::vector<FairMQChannel*>& channels) const override;
|
|
FairMQPollerPtr CreatePoller(const std::unordered_map<std::string, std::vector<FairMQChannel>>& channelsMap, const std::vector<std::string>& channelList) const override;
|
|
|
|
FairMQUnmanagedRegionPtr CreateUnmanagedRegion(const size_t size, FairMQRegionCallback callback = nullptr) const override;
|
|
|
|
fair::mq::Transport GetType() const override;
|
|
|
|
void Interrupt() override { FairMQSocketSHM::Interrupt(); }
|
|
void Resume() override { FairMQSocketSHM::Resume(); }
|
|
void Reset() override {}
|
|
|
|
~FairMQTransportFactorySHM() override;
|
|
|
|
private:
|
|
void SendHeartbeats();
|
|
void StartMonitor();
|
|
|
|
static fair::mq::Transport fTransportType;
|
|
std::string fDeviceId;
|
|
std::string fShmId;
|
|
void* fContext;
|
|
std::thread fHeartbeatThread;
|
|
std::atomic<bool> fSendHeartbeats;
|
|
std::unique_ptr<boost::interprocess::named_mutex> fShMutex;
|
|
fair::mq::shmem::DeviceCounter* fDeviceCounter;
|
|
std::unique_ptr<fair::mq::shmem::Manager> fManager;
|
|
};
|
|
|
|
#endif /* FAIRMQTRANSPORTFACTORYSHM_H_ */
|