146 lines
5.0 KiB
C++
146 lines
5.0 KiB
C++
#include "fsfw/events/eventmatching/EventMatchTree.h"
|
|
|
|
#include "fsfw/events/eventmatching/EventIdRangeMatcher.h"
|
|
#include "fsfw/events/eventmatching/ReporterRangeMatcher.h"
|
|
#include "fsfw/events/eventmatching/SeverityRangeMatcher.h"
|
|
|
|
EventMatchTree::EventMatchTree(StorageManagerIF* storageBackend, bool invertedMatch)
|
|
: MatchTree<EventMessage*>(end(), 1), factory(storageBackend), invertedMatch(invertedMatch) {}
|
|
|
|
EventMatchTree::~EventMatchTree() {}
|
|
|
|
bool EventMatchTree::match(EventMessage* number) {
|
|
if (invertedMatch) {
|
|
return !MatchTree<EventMessage*>::match(number);
|
|
} else {
|
|
return MatchTree<EventMessage*>::match(number);
|
|
}
|
|
}
|
|
|
|
ReturnValue_t EventMatchTree::addMatch(EventId_t idFrom, EventId_t idTo, bool idInverted,
|
|
object_id_t reporterFrom, object_id_t reporterTo,
|
|
bool reporterInverted) {
|
|
if (idFrom == 0) {
|
|
// Assuming all events shall be forwarded.
|
|
idTo = 0;
|
|
idInverted = true;
|
|
}
|
|
if (idTo == 0) {
|
|
idTo = idFrom;
|
|
}
|
|
iterator lastTest;
|
|
ReturnValue_t result = findOrInsertRangeMatcher<EventId_t, EventIdRangeMatcher>(
|
|
begin(), idFrom, idTo, idInverted, &lastTest);
|
|
if (result != returnvalue::OK) {
|
|
return result;
|
|
}
|
|
if (reporterFrom == 0) {
|
|
// No need to add another AND branch
|
|
return returnvalue::OK;
|
|
}
|
|
if (reporterTo == 0) {
|
|
reporterTo = reporterFrom;
|
|
}
|
|
return findOrInsertRangeMatcher<object_id_t, ReporterRangeMatcher>(
|
|
lastTest.left(), reporterFrom, reporterTo, reporterInverted, &lastTest);
|
|
}
|
|
|
|
ReturnValue_t EventMatchTree::removeMatch(EventId_t idFrom, EventId_t idTo, bool idInverted,
|
|
object_id_t reporterFrom, object_id_t reporterTo,
|
|
bool reporterInverted) {
|
|
iterator foundElement;
|
|
|
|
if (idFrom == 0) {
|
|
// Assuming a "forward all events" request.
|
|
idTo = 0;
|
|
idInverted = true;
|
|
}
|
|
if (idTo == 0) {
|
|
idTo = idFrom;
|
|
}
|
|
foundElement =
|
|
findRangeMatcher<EventId_t, EventIdRangeMatcher>(begin(), idFrom, idTo, idInverted);
|
|
if (foundElement == end()) {
|
|
return NO_MATCH; // Can't tell if too detailed or just not found.
|
|
}
|
|
if (reporterFrom == 0) {
|
|
if (foundElement.left() == end()) {
|
|
return removeElementAndReconnectChildren(foundElement);
|
|
} else {
|
|
return TOO_GENERAL_REQUEST;
|
|
}
|
|
}
|
|
if (reporterTo == 0) {
|
|
reporterTo = reporterFrom;
|
|
}
|
|
foundElement = findRangeMatcher<object_id_t, ReporterRangeMatcher>(
|
|
foundElement.left(), reporterFrom, reporterTo, reporterInverted);
|
|
if (foundElement == end()) {
|
|
return NO_MATCH;
|
|
} else {
|
|
return removeElementAndReconnectChildren(foundElement);
|
|
}
|
|
}
|
|
|
|
template <typename VALUE_T, typename INSERTION_T>
|
|
inline ReturnValue_t EventMatchTree::findOrInsertRangeMatcher(iterator start, VALUE_T idFrom,
|
|
VALUE_T idTo, bool inverted,
|
|
iterator* lastTest) {
|
|
bool attachToBranch = AND;
|
|
iterator iter = start;
|
|
while (iter != end()) {
|
|
INSERTION_T* matcher = static_cast<INSERTION_T*>(*iter);
|
|
attachToBranch = OR;
|
|
*lastTest = iter;
|
|
if ((matcher->rangeMatcher.lowerBound == idFrom) &&
|
|
(matcher->rangeMatcher.upperBound == idTo) &&
|
|
(matcher->rangeMatcher.inverted == inverted)) {
|
|
return returnvalue::OK;
|
|
} else {
|
|
iter = iter.right();
|
|
}
|
|
}
|
|
// Only reached if nothing was found.
|
|
SerializeableMatcherIF<EventMessage*>* newContent =
|
|
factory.generate<INSERTION_T>(idFrom, idTo, inverted);
|
|
if (newContent == NULL) {
|
|
return FULL;
|
|
}
|
|
Node* newNode = factory.generate<Node>(newContent);
|
|
if (newNode == NULL) {
|
|
// Need to make sure partially generated content is deleted, otherwise, that's a leak.
|
|
factory.destroy<INSERTION_T>(static_cast<INSERTION_T*>(newContent));
|
|
return FULL;
|
|
}
|
|
*lastTest = insert(attachToBranch, *lastTest, newNode);
|
|
if (*lastTest == end()) {
|
|
// This actaully never fails, so creating a dedicated returncode seems an overshoot.
|
|
return returnvalue::FAILED;
|
|
}
|
|
return returnvalue::OK;
|
|
}
|
|
|
|
template <typename VALUE_T, typename INSERTION_T>
|
|
EventMatchTree::iterator EventMatchTree::findRangeMatcher(iterator start, VALUE_T idFrom,
|
|
VALUE_T idTo, bool inverted) {
|
|
iterator iter = start;
|
|
while (iter != end()) {
|
|
INSERTION_T* matcher = static_cast<INSERTION_T*>(*iter);
|
|
if ((matcher->rangeMatcher.lowerBound == idFrom) &&
|
|
(matcher->rangeMatcher.upperBound == idTo) &&
|
|
(matcher->rangeMatcher.inverted == inverted)) {
|
|
break;
|
|
} else {
|
|
iter = iter.right(); // next OR element
|
|
}
|
|
}
|
|
return iter;
|
|
}
|
|
|
|
ReturnValue_t EventMatchTree::cleanUpElement(iterator position) {
|
|
factory.destroy(position.element->value);
|
|
// If deletion fails, delete element anyway, nothing we can do.
|
|
// SHOULDO: Throw event, or write debug output.
|
|
return factory.destroy(position.element);
|
|
}
|