（五）Netty之Selector选择器

_蔚然

已于 2022-04-12 18:51:01 修改

阅读量567

点赞数

分类专栏： Netty 文章标签： java

于 2022-01-07 19:03:46 首次发布

本文链接：https://blog.csdn.net/qq_38949960/article/details/122370572

版权

Netty 专栏收录该内容

7 篇文章 1 订阅

订阅专栏

基本介绍

Java的NIO，用非阻塞的IO方式，可以用一个线程，处理多个客户端连接，就会使用到Selector（选择器）
Selector能够检测多个注册通道上是否有事件发生（注意：多个Channel以事件的方式可以注册到同一个Selector），如果有事件发生，便获取时间然后针对每个事件进行相应的处理。这样就可以只用一个单线程去管理多个通道，也就是管理多个连接和请求
只有在连接真正有读写事件发生时，才会进行读写，就大大地减少了系统开销，并且不必为每个连接都创建一个线程，不用去维护多个线程
避免了多线程之间的上下文切换导致的开销

示意图

在这里插入图片描述

特点说明

Netty的IO线程NioEventLoop聚合了Selector(选择器，也叫多路复用器)，可以同时并发处理成百上千个客户端连接。
当线程从客户端Socket通道进行读写数据时，若没有数据可用时，该线程可以进行其他任务。
线程通常将非阻塞IO的空闲时间用于在其他通道上执行IO操作，所以单独的线程可以管理多个输入和输出通道。
由于读写操作都是非阻塞的，这就可以充分提升IO线程的运行效率，避免由于频繁I/O阻塞导致的线程挂起
一个I/O线程可以并发处理N个客户端连接和读写操作，这从根本上解决了传统同步阻塞I/O一连接一线程模型，架构的性能、弹性伸缩能力和可靠性都得到了极大的提升。

Selector

Selector类相关方法

Selector类是一个抽象类，常用方法和说明如下：

public abstract class Selector implements Closeable {

    public static Selector open();//得到一个选择器对象

    public abstract int select(long timeout) throws IOException;//监控所有注册的通道，当其中有IO操作可以进行时，将对应的SelectionKey加入到内部集合中并返回，参数用来设置超时时间

    public abstract Set<SelectionKey> selectedKeys();//从内部集合中得到所有的SelectionKey
}

在这里插入图片描述

注意事项

NIO中的ServerSocketChannel功能类似ServerSocket，SocketChannel功能类似Socket
Selector相关方法说明
slector.select();//阻塞
slector.select(1000);//阻塞1000毫秒，在1000毫秒后返回
slector.WAKEUP();//唤醒slector
slector.selectNow();//不阻塞，立马返回

NIO非阻塞网络编程原理分析图

NIO非阻塞网络编程相关的(Selector, SelectionKey, ServerSocketChannel和SocketChannel)关系梳理图
在这里插入图片描述
对上图说明

当客户端连接时，会通过ServerSocketChannel得到SocketChannel
将SocketChannel注册到Selector上，register(Selector sel, int ops)，一个Selector上可以注册多个SocketChannel
注册后返回一个SelectionKey，会和该Selector关联（集合）
Selector进行监听select方法，返回有事件发生的通道的个数
进一步得到各个SelectionKey（有事件发生）
在通过SelectionKey反向获取SocketChannel，方法channel()
可以通过得到的channel，完成业务处理

代码示例

编写一个NIO入门案例，实现服务器和客户端之间的数据简单通讯（非阻塞）

服务器端代码：

package com.example.demo.netty;

import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.Iterator;
import java.util.Set;

public class NIOServer {
    public static void main(String[] args) throws Exception {
        //创建ServerSocketChannel -> ServerSocket
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();

        //得到一个Selector对象
        Selector selector = Selector.open();

        //绑定一个端口 在服务器端监听
        serverSocketChannel.bind(new InetSocketAddress(6666));

        //设置为非阻塞
        serverSocketChannel.configureBlocking(false);

        //把serverSocketChannel 注册到 selector 事件为 OP_ACCEPT
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

        //循环等待客户端连接
        while (true) {
            //这里我们等待1秒，如果没有事件发生，返回
            if (selector.select(1000) == 0) {
                System.out.println("服务器等待了1秒，无连接");
                continue;
            }

            //如果返回的>0，就获取到相关的selectionKey集合
            //1.如果返回的>0，表示已经获取到关注的事件
            //2.selector.selectedKeys()返回关注事件的集合
            //  通过selectionKeys 反向获取通道
            Set<SelectionKey> selectionKeys = selector.selectedKeys();
            //遍历Set<SelectionKey>，使用迭代器遍历
            Iterator<SelectionKey> keyIterator = selectionKeys.iterator();

            while (keyIterator.hasNext()) {
                //获取到SelectionKey
                SelectionKey key = keyIterator.next();
                //根据key 对应的通道发生的事件做相应处理
                if (key.isAcceptable()) {   //如果是OP_ACCEPT，有新的客户端连接
                    //该客户端生成一个SocketChannel
                    SocketChannel socketChannel = serverSocketChannel.accept();
                    System.out.println("客户端连接成功 生成了一个 socketChannel " + socketChannel.hashCode());
                    //将SocketChannel注册到selector,关注事件为OP_READ
                    //同时给SocketChannel关联一个buffer
                    socketChannel.configureBlocking(false);
                    socketChannel.register(selector, SelectionKey.OP_READ, ByteBuffer.allocate(1024));
                }

                if (key.isReadable()) { //发生OP_READ
                    //通过key反向获取到对应channel
                    SocketChannel channel = (SocketChannel)key.channel();
                    //通过key获取到该channel关联的buffer
                    ByteBuffer buffer = (ByteBuffer) key.attachment();
                    channel.read(buffer);
                    System.out.println("from 客户端:" + new String(buffer.array()));
                }
                //手动从集合中移除当前操作的SelectionKey,防止重复操作
                keyIterator.remove();
            }
        }
    }
}

