Java

1. 概述

现在使用NIO的场景越来越多，很多技术框架都使用NIO技术，比如Tomcat，Jetty，Netty等。

传统IO基于字节流和字符流进行操作，而NIO基于Channel和Buffer进行操作，数据总是从Channel读取到Buffer中，或者从Buffer写入到Channel。Selector用于监听多个通道的事件。因此，单个线程可以监听多个数据通道。

Java NIO由以下几个核心部分组成：

Channel（通道）
Buffer（缓冲区）
Selector（选择器）

Channel，Buffer和Selector构成了核心的API，其他组件，如Pipe和FileLock只不过是与3个核心组件使用的工具类。

下面提供一个IO与NIO的文件操作：

public static void io() {
        InputStream in = null;
        try {
            in = new BufferedInputStream(new FileInputStream("d:\okhttp-3.9.0.jar"));
            byte[] buf = new byte[1024];
            int bytesRead = in.read(buf);
            while (bytesRead != -1) {
                for (int i = 0; i < bytesRead; i++) {
                    System.out.println((char) buf[i]);
                }
                bytesRead = in.read(buf);
            }

        } catch (IOException e) {

        } finally {
            try {
                if (in != null)
                    in.close();

            } catch (IOException e) {

            }
        }
    }

    public static void nio() {
        RandomAccessFile aFile = null;
        try {
            aFile = new RandomAccessFile("d:\okhttp-3.9.0.jar", "rw");
            FileChannel fileChannel = aFile.getChannel();
            ByteBuffer buf = ByteBuffer.allocate(1024);

            int bytesRead = fileChannel.read(buf);
            System.out.println(bytesRead);

            while (bytesRead != -1) {
                buf.flip();
                while (buf.hasRemaining())
                    System.out.println((char) buf.get());

                buf.compact();
                bytesRead = fileChannel.read(buf);
            }

        } catch (IOException e) {

        } finally {
            try {
                if (aFile != null)
                    aFile.close();

            } catch (IOException e) {

            }
        }
    }

1.1 Channel和Buffer

基本上，所有的IO在NIO都从一个Channel开始，数据可以从Channel读到Buffer中，也可以从Buffer写到Channel中。

Channel和Buffer有很多实现，NIO中的一些主要Channel实现：

FileChannel：从文件中读写数据；
DatagramChannel：能通过UDP读写网络中的数据；
SocketChannel：通过TCP读写网络中的数据；
ServerSocketChannel：可以监听新的TCP连接，对新连接都会创建一个SocketChannel。

Buffer的关键实现：

ByteBuffer
CharBuffer
DoubleBuffer
FloatBuffer
IntBuffer
LongBuffer
ShortBuffer

这些Buffer覆盖了IO发送的基本数据类型：byte，short，int，long，float，double和char。

NIO还有一个MappedByteBuffer，用于标示内存映射文件。

1.2 Selector

Selector允许单线程处理多个Channel。如果你额应用打开了多个连接（Channel），但每个连接的流量都很低，使用Selector就会很方便。

要使用Selector，得向Selector注册Channel，然后调用Selector的select()方法。这个方法会一直阻塞到某个注册的通道有事件就绪，一旦这个方法返回，线程就可以处理这些事件，事件的例子有如新连接，数据接收等。

2. Buffer

Buffer用于和Channel进行交互：数据从Channel读入Buffer，从Buffer写入到Channel中。

Buffer本质上是一块可以写入数据，然后可以从中读取数据的内存。这块内存被包装成NIO Buffer对象，并提供一组方法，用来方便的访问该块内存。

2.1 Buffer的基本用法

使用Buffer读写数据一般遵循以下4个步骤：

写入数据到Buffer；
调用flip方法；
从Buffer中读取数据；
调用clear或compact方法

当想Buffer写入数据时，Buffer会记录下谢了多少数据，一旦要读取数据，需要通过flip方法将Buffer从写模式切换到读模式。在读模式下，可以读取之前写入到Buffer的所有数据。

一旦读完了所有的数据，就需要清空缓冲区，让它可以再次被写入，有两种方法能清空缓冲区：调用clear和compact方法。clear方法会清空整个缓冲区。compact方法只会清除已经读过的数据。任何未读的数据都被移到缓冲区的起始处，新写入的数据将放到缓冲区未读数据的后面。

