Preparing for multistack

[anna.git] / include / anna / diameter.comm / Entity.hpp

diff --git a/include/anna/diameter.comm/Entity.hpp b/include/anna/diameter.comm/Entity.hpp
index 151f0ff..2335d82 100644 (file)
--- a/include/anna/diameter.comm/Entity.hpp
+++ b/include/anna/diameter.comm/Entity.hpp
@@ -19,6 +19,8 @@
 #include <anna/core/RuntimeException.hpp>
 #include <anna/core/util/Millisecond.hpp>
 
+#include <anna/diameter.comm/ClassCode.hpp>
+
 
 namespace anna {
 class DataBlock;
@@ -39,6 +41,7 @@ class Engine;
 class Server;
 class Response;
 class Message;
+class ClientSession;
 
 
 /**
@@ -131,6 +134,11 @@ public:
   */
   void setEngine(Engine *e) throw() { a_engine = e; }
 
+//  /**
+//  * Gets the diameter::comm::Engine
+//  * @return Diameter::comm::Engine
+//  */
+//  Engine *getEngine() const throw() { return a_engine; }
 
   /**
   * Add a server to the entity and create all the servers configured at #setSocketsPerDiameterServer within that server.