1 |
| |
2 |
| |
3 |
| |
4 |
| |
5 |
| |
6 |
| |
7 |
| |
8 |
| package org.codehaus.spice.netserve.connection.handlers; |
9 |
| |
10 |
| import java.io.IOException; |
11 |
| import java.net.Socket; |
12 |
| import java.util.HashSet; |
13 |
| import java.util.Set; |
14 |
| import org.codehaus.spice.netserve.connection.RequestHandler; |
15 |
| |
16 |
| |
17 |
| |
18 |
| |
19 |
| |
20 |
| |
21 |
| |
22 |
| public abstract class AbstractRequestHandler |
23 |
| implements RequestHandler |
24 |
| { |
25 |
| |
26 |
| |
27 |
| |
28 |
| private final Set m_activeRequests = new HashSet(); |
29 |
| |
30 |
| |
31 |
| |
32 |
| |
33 |
| private boolean m_shutdown; |
34 |
| |
35 |
| |
36 |
| |
37 |
| |
38 |
| |
39 |
| |
40 |
27
| public void handleConnection( Socket socket )
|
41 |
| { |
42 |
27
| performRequest( socket );
|
43 |
| } |
44 |
| |
45 |
| |
46 |
| |
47 |
| |
48 |
30
| public void shutdown( final long timeout )
|
49 |
| { |
50 |
30
| markAsShutdown();
|
51 |
30
| final Thread[] threads;
|
52 |
30
| synchronized( this )
|
53 |
| { |
54 |
30
| threads = (Thread[])m_activeRequests.
|
55 |
| toArray( new Thread[ m_activeRequests.size() ] ); |
56 |
| } |
57 |
30
| for( int i = 0; i < threads.length; i++ )
|
58 |
| { |
59 |
27
| final Thread thread = threads[ i ];
|
60 |
27
| thread.interrupt();
|
61 |
| } |
62 |
30
| final long now = System.currentTimeMillis();
|
63 |
30
| final long then = now + timeout;
|
64 |
| |
65 |
30
| while( System.currentTimeMillis() < then || 0 == timeout )
|
66 |
| { |
67 |
47
| synchronized( this )
|
68 |
| { |
69 |
47
| if( 0 == m_activeRequests.size() )
|
70 |
| { |
71 |
21
| return;
|
72 |
| } |
73 |
26
| try
|
74 |
| { |
75 |
26
| wait( timeout );
|
76 |
| } |
77 |
| catch( final InterruptedException ie ) |
78 |
| { |
79 |
| |
80 |
| } |
81 |
| } |
82 |
| } |
83 |
| } |
84 |
| |
85 |
| |
86 |
| |
87 |
| |
88 |
39
| protected void markAsShutdown()
|
89 |
| { |
90 |
39
| m_shutdown = true;
|
91 |
| } |
92 |
| |
93 |
| |
94 |
| |
95 |
| |
96 |
| |
97 |
| |
98 |
18
| protected boolean isShutdown()
|
99 |
| { |
100 |
18
| return m_shutdown;
|
101 |
| } |
102 |
| |
103 |
| |
104 |
| |
105 |
| |
106 |
| |
107 |
| |
108 |
| |
109 |
42
| protected void performRequest( final Socket socket )
|
110 |
| { |
111 |
42
| synchronized( this )
|
112 |
| { |
113 |
42
| m_activeRequests.add( Thread.currentThread() );
|
114 |
| } |
115 |
42
| setupThreadName( socket );
|
116 |
42
| try
|
117 |
| { |
118 |
42
| doPerformRequest( socket );
|
119 |
| } |
120 |
| catch( final Throwable t ) |
121 |
| { |
122 |
3
| errorHandlingConnection( socket, t );
|
123 |
| } |
124 |
| finally |
125 |
| { |
126 |
33
| endConnection( socket );
|
127 |
33
| synchronized( this )
|
128 |
| { |
129 |
33
| m_activeRequests.remove( Thread.currentThread() );
|
130 |
33
| notifyAll();
|
131 |
| } |
132 |
| } |
133 |
| } |
134 |
| |
135 |
| |
136 |
| |
137 |
| |
138 |
| |
139 |
| |
140 |
| |
141 |
| protected abstract void doPerformRequest( Socket socket ) |
142 |
| throws Exception; |
143 |
| |
144 |
| |
145 |
| |
146 |
| |
147 |
| |
148 |
| |
149 |
42
| protected void setupThreadName( final Socket socket )
|
150 |
| { |
151 |
42
| final String name = getThreadName( socket );
|
152 |
42
| Thread.currentThread().setName( name );
|
153 |
| } |
154 |
| |
155 |
| |
156 |
| |
157 |
| |
158 |
| |
159 |
| |
160 |
36
| protected void endConnection( final Socket socket )
|
161 |
| { |
162 |
36
| if( socket.isConnected() )
|
163 |
| { |
164 |
6
| try
|
165 |
| { |
166 |
6
| socket.close();
|
167 |
| } |
168 |
| catch( final IOException ioe ) |
169 |
| { |
170 |
3
| errorClosingConnection( socket, ioe );
|
171 |
| } |
172 |
| } |
173 |
| } |
174 |
| |
175 |
| |
176 |
| |
177 |
| |
178 |
| |
179 |
| |
180 |
| |
181 |
15
| protected Runnable createRunnable( final Socket socket )
|
182 |
| { |
183 |
15
| return new Runnable()
|
184 |
| { |
185 |
12
| public void run()
|
186 |
| { |
187 |
12
| performRequest( socket );
|
188 |
| } |
189 |
| }; |
190 |
| } |
191 |
| |
192 |
| |
193 |
| |
194 |
| |
195 |
| |
196 |
| |
197 |
| |
198 |
42
| protected String getThreadName( final Socket socket )
|
199 |
| { |
200 |
42
| if( socket.isConnected() )
|
201 |
| { |
202 |
3
| return "RequestHandler for " +
|
203 |
| socket.getInetAddress().getHostAddress() + ":" + |
204 |
| socket.getPort(); |
205 |
| } |
206 |
| else |
207 |
| { |
208 |
39
| return "RequestHandler for " + socket;
|
209 |
| } |
210 |
| } |
211 |
| |
212 |
| |
213 |
| |
214 |
| |
215 |
| |
216 |
| |
217 |
| |
218 |
6
| protected void errorHandlingConnection( final Socket socket,
|
219 |
| final Throwable t ) |
220 |
| { |
221 |
| } |
222 |
| |
223 |
| |
224 |
| |
225 |
| |
226 |
| |
227 |
| |
228 |
| |
229 |
3
| protected void errorClosingConnection( final Socket socket,
|
230 |
| final Throwable t ) |
231 |
| { |
232 |
3
| errorHandlingConnection( socket, t );
|
233 |
| } |
234 |
| } |