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