ACE/tests/Proactor_Test.h

   1
   2 //=============================================================================
   3 /**
   4  *  @file    Proactor_Test.h
   5  *
   6  *  Define class needed for generating templates. IBM C++ requires this to
   7  *  be in its own file for auto template instantiation.
   8  *
   9  *  @author @author Alexander Libman <alibman@baltimore.com>
  10  */
  11 //=============================================================================
  12
  13
  14 #ifndef ACE_TESTS_PROACTOR_TEST_H
  15 #define ACE_TESTS_PROACTOR_TEST_H
  16
  17 #include "ace/Synch_Traits.h"
  18 #include "ace/Thread_Mutex.h"
  19
  20 // forward declaration
  21 class TestData;
  22
  23 class Server : public ACE_Service_Handler
  24 {
  25 public:
  26   Server ();
  27   Server (TestData *tester, int id);
  28   ~Server ();
  29
  30   int id () { return this->id_; }
  31   size_t get_total_snd () { return this->total_snd_; }
  32   size_t get_total_rcv () { return this->total_rcv_; }
  33   long get_total_w   () { return this->total_w_; }
  34   long get_total_r   () { return this->total_r_; }
  35
  36   // This is called to pass the new connection's addresses.
  37   virtual void addresses (const ACE_INET_Addr& peer,
  38                           const ACE_INET_Addr& local);
  39
  40   /// This is called after the new connection has been accepted.
  41   virtual void open (ACE_HANDLE handle,
  42                      ACE_Message_Block &message_block);
  43   void cancel ();
  44
  45 protected:
  46   //// This is called when asynchronous <read> operation from the
  47   //// socket completes.
  48   /**
  49    * @name AIO callback handling
  50    *
  51    * These methods are called by the framework
  52    */
  53   virtual void handle_read_stream (const ACE_Asynch_Read_Stream::Result &result);
  54
  55   /// This is called when an asynchronous <write> to the socket
  56   /// completes.
  57   virtual void handle_write_stream (const ACE_Asynch_Write_Stream::Result &result);
  58
  59 private:
  60   int initiate_read_stream ();
  61   int initiate_write_stream (ACE_Message_Block &mb, size_t nbytes);
  62
  63   TestData *tester_;
  64   int id_;
  65
  66   ACE_Asynch_Read_Stream rs_;
  67   ACE_Asynch_Write_Stream ws_;
  68   ACE_HANDLE handle_;
  69   ACE_SYNCH_MUTEX lock_;
  70
  71   int io_count_;             // Number of currently outstanding I/O requests
  72   int flg_cancel_;
  73   size_t total_snd_;         // Number of bytes successfully sent
  74   size_t total_rcv_;         // Number of bytes successfully received
  75   int total_w_;              // Number of write operations
  76   int total_r_;              // Number of read operations
  77 };
  78
  79 // *******************************************
  80 //   Client
  81 // *******************************************
  82
  83 class Client : public ACE_Service_Handler
  84 {
  85 public:
  86   /// This is called after the new connection has been established.
  87   virtual void open (ACE_HANDLE handle,
  88                      ACE_Message_Block &message_block);
  89
  90   Client ();
  91   Client (TestData *tester, int id);
  92   ~Client ();
  93
  94   int id () { return this->id_; }
  95   size_t get_total_snd () { return this->total_snd_; }
  96   size_t get_total_rcv () { return this->total_rcv_; }
  97   int    get_total_w   () { return this->total_w_; }
  98   int    get_total_r   () { return this->total_r_; }
  99
 100   // This is called to pass the new connection's addresses.
 101   virtual void addresses (const ACE_INET_Addr& peer,
 102                           const ACE_INET_Addr& local);
 103
 104   /// This is called when asynchronous reads from the socket complete
 105   virtual void handle_read_stream (const ACE_Asynch_Read_Stream::Result &result);
 106
 107   /// This is called when asynchronous writes from the socket complete
 108   virtual void handle_write_stream (const ACE_Asynch_Write_Stream::Result &result);
 109
 110   void cancel ();
 111
 112 private:
 113   int initiate_read_stream ();
 114   int initiate_write_stream ();
 115   void close ();
 116
 117   TestData *tester_;
 118   int id_;
 119
 120   ACE_Asynch_Read_Stream rs_;
 121   ACE_Asynch_Write_Stream ws_;
 122   ACE_HANDLE handle_;
 123
 124   ACE_SYNCH_MUTEX lock_;
 125
 126   int io_count_;
 127   int stop_writing_;           // Writes are shut down; just read.
 128   int flg_cancel_;
 129   size_t total_snd_;
 130   size_t total_rcv_;
 131   int total_w_;
 132   int total_r_;
 133 };
 134
 135 #endif /* ACE_TESTS_PROACTOR_TEST_H */