2.2 capacity，position和limit

为了理解Buffer的工作原理，需要熟悉它的三个属性：

capacity：作为一块内存，Buffer有一个固定的大小值capacity，最多写入capacity个byte，long，char等类型。一旦Buffer满，需要将其清空（通过读数据或清除数据）才能继续写数据；
position：当写数据到Buffer中，position表示当前的位置。初始的position值为0，当一个byte，long等数据写到Buffer后，position会向前移动到下一个可插入数据的Buffer单元。position最大可为capacity -1。当读取数据时，也是从某个特定位置读，当将Buffer从写模式切换到读模式，position会被重置为0。当从Buffer的position处读取数据时，position向前移动到下一个可读的位置；
limit：在写模式下，Buffer的limit表示你最多能往Buffer里写多少数据，写模式下，limit等于Buffer的capacity；当切换Buffer到读模式时，limit表示你最多能读到多少数据。因此，当切换Buffer到读模式时，limit会被设置成写模式的position值。换句话说，你能读到之前写入的所有数据（limit被设置成已写数据的数量，这个值在写模式下就是position）。

position和limit的含义取决于Buffer处于读模式还是写模式。不管Buffer处在什么模式，capacity的含义总是一样的。

2.3 Buffer的分配和读写

2.3.1 分配

要想获得一个Buffer对象首先要进行分配，每一个Buffer类都有一个allocate方法：

ByteBuffer buff = ByteBuffer.allocate(48);

分配一个可存储1024字符的CharBuffer：

CharBuffer buf = CharBuffer.allocate(1024);

2.3.2 写

写数据到Buffer有两种方法：

从Channel写到Buffer：int byteRead = channel.read(buff)；
通过Buffer的put()方法写到Buffer里，其中put方法有很多版本，允许以不同方式把数据写入到Buffer中：buf.put(127)；

flip()方法

将Buffer从写模式切换到读模式，调用flip方法会将position设置为0，并将limit设置为之前position的值。

换句话说，position现在用于标记读的位置，limit标示之前写进了多少byte，char等。

2.3.3 读

从Buffer中读取数据有两种方式：

从Buffer读取数据到Channel：int byteWrite = channel.write(buf)；
使用get方法从Buffer中读取数据：byte abyte = buf.get()；

get方法有很多版本，允许以不同的方法从Buffer中读取数据。

2.4 API

2.4.1 flip

将Buffer从写模式切换到读模式，调用flip方法会将position设置为0，并将limit设置为之前position的值。

换句话说，position现在用于标记读的位置，limit标示之前写进了多少byte，char等。

2.4.2 rewind

将position设回0，所以可以重读Buffer里面的所有数据。limit保持不变，仍然标示能从Buffer里面读取多少个元素。

2.4.3 clear和compact

一旦读完Buffer中的数据，需要让Buffer准备好再次被写入，可以通过clear和compact方法来完成。

clear：如果调用的是clear方法，position将被设回0，limit被设置成capacity的值。换句话说，Buffer被清空了。Buffer中的数据并未清除，只是这些标记告诉我们可以从哪里开始往Buffer里写数据；如果Buffer中有一些未读的数据，调用clear方法，数据将被遗忘，意味着不再有任何标记告诉你哪些数据被读过，哪些还没有。

compact：如果Buffer中有一些未读的数据，且后续还需要这些数据，但是此时想要先写一些数据，那么使用compact；可以将所有未读的数据拷贝到Buffer起始处，然后将position设到最后一个未读元素正后面。limit属性依然像clear方法一样，设置成capacity。现在Buffer准备好写数据了，但是不会覆盖未读的数据。

2.4.4 mark和reset

通过调用Buffer.mark方法，可以标记Buffer中的一个特定position。之后可以通过调用Buffer.reset方法恢复到这个position。

2.4.5 equals和compareTo

可以使使用equals和compareTo比较两个Buffer。

equals

当满足下列条件时，标示两个Buffer相等，equals只是比较Buffer的一部分，不是每一个在它里面的元素都比较，实际上，它只比较Buffer中的剩余元素。

