1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.eclipse.jetty.io;
20
21 import java.io.IOException;
22 import java.net.Socket;
23 import java.nio.ByteBuffer;
24 import java.nio.channels.SelectionKey;
25 import java.nio.channels.SocketChannel;
26 import java.util.List;
27
28 import org.eclipse.jetty.util.log.Log;
29 import org.eclipse.jetty.util.log.Logger;
30 import org.eclipse.jetty.util.thread.Scheduler;
31
32 public class NetworkTrafficSelectChannelEndPoint extends SelectChannelEndPoint
33 {
34 private static final Logger LOG = Log.getLogger(NetworkTrafficSelectChannelEndPoint.class);
35
36 private final List<NetworkTrafficListener> listeners;
37
38 public NetworkTrafficSelectChannelEndPoint(SocketChannel channel, ManagedSelector selectSet, SelectionKey key, Scheduler scheduler, long idleTimeout, List<NetworkTrafficListener> listeners) throws IOException
39 {
40 super(channel, selectSet, key, scheduler, idleTimeout);
41 this.listeners = listeners;
42 }
43
44 @Override
45 public int fill(ByteBuffer buffer) throws IOException
46 {
47 int read = super.fill(buffer);
48 notifyIncoming(buffer, read);
49 return read;
50 }
51
52 @Override
53 public boolean flush(ByteBuffer... buffers) throws IOException
54 {
55 boolean flushed=true;
56 for (ByteBuffer b : buffers)
57 {
58 if (b.hasRemaining())
59 {
60 int position = b.position();
61 ByteBuffer view=b.slice();
62 flushed&=super.flush(b);
63 int l=b.position()-position;
64 view.limit(view.position()+l);
65 notifyOutgoing(view);
66 if (!flushed)
67 break;
68 }
69 }
70 return flushed;
71 }
72
73
74
75 @Override
76 public void onOpen()
77 {
78 super.onOpen();
79 if (listeners != null && !listeners.isEmpty())
80 {
81 for (NetworkTrafficListener listener : listeners)
82 {
83 try
84 {
85 listener.opened(getSocket());
86 }
87 catch (Exception x)
88 {
89 LOG.warn(x);
90 }
91 }
92 }
93 }
94
95 @Override
96 public void onClose()
97 {
98 super.onClose();
99 if (listeners != null && !listeners.isEmpty())
100 {
101 for (NetworkTrafficListener listener : listeners)
102 {
103 try
104 {
105 listener.closed(getSocket());
106 }
107 catch (Exception x)
108 {
109 LOG.warn(x);
110 }
111 }
112 }
113 }
114
115
116 public void notifyIncoming(ByteBuffer buffer, int read)
117 {
118 if (listeners != null && !listeners.isEmpty() && read > 0)
119 {
120 for (NetworkTrafficListener listener : listeners)
121 {
122 try
123 {
124 ByteBuffer view = buffer.asReadOnlyBuffer();
125 listener.incoming(getSocket(), view);
126 }
127 catch (Exception x)
128 {
129 LOG.warn(x);
130 }
131 }
132 }
133 }
134
135 public void notifyOutgoing(ByteBuffer view)
136 {
137 if (listeners != null && !listeners.isEmpty() && view.hasRemaining())
138 {
139 Socket socket=getSocket();
140 for (NetworkTrafficListener listener : listeners)
141 {
142 try
143 {
144 listener.outgoing(socket, view);
145 }
146 catch (Exception x)
147 {
148 LOG.warn(x);
149 }
150 }
151 }
152 }
153
154 }