1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.mina.example.proxy;
21
22 import java.nio.charset.Charset;
23
24 import org.apache.mina.core.buffer.IoBuffer;
25 import org.apache.mina.core.service.IoHandlerAdapter;
26 import org.apache.mina.core.session.IoSession;
27 import org.apache.mina.core.session.TrafficMask;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30
31
32
33
34
35
36
37
38
39 public abstract class AbstractProxyIoHandler extends IoHandlerAdapter {
40 private static final Charset CHARSET = Charset.forName("iso8859-1");
41 public static final String OTHER_IO_SESSION = AbstractProxyIoHandler.class.getName()+".OtherIoSession";
42
43 private final Logger logger = LoggerFactory.getLogger(getClass());
44
45 @Override
46 public void sessionCreated(IoSession session) throws Exception {
47 session.setTrafficMask(TrafficMask.NONE);
48 }
49
50 @Override
51 public void sessionClosed(IoSession session) throws Exception {
52 if (session.getAttribute( OTHER_IO_SESSION ) != null) {
53 IoSession sess = (IoSession) session.getAttribute(OTHER_IO_SESSION);
54 sess.setAttribute(OTHER_IO_SESSION, null);
55 sess.closeOnFlush();
56 session.setAttribute(OTHER_IO_SESSION, null);
57 }
58 }
59
60 @Override
61 public void messageReceived(IoSession session, Object message)
62 throws Exception {
63 IoBuffer rb = (IoBuffer) message;
64 IoBuffer wb = IoBuffer.allocate(rb.remaining());
65 rb.mark();
66 wb.put(rb);
67 wb.flip();
68 ((IoSession) session.getAttribute(OTHER_IO_SESSION)).write(wb);
69 rb.reset();
70 logger.info(rb.getString(CHARSET.newDecoder()));
71 }
72 }