使用Java实现NIO

以下是一个使用 Java NIO 实现 Reactor 模型的简单示例代码,并附有详细的注释:

 
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;

public class ReactorServer {
    public static void main(String[] args) throws IOException {
        // 创建 Selector 对象
        Selector selector = Selector.open();

        // 创建 ServerSocketChannel,并设置为非阻塞模式
        ServerSocketChannel serverSocket = ServerSocketChannel.open();
        serverSocket.bind(new InetSocketAddress("127.0.0.1", 8888));
        serverSocket.configureBlocking(false);

        // 注册 ServerSocketChannel 到 Selector,监听连接事件
        serverSocket.register(selector, SelectionKey.OP_ACCEPT);

        while (true) {
            // 调用 select 方法阻塞等待事件发生
            selector.select();

            // 获取触发的事件集合
            Set<SelectionKey> selectedKeys = selector.selectedKeys();
            Iterator<SelectionKey> keyIterator = selectedKeys.iterator();

            while (keyIterator.hasNext()) {
                SelectionKey key = keyIterator.next();

                if (key.isAcceptable()) {
                    // 处理连接事件
                    ServerSocketChannel server = (ServerSocketChannel) key.channel();
                    SocketChannel client = server.accept();
                    client.configureBlocking(false);
                    client.register(selector, SelectionKey.OP_READ);
                } else if (key.isReadable()) {
                    // 处理读取事件
                    SocketChannel client = (SocketChannel) key.channel();
                    ByteBuffer buffer = ByteBuffer.allocate(1024);
                    int bytesRead = client.read(buffer);
                    if (bytesRead == -1) {
                        // 客户端关闭连接
                        client.close();
                    } else if (bytesRead > 0) {
                        // 读取数据并进行处理
                        buffer.flip();
                        byte[] data = new byte[buffer.remaining()];
                        buffer.get(data);
                        System.out.println("Received data: " + new String(data));
                        // 可以在这里添加业务逻辑处理
                        // ...
                    }
                }

                // 从触发的事件集合中移除当前事件
                keyIterator.remove();
            }
        }
    }
}

  

在这个示例中,我们使用了 Java NIO 的 Selector、ServerSocketChannel 和 SocketChannel 等组件实现了一个简单的 Reactor 服务器。主要的事件循环部分通过调用 selector.select() 阻塞等待事件发生,然后遍历处理触发的事件。

希望这段代码能够帮助你更好地了解在 Java 中如何使用 NIO 实现 Reactor 模型。如果你有任何疑问或需要进一步解释,请随时告诉我。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章