（1）有相同的类型（byte，char，int等）；

（2）Buffer中剩余的byte，char等个个数相等；

（3）Buffer中所有剩余的byte，char等都相同。

注意：剩余元素是从position到limit之间的元素。

comparTo

比较两个Buffer剩余元素（byte，char等），如果满足下列条件，则认为一个Buffer“小于”另一Buffer：

（1）第一个不相等的元素小于另一个Buffer中对应的元素；

（2）所有元素都相等，但第一个Buffer的元素个数比另一个少；

3. Channel

JavaNIO通道类似流，但有些不同：

既可以从通道中读取数据，又可以写数据到通道，但流的读写通常是单向的；
通道可以异步的读写；
通道中的数据总是要先读到一个Buffer，或者总是要从一个Buffer中写入；

实例：通过FileChannel读取文件

import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;

public class TestFileChannel {

    public static void main(String[] args) throws IOException {
        RandomAccessFile aaf = new RandomAccessFile("D:\App.java", "rw");
        FileChannel fChannel = aaf.getChannel(); // 创建FileChannel
        ByteBuffer buff = ByteBuffer.allocate(48); // 创建Buffer对象
        
        int readBytes = fChannel.read(buff); // 读取文件
        while (readBytes != -1) { // 如果读到文件末尾，则返回-1
            System.out.println(readBytes);
            buff.flip(); // 将Buffer从写模式切换到读模式
            while (buff.hasRemaining()) {
                System.out.println((char)buff.get());
            }
            buff.clear();
            readBytes = fChannel.read(buff); // 循环读取文件
        }
        aaf.close();
    }

}

3.1 FileChannel

FileChannel是一个连接到文件的通道，可以通过文件通道读写文件。FileChannel无法设置为非阻塞模式，它总是运行在阻塞模式下。

3.1.1 文件操作

（1）打开FileChannel

在使用FileChannel之前，必须先打开它，但是我们无法直接打开一个FileChannel，需要通过使用一个InputStream，OutputStream或RandomAccessFile打开FileChannel来获取一个FileChannel实例。

RandomAccessFile aFile = new RandomAccessFile("data/nio-data.txt", "rw");
FileChannel inChannel = aFile.getChannel();

（2）读取数据

调用read方法从FileChannel中读取数据：

ByteBuffer buf = ByteBuffer.allocate(48);
int bytesRead = inChannel.read(buf);

首先，分配一个Buffer，从FileChannel中读取的数据将被读到Buffer中。

然后，调用read方法，该方法将数据从FileChannel读取到Buffer中，read方法返回的int值标示有多少字节被读到了Buffer中。如果返回-1，标示到了文件末尾。

（3）写入数据

使用write方法向FileChannel写数据，该方法的参数是一个Buffer：

String newData = "New String to write to file..." + System.currentTimeMillis();

ByteBuffer buf = ByteBuffer.allocate(48);
buf.clear();
buf.put(newData.getBytes());

buf.flip();

while(buf.hasRemaining()) {
	channel.write(buf);
}

注意：write是在while循环中调用的，因为无法保证write方法一次能向FileChannel写入多少字节，因此需要重复调用write方法，直到Buffer中已经没有尚未写入通道的字节。

（4）关闭

用完FileChannel必须关闭：

FileChannel.close　　

3.1.2 其他API

（1）position方法

有时候可能需要在FileChannel的某个特定位置进行数据的读写操作，可以通过调用position方法获取FileChannel的当前位置。也可以通过调用position(long pos)方法设置FileChannel的当前位置。

long pos = channel.position();
channel.position(pos +123);

如果将位置设置在文件结束符之后，然后试图从文件通道中读取数据，读方法将返回-1（文件结束标志）
如果将位置设置在文件结束符之后，然后向通道中写数据，文件将撑大到当前位置并写入数据。这可能导致“文件空洞”，磁盘上物理文件中写入的数据间有空隙。

（2）size方法

将返回所关联文件的大小。

（3）truncate方法

截取一个文件，截取文件时，文件将指定长度后面的部分删除。

FileChannel.truncate(1024);

（4）force方法

