一个Selector能够管理多个信道上的I/O操作代码,Java代码库,德仔网

代码语言
代码分类
【Java】一个Selector能够管理多个信道上的I/O操作
作者:Dezai.CN / 发布于2011/12/1/ 733
<div> interface TCPProtocal {
 //当有新客户端连接到服务器时，调用该函数
 void handleAccept(SelectionKey key) throws IOException;
 //当客户端有新的数据发送到服务器时，调用该函数
 void handleRead(SelectionKey key) throws IOException;
 //当可以写数据到客户端时条用该函数。
 void handleWrite(SelectionKey key) throws IOException;
 }
 
 class SelectorProtocal implements TCPProtocal {
 private int bufSize;
 public SelectorProtocal(int bufSize) {
 this.bufSize = bufSize;
 }
 @Override
 public void handleAccept(SelectionKey key) throws IOException {
 //1. 通过发生的事件(key)来获取发生事件的通道，一般而言，发生Accept事件的Channel
 //通常为Server端负责监听的SocketChannel。之后再通过该通道接收来自客户端的连接请求，
 //最后返回用于之后I/O操作的客户端信道。
 SocketChannel clntChannel = ((ServerSocketChannel)key.channel()).accept();
 //2. 同样将该客户端的SocketChannel设置为非阻塞模式。
 clntChannel.configureBlocking(false);
 //3. 将Accept返回的客户端SocketChannel也注册到产生当前事件的Selector实例中，
 //同时告知Selector，只是感知该SocketChannel的READ事件，即有数据到来是触发。
 //第三个参数是附件参数，这里可以放置任何对象和该SocketChannel关联，在该事件
 //发生时，可以通过key的attachment方法获取该附件对象。
 clntChannel.register(key.selector(), SelectionKey.OP_READ,ByteBuffer.allocate(bufSize)); 
 }
 @Override
 public void handleRead(SelectionKey key) throws IOException {
 //1. 同样通过发生的事件(key)获取产生事件的通道。
 SocketChannel clntChannel = (SocketChannel)key.channel();
 //2. 再通过key获取存储数据的ByteBuffer对象。
 ByteBuffer buf = (ByteBuffer)key.attachment();
 long bytesRead = clntChannel.read(buf);
 if (bytesRead == -1) {
 clntChannel.close();
 } else if (bytesRead > 0) {
 //这里将重新设置该key在Selector中感兴趣的事件。
 //由于通常在收到数据后可能需要返回应答数据给客户端，因此这里对该
 //SocketChannel增加了OP_WRITE写事件。
 key.interestOps(SelectionKey.OP_READ | SelectionKey.OP_WRITE);
 }
 }
 @Override
 public void handleWrite(SelectionKey key) throws IOException {
 ByteBuffer buf = (ByteBuffer)key.attachment();
 buf.flip();
 SocketChannel clntChannel = (SocketChannel)key.channel();
 clntChannel.write(buf);
 //如果所有应答数据都发送，则需要将该SocketChannel的OP_WRITE取消，而只是
 //注册该SocketChannel的OP_READ事件，以等待客户端有新的请求数据发送过来。
 if (!buf.hasRemaining()) {
 key.interestOps(SelectionKey.OP_READ);
 }
 buf.compact();
 }
 }
 
 public class MyTest {
 private static final int BUFSIZE = 256;
 private static final int TIMEOUT = 3000;
 public static void main(String[] args) throws IOException {
 Selector selector = Selector.open();
 //打开服务器监听SocketChannel
 ServerSocketChannel listenChannel = ServerSocketChannel.open();
 //绑定监听SocketChannel到指定的IP地址和端口
 listenChannel.socket().bind(new InetSocketAddress("10.1.24.199", 5050));
 listenChannel.configureBlocking(false);
 //该监听SocketChannel只能注册OP_ACCEPT接收客户端连接的事件。
 listenChannel.register(selector, SelectionKey.OP_ACCEPT);
 TCPProtocal p = new SelectorProtocal(BUFSIZE);
 while (true) {
 //给当前的Selector实例设置超时，从而避免了忙等待现象，也避免了
 //无限期阻塞的现象，当超时时，select函数返回0。
 if (selector.select(TIMEOUT) == 0) {
 System.out.print(".");
 continue;
 }
 //此时已经有感知的事件发生了，通过selector获取所有事件集合的迭代器
 Iterator<SelectionKey> keyIt = selector.selectedKeys().iterator();
 while (keyIt.hasNext()) {
 SelectionKey key = keyIt.next();
 //判断事件的类型，是否为accept、read或者write
 if (key.isAcceptable()) 
 p.handleAccept(key);
 if (key.isReadable())
 p.handleRead(key);
 if (key.isValid() && key.isWritable())
 p.handleWrite(key);
 //在处理完该事件之后需要从集合中主动移除该事件。
 keyIt.remove();
 }
 }
 }
 }
试试其它关键字
同语言下
可能有用的
Dezai.CN贡献的其它代码(4037)