NIO 源码分析(01) NIO 最简用法

一、服务端
二、客户端

Netty 系列目录(https://www.cnblogs.com/binarylei/p/10117436.html)

Java NIO 主要由三个部分组成：Channel、Buffer 和 Selector。在分析源码前最好对 NIO 的基本用法和 Linux NIO 在一个基本的了解。

NIO 入门

Linux NIO

本文会提供一个 NIO 最简使用示例，之后的源码分析都会基于该示例及其扩展进行。

一、服务端

public class Server implements Runnable {

    public static void main(String[] args) {

        new Thread(new Server(8765)).start();

    }

    // 1 多路复用器（管理所有的通道）

    private Selector selector;

    // 2 建立缓冲区

    private ByteBuffer readBuf = ByteBuffer.allocate(1024);

    private ByteBuffer writeBuf = ByteBuffer.allocate(1024);

    public Server(int port) {

        try {

            //1. 获取多路复用器

            this.selector = Selector.open();

            //2. 获取服务器端通道

            ServerSocketChannel ssChannel = ServerSocketChannel.open();

            //3. 切换成非阻塞模式

            ssChannel.configureBlocking(false);

            //4. 绑定端口号

            ssChannel.bind(new InetSocketAddress(port));

            //5. 把服务器通道注册到多路复用器上，并且监听阻塞事件

            ssChannel.register(this.selector, SelectionKey.OP_ACCEPT);

            System.out.println("Server start, port :" + port);

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    public void run() {

        while (true) {

            try {

                // 1. 阻塞到至少有一个通道在你注册的事件上就绪，返回的 int 值表示有多少通道已经就绪

                int wait = selector.select();

                if (wait == 0) continue;

                //2. 遍历多路复用器上已经准备就绪的通道

                Iterator<SelectionKey> iterator = this.selector.selectedKeys().iterator();

                while (iterator.hasNext()) {

                    SelectionKey key = iterator.next();

                    if (key.isValid()) {

                        if (key.isAcceptable()) {

                            // 3.1 如果为 accept 状态，就获取客户端通道并注册到selector上

                            this.accept(key);

                        } else if (key.isReadable()) {

                            // 3.2 如果为可读状态

                            this.read(key);

                        } else if (key.isWritable()) {

                            // 3.3 写数据

                            this.write(key);

                        }

                    }

                    // 4. 注意必须在处理完通道时自己移除

                    iterator.remove();

                }

            } catch (IOException e) {

                e.printStackTrace();

            }

        }

    }

    private void write(SelectionKey key) {

        //ServerSocketChannel ssc =  (ServerSocketChannel) key.channel();

        //ssc.register(this.seletor, SelectionKey.OP_WRITE);

    }

    private void read(SelectionKey key) {

        try {

            //1. 清空缓冲区旧的数据

            this.readBuf.clear();

            //2. 获取之前注册的socket通道对象

            SocketChannel sChannel = (SocketChannel) key.channel();

            //3. 读数据

            int len = sChannel.read(this.readBuf);

            if (len == -1) {

                key.channel().close();

                key.cancel();

                return;

            }

            //5 有数据则进行读取 读取之前要flip()复位

            this.readBuf.flip();

            byte[] bytes = new byte[this.readBuf.remaining()];

            this.readBuf.get(bytes);

            System.out.println("Server : " + new String(bytes).trim());

            //6. 可以写回给客户端数据

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    private void accept(SelectionKey key) {

        try {

            //1. 获取服务通道ServerSocketChannel

            ServerSocketChannel ssChannel = (ServerSocketChannel) key.channel();

            //2. 获取客户端通道SocketChannel

            SocketChannel sChannel = ssChannel.accept();

            //3. 客户端通道SocketChannel也要设置成非阻塞模式

            sChannel.configureBlocking(false);

            //4 注册到多路复用器上，并设置读取标识

            sChannel.register(this.selector, SelectionKey.OP_READ);

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

}

二、客户端

public static void main(String[] args) throws IOException {

    SocketChannel sChannel = null;

    try {

        //1. 获取通道

        sChannel = SocketChannel.open(

                new InetSocketAddress("127.0.0.1", 8765));

        //2. 切换成非阻塞模式

        sChannel.configureBlocking(false);

        //3. 分配缓冲区

        ByteBuffer buf = ByteBuffer.allocate(1024);

        //4. 把从控制台读到数据发送给服务器端

        Scanner scanner = new Scanner(System.in);

        while (true) {

            String str = scanner.next();

            SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd hh:mm:ss");

            buf.put((dateFormat.format(new Date()) + "：" + str).getBytes());

            buf.flip();// 非阻塞模式下，write()方法在尚未写出任何内容时可能就返回了

            while (buf.hasRemaining()) {

                sChannel.write(buf);

            }

            buf.clear();

        }

    } finally {

        sChannel.close();

    }

}

ok，完成！！！

每天用心记录一点点。内容也许不重要，但习惯很重要！

巴特西

NIO 源码分析(01) NIO 最简用法

一、服务端

二、客户端

最新文章

热门文章