88 lines
3.0 KiB
C++
88 lines
3.0 KiB
C++
#include "ResourceValueUpdate.h"
|
|
|
|
#include "common/validator/MessageValidator.h"
|
|
#include "Dispatcher.h"
|
|
#include "fd/NetworkFailureDetector.h"
|
|
namespace ubeeme {
|
|
namespace moversight {
|
|
|
|
/**
|
|
* @brief Constructor
|
|
* @param peerResources - resource value of the peer
|
|
* @param peerID - Id of the peer which will tell his new resource value
|
|
*/
|
|
ResourceValueUpdate::ResourceValueUpdate(const PeerResources & peerResources, PeerID aPeerID) : MulticastMessage("RVU", RVU), resources(peerResources), peerID(aPeerID) {
|
|
}
|
|
|
|
/**
|
|
* @brief Deconstructor
|
|
*/
|
|
ResourceValueUpdate::~ResourceValueUpdate() {
|
|
//dtor
|
|
}
|
|
|
|
/**
|
|
* @brief Copy constructor
|
|
* @param other Object to copy from
|
|
*/
|
|
ResourceValueUpdate::ResourceValueUpdate(const ResourceValueUpdate& other) : MulticastMessage(other) {
|
|
operator=(other);
|
|
//copy ctor
|
|
}
|
|
|
|
/**
|
|
* @brief Assignment operator
|
|
* @param other Object to assign from
|
|
* @return A reference to this
|
|
*/
|
|
ResourceValueUpdate & ResourceValueUpdate::operator=(const ResourceValueUpdate& rhs) {
|
|
if (this == &rhs) return *this; // handle self assignment
|
|
|
|
resources = rhs.resources;
|
|
peerID = rhs.peerID;
|
|
//assignment operator
|
|
return *this;
|
|
}
|
|
|
|
/**
|
|
* @brief method duplicate this messages
|
|
* @return ResourceValueUpdate
|
|
*/
|
|
ResourceValueUpdate *
|
|
ResourceValueUpdate::dup() const {
|
|
return new ResourceValueUpdate(*this);
|
|
}
|
|
|
|
/**
|
|
* @brief Handles this message. To handle a message, it has first successfully validated.
|
|
* @param dis A reference to the dispatcher, used to forward this message to the handling modules.
|
|
*/
|
|
void
|
|
ResourceValueUpdate::handleReceive(Dispatcher& dis) {
|
|
if (dis.getMessageValidator()->isValid(*this)) {
|
|
//notify the NFD about ongoing traffic
|
|
dis.getNetworkFailureDetector().handleMessage(this);
|
|
//handle the message
|
|
dis.getMessageTransfer().handleMessage(this);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @brief Handles the message delivering. To deliver a message, the message to each module which is interested in this message.
|
|
* @param dis A reference to the dispatcher, used to forward this message to the handling modules.
|
|
* @param missedPeers A list of peers, which have missed this message, as they are pending.
|
|
*/
|
|
void
|
|
ResourceValueUpdate::handleDeliver(Dispatcher& dis, const PeerIDList& missedPeers) {
|
|
MemberRegister & memberReg = dis.getMembershipService().getCurrentMemberRegister();
|
|
|
|
if (memberReg.contains(peerID)) {
|
|
Peer & peer = memberReg.getPeer(peerID);
|
|
peer.setPeerResources(resources);
|
|
memberReg.printMemberRegister();
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|