From c2ae73d1963409a496c2dbefb0ee1505b944c2c8 Mon Sep 17 00:00:00 2001 From: Steve Huston Date: Mon, 15 Mar 2004 21:12:55 +0000 Subject: ChangeLogTag:Mon Mar 15 15:42:33 2004 Steve Huston --- tests/Proactor_Test.h | 141 ++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 141 insertions(+) create mode 100644 tests/Proactor_Test.h (limited to 'tests/Proactor_Test.h') diff --git a/tests/Proactor_Test.h b/tests/Proactor_Test.h new file mode 100644 index 00000000000..6d36e5fe137 --- /dev/null +++ b/tests/Proactor_Test.h @@ -0,0 +1,141 @@ +// $Id$ + +// ============================================================================ +// +// = LIBRARY +// tests +// +// = FILENAME +// Proactor_Test.h +// +// = DESCRIPTION +// Define class needed for generating templates. IBM C++ requires this to +// be in its own file for auto template instantiation. +// +// = AUTHOR +// @author Alexander Libman +// +// ============================================================================ + +#ifndef ACE_TESTS_PROACTOR_TEST_H +#define ACE_TESTS_PROACTOR_TEST_H + +#include "ace/Synch_Traits.h" +#include "ace/Thread_Mutex.h" + +// forward declaration +class Acceptor; + +class Receiver : public ACE_Service_Handler +{ + friend class Acceptor; +public: + Receiver (Acceptor *acceptor = 0, int index = -1); + ~Receiver (void); + + size_t get_total_snd (void) { return this->total_snd_; } + size_t get_total_rcv (void) { return this->total_rcv_; } + long get_total_w (void) { return this->total_w_; } + long get_total_r (void) { return this->total_r_; } + + // This is called to pass the new connection's addresses. + virtual void addresses (const ACE_INET_Addr& peer, + const ACE_INET_Addr& local); + + /// This is called after the new connection has been accepted. + virtual void open (ACE_HANDLE handle, + ACE_Message_Block &message_block); + +protected: + /** + * @name AIO callback handling + * + * These methods are called by the framework + */ + /// This is called when asynchronous operation from the + /// socket completes. + virtual void handle_read_stream (const ACE_Asynch_Read_Stream::Result &result); + + /// This is called when an asynchronous to the socket + /// completes. + virtual void handle_write_stream (const ACE_Asynch_Write_Stream::Result &result); + +private: + int initiate_read_stream (void); + int initiate_write_stream (ACE_Message_Block &mb, size_t nbytes); + void cancel (); + + Acceptor *acceptor_; + int index_; + + ACE_Asynch_Read_Stream rs_; + ACE_Asynch_Write_Stream ws_; + ACE_HANDLE handle_; + ACE_SYNCH_MUTEX lock_; + + long io_count_; // Number of currently outstanding I/O requests + int flg_cancel_; + size_t total_snd_; // Number of bytes successfully sent + size_t total_rcv_; // Number of bytes successfully received + long total_w_; // Number of write operations + long total_r_; // Number of read operations +}; + +// ******************************************* +// Sender +// ******************************************* + +class Connector; + +class Sender : public ACE_Service_Handler +{ + friend class Connector; +public: + + /// This is called after the new connection has been established. + virtual void open (ACE_HANDLE handle, + ACE_Message_Block &message_block); + + Sender (Connector *connector = 0, int index = -1); + ~Sender (void); + + size_t get_total_snd (void) { return this->total_snd_; } + size_t get_total_rcv (void) { return this->total_rcv_; } + long get_total_w (void) { return this->total_w_; } + long get_total_r (void) { return this->total_r_; } + + // This is called to pass the new connection's addresses. + virtual void addresses (const ACE_INET_Addr& peer, + const ACE_INET_Addr& local); + + virtual void handle_read_stream (const ACE_Asynch_Read_Stream::Result &result); + // This is called when asynchronous reads from the socket complete + + virtual void handle_write_stream (const ACE_Asynch_Write_Stream::Result &result); + // This is called when asynchronous writes from the socket complete + +private: + int initiate_read_stream (void); + int initiate_write_stream (void); + void cancel (void); + void close (void); + + int index_; + Connector * connector_; + + ACE_Asynch_Read_Stream rs_; + ACE_Asynch_Write_Stream ws_; + ACE_HANDLE handle_; + + ACE_SYNCH_MUTEX lock_; + + long io_count_; + int stop_writing_; // Writes are shut down; just read. + int flg_cancel_; + size_t total_snd_; + size_t total_rcv_; + long total_w_; + long total_r_; +}; + +#endif /* ACE_TESTS_PROACTOR_TEST_H */ -- cgit v1.2.1