UdpRegistryClientReadThread.h

Go to the documentation of this file.

#ifndef _MADARA_UDP_REGISTRY_CLIENT_READ_THREAD_H_
#define _MADARA_UDP_REGISTRY_CLIENT_READ_THREAD_H_

#include <string>

#include "madara/utility/ScopedArray.h"
#include "madara/knowledge/ThreadSafeContext.h"
#include "madara/transport/BandwidthMonitor.h"
#include "madara/transport/QoSTransportSettings.h"
#include "madara/expression/ExpressionTree.h"
#include "madara/transport/Transport.h"
#include "madara/transport/MessageHeader.h"
#include "madara/threads/BaseThread.h"

#include "ace/Task.h"
#include "ace/Mutex.h"
#include "ace/Barrier.h"
#include "ace/Atomic_Op_T.h"
#include "ace/Thread_Mutex.h"

#include "ace/Synch.h"
#include "ace/INET_Addr.h"
#include "ace/SOCK_Dgram.h"

namespace madara
{
  namespace transport
  {
    class UdpRegistryClientReadThread : public threads::BaseThread
    {
    public:
      UdpRegistryClientReadThread (
        const TransportSettings & settings,
        const std::string & id,
        std::map <std::string, ACE_INET_Addr> & addresses,
        ACE_SOCK_Dgram & socket,
        BandwidthMonitor & send_monitor,
        BandwidthMonitor & receive_monitor,
        PacketScheduler & packet_scheduler);
      
      void init (knowledge::KnowledgeBase & knowledge);

      void cleanup (void);

      void run (void);

      void rebroadcast (
        const char * print_prefix,
        MessageHeader * header,
        const knowledge::KnowledgeMap & records);

    private:
      const QoSTransportSettings settings_;

      const std::string                                 id_;
      
      knowledge::ThreadSafeContext * context_;
      
      std::map <std::string, ACE_INET_Addr> addresses_;
      
      ACE_SOCK_Dgram       &             socket_;
      
#ifndef _MADARA_NO_KARL_
      madara::knowledge::CompiledExpression  on_data_received_;
#endif // _MADARA_NO_KARL_
      
      madara::utility::ScopedArray <char>      buffer_;

      BandwidthMonitor   &   send_monitor_;
      
      BandwidthMonitor   &   receive_monitor_;

      PacketScheduler    &   packet_scheduler_;
    };
  }
}

#endif // _MADARA_UDP_REGISTRY_CLIENT_READ_THREAD_H_