diff UnitTests/MessageWithDestination.cpp @ 140:4d863c7b2f44

message queues
author Sebastien Jodogne <s.jodogne@gmail.com>
date Wed, 10 Oct 2012 18:20:04 +0200
parents 3ad78369fcc4
children 24681d35bad9
line wrap: on
line diff
--- a/UnitTests/MessageWithDestination.cpp	Wed Oct 10 17:57:03 2012 +0200
+++ b/UnitTests/MessageWithDestination.cpp	Wed Oct 10 18:20:04 2012 +0200
@@ -1,14 +1,94 @@
 #include "../Core/IDynamicObject.h"
 
+#include "../Core/OrthancException.h"
+
+#include <memory>
+#include <map>
 #include <gtest/gtest.h>
 #include <string>
+#include <boost/thread.hpp>
+#include <boost/date_time/posix_time/posix_time_types.hpp>
 
 namespace Orthanc
 {
+  class SharedMessageQueue
+  {
+  private:
+    typedef std::list<IDynamicObject*>  Queue;
+
+    unsigned int maxSize_;
+    Queue queue_;
+    boost::mutex mutex_;
+    boost::condition_variable elementAvailable_;
+
+  public:
+    SharedMessageQueue(unsigned int maxSize = 0)
+    {
+      maxSize_ = maxSize;
+    }
+
+    ~SharedMessageQueue()
+    {
+      for (Queue::iterator it = queue_.begin(); it != queue_.end(); it++)
+      {
+        delete *it;
+      }
+    }
+
+    void Enqueue(IDynamicObject* message)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+
+      if (maxSize_ != 0 && queue_.size() > maxSize_)
+      {
+        // Too many elements in the queue: First remove the oldest
+        delete queue_.front();
+        queue_.pop_front();
+      }
+
+      queue_.push_back(message);
+      elementAvailable_.notify_one();
+    }
+
+    IDynamicObject* Dequeue(int32_t timeout)
+    {
+      boost::mutex::scoped_lock lock(mutex_);
+
+      // Wait for a message to arrive in the FIFO queue
+      while (queue_.empty())
+      {
+        if (timeout == 0)
+        {
+          elementAvailable_.wait(lock);
+        }
+        else
+        {
+          bool success = elementAvailable_.timed_wait
+            (lock, boost::posix_time::milliseconds(timeout));
+          if (!success)
+          {
+            throw OrthancException(ErrorCode_Timeout);
+          }
+        }
+      }
+
+      std::auto_ptr<IDynamicObject> message(queue_.front());
+      queue_.pop_front();
+
+      return message.release();
+    }
+
+    IDynamicObject* Dequeue()
+    {
+      return Dequeue(0);
+    }
+  };
+
+
   /**
    * This class represents a message that is to be sent to some destination.
    **/
-  class MessageWithDestination : public boost::noncopyable
+  class MessageToDispatch : public boost::noncopyable
   {
   private:
     IDynamicObject* message_;
@@ -20,14 +100,14 @@
      * \param message The content of the message (takes the ownership)
      * \param destination The destination of the message
      **/
-    MessageWithDestination(IDynamicObject* message,
-                           const char* destination)
+    MessageToDispatch(IDynamicObject* message,
+                      const char* destination)
     {
       message_ = message;
       destination_ = destination;
     }
 
-    ~MessageWithDestination()
+    ~MessageToDispatch()
     {
       if (message_)
       {
@@ -35,6 +115,46 @@
       }
     }
   };
+
+
+  class IDestinationContext : public IDynamicObject
+  {
+  public:
+    virtual void Handle(const IDynamicObject& message) = 0;
+  };
+
+
+  class IDestinationContextFactory : public IDynamicObject
+  {
+  public:
+    virtual IDestinationContext* Construct(const char* destination) = 0;
+  };
+
+
+  class MessageDispatcher
+  {
+  private:
+    typedef std::map<std::string, IDestinationContext*>  ActiveContexts;
+
+    std::auto_ptr<IDestinationContextFactory> factory_;
+    ActiveContexts activeContexts_;
+    SharedMessageQueue queue_;
+
+  public:
+    MessageDispatcher(IDestinationContextFactory* factory)  // takes the ownership
+    {
+      factory_.reset(factory);
+    }
+
+    ~MessageDispatcher()
+    {
+      for (ActiveContexts::iterator it = activeContexts_.begin(); 
+           it != activeContexts_.end(); it++)
+      {
+        delete it->second;
+      }
+    }
+  };
 }
 
 
@@ -43,8 +163,8 @@
 
 using namespace Orthanc;
 
-TEST(MessageWithDestination, A)
+TEST(MessageToDispatch, A)
 {
-  MessageWithDestination a(new DicomString("coucou"), "pukkaj");
+  MessageToDispatch a(new DicomString("coucou"), "pukkaj");
 }