X-Git-Url: https://git.teslayout.com/public/public/public/?a=blobdiff_plain;f=example%2Fdiameter%2Flauncher%2FMyLocalServer.cpp;h=c4f5db7f3359f3715ad39b401bee94480dfe4e8d;hb=d723d5bf571eb48c641b092058eaa38bb6c4fcc8;hp=4bc8ebab66ed466fc95666a194df760912514bf4;hpb=74730c7d0e4f713c932d6b06a07644d1c3c6c6c1;p=anna.git

diff --git a/example/diameter/launcher/MyLocalServer.cpp b/example/diameter/launcher/MyLocalServer.cpp
index 4bc8eba..c4f5db7 100644
--- a/example/diameter/launcher/MyLocalServer.cpp
+++ b/example/diameter/launcher/MyLocalServer.cpp
@@ -14,13 +14,14 @@
 #include <anna/diameter.comm/Response.hpp>
 #include <anna/diameter.comm/ClientSession.hpp>
 #include <anna/diameter.comm/Server.hpp>
+#include <anna/diameter.comm/OriginHost.hpp>
 
 // Process
-#include "MyLocalServer.hpp"
-#include "MyDiameterEngine.hpp"
-#include "MyDiameterEntity.hpp"
-#include "Launcher.hpp"
-#include "RealmNode.hpp"
+#include <MyLocalServer.hpp>
+#include <MyDiameterEngine.hpp>
+#include <MyDiameterEntity.hpp>
+#include <Launcher.hpp>
+#include <anna/testing/TestManager.hpp>
 
 
 void MyLocalServer::eventRequest(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message)
