基本介绍
- 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:代表写操作,值为4public 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();//关闭通道
}