Netty——网络编程 NIO（Selector处理read事件 客户端断开问题解决）代码示例

服务端代码

package com.example.nettytest.nio.day3;

import com.example.nettytest.nio.day1.ByteBufferUtil;
import lombok.extern.slf4j.Slf4j;

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;

/**
 * @description: Selector处理read事件（客户端断开问题） 代码示例
 * @author: xz
 * @create: 2022-09-04
 */
@Slf4j
public class Test4Server {
    public static void main(String[] args){
        nioSelectorReadServer();
    }
    /**
     * 1、客户端强制断开后，此方法报错：远程主机强迫关闭了一个现有的连接。
     * 2、客户端正常断开后，服务端会一直产生读事件。
     * */
    private static void nioSelectorReadServer(){
        try {
            // 1. 创建 selector, 管理多个 channel
            Selector selector = Selector.open();
            ServerSocketChannel ssc = ServerSocketChannel.open();
            ssc.configureBlocking(false);
            // 2. 建立 selector 和 channel 的联系（注册）
            // SelectionKey 就是将来事件发生后，通过它可以知道事件和哪个channel的事件
            SelectionKey sscKey = ssc.register(selector, 0, null);
            // key 只关注 accept 事件
            sscKey.interestOps(SelectionKey.OP_ACCEPT);
            log.debug("sscKey:{}", sscKey);
            ssc.bind(new InetSocketAddress(8080));
            while (true) {
                // 3. select 方法, 没有事件发生，线程阻塞，有事件，线程才会恢复运行
                // select 在事件未处理时，它不会阻塞, 事件发生后要么处理，要么取消，不能置之不理
                selector.select();
                // 4. 处理事件, selectedKeys 内部包含了所有发生的事件
                Iterator iter = selector.selectedKeys().iterator(); // accept, read
                while (iter.hasNext()) {
                    SelectionKey key = iter.next();
                    // 处理key 时，要从 selectedKeys 集合中删除，否则下次处理就会有问题
                    iter.remove();
                    log.debug("key: {}", key);
                    // 5. 区分事件类型
                    if (key.isAcceptable()) { // 如果是 accept
                        ServerSocketChannel channel = (ServerSocketChannel) key.channel();
                        SocketChannel sc = channel.accept();
                        sc.configureBlocking(false);
                        ByteBuffer buffer = ByteBuffer.allocate(16); // attachment
                        // 将一个 byteBuffer 作为附件关联到 selectionKey 上
                        SelectionKey scKey = sc.register(selector, 0, buffer);
                        scKey.interestOps(SelectionKey.OP_READ);
                        log.debug("{}", sc);
                        log.debug("scKey:{}", scKey);
                    } else if (key.isReadable()) { // 如果是 read
                        SocketChannel channel = (SocketChannel) key.channel(); // 拿到触发事件的channel
                        // 获取 selectionKey 上关联的附件
                        ByteBuffer buffer = (ByteBuffer) key.attachment();
                        channel.read(buffer); // 如果是正常断开，read 的方法的返回值是 -1
                        buffer.flip();
                        //打印可读取内容
                        ByteBufferUtil.debugRead(buffer);

                    }
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

客户端代码示例

package com.example.nettytest.nio.day3;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.nio.channels.SocketChannel;

/**
 * @description:
 * @author: xz
 * @create: 2022-09-04
 */
public class Test4Client {
    public static void main(String[] args) throws IOException {
        SocketChannel sc = SocketChannel.open();
        sc.connect(new InetSocketAddress("localhost", 8080));
        SocketAddress address = sc.getLocalAddress();
        System.out.println("waiting...");
        sc.close();
    }
}

工具类，打印输入、输出数据使用

package com.example.nettytest.nio.day1;

import io.netty.util.internal.StringUtil;

import java.nio.ByteBuffer;

import static io.netty.util.internal.MathUtil.isOutOfBounds;
import static io.netty.util.internal.StringUtil.NEWLINE;

public class ByteBufferUtil {
    private static final char[] BYTE2CHAR = new char[256];
    private static final char[] HEXDUMP_TABLE = new char[256 * 4];
    private static final String[] HEXPADDING = new String[16];
    private static final String[] HEXDUMP_ROWPREFIXES = new String[65536 >>> 4];
    private static final String[] BYTE2HEX = new String[256];
    private static final String[] BYTEPADDING = new String[16];

    static {
        final char[] DIGITS = "0123456789abcdef".toCharArray();
        for (int i = 0; i  4 & 0x0F];
            HEXDUMP_TABLE[(i