服务端:

public class NIOServer {private static final String HOST = "localhost";private static final int PORT = 10086;public static void main(String[] args) {ServerSocketChannel serverSocketChannel = null;ServerSocket serverSocket = null;Selector selector = null;try {serverSocketChannel = ServerSocketChannel.open();//工厂方法创建ServerSocketChannelserverSocket = serverSocketChannel.socket(); //获取channel对应的ServerSocketserverSocket.bind(new InetSocketAddress(HOST, PORT)); //绑定地址serverSocketChannel.configureBlocking(false); //设置ServerSocketChannel非阻塞模式selector = Selector.open();//工厂方法创建SelectorserverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);//通道注册选择器,接受连接就绪状态。while (true) {//循环检查if (selector.select() == 0) {//阻塞检查,当有就绪状态发生,返回键集合continue;}// if (selector.select(2000) == 0) {//     //在等待信道准备的同时,也可以异步地执行其他任务,  这里打印*//     System.out.print("*");//     continue;// }Iterator<SelectionKey> it = selector.selectedKeys().iterator(); //获取就绪键遍历对象。while (it.hasNext()) {SelectionKey selectionKey = it.next();//处理就绪状态if (selectionKey.isAcceptable()) {ServerSocketChannel schannel = (ServerSocketChannel) selectionKey.channel();//只负责监听,阻塞,管理,不发送、接收数据SocketChannel socketChannel = schannel.accept();//就绪后的操作,刚到达的socket句柄if (null == socketChannel) {continue;}socketChannel.configureBlocking(false);socketChannel.register(selector, SelectionKey.OP_READ); //告知选择器关心的通道,准备好读数据} else if (selectionKey.isReadable()) {SocketChannel socketChannel = (SocketChannel) selectionKey.channel();ByteBuffer byteBuffer = ByteBuffer.allocate(4 * 1024);System.out.println(socketChannel.getRemoteAddress());int read = socketChannel.read(byteBuffer);if (read == -1) {//如果不关闭会一直产生isReadable这个消息
                            selectionKey.cancel();socketChannel.close();} else {StringBuilder result = new StringBuilder();while (read > 0) {//确保读完
                                byteBuffer.flip();result.append(new String(byteBuffer.array()));byteBuffer.clear();//每次清空 对应上面flip()read = socketChannel.read(byteBuffer);}System.out.println("server receive: " + result.toString());socketChannel.register(selector, SelectionKey.OP_WRITE);}} else if (selectionKey.isWritable()) {SocketChannel socketChannel = (SocketChannel) selectionKey.channel();String sendStr = "server send data: " + Math.random();ByteBuffer send = ByteBuffer.wrap(sendStr.getBytes());while (send.hasRemaining()) {socketChannel.write(send);}socketChannel.register(selector, SelectionKey.OP_READ);System.out.println(sendStr);}it.remove();}}} catch (IOException e) {e.printStackTrace();}}
}

客户端:

public class NIOClient {public static void main(String[] args) throws Exception {SocketChannel clntChan = SocketChannel.open();clntChan.configureBlocking(false);if (!clntChan.connect(new InetSocketAddress("localhost", 10086))) {//不断地轮询连接状态,直到完成连接while (!clntChan.finishConnect()) {//在等待连接的时间里,可以执行其他任务,以充分发挥非阻塞IO的异步特性//这里为了演示该方法的使用,只是一直打印"."System.out.print(".");}}//为了与后面打印的"."区别开来,这里输出换行符System.out.print("\n");//分别实例化用来读写的缓冲区ByteBuffer writeBuf = ByteBuffer.wrap("send send send".getBytes());ByteBuffer readBuf = ByteBuffer.allocate("send".getBytes().length - 1);while (writeBuf.hasRemaining()) {//如果用来向通道中写数据的缓冲区中还有剩余的字节,则继续将数据写入信道
            clntChan.write(writeBuf);}Thread.sleep(10000);StringBuffer stringBuffer = new StringBuffer();//如果read()接收到-1,表明服务端关闭,抛出异常while ((clntChan.read(readBuf)) > 0) {readBuf.flip();stringBuffer.append(new String(readBuf.array(), 0, readBuf.limit()));readBuf.clear();}//打印出接收到的数据System.out.println("Client Received: " + stringBuffer.toString());//关闭信道
        clntChan.close();}
}

注意当客户端断开socket的时候需要处理,不然服务端对应的channel会一直处于readable的状态,会造成死循环。

当客户端调用:clntChan.close();

这一句是正常关闭代码,它会传给服务端关闭的指令(也就是数据)。

而服务端的socketChannel.read(byteBuffer)=-1则代表收到这个关闭指令(数据),可以依据-1来关闭服务端的channel,不过仍有可能有问题,可能是客户端调用socketChannel.close()的时候,服务端这边的bytebuffer已经满了,那么num只能返回0而不能返回-1。具体该怎么写你应该根据你自己的业务来处理。

当socketChannel为阻塞方式时(默认就是阻塞方式)read函数,不会返回0,阻塞方式的socketChannel,若没有数据可读,或者缓冲区满了,就会阻塞,直到满足读的条件,所以一般阻塞方式的read是比较简单的,不过阻塞方式的socketChannel的问题也是显而易见的。这里我结合基于NIO 写ftp服务器调试过程中碰到的问题,总结一下非阻塞场景下的read碰到的问题。注意:这里的场景都是基于客户端以阻塞socket的方式发送数据。

1、read什么时候返回-1

read返回-1说明客户端的数据发送完毕,并且主动的close socket。所以在这种场景下,(服务器程序)你需要关闭socketChannel并且取消key,最好是退出当前函数。注意,这个时候服务端要是继续使用该socketChannel进行读操作的话,就会抛出“远程主机强迫关闭一个现有的连接”的IO异常。

2、read什么时候返回0

其实read返回0有3种情况,一是某一时刻socketChannel中当前(注意是当前)没有数据可以读,这时会返回0,其次是bytebuffer的position等于limit了,即bytebuffer的remaining等于0,这个时候也会返回0,最后一种情况就是客户端的数据发送完毕了(注意看后面的程序里有这样子的代码),这个时候客户端想获取服务端的反馈调用了recv函数,若服务端继续read,这个时候就会返回0。

-------------------------------------------------------------------------------------------------

实际写代码过程中观察发现,如果客户端发送数据后不关闭channel,同时服务端收到数据后反倒再次发给客户端,那么此时客户端read方法永远返回0.

SelectionKey.OP_WRITE

当时有一个连接进来后,如果注册了SelectionKey.OP_WRITE消息,selectionKey.isWritable()会一直返回true知道写缓冲区写满。所以这点就值注册了SelectionKey.OP_READ.

OP_WRITE事件的就绪条件并不是发生在调用channel的write方法之后,而是在当底层缓冲区有空闲空间的情况下。因为写缓冲区在绝大部分时候都是有空闲空间的,所以如果你注册了写事件,这会使得写事件一直处于就就绪,选择处理现场就会一直占用着CPU资源。所以,只有当你确实有数据要写时再注册写操作,并在写完以后马上取消注册。
其实,在大部分情况下,我们直接调用channel的write方法写数据就好了,没必要都用OP_WRITE事件。那么OP_WRITE事件主要是在什么情况下使用的了?
其实OP_WRITE事件主要是在发送缓冲区空间满的情况下使用的。如:
while (buffer.hasRemaining()) {int len = socketChannel.write(buffer);   if (len == 0) {selectionKey.interestOps(selectionKey.interestOps() | SelectionKey.OP_WRITE);selector.wakeup();break;}
}

当buffer还有数据,但缓冲区已经满的情况下,socketChannel.write(buffer)会返回已经写出去的字节数,此时为0。那么这个时候我们就需要注册OP_WRITE事件,这样当缓冲区又有空闲空间的时候就会触发OP_WRITE事件,这是我们就可以继续将没写完的数据继续写出了。
而且在写完后,一定要记得将OP_WRITE事件注销:
selectionKey.interestOps(sk.interestOps() & ~SelectionKey.OP_WRITE);
注意,这里在修改了interest之后调用了wakeup();方法是为了唤醒被堵塞的selector方法,这样当while中判断selector返回的是0时,会再次调用selector.select()。而selectionKey的interest是在每次selector.select()操作的时候注册到系统进行监听的,所以在selector.select()调用之后修改的interest需要在下一次selector.select()调用才会生效。

nio的select()的时候,只要数据通道允许写,每次select()返回的OP_WRITE都是true。所以在nio的写数据里面,我们在每次需要写数据之前把数据放到缓冲区,并且注册OP_WRITE,对selector进行wakeup(),这样这一轮select()发现有OP_WRITE之后,将缓冲区数据写入channel,清空缓冲区,并且反注册OP_WRITE,写数据完成。这里面需要注意的是,每个SocketChannel只对应一个SelectionKey,也就是说,在上述的注册和反注册OP_WRITE的时候,不是通过channel.register()和key.cancel()做到的,而是通过key.interestOps()做到的。

public void write(MessageSession session, ByteBuffer buffer) throws ClosedChannelException {SelectionKey key = session.key();if((key.interestOps() & SelectionKey.OP_WRITE) == 0) {key.interestOps(key.interestOps() | SelectionKey.OP_WRITE);}try {writebuf.put(buffer);} catch(Exception e) {System.out.println("want put:"+buffer.remaining()+", left:"+writebuf.remaining());e.printStackTrace();}selector.wakeup();
}

while(true) {selector.select();.....if(key.isWritable()) {MessageSession session = (MessageSession)key.attachment();//System.out.println("Select a write");synchronized(session) {writebuf.flip();SocketChannel channel = (SocketChannel)key.channel();int count = channel.write(writebuf);//System.out.println("write "+count+" bytes");
          writebuf.clear();key.interestOps(SelectionKey.OP_READ);}}......}

要点一:不推荐直接写channel,而是通过缓存和attachment传入要写的数据,改变interestOps()来写数据;

要点二:每个channel只对应一个SelectionKey,所以,只能改变interestOps(),不能register()和cancel()。

参考:https://www.cnblogs.com/burgeen/p/3618059.html

转载于:https://www.cnblogs.com/grasp/p/10607271.html

Java Socket NIO相关推荐

