学习完nio的一个小笔记吧

这是一个nio网络通信服务端的demo，主要就学习了selector的一些用法，以及它里面的事件类型
selector是对nio的一个优化，它能保证既能高效处理线程中的事件，又能保证线程不会一直占用cpu

其中我认为最重要的是selector的执行流程，机制

将 channel和 selector建立联系。（联系就是指客户端向服务端发送的某个事件会被selector给监听到）。
当客户端给服务器发送了相应事件后，selector就会将这个事件的 selectionKey加入到集合 selectionKeys中，并且处理该事件。
在处理事件的时候，可以根据事件类型来进行处理。比如说使用 channel.accept()、channel.read(xx)来处理。
对一些客户端异常关闭，或者说正常关闭要特殊处理。客户端关闭，可以直接调 key.cancel()将该key从 selector中删除。

服务端

import lombok.extern.slf4j.Slf4j;

import java.io.IOException;

import java.net.InetSocketAddress;

import java.nio.ByteBuffer;

import java.nio.channels.*;

import java.util.Iterator;

import java.util.Set;

import static com.pzistart.utils.ByteBufferUtil.debugRead;

/**

 * @author Pzi

 * @create 2022-09-12 17:47

 */

@Slf4j

public class ServerUp {

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

        ServerSocketChannel ssc = ServerSocketChannel

                .open()

                .bind(new InetSocketAddress(8080));

        //  1.创建 selector，管理多个 channel

        Selector selector = Selector.open();

        //  2.开启非阻塞模式，并且将ssc注册到selector中

        ssc.configureBlocking(false);

        //  3.建立 ssc(channel) 和 selector 的联系

        SelectionKey sscKey = ssc.register(selector, SelectionKey.OP_ACCEPT);

        log.debug("regist key:{}", sscKey);

        while (true) {

            //  如果selector监听到有事件发生，那么就向 selectionKeys 这个集合中加入key。

            //  并且执行下面的操作，否则就阻塞线程

            int count = selector.select();

            // selectionKeys中包含所有的事件

            Set<SelectionKey> selectionKeys = selector.selectedKeys();

            //  4.处理事件

            Iterator<SelectionKey> iterator = selectionKeys.iterator();

            while (iterator.hasNext()) {

                SelectionKey key = iterator.next();

                //  在selectionKeys中即使是处理完毕事件，也不会自动移除key。

                //  导致下一次处理事件的时候，还会找到事件类型为 accept 的那个 SelectionKey

                //  但是由于accept事件已经被消耗，所以获取不到对应的 ServerSocketChannel。所以在用完该key，就要立马移除集合中该key

                iterator.remove();

                //  如果客户端发过来的是 accept相关的事件，那么就会被 selector 识别并且将该时间加入到 selectionKeys集合中。从而在下面的事件处理分支进行处理。

                //  如果客户端发过来的是 read相关的之间，同样的事件处理方式

                if (key.isAcceptable()) {

                    log.debug("key{}", key);

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

                    //  调用accept()方法处理事件

                    SocketChannel sc = c.accept();

                    sc.configureBlocking(false);

                    //  将sc注册到selector中，建立sc(channel)和selector的联系

                    SelectionKey scKey = sc.register(selector, SelectionKey.OP_READ);

                    log.debug("connected... {}", sc);

                    // 也可以调用cancel()方法处理事件 key.cancel();

                } else if (key.isReadable()) {

                    try {

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

                        //  将sc注册到selector中

                        ByteBuffer buffer = ByteBuffer.allocate(16);

                        int read = channel.read(buffer);

                        if (read == -1) {

                            key.cancel();

                        } else {

                            buffer.flip();

                            debugRead(buffer);

                            buffer.clear();

                        }

                    } catch (Exception e) {

                        e.printStackTrace();

                        //  在客户端异常断开连接时，客户端会向服务器发送一个 read()事件

                        //  read()事件没有得到处理，那么就会多次被 selector监听到，从而循环抛异常。

                        //  解决方法就是将该 key删除，就是直接从 selector中反注册，那么 selector自然不会监听到该 channel的相应事件

                        //  key.cancel();

                    }

                }

            }

        }

    }

}

客户端

package com.pzistart.netcoding.bio;

import java.io.IOException;

import java.net.InetSocketAddress;

import java.nio.channels.SocketChannel;

/**

 * @author Pzi

 * @create 2022-09-12 15:04

 */

public class Client {

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

        SocketChannel sc = SocketChannel.open();

        sc.connect(new InetSocketAddress("localhost", 8080));

        System.out.println("ccc");

    }

}

巴特西

学习完nio的一个小笔记吧

其中我认为最重要的是selector的执行流程，机制

服务端

客户端

最新文章

热门文章