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