@@ -28,7 +29,8 @@ throw(anna::RuntimeException) {
   LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventRequest", ANNA_FILE_LOCATION));
   // Performance stats:
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
-  RealmNode * my_node = my_app.getRealmNode(getEngine()->getRealm());
+  anna::diameter::comm::OriginHost *my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
+  anna::diameter::codec::Engine *codecEngine = my_node->getCodecEngine();
 
   // CommandId:
   anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
@@ -57,7 +59,7 @@ throw(anna::RuntimeException) {
     msg->forwardEndToEnd(); // end-to-end will be kept
     msg->setBody(message);
     msg->setRequestServerSessionKey(serverSession->getKey());
-    bool success = entity->send(msg, my_node->getEntity()->getBalance());
+    bool success = entity->send(msg);
 
     // Detailed log:
     if(my_node->logEnabled()) {
@@ -67,14 +69,18 @@ throw(anna::RuntimeException) {
       my_node->writeLogFile(message, (success ? "fwd2e" : "fwd2eError"), detail); // forwarded
     }
 
+
+    // Testing:
+    anna::testing::TestManager::instantiate().receiveMessage(message, serverSession);
+
     return;
   }
 
   // Error analisys:
   bool analysisOK = true; // by default
   anna::diameter::codec::Message *answer_message = NULL;
-  anna::diameter::codec::Message codecMsg(a_codecEngine);
-  anna::diameter::codec::Message codecAnsMsg(a_codecEngine);
+  anna::diameter::codec::Message codecMsg;
+  anna::diameter::codec::Message codecAnsMsg;
 
   CommandLine& cl(anna::CommandLine::instantiate());
   if(!cl.exists("ignoreErrors")) {  // Error analysis
@@ -84,7 +90,7 @@ throw(anna::RuntimeException) {
     // Decode
     try { codecMsg.decode(message, answer_message); } catch(anna::RuntimeException &ex) { ex.trace(); }
 
-    answer_message->setStandardToAnswer(codecMsg, my_node->getMyDiameterEngine()->getHost(), my_node->getMyDiameterEngine()->getRealm());
+    answer_message->setStandardToAnswer(codecMsg, my_node->getCommEngine()->getOriginHostName(), my_node->getCommEngine()->getOriginRealmName());
     analysisOK = (answer_message->getResultCode() == anna::diameter::helpers::base::AVPVALUES__Result_Code::DIAMETER_SUCCESS);
   }
 
@@ -94,19 +100,25 @@ throw(anna::RuntimeException) {
       answer_message = programmed_answer;
       // Prepare answer:
       my_app.getCommunicator()->prepareAnswer(answer_message, message);
-    } else return; // nothing done
+    } else {
+
+      // Testing:
+      anna::testing::TestManager::instantiate().receiveMessage(message, serverSession);
+
+      return; // nothing done
+    }
   }
 
-  anna::diameter::codec::Engine::ValidationMode::_v backupVM = a_codecEngine->getValidationMode();
+  anna::diameter::codec::Engine::ValidationMode::_v backupVM = codecEngine->getValidationMode();
 
   if(!analysisOK)
-    a_codecEngine->setValidationMode(anna::diameter::codec::Engine::ValidationMode::Never);
+    codecEngine->setValidationMode(anna::diameter::codec::Engine::ValidationMode::Never);
 
+  anna::diameter::comm::Message *msg;
   try {
-    anna::diameter::comm::Message *msg = my_node->createCommMessage();
+    msg = my_node->createCommMessage();
     msg->setBody(answer_message->code());
     /* response = NULL =*/serverSession->send(msg);
-    my_node->releaseCommMessage(msg);
 
     if(my_node->logEnabled()) my_node->writeLogFile(*answer_message, "sent2c", serverSession->asString());
   } catch(anna::RuntimeException &ex) {
@@ -114,18 +126,24 @@ throw(anna::RuntimeException) {
     if(my_node->logEnabled()) my_node->writeLogFile(*answer_message, "send2cError", serverSession->asString());
   }
 
+  // release msg
+  my_node->releaseCommMessage(msg);
+
   // Restore validation mode
-  a_codecEngine->setValidationMode(backupVM);
+  codecEngine->setValidationMode(backupVM);
 
   // Pop front the reacting answer:
   if(analysisOK && programmed) a_reactingAnswers.nextMessage(code);
+
+  // Testing:
+  anna::testing::TestManager::instantiate().receiveMessage(message, serverSession);
 }
 
 void MyLocalServer::eventResponse(const anna::diameter::comm::Response &response)
 throw(anna::RuntimeException) {
   LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventResponse", ANNA_FILE_LOCATION));
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
-  RealmNode * my_node = my_app.getRealmNode(getEngine()->getRealm());
+  anna::diameter::comm::OriginHost * my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
   anna::diameter::comm::ClassCode::_v code = response.getClassCode();
   anna::diameter::comm::Response::ResultCode::_v result = response.getResultCode();
   anna::diameter::comm::Message* request = const_cast<anna::diameter::comm::Message*>(response.getRequest());
@@ -136,6 +154,7 @@ throw(anna::RuntimeException) {
   bool contextExpired = (result == anna::diameter::comm::Response::ResultCode::Timeout);
   bool isUnavailable = (result == anna::diameter::comm::Response::ResultCode::DiameterUnavailable);
   bool isOK = (result == anna::diameter::comm::Response::ResultCode::Success);
+
   // CommandId:
   anna::diameter::CommandId request_cid = request->getCommandId();
   LOGDEBUG
@@ -179,13 +198,14 @@ throw(anna::RuntimeException) {
     anna::diameter::comm::Entity *entity = my_node->getEntity();
 
     if(entity) {
-      anna::diameter::comm::ClientSession *usedClientSession = my_node->getMyDiameterEngine()->findClientSession(request->getRequestClientSessionKey());
+      anna::diameter::comm::ClientSession *usedClientSession = my_node->getCommEngine()->findClientSession(request->getRequestClientSessionKey());
       std::string detail;
 
       if(my_node->logEnabled()) detail = usedClientSession ? usedClientSession->asString() : "<null client session>";  // this should not happen
+      anna::diameter::comm::Message *msg;
 
       try {
-        anna::diameter::comm::Message *msg = my_node->createCommMessage();
+        msg = my_node->createCommMessage();
         msg->forwardEndToEnd(); // end-to-end will be kept
         msg->setBody(*message);
 
@@ -196,16 +216,20 @@ throw(anna::RuntimeException) {
         //msg->setRequestClientSessionKey(request->getRequestClientSessionKey());
         //bool success = entity->send(msg);
 
-        my_node->releaseCommMessage(msg);
-        my_node->releaseCommMessage(request);
-
         if(my_node->logEnabled()) my_node->writeLogFile(*message, "fwd2e", detail);  // forwarded
       } catch(anna::RuntimeException &ex) {
         ex.trace();
         if(my_node->logEnabled()) my_node->writeLogFile(*message, "fwd2eError", detail);  // forwarded
       }
+
+      // release msgs
+      my_node->releaseCommMessage(msg);
+      my_node->releaseCommMessage(request);
     }
   }
+
+  // Testing:
+  if(isOK) anna::testing::TestManager::instantiate().receiveMessage(*message, serverSession);
 }
 
 void MyLocalServer::eventUnknownResponse(anna::diameter::comm::ServerSession *serverSession, const anna::DataBlock &message)
@@ -213,7 +237,7 @@ throw(anna::RuntimeException) {
   LOGMETHOD(anna::TraceMethod tm("launcher::MyLocalServer", "eventUnknownResponse", ANNA_FILE_LOCATION));
   // Performance stats:
   Launcher& my_app = static_cast <Launcher&>(anna::app::functions::getApp());
-  RealmNode * my_node = my_app.getRealmNode(getEngine()->getRealm());
+  anna::diameter::comm::OriginHost *my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
   // CommandId:
   anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
   LOGDEBUG
@@ -235,7 +259,7 @@ 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());
-  RealmNode * my_node = my_app.getRealmNode(getEngine()->getRealm());
+  anna::diameter::comm::OriginHost *my_node = my_app.getOriginHost(getEngine()->getOriginHostName());
   // CommandId:
   anna::diameter::CommandId cid = anna::diameter::codec::functions::getCommandId(message);
   LOGDEBUG
@@ -249,5 +273,9 @@ throw(anna::RuntimeException) {
     anna::Logger::debug(msg, ANNA_FILE_LOCATION);
   );
 
+  // Write reception
   if(my_node->logEnabled()) my_node->writeLogFile(message, "recvfc", serverSession->asString());
+
+  // Testing:
+  anna::testing::TestManager::instantiate().receiveMessage(message, serverSession);
 }