100 lines
3.0 KiB
C++
100 lines
3.0 KiB
C++
|
#include <framework/ipc/MessageQueueSender.h>
|
||
|
#include <framework/health/HealthHelper.h>
|
||
|
#include <framework/ipc/MessageQueueSender.h>
|
||
|
#include <framework/serviceinterface/ServiceInterfaceStream.h>
|
||
|
HealthHelper::HealthHelper(HasHealthIF* owner, object_id_t objectId) :
|
||
|
healthTable(NULL), eventSender(NULL), objectId(objectId), parentQueue(
|
||
|
0), owner(owner) {
|
||
|
}
|
||
|
|
||
|
HealthHelper::~HealthHelper() {
|
||
|
}
|
||
|
|
||
|
ReturnValue_t HealthHelper::handleHealthCommand(CommandMessage* message) {
|
||
|
switch (message->getCommand()) {
|
||
|
case HealthMessage::HEALTH_SET:
|
||
|
handleSetHealthCommand(message);
|
||
|
return HasReturnvaluesIF::RETURN_OK;
|
||
|
case HealthMessage::HEALTH_ANNOUNCE: {
|
||
|
eventSender->forwardEvent(HasHealthIF::HEALTH_INFO, getHealth(),
|
||
|
getHealth());
|
||
|
}
|
||
|
return HasReturnvaluesIF::RETURN_OK;
|
||
|
default:
|
||
|
return HasReturnvaluesIF::RETURN_FAILED;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
HasHealthIF::HealthState HealthHelper::getHealth() {
|
||
|
return healthTable->getHealth(objectId);
|
||
|
}
|
||
|
|
||
|
ReturnValue_t HealthHelper::initialize(MessageQueueId_t parentQueue) {
|
||
|
setParentQeueue(parentQueue);
|
||
|
return initialize();
|
||
|
}
|
||
|
|
||
|
void HealthHelper::setParentQeueue(MessageQueueId_t parentQueue) {
|
||
|
this->parentQueue = parentQueue;
|
||
|
}
|
||
|
|
||
|
ReturnValue_t HealthHelper::initialize() {
|
||
|
healthTable = objectManager->get<HealthTableIF>(objects::HEALTH_TABLE);
|
||
|
eventSender = objectManager->get<EventReportingProxyIF>(objectId);
|
||
|
if ((healthTable == NULL) || eventSender == NULL) {
|
||
|
return HasReturnvaluesIF::RETURN_FAILED;
|
||
|
}
|
||
|
ReturnValue_t result = healthTable->registerObject(objectId,
|
||
|
HasHealthIF::HEALTHY);
|
||
|
if (result != HasReturnvaluesIF::RETURN_OK) {
|
||
|
return result;
|
||
|
}
|
||
|
return HasReturnvaluesIF::RETURN_OK;
|
||
|
}
|
||
|
|
||
|
void HealthHelper::setHealth(HasHealthIF::HealthState health) {
|
||
|
HasHealthIF::HealthState oldHealth = getHealth();
|
||
|
eventSender->forwardEvent(HasHealthIF::HEALTH_INFO, health, oldHealth);
|
||
|
if (health != oldHealth) {
|
||
|
healthTable->setHealth(objectId, health);
|
||
|
informParent(health, oldHealth);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void HealthHelper::informParent(HasHealthIF::HealthState health,
|
||
|
HasHealthIF::HealthState oldHealth) {
|
||
|
if (parentQueue == 0) {
|
||
|
return;
|
||
|
}
|
||
|
CommandMessage message;
|
||
|
MessageQueueSender sender(parentQueue);
|
||
|
HealthMessage::setHealthMessage(&message, HealthMessage::HEALTH_INFO,
|
||
|
health, oldHealth);
|
||
|
if (sender.sendToDefault(&message, owner->getCommandQueue())
|
||
|
!= HasReturnvaluesIF::RETURN_OK) {
|
||
|
debug << "HealthHelper::informParent: sending health reply failed."
|
||
|
<< std::endl;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void HealthHelper::handleSetHealthCommand(CommandMessage* message) {
|
||
|
ReturnValue_t result = owner->setHealth(HealthMessage::getHealth(message));
|
||
|
if (message->getSender() == 0) {
|
||
|
return;
|
||
|
}
|
||
|
CommandMessage reply;
|
||
|
if (result == HasReturnvaluesIF::RETURN_OK) {
|
||
|
HealthMessage::setHealthMessage(&reply,
|
||
|
HealthMessage::REPLY_HEALTH_SET);
|
||
|
} else {
|
||
|
reply.setReplyRejected(result, message->getCommand());
|
||
|
}
|
||
|
MessageQueueSender sender(message->getSender());
|
||
|
if (sender.sendToDefault(&reply, owner->getCommandQueue())
|
||
|
!= HasReturnvaluesIF::RETURN_OK) {
|
||
|
debug
|
||
|
<< "HealthHelper::handleHealthCommand: sending health reply failed."
|
||
|
<< std::endl;
|
||
|
}
|
||
|
}
|