Skip to content

Java NIO之Selector（选择器）

Jump to bottom
MySelf edited this page Dec 18, 2018 · 3 revisions

一 Selector（选择器）介绍

Selector 一般称 为选择器 ，当然你也可以翻译为 多路复用器 。它是Java NIO核心组件中的一个，用于检查一个或多个NIO Channel（通道）的状态是否处于可读、可写。如此可以实现单线程管理多个channels,也就是可以管理多个网络链接。

使用Selector的好处在于： 使用更少的线程来就可以来处理通道了， 相比使用多个线程，避免了线程上下文切换带来的开销。

二 Selector（选择器）的使用方法介绍

1.Selector的创建

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

Selector selector = Selector.open();

2、注册Channel到Selector

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

Channel必须是费阻塞的 所以FileChannel不适用Selector，因为FileChannel不能切换为非阻塞模式，更准确的来说是因为FileChannel没有继承SelectableChannel。Socket Channel可以正常使用。

SelectableChannel抽象类有一个**configureBlocking（）**方法用于使通道处于阻塞模式或非阻塞模式。

abstract SelectableChannel configureBlocking(boolean block)

注意：

SelectableChannel抽象类的configureBlocking（）方法是由AbstractSelectableChannel抽象类实现的，SocketChannel、ServerSocketChannel、DatagramChannel都是直接继承了AbstractSelectableChannel抽象类。

register（）方法的第二个参数。这是一个“interest集合”，意思是在通过Selector监听Channel时对什么事件感兴趣。可以监听四种不同类型的事件：

  • Connect
  • Accept
  • Read
  • Write

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

这四种事件用SelectionKey的四个常理来表示：

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

如果你对不止一种事件感兴趣，使用或与运算符即可，如下：

int interestSet = SelectionKey.OP_READ | SelectionKey.OP_WRITE;

3、SelectionKey介绍

一个SelectionKey键表示了一个特定的通道对象和一个特定的选择器对象之间的注册关系。

* key.attachment();//返回SelectionKey的attachment，attachment可以在注册channel的时候指定。
* key.channel();//返回该SelectionKey对应的channel
* key.selector();//返回该SelectionKey对应的Selector
* key.interestOps();//返回代表需要Selector监控的IO操作的bit mask
* key.readyOps();//返回一个bit mask，代表在相应channel上可以进行的IO操作

key.interestOps();

我们可以通过以下方法来判断Selector是否对Channel的某种事件感兴趣

int interestSet = selectionKey.interestOps();
boolean isInterestedInAccept = (interestSet & SelectionKey.OP_ACCEPT) == SelectionKey.OP_ACCEPT;
boolean isInterestedInConnect = interestSet & SelectionKey.OP_CONNECT;
boolean isInterestedInRead = interestSet & SelectionKey.OP_READ;
boolean isInterestedInWrite = interestSet & SelectionKey.OP_WRITE;

key.readyOps();

ready集合是通道已经准备就绪的操作的集合。Java中定义以下几个方法用来检查这些操作是否就绪。

//创建ready集合的方法
int readySet = selectionKey.readOps();
//检查这些操作是否就绪的方法
boolean isReadable();//是否可读
boolean isWritable();//是否可写
boolean isConnectable();//是否可连接
boolean isAcceptable();//是否可接收

从SelectionKey访问Channel和Selection很简单。如下：

Channel channel = key.channel();
Selector selector = key.selector();
key.attachment();

可以将一个对象或者更多信息附着到SelectionKey上，这样就能方便的识别某个给定的通道。例如，可以附加与通道一起使用的Buffer，或是包含聚集数据的某个对象。使用这个方法如下：

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

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

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

4、从Selector中选择channel

选择器维护注册过的通道的集合，并且这种注册关系都被封装在SelectionKey当中

Selector维护的三种类型SelectionKey集合：

  • 已注册的键的集合（Registered Key set）

所有与选择器关联的通道所生成的键的集合称为已经注册的键的集合。并不是所有注册过的键都仍然有效。这个集合通过Keys()方法返回，并且可能是空的。这个已注册的键的集合不是可以直接修改的；试图这么做的话讲引发UnsupportedOperationException。

  • 已选择的键的集合（Selected key set）

所有与选择器关联的通道所生成的键的集合称为已经注册的键的集合。并不是所有注册过的键都仍然有效。这个集合通过keys()方法返回，并且可能是空的。这个已注册的键的集合不是可以直接修改的；试图这么做的话将引发UnsupportedOperationException。

  • 已取消的键的集合（Cancelled Key set）

已注册的键的集合的子集，这个集合包含了cancel()方法被调用过的键（这个键已经被无效化），但它们还没有被注销。这个集合是选择器对象的私有成员，因而无法直接访问。

注意：

当键被取消（可以通过isValid（）方法来判断）时，它将被放在相关的选择权的已取消的键的集合里。注册不会立即被取消，但键会立即失效。当再次调用select（）方法时（或者一个正在进行的select（）调用结束时），已取消的键的集合中的被取消的键将被清理掉，并且相应的注销也将完成。通道会被注销，而新的SelectionKey将被返回。当通道关闭时，所有相关的键会自动取消（记住，一个通道可以被注册到多个选择器上）。当选择器关闭时，所有被注册到该选择器的通道都将被注销，并且相关的键将立即被无效化（取消）。一旦键被无效化，调用它的与选择相关的方法就将抛出CancelledKeyException。

