ACE/examples/APG/Threads/Message_Receiver.h

   1 /**
   2  * Sample code from The ACE Programmer's Guide,
   3  * copyright 2003 Addison-Wesley. All Rights Reserved.
   4  */
   5
   6 #ifndef __MESSAGE_RECEIVER_H_
   7 #define __MESSAGE_RECEIVER_H_
   8
   9 #include "ace/Log_Msg.h"
  10 #include "ace/Message_Block.h"
  11 #include "ace/SOCK_Stream.h"
  12 #include "ace/Svc_Handler.h"
  13 #include "ace/Synch.h"
  14 #include "ace/Task.h"
  15
  16 // Listing 1 code/ch12
  17 struct DeviceCommandHeader
  18 {
  19   int length_;
  20   int deviceId_;
  21 };
  22 // Listing 1
  23
  24 class HA_Device_Repository
  25 {
  26 public:
  27   HA_Device_Repository ();
  28
  29   int update_device (int device_id, char *commands);
  30
  31 private:
  32   ACE_Task_Base *owner_;
  33 };
  34
  35 HA_Device_Repository::HA_Device_Repository ()
  36 { }
  37
  38 int
  39 HA_Device_Repository::update_device (int, char *)
  40 {
  41   return 0;
  42 }
  43
  44 class HA_CommandHandler : public ACE_Task<ACE_MT_SYNCH>
  45 {
  46 public:
  47   HA_CommandHandler (HA_Device_Repository &rep) : rep_(rep)
  48   { }
  49
  50   virtual int svc();
  51
  52 private:
  53   HA_Device_Repository &rep_;
  54 };
  55
  56 class Message_Receiver :
  57   public ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_MT_SYNCH>
  58 {
  59 public:
  60   Message_Receiver () : handler_(0)
  61   {
  62     ACE_ASSERT(0);
  63   }
  64
  65   Message_Receiver (HA_CommandHandler *ch) : handler_(ch)
  66   { }
  67
  68   ACE_Message_Block *shut_down_message ();
  69
  70   virtual int handle_input (ACE_HANDLE fd);
  71
  72   virtual int handle_close (ACE_HANDLE = ACE_INVALID_HANDLE,
  73                             ACE_Reactor_Mask = ACE_Event_Handler::ALL_EVENTS_MASK)
  74   {
  75     this->peer ().close ();
  76     delete this;
  77     return 0;
  78   }
  79
  80 private:
  81   int read_header (DeviceCommandHeader *dch);
  82   int copy_payload (ACE_Message_Block *mb, int payload_length);
  83
  84 private:
  85   HA_CommandHandler *handler_;
  86 };
  87
  88 #endif /* __MESSAGE_RECEIVER_H */