将通道里尚未写入磁盘的数据轻质写到磁盘上，出于性能方面的考虑，操作系统会将数据缓存在内存中，所以无法保证写入到FileChannel里的数据一定会即时写到磁盘上。要保证这一点，需要调用force方法。

force方法有一个boolean类型的参数：指明是否同时将文件元数据（权限信息等）写到磁盘上。

（5）transferFrom/transferTo

在JavaNIO中，如果两个通道中有一个是FileChannel，那你可以直接将数据从一个channel传输到另一个channel。

FileChannel的transferForm方法可以将数据从源通道传输到FileChannel中。

RandomAccessFile fromFile = new RandomAccessFile("fromFile.txt", "rw");
FileChannel      fromChannel = fromFile.getChannel();

RandomAccessFile toFile = new RandomAccessFile("toFile.txt", "rw");
FileChannel      toChannel = toFile.getChannel();

long position = 0;
long count = fromChannel.size();

toChannel.transferFrom(position, count, fromChannel);

注意：在SocketChannel的实现中，SocketChannel只会传输此刻准备好的数据。因此，SocketChannel可能不会将请求的所有数据全部传输到FileChannel中。

transferTo方法将数据从FileChannel传输到其他的Channel中。

RandomAccessFile fromFile = new RandomAccessFile("fromFile.txt", "rw");
FileChannel      fromChannel = fromFile.getChannel();

RandomAccessFile toFile = new RandomAccessFile("toFile.txt", "rw");
FileChannel      toChannel = toFile.getChannel();

long position = 0;
long count = fromChannel.size();

fromChannel.transferTo(position, count, toChannel);

3.2 SocketChannel

JavaNIO的SocketChannel是一个连接到TCP网络套接字的通道，可以通过以下2种方式创建SocketChannel：

打开一个SocketChannel并连接到互联网上的某台服务器；
一个新连接达到ServerSocketChannel时，会创建一个SocketChannel；

3.2.1 打开SocketChannel

SocketChannel socketChannel = SocketChannel.open();
socketChannel.connect(new InetSocketAddress("http://jenkov.com", 80));

3.2.2 关闭SocketChannel　　

当用完SocketChannel之后调用close关闭：

socketChannel.close();

3.2.3 从SocketChannel读取数据

要从SocketChannel中读取数据，调用read方法

ByteBuffer buf = ByteBuffer.allocate(48);
int bytesRead = socketChannel.read(buf);

首先，分配一个Buffer，从SocketChannel读取到的数据将会放到这个Buffer中。

然后，调用SocketChannel.read，该方法将数据从SocketChannel读到Buffer中。read方法返回的int值表示读了多少字节进Buffer里面。如果返回-1，表示已经读到了流的末尾（连接关闭了）。

3.2.4 写入SocketChannel　　

写数据到SocketChannel用的是write方法，该方法以一个Buffer为参数。

String newData = "New String to write to file..." + System.currentTimeMillis();

ByteBuffer buf = ByteBuffer.allocate(48);
buf.clear();
buf.put(newData.getBytes());

buf.flip();

while(buf.hasRemaining()) {
    channel.write(buf);
}

注意：SocketChannel.write方法的调用是在一个while循环中，write方法无法保证能写多少字节到SocketChannel。所以，重复调用write直到Buffer没有要写的字节为止。

3.2.5 非阻塞模式

可以设置SocketChannel为非阻塞模式（non-blocking mode），设置周，就可以在异步模式下调用connect，read和write。

connect：如果SocketChannel在非阻塞模式下，此时调用connect，该方法可能在连接建立之前就返回了。为了确定连接是否建立，可以调用finishConnect方法。

socketChannel.configureBlocking(false);
socketChannel.connect(new InetSocketAddress("http://jenkov.com", 80));

while(! socketChannel.finishConnect() ){
    //wait, or do something else...
}

write：非阻塞模式下，write方法在尚未写出任何内容时可能就返回了，所以需要在循环中调用write。

read：非阻塞模式下，read方法在尚未读到任何数据时可能就返回了。所以需要关注它的int返回值，它会告诉你读取了多少字节。

非阻塞模式与Selector：非阻塞模式与选择器搭配会工作的更好，通过将一个或多个SocketChannel注册到Selector，可以询问选择器哪个通道已经准备好了读取，写入等。

