X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=example%2Fdiameter%2Flauncher%2FProgrammedAnswers.cpp;h=b2b89b2e2030226bee87bc6e12ea4d74f1a2bb33;hb=5e4f1d9d5902d86a37ecf88b372d9089d417e663;hp=e69de29bb2d1d6434b8b29ae775ad8c2e48c5391;hpb=68a8fe342d8337ec1a86b57920bc7dcf7faf2413;p=anna.git

diff --git a/example/diameter/launcher/ProgrammedAnswers.cpp b/example/diameter/launcher/ProgrammedAnswers.cpp
index e69de29..b2b89b2 100644
--- a/example/diameter/launcher/ProgrammedAnswers.cpp
+++ b/example/diameter/launcher/ProgrammedAnswers.cpp
@@ -0,0 +1,146 @@
+// 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();
+    anna::diameter::codec::Engine *engine;
+
+    for (reacting_answers_const_iterator it = a_deques.begin(); it != a_deques.end(); it++) {
+      anna::diameter::codec::Message *message = *(it->second->begin());
+      engine = em.getCodecEngine(message->getApplicationId());
+      if (engine) {
+        engine->releaseMessage(message);
+        delete (it->second);
+      }
+      else {
+        LOGWARNING(anna::Logger::warning("Cannot release a message for which i don't know the codec engine (check the registered stack id regarding the message application id) !", ANNA_FILE_LOCATION));
+      }
+    }
+    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)
+  anna::diameter::codec::Engine *engine;
+
+  try {
+    reacting_answers_const_iterator it = a_deques.find(code);
+    if (it != a_deques.end()) {
+      if (!it->second->empty()) {
+        anna::diameter::codec::Message *message = *(it->second->begin());
+        if (a_rotate) {
+          addMessage(code, message);
+        } else {
+          engine = anna::diameter::codec::EngineManager::instantiate().getCodecEngine(message->getApplicationId());
+          if (engine) {
+            engine->releaseMessage(message);
+          }
+          else {
+            LOGWARNING(anna::Logger::warning("Cannot release a message for which i don't know the codec engine (check the registered stack id regarding the message application id) !", ANNA_FILE_LOCATION));
+            return;
+          }
+        }
+        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;
+}