  1. java socket nio 阻塞_Java NIO实现非阻塞式socket通信

    博主知识水平有限,只能提供一个个人的狭隘的理解,如果有新人读到这儿,建议看一下其他教程或者API,如果不明白,再来看一下:如果有dalao读到这儿,希望能指出理解中的问题~谢谢 Java提供了用于网络 ...

  2. Java Socket实战之六 使用NIO包实现Socket通信

    2019独角兽企业重金招聘Python工程师标准>>> 本文地址:http://blog.csdn.net/kongxx/article/details/7288896 Java S ...

  3. 理解Java的NIO

    同步与阻塞 同步和异步是针对应用程序和内核的交互而言的. 同步:执行一个操作之后,进程触发IO操作并等待(阻塞)或者轮询的去查看IO的操作(非阻塞)是否完成,等待结果,然后才继续执行后续的操作. 异步 ...

  4. 基于Java的RDMA高性能通信库(六):SDP - Java Socket Direct Protocol

    目录 1.Java网络编程和套接字API的历史 2.InfiniBand 高速网络通信技术 3. Java 网络协议栈API 4.Java 7 SDP 远程直接内存存取(RDMA) 4.1 Java7 ...

  5. JAVA Socket 底层是怎样基于TCP/IP 实现的???

    首先必须明确:TCP/IP模型中有四层结构:       应用层(Application Layer).传输层(Transport  Layer).网络层(Internet Layer  ).链路层( ...

  6. 学习socket nio 之 mina实例

    1:mina之浅谈 mina是apache基于socket nio的一套框架,可以从apache官网下载jar包和源码.试用起来非常方便,也非常强大.如果对socket nio还不是很了解的话:请看一 ...

  7. 【Java】NIO中Selector的select方法源码分析

    该篇博客的有些内容和在之前介绍过了,在这里再次涉及到的就不详细说了,如果有不理解请看[Java]NIO中Channel的注册源码分析, [Java]NIO中Selector的创建源码分析 Select ...

  8. java socket android_Android:这是一份很详细的Socket使用攻略

    前言 Socket的使用在 Android网络编程中非常重要 今天我将带大家全面了解 Socket 及 其使用方法 目录 示意图 1.网络基础 阅读本文前,请先了解 关于计算机网络基础,如计算机体系结 ...

  9. java之NIO(Channel,Buffer,Selector)

    java之NIO 1 什么是NIO Java NIO (New IO,Non-Blocking IO)是从Java 1.4版本开始引入的一套新的IO API.NIO支持面向缓冲区的.基于通道的IO操作 ...

最新文章

  1. minor gc和Major GC,Full GC的触发条件
  2. 人生之精华,胜读十年书
  3. Flutter 学习(V2EX)
  4. Android性能优化 - 消除卡顿
  5. RocketMQ(四):生产者消费者实践
  6. OpenStack 之Nova添加扩展API流程,附带资源的查找功能
  7. android平台上持久化存储3种手段_深入学习Redis :持久化
  8. (原创)3.2 AddOwner和OverrideMetadata的区别
  9. python装饰器实现用户密码认证(简单初形)
  10. 《高效能人士的7个习惯》读后摘记
  11. StringTokenizer字符串分解器
  12. linux输入子系统概述,4. Linux - 输入子系统框架详解
  13. java随机生成名字_java随机生成一个名字和对应拼音的方法
  14. 小牛电动为何沉迷于打造“社区文化”?
  15. 小写数字转大写金额php,php 金额小写数字转大写汉字
  16. 您有一份《七夕安全须知》待查收
  17. 快手程序员爆料:有大厂背书就是好,这就是光环效应
  18. 机器学习(MATLAB实现)——SVM支持向量机(一)
  19. windows上ppt转图片的方法
  20. 【BSP视频教程】STM32H7视频教程第2期:STM32H7四通八达的总线矩阵,从系统框架整体把控H7

热门文章

  1. mysql配置环境变量(win 10)_mysql配置环境变量(win 10)
  2. springboot 上传文件_基于SpringBoot的文件上传
  3. excel记账本模板_原来这才是老板最喜欢看的财务报表!这些模板送你,录入自动生成...
  4. 数据解析学习笔记(正则解析、bs4解析、xpath解析)
  5. LeetCode 2191. 将杂乱无章的数字排序(自定义排序)
  6. fastapi 响应模型 / 响应状态码 / 表单参数
  7. python 代码文件路径注意事项
  8. LeetCode MySQL 180. 连续出现的数字(cast)
  9. LeetCode 1049. 最后一块石头的重量 II(DP)
  10. win10版本查看_想知道电脑中安装的win10版本号,用这3招就对了,一键查看