select()方法介绍：

在刚初始化的Selector对象中，这三个集合都是空的，通过Selector的select()方法可以选择已经准备就绪的通道（这些通道包含你感兴趣的事件）。比如你对读就绪的通道感兴趣，那么select()方法就会返回读事件已经就绪的那些通道。下面是Selector几个重载的select（）方法：

  • int select（）：阻塞到至少有一个通道在你注册的事件上就绪了。
  • int select（long timeout）：和select（）一样，但最长阻塞时间为timeout毫秒。
  • int selectNow（）：非阻塞，只要有通道就绪就离开返回。

select()方法返回的int值表示有多少通道已经就绪，是自上次调用select()方法后有多少通道变成就绪状态。之前在select()调用时进入就绪的通道不会在本次调用中被记入，而在前一次select()调用进入就绪但现在已经不在处于就绪的通道也不会被记入。

例如：首次调用select()方法，如果有一个通道变成就绪状态，返回了1，若再次调用select(）方法，如果另一个通道就绪了，它会再次返回1.如果对第一个就绪的channel没有做任何操作，现在就有两个就绪的通道，但在每次select方法调用之间，只有一个通道就绪了。

一旦调用select(）方法，并且返回值不为0时，则可以通过调用Selector的SelectedKeys()方法来访问已选择键集合。

5、停止选择的方法

选择器执行选择的过程，系统底层会依次询问每个通道是否已经就绪，这个过程可能会造成调用线程进入阻塞状态，那么我们有以下三种方式可以唤醒在select()方法中阻塞的线程。

  • wakeup()方法：通过调用Selector对象的wakeup()方法让处在阻塞状态的select()方法立刻返回

该方法使得选择器上的第一个还没有返回的选择操作立即返回。如果当前没有进行中的选择操作，那么下一次对select()方法的一次调用将立即返回。

  • close()方法：通过close()方法关闭Selector

该方法使得任何一个在选择操作中阻塞的线程都被唤醒，同时使得注册到该Selector的所有Channel被注销，所有的键将被取消，但是Channel本身并不会关闭。

Demo

package com.github.myself;

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;
import java.util.Set;

/**
 * Created by MySelf on 2018/12/17.
 */
public class WebServer {

    public static void main(String[] args) {
        try {
            ServerSocketChannel ssc = ServerSocketChannel.open();
            ssc.socket().bind(new InetSocketAddress("127.0.0.1",8900));
            //设定为非阻塞
            ssc.configureBlocking(false);

            Selector selector = Selector.open();
            //注册channel，并且指定感兴趣的事件是Accept(接收)
            ssc.register(selector, SelectionKey.OP_ACCEPT);

            ByteBuffer readBuff = ByteBuffer.allocate(1024);
            ByteBuffer writeBuff = ByteBuffer.allocate(128);
            writeBuff.put("received".getBytes());
            writeBuff.flip();
            int a = 1;

            while (true){
                int nReady = selector.select();
                Set<SelectionKey> keys = selector.selectedKeys();
                Iterator<SelectionKey> it = keys.iterator();

                while (it.hasNext()){
                    a++;
                    SelectionKey key = it.next();
                    it.remove();
                    System.out.println("----");
                    if (key.isAcceptable()) {
                        //创建新的连接，并且把链接注册到selector上
                        //声明这个channel只对读操作感兴趣
                        SocketChannel socketChannel = ssc.accept();
                        socketChannel.configureBlocking(false);
                        socketChannel.register(selector, SelectionKey.OP_READ);
                        System.out.println("Acceptable");
                    }else if (key.isConnectable()){
                        System.out.println("Connect");
                    }else if(key.isReadable()){
                        SocketChannel socketChannel = (SocketChannel)key.channel();
                        readBuff.clear();
                        socketChannel.read(readBuff);
                        readBuff.flip();
                        System.out.println("received:"+new String(readBuff.array()));
                        key.interestOps(SelectionKey.OP_WRITE);
                        System.out.println("Readable");
                    }else if(key.isWritable()){
                        writeBuff.rewind();
                        SocketChannel socketChannel = (SocketChannel)key.channel();
                        socketChannel.write(writeBuff);
                        key.interestOps(SelectionKey.OP_READ);
                        System.out.println("Writable");
                    }
                }
            }
        }catch (IOException e){
            e.printStackTrace();
        }
    }
}

客户端

package com.github.myself;

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

/**
 * Created by MySelf on 2018/12/17.
 */
public class WebClient {

    public static void main(String[] args) throws IOException {
        try {
            SocketChannel socketChannel = SocketChannel.open();
            socketChannel.connect(new InetSocketAddress("127.0.0.1",8900));

            ByteBuffer writeBuffer = ByteBuffer.allocate(32);
            ByteBuffer readBuffer = ByteBuffer.allocate(32);

            writeBuffer.put("hello".getBytes());
            writeBuffer.flip();
            while (true){
//            for (int i=0;i<5;i++){
                writeBuffer.rewind();
                socketChannel.write(writeBuffer);
                readBuffer.clear();
                socketChannel.read(readBuffer);
                readBuffer.flip();
                System.out.println(new String(readBuffer.array()));
            }
        }catch (IOException e){
            e.printStackTrace();
        }
    }

}

公众号：UncleCatMySelf Image text

Clone this wiki locally