+// ANNA - Anna is Not Nothingness Anymore //
+// //
+// (c) Copyright 2005-2015 Eduardo Ramos Testillano & Francisco Ruiz Rayo //
+// //
+// See project site at http://redmine.teslayout.com/projects/anna-suite //
+// See accompanying file LICENSE or copy at http://www.teslayout.com/projects/public/anna.LICENSE //
+
+// Standard
+#include <string>
+#include <fstream>
+
+// Project
+#include <anna/diameter/codec/Message.hpp>
+#include <anna/diameter/codec/Engine.hpp>
+#include <anna/diameter/codec/EngineManager.hpp>
+
+// Process
+#include <ProgrammedAnswers.hpp>
+
+
+void ProgrammedAnswers::clear() throw () {
+ try {
+ anna::diameter::codec::EngineManager &em = anna::diameter::codec::EngineManager::instantiate();
+ for (reacting_answers_const_iterator it = a_deques.begin(); it != a_deques.end(); it++) {
+ anna::diameter::codec::Message *message = *(it->second->begin());
+ em.getCodecEngine(message->getApplicationId())->releaseMessage(message);
+ delete (it->second);
+ }
+ a_deques.clear();
+ }
+ catch (anna::RuntimeException &ex) {
+ ex.trace();
+ }
+}
+
+void ProgrammedAnswers::dump() throw () {
+ std::string outfilename, xmlString;
+ for (reacting_answers_const_iterator it = a_deques.begin();
+ it != a_deques.end(); it++) {
+ int sequence = 1;
+ for (codec_messages_deque_const_iterator itm = it->second->begin();
+ itm != it->second->end(); itm++) {
+ // programmed_answer.<code>.<sequence>
+ outfilename = "programmed_answer.";
+ outfilename += anna::functions::asString(it->first);
+ outfilename += ".";
+ outfilename += anna::functions::asString(sequence++);
+ outfilename += ".xml";
+ std::ofstream outfile(outfilename.c_str(), std::ifstream::out);
+ xmlString = (*itm)->asXMLString();
+ outfile.write(xmlString.c_str(), xmlString.size());
+ outfile.close();
+ }
+ }
+}
+
+void ProgrammedAnswers::addMessage(int code, anna::diameter::codec::Message *message) throw () {
+ if (!message)
+ return; // just in case
+
+ reacting_answers_const_iterator it = a_deques.find(code);
+ if (it != a_deques.end()) {
+ it->second->push_back(message);
+ } else {
+ codec_messages_deque *deque = new codec_messages_deque;
+ a_deques[code] = deque;
+ deque->push_back(message);
+ }
+}
+
+anna::diameter::codec::Message* ProgrammedAnswers::getMessage(int code) const throw () { //get the front message (begin()), returns NULL if deque is empty
+ anna::diameter::codec::Message *result = NULL;
+ reacting_answers_const_iterator it = a_deques.find(code);
+ if (it != a_deques.end()) {
+ if (!it->second->empty())
+ result = *(it->second->begin());
+ }
+ return result;
+}
+
+void ProgrammedAnswers::nextMessage(int code) throw () { //pops the deque and release the message (when deque is not empty: deque::empty)
+ try {
+ reacting_answers_const_iterator it = a_deques.find(code);
+ if (it != a_deques.end()) {
+ if (!it->second->empty()) {
+ if (a_rotate) {
+ addMessage(code, *(it->second->begin()));
+ } else {
+ anna::diameter::codec::EngineManager &em = anna::diameter::codec::EngineManager::instantiate();
+ anna::diameter::codec::Message *message = *(it->second->begin());
+ em.getCodecEngine(message->getApplicationId())->releaseMessage(message);
+ }
+ it->second->pop_front();
+ }
+ }
+ }
+ catch (anna::RuntimeException &ex) {
+ ex.trace();
+ }
+}
+
+std::string ProgrammedAnswers::asString(const char *queueName) const throw () {
+ std::string result = "";
+ std::string aux = "FIFO QUEUE '";
+ aux += queueName;
+ aux += "', Rotation ";
+ aux += a_rotate ? "enabled" : "disabled";
+ result += anna::functions::highlightJustify(aux);
+ if (a_deques.size() != 0) {
+ for (reacting_answers_const_iterator it = a_deques.begin();
+ it != a_deques.end(); it++) {
+ if (it->second->size() != 0) {
+ aux = "Answer code ";
+ aux += anna::functions::asString(it->first);
+ result += anna::functions::highlightJustify(aux,
+ anna::functions::TextHighlightMode::OverAndUnderline,
+ anna::functions::TextJustifyMode::Left, '-');
+ for (codec_messages_deque_const_iterator itm = it->second->begin();
+ itm != it->second->end(); itm++) {
+ result += (*itm)->asXMLString();
+ result += "\n";
+ }
+ result += "\n";
+ }
+ }
+ } else {
+ result = "No ocurrences found\n\n";
+ }
+ return result;
+}