1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.mina.handler.demux;
21
22 import java.util.Collections;
23 import java.util.Map;
24 import java.util.Set;
25 import java.util.concurrent.ConcurrentHashMap;
26
27 import org.apache.mina.core.service.IoHandler;
28 import org.apache.mina.core.service.IoHandlerAdapter;
29 import org.apache.mina.core.session.IoSession;
30 import org.apache.mina.core.session.UnknownMessageTypeException;
31 import org.apache.mina.util.IdentityHashSet;
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 public class DemuxingIoHandler extends IoHandlerAdapter {
80
81 private final Map<Class<?>, MessageHandler<?>> receivedMessageHandlerCache =
82 new ConcurrentHashMap<Class<?>, MessageHandler<?>>();
83
84 private final Map<Class<?>, MessageHandler<?>> receivedMessageHandlers =
85 new ConcurrentHashMap<Class<?>, MessageHandler<?>>();
86
87 private final Map<Class<?>, MessageHandler<?>> sentMessageHandlerCache =
88 new ConcurrentHashMap<Class<?>, MessageHandler<?>>();
89
90 private final Map<Class<?>, MessageHandler<?>> sentMessageHandlers =
91 new ConcurrentHashMap<Class<?>, MessageHandler<?>>();
92
93 private final Map<Class<?>, ExceptionHandler<?>> exceptionHandlerCache =
94 new ConcurrentHashMap<Class<?>, ExceptionHandler<?>>();
95
96 private final Map<Class<?>, ExceptionHandler<?>> exceptionHandlers =
97 new ConcurrentHashMap<Class<?>, ExceptionHandler<?>>();
98
99
100
101
102 public DemuxingIoHandler() {
103
104 }
105
106
107
108
109
110
111
112
113 @SuppressWarnings("unchecked")
114 public <E> MessageHandler<? super E> addReceivedMessageHandler(Class<E> type,
115 MessageHandler<? super E> handler) {
116 receivedMessageHandlerCache.clear();
117 return (MessageHandler<? super E>) receivedMessageHandlers.put(type, handler);
118 }
119
120
121
122
123
124
125
126 @SuppressWarnings("unchecked")
127 public <E> MessageHandler<? super E> removeReceivedMessageHandler(Class<E> type) {
128 receivedMessageHandlerCache.clear();
129 return (MessageHandler<? super E>) receivedMessageHandlers.remove(type);
130 }
131
132
133
134
135
136
137
138
139 @SuppressWarnings("unchecked")
140 public <E> MessageHandler<? super E> addSentMessageHandler(Class<E> type,
141 MessageHandler<? super E> handler) {
142 sentMessageHandlerCache.clear();
143 return (MessageHandler<? super E>) sentMessageHandlers.put(type, handler);
144 }
145
146
147
148
149
150
151
152 @SuppressWarnings("unchecked")
153 public <E> MessageHandler<? super E> removeSentMessageHandler(Class<E> type) {
154 sentMessageHandlerCache.clear();
155 return (MessageHandler<? super E>) sentMessageHandlers.remove(type);
156 }
157
158
159
160
161
162
163
164
165 @SuppressWarnings("unchecked")
166 public <E extends Throwable>
167 ExceptionHandler<? super E> addExceptionHandler(
168 Class<E> type, ExceptionHandler<? super E> handler) {
169 exceptionHandlerCache.clear();
170 return (ExceptionHandler<? super E>) exceptionHandlers.put(type, handler);
171 }
172
173
174
175
176
177
178
179 @SuppressWarnings("unchecked")
180 public <E extends Throwable> ExceptionHandler<? super E>
181 removeExceptionHandler(Class<E> type) {
182 exceptionHandlerCache.clear();
183 return (ExceptionHandler<? super E>) exceptionHandlers.remove(type);
184 }
185
186
187
188
189
190 @SuppressWarnings("unchecked")
191 public <E> MessageHandler<? super E> getMessageHandler(Class<E> type) {
192 return (MessageHandler<? super E>) receivedMessageHandlers.get(type);
193 }
194
195
196
197
198
199 public Map<Class<?>, MessageHandler<?>> getReceivedMessageHandlerMap() {
200 return Collections.unmodifiableMap(receivedMessageHandlers);
201 }
202
203
204
205
206
207 public Map<Class<?>, MessageHandler<?>> getSentMessageHandlerMap() {
208 return Collections.unmodifiableMap(sentMessageHandlers);
209 }
210
211
212
213
214
215 public Map<Class<?>, ExceptionHandler<?>> getExceptionHandlerMap() {
216 return Collections.unmodifiableMap(exceptionHandlers);
217 }
218
219
220
221
222
223
224
225
226
227 @Override
228 public void messageReceived(IoSession session, Object message)
229 throws Exception {
230 MessageHandler<Object> handler = findReceivedMessageHandler(message.getClass());
231 if (handler != null) {
232 handler.handleMessage(session, message);
233 } else {
234 throw new UnknownMessageTypeException(
235 "No message handler found for message type: " +
236 message.getClass().getSimpleName());
237 }
238 }
239
240
241
242
243
244
245
246
247 @Override
248 public void messageSent(IoSession session, Object message) throws Exception {
249 MessageHandler<Object> handler = findSentMessageHandler(message.getClass());
250 if (handler != null) {
251 handler.handleMessage(session, message);
252 } else {
253 throw new UnknownMessageTypeException(
254 "No handler found for message type: " +
255 message.getClass().getSimpleName());
256 }
257 }
258
259
260
261
262
263
264
265
266
267
268 @Override
269 public void exceptionCaught(IoSession session, Throwable cause) throws Exception {
270 ExceptionHandler<Throwable> handler = findExceptionHandler(cause.getClass());
271 if (handler != null) {
272 handler.exceptionCaught(session, cause);
273 } else {
274 throw new UnknownMessageTypeException(
275 "No handler found for exception type: " +
276 cause.getClass().getSimpleName());
277 }
278 }
279
280 protected MessageHandler<Object> findReceivedMessageHandler(Class<?> type) {
281 return findReceivedMessageHandler(type, null);
282 }
283
284 protected MessageHandler<Object> findSentMessageHandler(Class<?> type) {
285 return findSentMessageHandler(type, null);
286 }
287
288 protected ExceptionHandler<Throwable> findExceptionHandler(Class<? extends Throwable> type) {
289 return findExceptionHandler(type, null);
290 }
291
292 @SuppressWarnings("unchecked")
293 private MessageHandler<Object> findReceivedMessageHandler(
294 Class type, Set<Class> triedClasses) {
295
296 return (MessageHandler<Object>) findHandler(
297 receivedMessageHandlers, receivedMessageHandlerCache, type, triedClasses);
298 }
299
300 @SuppressWarnings("unchecked")
301 private MessageHandler<Object> findSentMessageHandler(
302 Class type, Set<Class> triedClasses) {
303
304 return (MessageHandler<Object>) findHandler(
305 sentMessageHandlers, sentMessageHandlerCache, type, triedClasses);
306 }
307
308 @SuppressWarnings("unchecked")
309 private ExceptionHandler<Throwable> findExceptionHandler(
310 Class type, Set<Class> triedClasses) {
311
312 return (ExceptionHandler<Throwable>) findHandler(
313 exceptionHandlers, exceptionHandlerCache, type, triedClasses);
314 }
315
316 @SuppressWarnings("unchecked")
317 private Object findHandler(
318 Map handlers, Map handlerCache,
319 Class type, Set<Class> triedClasses) {
320
321 Object handler = null;
322
323 if (triedClasses != null && triedClasses.contains(type)) {
324 return null;
325 }
326
327
328
329
330 handler = handlerCache.get(type);
331 if (handler != null) {
332 return handler;
333 }
334
335
336
337
338 handler = handlers.get(type);
339
340 if (handler == null) {
341
342
343
344
345 if (triedClasses == null) {
346 triedClasses = new IdentityHashSet<Class>();
347 }
348 triedClasses.add(type);
349
350 Class[] interfaces = type.getInterfaces();
351 for (Class element : interfaces) {
352 handler = findHandler(handlers, handlerCache, element, triedClasses);
353 if (handler != null) {
354 break;
355 }
356 }
357 }
358
359 if (handler == null) {
360
361
362
363
364 Class superclass = type.getSuperclass();
365 if (superclass != null) {
366 handler = findHandler(handlers, handlerCache, superclass, null);
367 }
368 }
369
370
371
372
373
374
375 if (handler != null) {
376 handlerCache.put(type, handler);
377 }
378
379 return handler;
380 }
381 }