3.3 ServerSocketChannel

JavaNIO中的ServerSocketChannel是一个可以监听新进来的TCP连接的通道，就像标准IO的ServerSocket。

ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();

serverSocketChannel.socket().bind(new InetSocketAddress(9999));

while(true){
    SocketChannel socketChannel =
            serverSocketChannel.accept();

    //do something with socketChannel...
}

3.3.1 打开ServerSocketChannel

通过调用ServerSocketChannel.open方法来打开。

ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();

3.3.2 关闭ServerSocketChannel

serverSocketChannel.close();

3.3.3 监听连接

通过ServerSocketChannel.accept方法监听新的连接，当accept方法返回的时候，它返回一个包含新进来的连接的SocketChannel。因此，accept方法会一直阻塞到新连接到达。

通常不会仅仅监听一个连接，在while循环中调用accept方法，如下面：

while(true){
    SocketChannel socketChannel =
            serverSocketChannel.accept();

    //do something with socketChannel...
}

3.3.4 非阻塞模式

ServerSocketChannel可以设置成非阻塞模式，在非阻塞模式下，accept方法会立刻返回，如果还没有新进来的连接，返回的将是null，因此，需要检查返回的SocketChannel是否是null。

ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();

serverSocketChannel.socket().bind(new InetSocketAddress(9999));
serverSocketChannel.configureBlocking(false);

while(true){
    SocketChannel socketChannel =
            serverSocketChannel.accept();

    if(socketChannel != null){
        //do something with socketChannel...
    }
}

3.4 DatagramChannel

JavaNIO中的DatagramChannel是一个收发UDP包的通道。

因为UDP是无连接的网络协议，所以不像其他通道那样读取和写入。它发送和接收的是数据包。

3.4.1 打开DatagramChannel

DatagramChannel channel = DatagramChannel.open();
channel.socket().bind(new InetSocketAddress(9999));

3.4.2 接收数据

同receive方法从DatagramChannel接收数据，如：

ByteBuffer buf = ByteBuffer.allocate(48);
buf.clear();
channel.receive(buf);

receive方法会将接收到的数据包内容复制到指定的Buffer，如果Buffer容不下收到的数据，多出的数据将被丢弃。

3.4.3 发送数据

通过send方法从DatagramChannel发送数据：

String newData = "New String to write to file..." + System.currentTimeMillis();

ByteBuffer buf = ByteBuffer.allocate(48);
buf.clear();
buf.put(newData.getBytes());
buf.flip();

int bytesSent = channel.send(buf, new InetSocketAddress("jenkov.com", 80));

这个例子发送一串字符串到jenkov.com服务器的UDP端口80。因为服务端并没有监控这个端口，所以什么也不会发送。也不会通知你发出的数据包师傅已收到，因为UDP在数据传送方面没有任何保证。

3.4.4 连接到特定的地址

可以将DatagramChannel连接到网络中的特定地址，由于UDP是无连接的，连接到特定地址并不会像TCP通道那样创建一个真正的连接。而是锁住DatagramChannel，让其只能从特定地址收发数据。

channel.connect(new InetSocketAddress("jenkov.com", 80))；

当连接后，也可以使用read和write方法，就像在用传统的通道一样，只是在数据传送方面没有任何保证。

int bytesRead = channel.read(buf);
int bytesWritten = channel.write(but);

3.5 Pipe

JavaNIO管道是2个线程之间的单向数据连接。Pipe有一个source通道和一个sink通道。数据会被写到sink通道，从source通道读取。

pipe的原理：

3.5.1 创建管道

Pipe pipe = Pipe.open();

3.5.2 写数据

要向管道写数据，需要访问sink通道：

Pipe.SinkChannel sinkChannel = pipe.sink();

String newData = "New String to write to file..." + System.currentTimeMillis();
ByteBuffer buf = ByteBuffer.allocate(48);
buf.clear();
buf.put(newData.getBytes());

buf.flip();

while(buf.hasRemaining()) {
    sinkChannel.write(buf);
}

通过调用SinkChannel的write方法，将数据写入SinkChannel。

3.5.3 读数据

从读取管道的数据，需要访问source通道：

Pipe.SourceChannel sourceChannel = pipe.source();

