• Netty——NIO(Selector处理read事件 客户端断开问题解决)代码示例


    一、未处理客户端断开问题的代码示例

    1.1、服务端代码示例

    • 服务端代码

      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<SelectionKey> 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();
              }
          }
      }
      
      • 1
      • 2
      • 3
      • 4
      • 5
      • 6
      • 7
      • 8
      • 9
      • 10
      • 11
      • 12
      • 13
      • 14
      • 15
      • 16
      • 17
      • 18
      • 19
      • 20
      • 21
      • 22
      • 23
      • 24
      • 25
      • 26
      • 27
      • 28
      • 29
      • 30
      • 31
      • 32
      • 33
      • 34
      • 35
      • 36
      • 37
      • 38
      • 39
      • 40
      • 41
      • 42
      • 43
      • 44
      • 45
      • 46
      • 47
      • 48
      • 49
      • 50
      • 51
      • 52
      • 53
      • 54
      • 55
      • 56
      • 57
      • 58
      • 59
      • 60
      • 61
      • 62
      • 63
      • 64
      • 65
      • 66
      • 67
      • 68
      • 69
      • 70
      • 71
      • 72
      • 73
      • 74
      • 75
      • 76
      • 77
      • 78
      • 79
      • 80

    1.2、客户端代码示例

    • 客户端代码示例

      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();
          }
      }
      
      • 1
      • 2
      • 3
      • 4
      • 5
      • 6
      • 7
      • 8
      • 9
      • 10
      • 11
      • 12
      • 13
      • 14
      • 15
      • 16
      • 17
      • 18
      • 19
      • 20
      • 21

    1.3、工具类

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

      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 < 256; i++) {
                  HEXDUMP_TABLE[i << 1] = DIGITS[i >>> 4 & 0x0F];
                  HEXDUMP_TABLE[(i << 1) + 1] = DIGITS[i & 0x0F];
              }
      
              int i;
      
              // Generate the lookup table for hex dump paddings
              for (i = 0; i < HEXPADDING.length; i++) {
                  int padding = HEXPADDING.length - i;
                  StringBuilder buf = new StringBuilder(padding * 3);
                  for (int j = 0; j < padding; j++) {
                      buf.append("   ");
                  }
                  HEXPADDING[i] = buf.toString();
              }
      
              // Generate the lookup table for the start-offset header in each row (up to 64KiB).
              for (i = 0; i < HEXDUMP_ROWPREFIXES.length; i++) {
                  StringBuilder buf = new StringBuilder(12);
                  buf.append(NEWLINE);
                  buf.append(Long.toHexString(i << 4 & 0xFFFFFFFFL | 0x100000000L));
                  buf.setCharAt(buf.length() - 9, '|');
                  buf.append('|');
                  HEXDUMP_ROWPREFIXES[i] = buf.toString();
              }
      
              // Generate the lookup table for byte-to-hex-dump conversion
              for (i = 0; i < BYTE2HEX.length; i++) {
                  BYTE2HEX[i] = ' ' + StringUtil.byteToHexStringPadded(i);
              }
      
              // Generate the lookup table for byte dump paddings
              for (i = 0; i < BYTEPADDING.length; i++) {
                  int padding = BYTEPADDING.length - i;
                  StringBuilder buf = new StringBuilder(padding);
                  for (int j = 0; j < padding; j++) {
                      buf.append(' ');
                  }
                  BYTEPADDING[i] = buf.toString();
              }
      
              // Generate the lookup table for byte-to-char conversion
              for (i = 0; i < BYTE2CHAR.length; i++) {
                  if (i <= 0x1f || i >= 0x7f) {
                      BYTE2CHAR[i] = '.';
                  } else {
                      BYTE2CHAR[i] = (char) i;
                  }
              }
          }
      
          /**
           * 打印所有内容
           * @param buffer
           */
          public static void debugAll(ByteBuffer buffer) {
              int oldlimit = buffer.limit();
              buffer.limit(buffer.capacity());
              StringBuilder origin = new StringBuilder(256);
              appendPrettyHexDump(origin, buffer, 0, buffer.capacity());
              System.out.println("+--------+-------------------- all ------------------------+----------------+");
              System.out.printf("position: [%d], limit: [%d]\n", buffer.position(), oldlimit);
              System.out.println(origin);
              buffer.limit(oldlimit);
          }
      
          /**
           * 打印可读取内容
           * @param buffer
           */
          public static void debugRead(ByteBuffer buffer) {
              StringBuilder builder = new StringBuilder(256);
              appendPrettyHexDump(builder, buffer, buffer.position(), buffer.limit() - buffer.position());
              System.out.println("+--------+-------------------- read -----------------------+----------------+");
              System.out.printf("position: [%d], limit: [%d]\n", buffer.position(), buffer.limit());
              System.out.println(builder);
          }
      
          public static void main(String[] args) {
              ByteBuffer buffer = ByteBuffer.allocate(10);
              buffer.put(new byte[]{97, 98, 99, 100});
              debugAll(buffer);
          }
      
          private static void appendPrettyHexDump(StringBuilder dump, ByteBuffer buf, int offset, int length) {
              if (isOutOfBounds(offset, length, buf.capacity())) {
                  throw new IndexOutOfBoundsException(
                          "expected: " + "0 <= offset(" + offset + ") <= offset + length(" + length
                                  + ") <= " + "buf.capacity(" + buf.capacity() + ')');
              }
              if (length == 0) {
                  return;
              }
              dump.append(
                      "         +-------------------------------------------------+" +
                              NEWLINE + "         |  0  1  2  3  4  5  6  7  8  9  a  b  c  d  e  f |" +
                              NEWLINE + "+--------+-------------------------------------------------+----------------+");
      
              final int startIndex = offset;
              final int fullRows = length >>> 4;
              final int remainder = length & 0xF;
      
              // Dump the rows which have 16 bytes.
              for (int row = 0; row < fullRows; row++) {
                  int rowStartIndex = (row << 4) + startIndex;
      
                  // Per-row prefix.
                  appendHexDumpRowPrefix(dump, row, rowStartIndex);
      
                  // Hex dump
                  int rowEndIndex = rowStartIndex + 16;
                  for (int j = rowStartIndex; j < rowEndIndex; j++) {
                      dump.append(BYTE2HEX[getUnsignedByte(buf, j)]);
                  }
                  dump.append(" |");
      
                  // ASCII dump
                  for (int j = rowStartIndex; j < rowEndIndex; j++) {
                      dump.append(BYTE2CHAR[getUnsignedByte(buf, j)]);
                  }
                  dump.append('|');
              }
      
              // Dump the last row which has less than 16 bytes.
              if (remainder != 0) {
                  int rowStartIndex = (fullRows << 4) + startIndex;
                  appendHexDumpRowPrefix(dump, fullRows, rowStartIndex);
      
                  // Hex dump
                  int rowEndIndex = rowStartIndex + remainder;
                  for (int j = rowStartIndex; j < rowEndIndex; j++) {
                      dump.append(BYTE2HEX[getUnsignedByte(buf, j)]);
                  }
                  dump.append(HEXPADDING[remainder]);
                  dump.append(" |");
      
                  // Ascii dump
                  for (int j = rowStartIndex; j < rowEndIndex; j++) {
                      dump.append(BYTE2CHAR[getUnsignedByte(buf, j)]);
                  }
                  dump.append(BYTEPADDING[remainder]);
                  dump.append('|');
              }
      
              dump.append(NEWLINE +
                      "+--------+-------------------------------------------------+----------------+");
          }
      
          private static void appendHexDumpRowPrefix(StringBuilder dump, int row, int rowStartIndex) {
              if (row < HEXDUMP_ROWPREFIXES.length) {
                  dump.append(HEXDUMP_ROWPREFIXES[row]);
              } else {
                  dump.append(NEWLINE);
                  dump.append(Long.toHexString(rowStartIndex & 0xFFFFFFFFL | 0x100000000L));
                  dump.setCharAt(dump.length() - 9, '|');
                  dump.append('|');
              }
          }
      
          public static short getUnsignedByte(ByteBuffer buffer, int index) {
              return (short) (buffer.get(index) & 0xFF);
          }
      }
      
      • 1
      • 2
      • 3
      • 4
      • 5
      • 6
      • 7
      • 8
      • 9
      • 10
      • 11
      • 12
      • 13
      • 14
      • 15
      • 16
      • 17
      • 18
      • 19
      • 20
      • 21
      • 22
      • 23
      • 24
      • 25
      • 26
      • 27
      • 28
      • 29
      • 30
      • 31
      • 32
      • 33
      • 34
      • 35
      • 36
      • 37
      • 38
      • 39
      • 40
      • 41
      • 42
      • 43
      • 44
      • 45
      • 46
      • 47
      • 48
      • 49
      • 50
      • 51
      • 52
      • 53
      • 54
      • 55
      • 56
      • 57
      • 58
      • 59
      • 60
      • 61
      • 62
      • 63
      • 64
      • 65
      • 66
      • 67
      • 68
      • 69
      • 70
      • 71
      • 72
      • 73
      • 74
      • 75
      • 76
      • 77
      • 78
      • 79
      • 80
      • 81
      • 82
      • 83
      • 84
      • 85
      • 86
      • 87
      • 88
      • 89
      • 90
      • 91
      • 92
      • 93
      • 94
      • 95
      • 96
      • 97
      • 98
      • 99
      • 100
      • 101
      • 102
      • 103
      • 104
      • 105
      • 106
      • 107
      • 108
      • 109
      • 110
      • 111
      • 112
      • 113
      • 114
      • 115
      • 116
      • 117
      • 118
      • 119
      • 120
      • 121
      • 122
      • 123
      • 124
      • 125
      • 126
      • 127
      • 128
      • 129
      • 130
      • 131
      • 132
      • 133
      • 134
      • 135
      • 136
      • 137
      • 138
      • 139
      • 140
      • 141
      • 142
      • 143
      • 144
      • 145
      • 146
      • 147
      • 148
      • 149
      • 150
      • 151
      • 152
      • 153
      • 154
      • 155
      • 156
      • 157
      • 158
      • 159
      • 160
      • 161
      • 162
      • 163
      • 164
      • 165
      • 166
      • 167
      • 168
      • 169
      • 170
      • 171
      • 172
      • 173
      • 174
      • 175
      • 176
      • 177
      • 178
      • 179
      • 180
      • 181
      • 182
      • 183

    1.4、客户端强制断开测试

    • 先启动服务端
      在这里插入图片描述
    • debug模式启动客户端【再System.out.println(“waiting…”)代码位置加断点】
      在这里插入图片描述
    1.4.1、第一种情况:强制停止客户端测试
    • 强制停止客户端,查看服务端控制台输出
      在这里插入图片描述
    1.4.2、第一种情况结果:强制停止客户端,服务端控制台报错
    • 如下图所示:客户端强制断开后,控制台报错:远程主机强迫关闭了一个现有的连接。

      在这里插入图片描述

    1.4.3、第二种情况:正常停止客户端测试
    • 正常停止客户端,查看服务端控制台输出

    在这里插入图片描述

    1.4.4、第二种情况结果:正常停止客户端测试,服务端会一直产生读事件。
    • 如下图所示:客户端正常断开后,服务端会一直产生读事件。

      在这里插入图片描述

    二、处理客户端断开问题的代码示例

    2.1、修改服务端代码示例

    • 再处理read事件代码中,代码修如下:

       try {
           SocketChannel channel = (SocketChannel) key.channel(); // 拿到触发事件的channel
            // 获取 selectionKey 上关联的附件
            ByteBuffer buffer = (ByteBuffer) key.attachment();
            int read=channel.read(buffer); // 如果是正常断开,read 的方法的返回值是 -1
            if(read == -1) {
                key.cancel();
            } else {
                buffer.flip();
                //打印可读取内容
                ByteBufferUtil.debugRead(buffer);
            }
        } catch (IOException e) {
            e.printStackTrace();
            // 因为客户端断开了,因此需要将 key 取消(从 selector 的 keys 集合中真正删除 key)
            key.cancel();
        }
      
      • 1
      • 2
      • 3
      • 4
      • 5
      • 6
      • 7
      • 8
      • 9
      • 10
      • 11
      • 12
      • 13
      • 14
      • 15
      • 16
      • 17

    2.2、 服务端修改后的完整代码

    • 修改后的完整代码示例

      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();
          }
      	/**
           * 3、客户端断开后报错 远程主机强迫关闭了一个现有的连接问题
           * */
          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<SelectionKey> 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
                              try {
                                  SocketChannel channel = (SocketChannel) key.channel(); // 拿到触发事件的channel
                                  // 获取 selectionKey 上关联的附件
                                  ByteBuffer buffer = (ByteBuffer) key.attachment();
                                  int read=channel.read(buffer); // 如果是正常断开,read 的方法的返回值是 -1
                                  if(read == -1) {
                                      key.cancel();
                                  } else {
                                      buffer.flip();
                                      //打印可读取内容
                                      ByteBufferUtil.debugRead(buffer);
                                  }
                              } catch (IOException e) {
                                  e.printStackTrace();
                                  // 因为客户端断开了,因此需要将 key 取消(从 selector 的 keys 集合中真正删除 key)
                                  key.cancel();
                              }
      
                          }
                      }
                  }
              } catch (IOException e) {
                  e.printStackTrace();
              }
          }
      }
      
      • 1
      • 2
      • 3
      • 4
      • 5
      • 6
      • 7
      • 8
      • 9
      • 10
      • 11
      • 12
      • 13
      • 14
      • 15
      • 16
      • 17
      • 18
      • 19
      • 20
      • 21
      • 22
      • 23
      • 24
      • 25
      • 26
      • 27
      • 28
      • 29
      • 30
      • 31
      • 32
      • 33
      • 34
      • 35
      • 36
      • 37
      • 38
      • 39
      • 40
      • 41
      • 42
      • 43
      • 44
      • 45
      • 46
      • 47
      • 48
      • 49
      • 50
      • 51
      • 52
      • 53
      • 54
      • 55
      • 56
      • 57
      • 58
      • 59
      • 60
      • 61
      • 62
      • 63
      • 64
      • 65
      • 66
      • 67
      • 68
      • 69
      • 70
      • 71
      • 72
      • 73
      • 74
      • 75
      • 76
      • 77
      • 78
      • 79
      • 80
      • 81
      • 82
      • 83
      • 84
      • 85
      • 86
      • 87
      • 88

    2.3、客户端强制断开测试

    • 先启动服务端
      在这里插入图片描述
    • debug模式启动客户端【再System.out.println(“waiting…”)代码位置加断点】
      在这里插入图片描述
    2.3.1、第一种情况:强制停止客户端测试
    • 强制停止客户端,查看服务端控制台输出
      在这里插入图片描述
    2.3.2、第一种情况结果:强制停止客户端,服务端控制台无报错
    • 如下图所示:客户端强制断开后,服务端控制台无报错信息。
      在这里插入图片描述
    2.3.3、第二种情况:正常停止客户端测试
    • 正常停止客户端,查看服务端控制台输出
      在这里插入图片描述
    2.3.4、第二种情况结果:正常停止客户端测试,服务端无报错。
    • 如下图所示:客户端正常断开后,服务端无报错信息。
      在这里插入图片描述
  • 相关阅读:
    与字节、小米、移动云等企业一起揭秘 RocketMQ 实践之道
    matlab中实现画函数图像添加坐标轴
    月薪近万,2次转行失败,家政小哥选择软件测试,这次,他成功了
    如何做好网站SEO优化,网站SEO关键词优化的5个技巧
    Springboot 整合 SpringCache 使用 Redis 作为缓存
    acwing算法基础之基础算法--区间合并算法
    小程序的基本使用
    【Linux】 reboot 命令使用
    编写 bzt 脚本的正确姿势
    无线传感器网络的Z-SEP路由协议及对比(Matlab代码实现)
  • 原文地址:https://blog.csdn.net/li1325169021/article/details/126690068