/** {@inheritDoc} */
@Override
public long write(final SelectionKey key) throws IOException {
final DatagramChannel channel = (DatagramChannel) key.channel();
final MessageBufferConsumer<ByteBuffer> chnIn = getChannelInput();
long n = 0;
long k = chnIn.remaining();
for (; k >= 0; k--) {
final SocketAddress address;
try {
final long sequence = chnIn.acquire();
try {
final ByteBuffer msg = chnIn.get(sequence);
sendBuffer.clear();
codec.put(msg, sendBuffer);
msg.clear();
sendBuffer.flip();
address = (SocketAddress) chnIn.attachment(sequence);
} finally {
chnIn.release(sequence);
}
} catch (final InterruptedException e) {
throw new IOException(e);
}
limiter.send(sendBuffer.remaining());
do {
n += channel.send(sendBuffer, address);
} while (sendBuffer.remaining() > 0);
}
if (chnIn.remaining() == 0) {
disableWriter();
}
return n;
}
/** {@inheritDoc} */
@Override
public long read(final SelectionKey key) throws IOException {
final DatagramChannel channel = (DatagramChannel) key.channel();
final MessageBufferProducer<ByteBuffer> chnOut = getChannelOutput();
final int start = receiveBuffer.position();
final SocketAddress source = channel.receive(receiveBuffer);
if (source == null) {
return 0;
}
final int n = receiveBuffer.position() - start;
limiter.receive(n);
receiveBuffer.flip();
while (codec.hasNext(receiveBuffer)) {
try {
final long sequence = chnOut.acquire();
try {
final ByteBuffer buffer = chnOut.get(sequence);
buffer.clear();
codec.get(receiveBuffer, buffer);
buffer.flip();
chnOut.attach(sequence, source);
} finally {
chnOut.release(sequence);
}
} catch (final InterruptedException e) {
throw new IOException(e);
}
}
if (receiveBuffer.remaining() > 0) {
receiveBuffer.compact();
} else {
receiveBuffer.clear();
}
return n;
}