private void write(SelectionKey key) throws IOException {
SocketChannel ch = (SocketChannel) key.channel();
int count = 0;
synchronized (pendingData) {
List queue = (List) pendingData.get(ch);
/**
* Null exception alert: This can happen once a connection was established. Check send() for
* more information.
*/
if (queue == null) return;
while (!queue.isEmpty()) {
ByteBuffer buf = (ByteBuffer) queue.get(0);
ch.write(buf);
count += buf.capacity() - buf.remaining();
if (buf.remaining() > 0) break;
queue.remove(0);
}
if (queue.isEmpty()) key.interestOps(SelectionKey.OP_READ);
}
if (!listener.handleWrite(ch, count)) close(ch);
}
public void accept(SelectionKey key) throws IOException {
ServerSocketChannel ch = (ServerSocketChannel) key.channel();
SocketChannel s_ch = ch.accept();
s_ch.configureBlocking(false);
s_ch.register(selector, SelectionKey.OP_READ);
if (!listener.handleConnection(s_ch)) {
s_ch.close();
key.cancel();
}
}
private void read(SelectionKey key) throws IOException {
SocketChannel ch = (SocketChannel) key.channel();
ByteBuffer buffer = ByteBuffer.allocate(1024);
int count;
try {
count = ch.read(buffer);
} catch (IOException e) {
close(ch);
return;
}
buffer.flip();
if (count == -1 || (listener != null && !listener.handleRead(ch, buffer, count))) close(ch);
}
public void close(SocketChannel ch) {
if (!listener.handleConnectionClose(ch)) return;
try {
ch.close();
} catch (IOException e) {
e.printStackTrace();
}
ch.keyFor(selector).cancel();
synchronized (changeRequests) {
Iterator changes = changeRequests.iterator();
while (changes.hasNext()) {
ChangeRequest req = (ChangeRequest) changes.next();
if (req.socket == ch) {
changeRequests.remove(req);
break;
}
}
}
}