ByteBuffer buf = ByteBuffer.allocate(48);

int bytesRead = sourceChannel.read(buf);

调用souce通道的read方法来读取数据。

3.x Scatter、Gather

JavaNIO支持scather/gather，用于描述从Channel中读取或写入到Channel的操作。

scather（分散）从Channel中读取是指在读操作时将读取的数据写入多个Buffer中。因此，Channel将从Channel中读取的数据分散到多个Buffer中。

ByteBuffer header = ByteBuffer.allocate(128);
ByteBuffer body   = ByteBuffer.allocate(1024);

ByteBuffer[] bufferArray = { header, body };

channel.read(bufferArray);

注意Buffer首先被插入到数组，然后再讲数据作为channel.read的输入参数。read方法按照Buffer在数组中的顺序将从channel中读取的数据写入到Buffer，当一个Buffer写满后，Channel紧接着向另一个Buffer中写。

scatter在移动到下一个Buffer前，必须填满当前的Buffer，这意味着它不适用于动态消息。

gather（聚集）写入Channel是指在写操作时将多个Buffer的数据写入同一个Channel，因此，Channel将多个Buffer中的数据聚集后发送到Channel。

ByteBuffer header = ByteBuffer.allocate(128);
ByteBuffer body   = ByteBuffer.allocate(1024);

//write data into buffers

ByteBuffer[] bufferArray = { header, body };

channel.write(bufferArray);

buffers数组是write方法的入参，write方法会按照buffer在数组中的顺序，将数据写入到channel，注意只有position和limit之间的数据才会被写入。
与scatter相反，gather能较好的处理动态消息。

4. Selector

Selector选择器是JavaNIO中能够检测一到多个NIO通道，并能够知晓通道是否为诸如读写事件做好准备的组件。这样，一个单独的线程可以管理多个channel，从而管理多个网络连接。

4.1 why Selector？

仅用单个线程来处理多个Channels的好处是，只需要更少的线程来处理通道。事实上，可以只用一个线程处理所有的通道。对于操作系统来说，线程之间上下文切换的开销很大，而且每个线程都要占用系统的一些资源。因此，使用的线程越少越好。

4.2 Selector操作

4.2.1 Selector的创建

通过调用Selector.open()方法创建一个Selector，如下：

Selector selector = Selector.open();

4.2.2 向Selector注册通道

为了将Channel和Selector配合使用，必须将Channel注册到Selector上，通过SelectableChannel.register()方法来实现，如下：

channel.configureBlocking(false);
SelectionKey key = channel.register(selector, Selectionkey.OP_READ);

与Selector一起使用时，Channel必须处于非阻塞模式下。这意味着不能讲FileChannel与Selector一起使用，因为FileChannel不能切换到非阻塞模式，而套接字通道可以。

注意register()方法的第二个参数，这是一个interest集合，意思是在通多Selector监听Channel时对什么事件感兴趣，可以监听4中不同类型的事件。

Connect
Accept
Read
Write

通道触发一个事件的意思是该事件已经就绪。所以，某个Channel成功连接到另一个服务器称为“连接就绪”；一个ServerSocketChannel准备好接收新进入的连接称为“接收就绪”；一个有数据可读的通道可以说是“读就绪”；等待写数据的通道可以说是“写就绪”。

这4中事件可用SelectionKey的4个常量来表示：

SelectionKey.OP_CONNECT
SelectionKey.OP_ACCEPT
SelectionKey.OP_READ
SelectionKey.OP_WRITE

如果对不止一种事件感兴趣，可以用“位或”操作符将常量连接起来，比如：

int interestSet = SelectionKey.OP_READ | SelectionKey.OP_WRITE;

当向Selector注册Channel时，register方法会返回一个SelectionKey对象，这个对象包含了一些你感兴趣的属性：

interest集合：interest集合是你所选择的感兴趣的事件集合，可以通过SelectionKey读写interest集合。
ready集合：ready集合是通道已经准备就绪的操作的集合，在一次选择之后，你会先访问这个ready set。
Channel：SelectionKey.channel()
Selector：SelectionKey.selector()
附加的对象（可选）：可以将一个对象或者更多信息附着到SelectionKey上，这样就能方便的识别某个给定的通道。例如，可以附加与通道一起使用的Buffer，或是包含聚集数据的某个对象。

