Fix xml representations

[anna.git] / source / diameter.comm / Entity.cpp

diff --git a/source/diameter.comm/Entity.cpp b/source/diameter.comm/Entity.cpp
index 25a879c..fc07148 100644 (file)
--- a/source/diameter.comm/Entity.cpp
+++ b/source/diameter.comm/Entity.cpp
@@ -39,6 +39,8 @@ void Entity::initialize() throw() {
   a_description = "";
   a_category = 0;
   a_lastUsedResource = NULL;
+  a_balance = false;
+  a_sessionBasedModelsType = SessionBasedModelsType::SessionIdLowPart;
 }
 
 
@@ -74,6 +76,9 @@ throw(anna::RuntimeException) {
 
 
 int Entity::readSocketId(const Message* message, int maxClientSessions) const throw() {
+
+  if(a_sessionBasedModelsType == SessionBasedModelsType::RoundRobin) return -1;  // IEC also would return -1
+
   try {
     // Service-Context-Id:
     anna::diameter::helpers::dcca::ChargingContext::_v chargingContext;
@@ -88,12 +93,18 @@ int Entity::readSocketId(const Message* message, int maxClientSessions) const th
       std::string diameterIdentity, optional;
       anna::U32 high, low;
       anna::diameter::helpers::base::functions::decodeSessionId(sid, diameterIdentity, high, low /* context-teid */, optional);
-      return (low % maxClientSessions);
+
+      if(a_sessionBasedModelsType == SessionBasedModelsType::SessionIdLowPart) return (low % maxClientSessions);
+
+      if(a_sessionBasedModelsType == SessionBasedModelsType::SessionIdHighPart) return (high % maxClientSessions);
+
+      if(a_sessionBasedModelsType == SessionBasedModelsType::SessionIdOptionalPart) return (atoi(optional.c_str()) % maxClientSessions);
     }
-    //case anna::diameter::helpers::dcca::ChargingContext::SMS:
-    //case anna::diameter::helpers::dcca::ChargingContext::MMS:
-    //default:
-    //   return -1; // IEC model and Unknown traffic types
+    case anna::diameter::helpers::dcca::ChargingContext::SMS:
+    case anna::diameter::helpers::dcca::ChargingContext::MMS:
+    case anna::diameter::helpers::dcca::ChargingContext::Unknown:
+    default:
+       return -1; // IEC model and Unknown traffic types
     }
   } catch(anna::RuntimeException &ex) {
     LOGDEBUG(
@@ -107,7 +118,7 @@ int Entity::readSocketId(const Message* message, int maxClientSessions) const th
 }
 
 
-bool Entity::send(const Message* message, bool balance) throw(anna::RuntimeException) {
+bool Entity::send(const Message* message) throw(anna::RuntimeException) {
   LOGMETHOD(anna::TraceMethod tttm("diameter::comm::Entity", "send", ANNA_FILE_LOCATION));
   assertReady();
   // Carried socket id (forwarding/proxy features):
@@ -136,7 +147,7 @@ bool Entity::send(const Message* message, bool balance) throw(anna::RuntimeExcep
   ////////////////////////////////////////////////////////////////////////////////////////
 
   // Balance
-  if(balance) {
+  if(a_balance) {
     for(int k = 0; k < getMaxServers(); k++) {   // try round-robin only over one cycle,
       // no matter where you are: don't repeat same server
       if(a_deliveryIterator == end()) a_deliveryIterator = begin();
@@ -262,6 +273,11 @@ void Entity::close(bool destroy) throw(anna::RuntimeException) {
     a_engine->closeServer(*it, destroy);
 }
 
+const char* Entity::asText(const SessionBasedModelsType::_v sbmt)
+throw() {
+  static const char* text [] = { "RoundRobin", "SessionIdOptionalPart", "SessionIdHighPart", "SessionIdLowPart" };
+  return text [sbmt];
+}
 
 socket_v Entity::getAddressPortList() const throw() {
   socket_v result;
@@ -323,19 +339,24 @@ void Entity::eventPeerShutdown(const ClientSession* clientSession) throw() {
 void Entity::eventRequestRetransmission(const ClientSession* clientSession, Message *request) throw() {
   LOGWARNING(
     std::string msg(clientSession->asString());
-    msg += " | eventRequestRetransmission";
+
+    HopByHop hopByHop = codec::functions::getHopByHop(request->getBody()); // context identification
+    int retries = request->getRetries();
+
+    msg += anna::functions::asString(" | eventRequestRetransmission: request with application HopByHop: %u; remaining %d retries", hopByHop, retries);
     anna::Logger::warning(msg, ANNA_FILE_LOCATION);
   );
 }
 
 std::string Entity::asString() const throw() {
   std::string result("diameter::comm::Entity { ");
-  std::string realm = a_engine->getRealm();
+  std::string originRealm = a_engine->getOriginRealmName();
+  std::string originHost = a_engine->getOriginHostName();
 
-  if(realm != "") {
-    result += "Parent Engine (realm): ";
-    result += realm;
-  }
+  result += "Parent Engine Origin-Realm: ";
+  result += (originRealm != "") ? originRealm:"[not configured]";
+  result += " | Parent Engine Origin-Host: ";
+  result += (originHost != "") ? originHost:"[not configured]";
 
   result += " | Category: ";
   result += anna::functions::asString(a_category);
@@ -360,6 +381,8 @@ std::string Entity::asString() const throw() {
   result += a_lastOutgoingActivityTime.asString();
   result += " | Hidden: ";
   result += (hidden() ? "yes" : "no");
+  result += " | SessionBasedModelsType: ";
+  result += asText(a_sessionBasedModelsType);
   result += "\n";
 
   for(std::vector<Server*>::const_iterator it = begin(); it != end(); it++) {
@@ -372,9 +395,11 @@ std::string Entity::asString() const throw() {
 
 anna::xml::Node* Entity::asXML(anna::xml::Node* parent) const throw() {
   anna::xml::Node* result = parent->createChild("diameter.Entity");
-  std::string realm = a_engine->getRealm();
+  std::string originRealm = a_engine->getOriginRealmName();
+  std::string originHost = a_engine->getOriginHostName();
 
-  if(realm != "") result->createAttribute("ParentEngineRealm", realm);
+  if(originRealm != "") result->createAttribute("ParentEngineOriginRealm", originRealm);
+  if(originHost != "") result->createAttribute("ParentEngineOriginHost", originHost);
 
   result->createAttribute("Category", anna::functions::asString(a_category));
 
@@ -388,6 +413,7 @@ anna::xml::Node* Entity::asXML(anna::xml::Node* parent) const throw() {
   result->createAttribute("LastIncomingActivityTime", a_lastIncomingActivityTime.asString());
   result->createAttribute("LastOutgoingActivityTime", a_lastOutgoingActivityTime.asString());
   result->createAttribute("Hidden", hidden() ? "yes" : "no");
+  result->createAttribute("SessionBasedModelsType:", asText(a_sessionBasedModelsType));
   anna::xml::Node* servers = result->createChild("Entity.Servers");
 
   for(std::vector<Server*>::const_iterator it = begin(); it != end(); it++)