NIO(NO-Blocking I/O)模型

NIO 的一些关键概念和组件:

Channels(通道):代表到实体(如文件、套接字等)的开放连接,可以通过通道读写数据。

Buffers(缓冲区):是数据容器,用于存储不同类型的数据。通道读取数据到缓冲区,或者从缓冲区写入数据到通道。

Selectors(选择器):如果应用需要处理多个通道,选择器可以监控多个通道的 I/O 事件,如连接请求、数据到达等。这使得单个线程能够管理多个通道,从而提高效率。

⼯作原理可以分为以下⼏个步骤:

通过 FileChannel 、 SocketChannel 、 ServerSocketChannel 等类的静态⽅法打开 Channel;

open() 打开⼀个通道,创建⼀个或多个 Buffer ,⽤于读取或写⼊数据;

将数据写⼊ Buffer ,然后将 Buffer 中的数据写⼊ Channel ;

从 Channel 读取数据: 将 Channel 中的数据读取到 Buffer 中;

注册 Channel 到 Selector: 通过 Selector 监听⼀个或多个 Channel ,当 Channel 上发⽣感兴趣的事件时, Selector 将通知程序;

处理事件: 在⼀个循环中调⽤ Selector 的 select() ⽅法,该⽅法会阻塞直到⾄少⼀个注册的 Channel发⽣了感兴趣的事件。然后通过迭代 selectedKeys() 获取SelectionKey ,从⽽得知哪个 Channel 上发⽣了事件;

java 复制代码
//简单使用
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.nio.ByteBuffer;

public class NioExample {
    public static void main(String[] args) {
        String filePath = "/path/to/file.txt";
        
        try {
            // 创建一个通道
            java.nio.channels.SeekableByteChannel channel = Files.newByteChannel(Paths.get(filePath));
            
            // 创建一个缓冲区
            ByteBuffer buffer = ByteBuffer.allocate(1024); // 1KB buffer
            
            // 从通道读取数据到缓冲区
            while (channel.read(buffer) > 0) {
                // 切换为读模式
                buffer.flip();
                
                // 读取数据
                while (buffer.hasRemaining()) {
                    System.out.print((char) buffer.get());
                }
                
                // 清空缓冲区,准备下一次读取
                buffer.clear();
            }
            
            // 关闭通道
            channel.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}
java 复制代码
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.*;
import java.util.Iterator;
import java.util.Set;

public class SimpleNioServer {
    public void startServer(int port) throws IOException {
        Selector selector = Selector.open();
        ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
        serverSocketChannel.bind(new InetSocketAddress(port));
        serverSocketChannel.configureBlocking(false);
        serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);

        while (true) {
            selector.select(); // Wait for an event
            Set<SelectionKey> selectedKeys = selector.selectedKeys();
            Iterator<SelectionKey> iter = selectedKeys.iterator();

            while (iter.hasNext()) {
                SelectionKey key = iter.next();
                
                if (key.isAcceptable()) {
                    registerClient(serverSocketChannel, selector);
                }
                
                if (key.isReadable()) {
                    answerClient(key);
                }

                iter.remove();
            }
        }
    }

    private void registerClient(ServerSocketChannel serverSocketChannel, Selector selector) throws IOException {
        SocketChannel clientChannel = serverSocketChannel.accept();
        clientChannel.configureBlocking(false);
        clientChannel.register(selector, SelectionKey.OP_READ);
        System.out.println("New client connected: " + clientChannel);
    }

    private void answerClient(SelectionKey key) throws IOException {
        SocketChannel clientChannel = (SocketChannel) key.channel();
        ByteBuffer buffer = ByteBuffer.allocate(256);
        int read = clientChannel.read(buffer);

        if (read == -1) {
            clientChannel.close();
            System.out.println("Client disconnected: " + clientChannel);
            return;
        }

        buffer.flip();
        while (buffer.hasRemaining()) {
            clientChannel.write(buffer);
        }
        buffer.clear();
    }

    public static void main(String[] args) {
        SimpleNioServer server = new SimpleNioServer();
        try {
            server.startServer(8080);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

在这个例子中,startServer 方法会启动服务器,监听指定端口上的连接请求。当 selector.select() 调用返回时,意味着至少有一个通道准备好进行操作。如果事件是 OP_ACCEPT(即有新的连接请求),registerClient 方法会被调用以接受连接并注册新的客户端通道以读取数据。如果事件是 OP_READ(即有数据可读),answerClient 方法会被调用以读取数据,并将数据回显给客户端。

相关推荐
MacroZheng9 分钟前
完美替代 Navicat!这款内置 AI 的数据库工具,太香了!
java·后端·mysql
SamDeepThinking35 分钟前
从REST到gRPC,一个API选型的思考框架
java·后端·程序员
Zane199438 分钟前
接口都能写默认实现了,为什么还需要抽象类
java·后端
未秃头的程序猿40 分钟前
读Spring AI源码的方法——不是硬啃,是带着问题去翻
java·后端·spring
用户3126874877201 小时前
别再 Thread.sleep 硬等了!并发工具类到底怎么选?
java
Java内核笔记1 小时前
万字长文剖析 Spring Boot 4 自动配置机制源码:从 @EnableAutoConfiguration 到条件装配
java·后端
evans在进步1 小时前
Spring MVC 核心机制详解:全局异常处理、请求跳转与数据绑定
java·spring·mvc
weixin_BYSJ19871 小时前
springboot小区物业管理小程序---附源码45555
java·javascript·spring boot·python·django·flask·php
coder_lorraine1 小时前
Halo 2.x 回收站文章无法删除?一篇文章教你彻底解决
java·运维
索隆zoro1 小时前
Army 进阶:DML 操作体系与 Spring AI 集成
java