1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.mina.transport.socket.nio.support;
21
22 import java.util.Queue;
23
24 import org.apache.mina.common.ByteBuffer;
25 import org.apache.mina.common.IoFilter.WriteRequest;
26 import org.apache.mina.common.IoFilterChain;
27 import org.apache.mina.common.IoSession;
28 import org.apache.mina.common.support.AbstractIoFilterChain;
29
30
31
32
33
34
35 class DatagramFilterChain extends AbstractIoFilterChain {
36
37 DatagramFilterChain(IoSession parent) {
38 super(parent);
39 }
40
41 @Override
42 protected void doWrite(IoSession session, WriteRequest writeRequest) {
43 DatagramSessionImpl s = (DatagramSessionImpl) session;
44 Queue<WriteRequest> writeRequestQueue = s.getWriteRequestQueue();
45
46
47
48 ByteBuffer buffer = (ByteBuffer) writeRequest.getMessage();
49 buffer.mark();
50
51 int remaining = buffer.remaining();
52 if (remaining == 0) {
53 s.increaseScheduledWriteRequests();
54 } else {
55 s.increaseScheduledWriteBytes(buffer.remaining());
56 }
57
58 writeRequestQueue.add(writeRequest);
59
60 if (session.getTrafficMask().isWritable()) {
61 s.getManagerDelegate().flushSession(s);
62 }
63 }
64
65 @Override
66 protected void doClose(IoSession session) {
67 DatagramSessionImpl s = (DatagramSessionImpl) session;
68 DatagramService manager = s.getManagerDelegate();
69 if (manager instanceof DatagramConnectorDelegate) {
70 manager.closeSession(s);
71 } else {
72 ((DatagramAcceptorDelegate) manager).getListeners()
73 .fireSessionDestroyed(session);
74 session.getCloseFuture().setClosed();
75 }
76 }
77 }