客户端代码：

package com.example.demo.netty;

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

public class NIOClient {
    public static void main(String[] args) throws Exception {
        //得到一个网络通道
        SocketChannel socketChannel = SocketChannel.open();
        //设置非阻塞
        socketChannel.configureBlocking(false);
        //提供服务器端的ip和端口
        InetSocketAddress inetSocketAddress = new InetSocketAddress("127.0.0.1", 6666);
        //连接服务器
        if (!socketChannel.connect(inetSocketAddress)) {
            while (!socketChannel.finishConnect()) {
                System.out.println("因为连接需要事件，客户端不会一直阻塞，可以做其他工作..");
            }
            //如果连接成功，就发送数据
            String str = "嗨,Netty!";
            //wraps a byte array into a buffer
            ByteBuffer buffer = ByteBuffer.wrap(str.getBytes());
            //发送数据，将buffer数据写入channel
            socketChannel.write(buffer);
            System.in.read();
        }
    }
}

SelectionKey

SelectionKey，标识Selector和网络通道的注册关系，共四种：
int OP_ACCEPT：有新的网络连接可以accept，值为16
int OP_CONNECT：代表连接已经建立，值为8
int OP_READ：代表读操作，值为1
int OP_WRITE：代表写操作，值为4

public static final int OP_READ = 1 << 0;
public static final int OP_WRITE = 1 << 2;
public static final int OP_CONNECT = 1 << 3;
public static final int OP_ACCEPT = 1 << 4;
SelectionKey相关方法

public abstract class SelectionKey {
	 public abstract Selector selector();//得到与之关联的Selector对象
	 public abstract SelectableChannel channel();//得到与之关联的通道
	 public final Object attachment();//得到与之关联的共享数据
	 public abstract SelectionKey interestOps(int ops);//设置或改变监听事件
	 public final boolean isAcceptable();//是否可以accept
	 public final boolean isReadable();//是否可以读
	 public final boolean isWritable();//是否可以写
}

在这里插入图片描述

ServerSocketChannel

ServerSocketChannel在服务器端监听薪的客户端Socket连接
相关方法如下：

public abstract class ServerSocketChannel
    extends AbstractSelectableChannel
    implements NetworkChannel
{
	public static ServerSocketChannel open();//得到一个ServerSocketChannel通道
	public final ServerSocketChannel bind(SocketAddress local);//设置服务器端端口号
	public final SelectableChannel configureBlocking(boolean block);//设置阻塞或非阻塞模式，取值false标识采用非阻塞模式
	public abstract SocketChannel accept();//接受一个连接，返回代表这个连接的通道对象
	public final SelectionKey register(Selector sel, int ops);//注册一个选择器并设置监听事件
}

在这里插入图片描述

SocketChannel

SocketChannel，网络IO通道，具体负责进行读写操作。NIO把缓冲区的数据写入通道，或者把通道里的数据读到缓冲区
相关方法如下：

public abstract class SocketChannel
    extends AbstractSelectableChannel
    implements ByteChannel, ScatteringByteChannel, GatheringByteChannel, NetworkChannel
{
	public static SocketChannel open();//得到一个SocketChannel通道
	public final SelectableChannel configureBlocking(boolean block);//设置阻塞或非阻塞模式，取值false标识采用非阻塞模式
	public abstract boolean connect(SocketAddress remote);//连接服务器
	public abstract boolean finishConnect();//如果上面的方法连接失败，接下来就要通过该方法完成连接操作
	public abstract int write(ByteBuffer src);//往通道里写数据
	public abstract int read(ByteBuffer dst);//从通道里读数据
	public abstract SelectionKey register(Selector sel, int ops, Object att);//注册一个选择器并设置监听事件，最后一个参数可以设置共享数据
	public final void close();//关闭通道
}