selectionKey.attach(theObject);
Object attachedObj = selectionKey.attachment();

还可以在用register方法向Selector注册Channel的时候附加对象，如：

SelectionKey key = channel.register(selector, SelectionKey.OP_READ, theObject);

4.2.3 通过Selector选择通道

一旦向Selector注册了一或多个通道，就可以调用几个重载的select方法，这些方法返回你感兴趣的事件（如：连接，接受，读或写）已经准备就绪的那些通道。换句话说，如果你对“读就绪”的通道感兴趣，select方法会返回读事件已经就绪的那些通道。

下面是select方法：

int select() // 阻塞到至少一个通道在你注册的时间上就绪了

int select(long timeout) // 和select一样，除了最长会阻塞timeout毫秒

int selectNow() // 不会阻塞，不管什么通道就绪都立刻返回。如果自从前一次选择操作后，没有通道变成可选的，则此方法直接返回零

select方法返回的int值标示有多少通道已经就绪，即自上次调用select方法后有多少通道变成就绪状态。

如果调用select方法，因为有一个通道变成就绪状态，返回了1，若再次调用select方法如果另一个通道就绪了，它会再次返回1。

一旦调用了select方法，并且返回值表示有一个或更多个通道就绪了，然后可以通过调用selector的selectedKeys方法，访问“已选择键集”（selected key set）中的就绪通道。如下：

Set selectedKeys = selector.selectedKeys();

可以遍历这个已选择的键集合来访问就绪的通道，如下：

Set selectedKeys = selector.selectedKeys();
Iterator keyIterator = selectedKeys.iterator();
while(keyIterator.hasNext()) {
    SelectionKey key = keyIterator.next();
    if(key.isAcceptable()) {
        // a connection was accepted by a ServerSocketChannel.
    } else if (key.isConnectable()) {
        // a connection was established with a remote server.
    } else if (key.isReadable()) {
        // a channel is ready for reading
    } else if (key.isWritable()) {
        // a channel is ready for writing
    }
    keyIterator.remove();
}

这个循环遍历已选择键集中的每个键，并检测各个键所对应的通道的就绪事件。

注意每次迭代末尾的key.remove调用。Selector不会字节从已选择键集中移除SelectionKey实例。必须在处理完通道时自己移除。下次改通道变成就绪时，Selector会再次将其放入已选择键集中。

SelectionKey.channel方法返回的通道需要转型成你要处理的类型，如：ServerSocketChannel或SocketChannel。

某个线程调用select方法后阻塞了，即使没有通道已经就绪，也有办法让其从select方法返回。只要让其他线程在第一个线程调用select方法的那个对象上调用Selector.wakeup方法即可。阻塞在select方法上的线程会立刻返回。

如果在其他线程调用了wakeup方法，但当前没有线程阻塞在select方法上，下个调用select方法的线程会立刻醒来。

用完Selector后调用其close方法会关闭该Selector，且使注册到该Selector上的所有selectionKey实例无效，通道本身并不会关闭。

完整demo：

Selector selector = Selector.open();
channel.configureBlocking(false);
SelectionKey key = channel.register(selector, SelectionKey.OP_READ);
while(true) {
  int readyChannels = selector.select();
  if(readyChannels == 0) continue;
  Set selectedKeys = selector.selectedKeys();
  Iterator keyIterator = selectedKeys.iterator();
  while(keyIterator.hasNext()) {
    SelectionKey key = keyIterator.next();
    if(key.isAcceptable()) {
        // a connection was accepted by a ServerSocketChannel.
    } else if (key.isConnectable()) {
        // a connection was established with a remote server.
    } else if (key.isReadable()) {
        // a channel is ready for reading
    } else if (key.isWritable()) {
        // a channel is ready for writing
    }
    keyIterator.remove();
  }
}

5. 实例

服务器实例：

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.Iterator;

public class NioServerChannel {
    private static final int BUF_SIZE = 1024;
    private static final int PORT = 8080;
    private static final int TIMEOUT = 3000;

