mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-15 17:41:45 +00:00
FairMQ: Introduce callbacks for the FairMQUnmanagedRegion.
Callbacks are called when the data buffer of the message assiciated with the corresponding region is no longer needed by the transport. Example in examples/advanced/Region/
This commit is contained in:
committed by
Mohammad Al-Turany
parent
378c47c5e5
commit
58a312b730
@@ -11,6 +11,7 @@
|
||||
|
||||
#include "FairMQUnmanagedRegion.h"
|
||||
#include "FairMQLogger.h"
|
||||
#include "Manager.h"
|
||||
|
||||
#include <boost/interprocess/shared_memory_object.hpp>
|
||||
#include <boost/interprocess/mapped_region.hpp>
|
||||
@@ -21,59 +22,23 @@
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
|
||||
struct RemoteRegion // todo: better name?
|
||||
{
|
||||
RemoteRegion(std::string regionIdStr, uint64_t size)
|
||||
: fRegionName(regionIdStr)
|
||||
, fShmemObject(boost::interprocess::open_or_create, regionIdStr.c_str(), boost::interprocess::read_write)
|
||||
{
|
||||
if (size > 0)
|
||||
{
|
||||
fShmemObject.truncate(size);
|
||||
}
|
||||
fRegion = boost::interprocess::mapped_region(fShmemObject, boost::interprocess::read_write); // TODO: add HUGEPAGES flag here
|
||||
}
|
||||
|
||||
RemoteRegion() = delete;
|
||||
|
||||
RemoteRegion(const RemoteRegion& rr) = default;
|
||||
RemoteRegion(RemoteRegion&& rr) = default;
|
||||
|
||||
~RemoteRegion()
|
||||
{
|
||||
if (boost::interprocess::shared_memory_object::remove(fRegionName.c_str()))
|
||||
{
|
||||
LOG(DEBUG) << "destroyed region " << fRegionName;
|
||||
}
|
||||
}
|
||||
|
||||
std::string fRegionName;
|
||||
boost::interprocess::shared_memory_object fShmemObject;
|
||||
boost::interprocess::mapped_region fRegion;
|
||||
};
|
||||
|
||||
class FairMQUnmanagedRegionSHM : public FairMQUnmanagedRegion
|
||||
{
|
||||
friend class FairMQSocketSHM;
|
||||
friend class FairMQMessageSHM;
|
||||
|
||||
public:
|
||||
FairMQUnmanagedRegionSHM(const size_t size);
|
||||
FairMQUnmanagedRegionSHM(fair::mq::shmem::Manager& manager, const size_t size, FairMQRegionCallback callback = nullptr);
|
||||
|
||||
virtual void* GetData() const override;
|
||||
virtual size_t GetSize() const override;
|
||||
void* GetData() const override;
|
||||
size_t GetSize() const override;
|
||||
|
||||
static boost::interprocess::mapped_region* GetRemoteRegion(uint64_t regionId);
|
||||
|
||||
virtual ~FairMQUnmanagedRegionSHM();
|
||||
~FairMQUnmanagedRegionSHM() override;
|
||||
|
||||
private:
|
||||
static std::atomic<bool> fInterrupted;
|
||||
fair::mq::shmem::Manager& fManager;
|
||||
boost::interprocess::mapped_region* fRegion;
|
||||
uint64_t fRegionId;
|
||||
std::string fRegionIdStr;
|
||||
bool fRemote;
|
||||
static std::unordered_map<uint64_t, RemoteRegion> fRemoteRegionMap;
|
||||
};
|
||||
|
||||
#endif /* FAIRMQUNMANAGEDREGIONSHM_H_ */
|
Reference in New Issue
Block a user