ADD: added methode for message handling in the directCommsClient and Server

include/DirectCommunicationClient.hpp

@@ -7,6 +7,7 @@
 #include "zmq.hpp"
 #include <atomic>
 #include <cstdint>
+#include <functional>
 #include <memory>
 #include <string>
 #include <sys/types.h>
@@ -27,6 +28,9 @@ namespace DirectCommunication
             void sendMessage(std::string msg);
             std::string getLatestMessage();
 
+            void registerMessageCallback(std::function<void(std::string)> MessageHandle);
+
+
         private:
             ushort port_;
             std::string serverAddress_;
@@ -40,6 +44,8 @@ namespace DirectCommunication
 
             std::atomic_bool isConnected = true;
 
+            std::function<void(std::string)> MessageHandle_ = nullptr;
+
 
             WHISPER::threadSafeQueue<std::string> receivedMessages_;
 

include/DirectCommunicationServer.hpp

@@ -7,6 +7,7 @@
 #include "zmq.hpp"
 #include <atomic>
 #include <cstdint>
+#include <functional>
 #include <memory>
 #include <string>
 #include <sys/types.h>
@@ -27,6 +28,8 @@ namespace DirectCommunication
             void sendMessage(std::string msg);
             std::string getLatestMessage();
 
+            void registerMessageCallback(std::function<void(std::string)> MessageHandle);
+
             int countClients();
 
         private:
@@ -44,6 +47,8 @@ namespace DirectCommunication
 
             bool hasClient(std::uint32_t clientId);
 
+            std::function<void(std::string)> MessageHandle_ = nullptr;
+
             
     };
 

src/DirectCommunicationClient.cpp

@@ -36,6 +36,11 @@ namespace DirectCommunication
         socket_.close();
     }
 
+     void DirectCommunicationClient::registerMessageCallback(std::function<void(std::string)> MessageHandle)
+    {
+        MessageHandle_ = MessageHandle;
+    }
+
     void  DirectCommunicationClient::sendMessage(std::string msg)
     {
        
@@ -76,9 +81,16 @@ namespace DirectCommunication
                     socket_.disconnect("tcp://"+serverAddress_+":"+std::to_string(port_));
 
                 }else 
+                {
+                    if (MessageHandle_ != nullptr) 
+                    {
+                        MessageHandle_(msg.to_string());
+                    }
+                    else 
                     {
                         receivedMessages_.addElement(msg.to_string());
                     }
+                }
 
 
             }

src/DirectCommunicationServer.cpp

@@ -35,6 +35,12 @@ namespace DirectCommunication
         socket_.close();
     }
 
+    void DirectCommunicationServer::registerMessageCallback(std::function<void(std::string)> MessageHandle)
+    {
+        MessageHandle_ = MessageHandle;
+    }
+
+
     void  DirectCommunicationServer::sendMessage(std::string msg)
     {
         if (connectedClients_.size() > 0) 
@@ -84,11 +90,18 @@ namespace DirectCommunication
                    auto it =  std::find(connectedClients_.begin(), connectedClients_.end(), msg.routing_id());
                    connectedClients_.erase(it);
                 }else 
+                {   
+                    if (MessageHandle_ != nullptr) 
+                    {
+                        MessageHandle_(msg.to_string());
+                    }
+                    else 
                     {
                         receivedMessages_.addElement(msg.to_string());
                     }
                 }
             }
+        }
 
     }