    public static void handleAccept(SelectionKey key) throws IOException {
        System.out.println("accept");
        ServerSocketChannel ssc = (ServerSocketChannel) key.channel();
        // ServerSocketChannel可以设置成非阻塞模式，在非阻塞模式下，accept方法会立刻返回，如果还没有新进来的连接
        // 返回的将会是null，因此需要检查返回的SocketChannel是否是null。
        SocketChannel sc = ssc.accept();
        if (sc != null) {
            sc.configureBlocking(false);
            sc.register(key.selector(), SelectionKey.OP_READ, ByteBuffer.allocate(BUF_SIZE));
        }
    }

    public static void handleRead(SelectionKey key) throws IOException {
        System.out.println("read");
        SocketChannel sc = (SocketChannel) key.channel();
        ByteBuffer buf = (ByteBuffer) key.attachment();
        long byteRead = sc.read(buf);
        while (byteRead > 0) {
            buf.flip();
            while (buf.hasRemaining())
                System.out.print((char) buf.get());

            System.out.println();
            buf.clear();
            byteRead = sc.read(buf);
        }

        if (byteRead == -1) {
            sc.close();
        }

    }

    public static void handleWrite(SelectionKey key) throws IOException {
        System.out.println("write");
        ByteBuffer buf = (ByteBuffer) key.attachment();
        buf.flip();
        SocketChannel sc = (SocketChannel) key.channel();
        while (buf.hasRemaining())
            sc.write(buf);

        buf.compact();
    }

    public static void main(String[] args) {
        Selector selector = null;
        ServerSocketChannel ssc = null;
        try {
            selector = Selector.open();
            ssc = ServerSocketChannel.open();
            ssc.socket().bind(new InetSocketAddress(PORT));
            ssc.configureBlocking(false);
            // 为了将Channel和Selector配合使用，必须将Channel注册到Selector上
            // 通过SelectableChannel.register()方法来实现。
            // 与Selector一起使用时，Channel必须处于非阻塞模式下，这意味着不能讲FileChannel与Selector一起使用，因为FileChannel不能切换到非阻塞模式，而套接字通道可以
            ssc.register(selector, SelectionKey.OP_ACCEPT);

            while (true) {
                if (selector.select(TIMEOUT) == 0) {
                    System.out.println("==");
                    continue;
                }

                Iterator<SelectionKey> it = selector.selectedKeys().iterator();
                while (it.hasNext()) {
                    SelectionKey key = it.next();

                    if (key.isAcceptable()) {
                        handleAccept(key);
                    }

                    if (key.isReadable()) {
                        handleRead(key);
                    }

                    if (key.isWritable() && key.isValid()) {
                        handleWrite(key);
                    }

                    if (key.isConnectable()) {
                        System.out.println("isConeectable = true");
                    }

                    it.remove();
                }

            }

        } catch (Exception e) {
            System.out.println(e);
        } finally {
            try {
                if (selector != null)
                    selector.close();

                if (ssc != null)
                    ssc.close();

            } catch (Exception e) {
                System.out.println(e);
            }
        }
    }

}

客户端实例：

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SocketChannel;
import java.util.concurrent.TimeUnit;

public class NioClientChannel {

    public static void main(String[] args) {
        ByteBuffer buffer = ByteBuffer.allocate(1024);
        SocketChannel socketChannel = null;
        try {
            socketChannel = SocketChannel.open();
            socketChannel.configureBlocking(false);
            socketChannel.connect(new InetSocketAddress("127.0.0.1", 8080));

            if (socketChannel.finishConnect()) {
                int i = 0;
                while (true) {
                    TimeUnit.SECONDS.sleep(1);
                    String info = "I'm " + i++ + "-th information from client";
                    buffer.clear();
                    buffer.put(info.getBytes());
                    buffer.flip();
                    while (buffer.hasRemaining()) {
                        System.out.println(buffer);
                        socketChannel.write(buffer);
                    }
                }
            }

        } catch (IOException | InterruptedException e) {
            System.out.println(e);
        } finally {
            try {
                if (socketChannel != null)
                    socketChannel.close();
            } catch (Exception e) {
                System.out.println(e);
            }
        }

    }

}

6. 参考资料

http://www.importnew.com/19816.html

http://ifeve.com/overview/