Improv. presentetion of answering queues. New management operations for answering...
[anna.git] / example / diameter / launcher / main.cpp
index 7baab28..ab5ea0a 100644 (file)
@@ -113,11 +113,15 @@ typedef std::map < int /* message code */, codec_messages_deque* >::iterator  re
 typedef std::map < int /* message code */, codec_messages_deque* >::const_iterator  reacting_answers_const_iterator;
 
   reacting_answers_container a_deques;
+  bool a_rotate;
 
   public:
-    ProgrammedAnswers() {};
+    ProgrammedAnswers() { a_rotate = false; }
     ~ProgrammedAnswers() { clear(); }
 
+    bool rotate() const const throw() { return a_rotate; }
+    void rotate(bool r) throw() { a_rotate = r; }
+
     void clear () throw() {
       for (reacting_answers_const_iterator it = a_deques.begin(); it != a_deques.end(); it++) {
         anna::diameter::codec::Engine *engine = anna::functions::component <Engine> (ANNA_FILE_LOCATION);
@@ -172,25 +176,42 @@ typedef std::map < int /* message code */, codec_messages_deque* >::const_iterat
       if (it != a_deques.end()) {
         if (!it->second->empty()) {
           anna::diameter::codec::Engine *engine = anna::functions::component <Engine> (ANNA_FILE_LOCATION);
-          engine->releaseMessage(*(it->second->begin()));
+          if (a_rotate) {
+            addMessage(code, *(it->second->begin()));
+          }
+          else {
+            engine->releaseMessage(*(it->second->begin()));
+          }
           it->second->pop_front();
         }
       }
     }
 
-    std::string asString() const throw() {
-      std::string result = "No ocurrences found\n\n";
+    std::string 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++) {
-          result += "Answer code .............................................................. ";
-          result += anna::functions::asString(it->first); result += "\n";
-          for(codec_messages_deque_const_iterator itm = it->second->begin(); itm != it->second->end(); itm++) {
-            result += (*itm)->asXMLString();
+          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";
           }
-          result += "\n";
         }
       }
+      else {
+        result = "No ocurrences found\n\n";
+      }
       return result;
     }
 };
