ACE/protocols/ace/RMCast/Retransmit.h

   1 // author    : Boris Kolpackov <boris@kolpackov.net>
   2 #ifndef ACE_RMCAST_RETRANSMIT_H
   3 #define ACE_RMCAST_RETRANSMIT_H
   4
   5 #include "ace/Hash_Map_Manager.h"
   6 #include "ace/Thread_Manager.h"
   7
   8 #include "Stack.h"
   9 #include "Protocol.h"
  10 #include "Bits.h"
  11 #include "Parameters.h"
  12
  13 namespace ACE_RMCast
  14 {
  15   class Retransmit : public Element
  16   {
  17   public:
  18     Retransmit (Parameters const& params);
  19
  20     virtual void
  21     out_start (Out_Element* out);
  22
  23     virtual void
  24     out_stop ();
  25
  26   public:
  27     virtual void
  28     send (Message_ptr m);
  29
  30     virtual void
  31     recv (Message_ptr m);
  32
  33   private:
  34     struct Descr
  35     {
  36       // Shouldn't be available but ACE_Hash_Map needs it.
  37       //
  38       Descr ()
  39           : msg_ (), count_ (0)
  40       {
  41       }
  42
  43       Descr (Message_ptr msg)
  44           : msg_ (msg), count_ (0)
  45       {
  46       }
  47
  48       unsigned long
  49       inc ()
  50       {
  51         return ++count_;
  52       }
  53
  54       void
  55       reset ()
  56       {
  57         count_ = 0;
  58       }
  59
  60       Message_ptr
  61       message () const
  62       {
  63         return msg_->clone ();
  64       }
  65
  66     private:
  67       Message_ptr msg_;
  68       unsigned long count_;
  69     };
  70
  71     typedef
  72     ACE_Hash_Map_Manager<u64, Descr, ACE_Null_Mutex>
  73     Queue;
  74
  75   private:
  76     void
  77     track ();
  78
  79     static ACE_THR_FUNC_RETURN
  80     track_thunk (void* obj);
  81
  82   private:
  83     Parameters const& params_;
  84
  85     Queue queue_;
  86     Mutex mutex_;
  87     Condition cond_;
  88
  89     bool stop_;
  90     ACE_Thread_Manager tracker_mgr_;
  91   };
  92 }
  93
  94
  95 #endif  // ACE_RMCAST_RETRANSMIT_H