@@ -296,6 +317,7 @@ class MyDiameterEntity : public anna::diameter::comm::Entity {
   void eventResponse(const anna::diameter::comm::Response&) throw(anna::RuntimeException);
   void eventRequest(anna::diameter::comm::ClientSession *, const anna::DataBlock&) throw(anna::RuntimeException);
   void eventUnknownResponse(anna::diameter::comm::ClientSession *, const anna::DataBlock&) throw(anna::RuntimeException);
+  void eventDPA(anna::diameter::comm::ClientSession *, const anna::DataBlock&) throw(anna::RuntimeException);
 
   // Reimplementation
   int readSocketId(const anna::diameter::comm::Message* message, int maxClientSessions) const throw();
@@ -306,6 +328,7 @@ class MyLocalServer : public anna::diameter::comm::LocalServer {
   void eventResponse(const anna::diameter::comm::Response&) throw(anna::RuntimeException);
   void eventRequest(anna::diameter::comm::ServerSession *, const anna::DataBlock&) throw(anna::RuntimeException);
   void eventUnknownResponse(anna::diameter::comm::ServerSession *, const anna::DataBlock&) throw(anna::RuntimeException);
+  void eventDPA(anna::diameter::comm::ServerSession *, const anna::DataBlock&) throw(anna::RuntimeException);
 };
 
 class MyDiameterEngine : public anna::diameter::comm::Engine {
@@ -849,9 +872,12 @@ std::string Launcher::help() const throw() {
   result += "\n                           as many queues as different message codes have been received.";
   result += "\nanswerxml|[source_file]    Same as 'answerxml2c'.";
   result += "\nanswerxml(2e/2c)           List programmed answers (to entity/client) if no parameter provided.";
-  result += "\nanswerxml(2e/2c)|dump      Write programmed answers (to entity/client) to file 'programmed_answer.<code>.<sequence>',";
+  result += "\nanswerxml(2e/2c)|dump      Write programmed answers (to entity/client) to file 'programmed_answer.<message code>.<sequence>',";
   result += "\n                           where 'sequence' is the order of the answer in each FIFO code-queue of programmed answers.";
   result += "\nanswerxml(2e/2c)|clear     Clear programmed answers (to entity/client).";
+  result += "\nanswerxml(2e/2c)|exhaust   Disable the corresponding queue rotation, which is the default behaviour.";
+  result += "\nanswerxml(2e/2c)|rotate    Enable the corresponding queue rotation, useful in performance tests.";
+  result += "\n                           Rotation consists in add again to the queue, each element retrieved for answering.";
   result += "\n";
   result += "\nSend operations are available using hexadecimal content (hex formatted files) which also allow to test";
   result += "\nspecial scenarios (protocol errors):";
@@ -861,8 +887,8 @@ std::string Launcher::help() const throw() {
   result += "\nsendhex|<source_file>      Same as 'sendhex2e'.";
   result += "\n";
   result += "\nAnswer programming in hexadecimal is not really neccessary (you could use send primitives) and also";
-  result += "\nis intended to be used with decoded messages in order to replace things like hop by hop, end to end,";
-  result += "\nsubscriber id, session id, etc. Anyway you could use 'decode' operation and then program the xml created.";
+  result += "\n is intended to be used with decoded messages in order to replace things like hop by hop, end to end,";
+  result += "\n subscriber id, session id, etc. Anyway you could use 'decode' operation and then program the xml created.";
   result += "\n";
   result += "\nIf a request is received, answer map (built with 'answerxml<[2c] or 2e>' operations) will be";
   result += "\n checked to find a corresponding programmed answer to be replied(*). If no ocurrence is found,";
@@ -870,6 +896,11 @@ std::string Launcher::help() const throw() {
   result += "\n or nothing but trace when no peer at that side is configured. Answer to client have sense when";
   result += "\n diameter server socket is configured, answer to entity have sense when entity does.";
   result += "\n";
+  result += "\nIn the most complete situation (process with both client and server side) there are internally";
+  result += "\n two maps with N FIFO queues, one for each different message code within programmed answers.";
+  result += "\nOne map is for answers towards the client, and the other is to react entity requests. Then in";
+  result += "\n each one we could program different answers corresponding to different request codes received.";
+  result += "\n";
   result += "\n(*) sequence values (hop-by-hop and end-to-end), Session-Id and Subscription-Id avps, are mirrored";
   result += "\n    to the peer which sent the request. If user wants to test a specific answer without changing it,";
   result += "\n    use sendxml/sendhex operations better than programming.";
@@ -2049,11 +2080,13 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
       throw anna::RuntimeException("Operation not applicable (no own diameter server has been configured)", ANNA_FILE_LOCATION);
 
     if(param1 == "") { // programmed answers FIFO's to stdout
-      std::cout << std::endl << std::endl;
-      std::cout << "          ------------- CURRENT PROGRAMMED ANSWERS TO CLIENT -------------\n\n";
-      std::cout << G_reactingAnswers2C.asString() << std::endl;
+      std::cout << G_reactingAnswers2C.asString("ANSWERS TO CLIENT") << std::endl;
       response_content = "Programmed answers dumped on stdout\n";
       return;
+    } else if (param1 == "rotate") {
+      G_reactingAnswers2C.rotate(true);
+    } else if (param1 == "exhaust") {
+      G_reactingAnswers2C.rotate(false);
     } else if (param1 == "clear") {
       G_reactingAnswers2C.clear();
     } else if (param1 == "dump") {
@@ -2080,11 +2113,13 @@ void Launcher::eventOperation(const std::string &operation, std::string &respons
       throw anna::RuntimeException("Operation not applicable (no diameter entity has been configured)", ANNA_FILE_LOCATION);
 
     if(param1 == "") { // programmed answers FIFO's to stdout
-      std::cout << std::endl << std::endl;
-      std::cout << "          ------------- CURRENT PROGRAMMED ANSWERS TO ENTITY -------------\n\n";
-      std::cout << G_reactingAnswers2E.asString() << std::endl;
+      std::cout << G_reactingAnswers2E.asString("ANSWERS TO ENTITY") << std::endl;
       response_content = "Programmed answers dumped on stdout\n";
       return;
+    } else if (param1 == "rotate") {
+      G_reactingAnswers2C.rotate(true);
+    } else if (param1 == "exhaust") {
+      G_reactingAnswers2C.rotate(false);
     } else if (param1 == "clear") {
       G_reactingAnswers2E.clear();
     } else if (param1 == "dump") {
@@ -2408,8 +2443,27 @@ throw(anna::RuntimeException) {
   if(my_app.logEnabled()) my_app.writeLogFile(message, "recvfe-ans-unknown", clientSession->asString());
 }
 
+void MyDiameterEntity::eventDPA(anna::diameter::comm::ClientSession *clientSession, const anna::DataBlock &message)
+throw(anna::RuntimeException) {
+  LOGMETHOD(anna::TraceMethod tm("launcher::MyDiameterEntity", "eventDPA", ANNA_FILE_LOCATION));
+  // Performance stats:
+  Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
+  // CommandId:
+  anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
+  LOGDEBUG
+  (
+    std::string msg = "Disconnect-Peer-Answer received from entity: ";
+    msg += anna::diameter::functions::commandIdAsPairString(cid);
+    msg += " | DiameterServer: ";
+    msg += anna::functions::socketLiteralAsString(clientSession->getAddress(), clientSession->getPort());
+    msg += " | EventTime: ";
+    msg += anna::time::functions::currentTimeAsString();
+    anna::Logger::debug(msg, ANNA_FILE_LOCATION);
+  );
 
-
+  // Write reception
+  if(my_app.logEnabled()) my_app.writeLogFile(message, "recvfe", clientSession->asString());
+}
 
 void MyLocalServer::eventRequest(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message)
 throw(anna::RuntimeException) {
@@ -2609,6 +2663,26 @@ throw(anna::RuntimeException) {
   if(my_app.logEnabled()) my_app.writeLogFile(message, "recvfc-ans-unknown", serverSession->asString());
 }
 
+void MyLocalServer::eventDPA(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message)
+throw(anna::RuntimeException) {
+  LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventDPA", ANNA_FILE_LOCATION));
+  // Performance stats:
+  Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
+  // CommandId:
+  anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
+  LOGDEBUG
+  (
+    std::string msg = "Disconnect-Peer-Answer response received from client: ";
+    msg += anna::diameter::functions::commandIdAsPairString(cid);
+    msg += " | DiameterServer: ";
+    msg += anna::functions::socketLiteralAsString(serverSession->getAddress(), serverSession->getPort());
+    msg += " | EventTime: ";
+    msg += anna::time::functions::currentTimeAsString();
+    anna::Logger::debug(msg, ANNA_FILE_LOCATION);
+  );
+
+  if(my_app.logEnabled()) my_app.writeLogFile(message, "recvfc", serverSession->asString());
+}
 
 anna::xml::Node* Launcher::asXML(anna::xml::Node* parent) const
 throw() {