package com.java.NIO; 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.Date; import java.util.Iterator; import java.util.Scanner; import org.junit.Test; /** * * @author fliay * * 一、使用NIO 完成网络通信的三个核心 * 1.通道(channel):负责链接 * java.nio.channels.Channel 接口 * |--SelecttableChannel * |--SocketChannel * |--serverSocketChannel * |--DatagramChannel * * |--Pipe.SinkChannel * |--Pipe.SourceChannel * 2.缓冲区(Buffer):负责数据的存取 * * 3.寻则其(Selector): 是SelectableChannel的多路复用器,用于监控SelectableChannel的IO状况 * * */ public class TestNoBlockingNIO { @Test//客户端 public void client() throws IOException{ //1.获取通道 SocketChannel sChannel = SocketChannel.open(new InetSocketAddress("127.0.0.1", 1008)); //2.切换非阻塞模式 sChannel.configureBlocking(false); //3.分配指定大小的缓冲区 ByteBuffer buf = ByteBuffer.allocate(1024); //4.发送数据给服务器 Scanner input = new Scanner(System.in); while(input.hasNext()){ String str = input.nextLine(); buf.put((new Date().toString()+":\n"+str).getBytes()); buf.flip(); sChannel.write(buf); buf.clear(); } //5.关闭通道 sChannel.close(); } @Test //服务器 public void Server() throws IOException{ //1.获取通道 ServerSocketChannel ssChannel = ServerSocketChannel.open(); //2.切换非阻塞模式 ssChannel.configureBlocking(false); //3.绑定连接 ssChannel.bind(new InetSocketAddress(1008)); //4.获取选择器 Selector selector = Selector.open(); //5.将通道注册到选择器上,并且指定“监听接收事件” ssChannel.register(selector, SelectionKey.OP_ACCEPT); //6.轮训式的获获取选择器上的以及“准备就绪”的事件 while(selector.select()>0){ //7.获取当前选择器中所有注册的“选择键(已就绪的监听事件)” Iterator<SelectionKey> it = selector.selectedKeys().iterator(); while(it.hasNext()){ //8.获取就绪的事件 SelectionKey sk = it.next(); //9.判断具体是什么事件准备就绪 if(sk.isAcceptable()){ //10. 若“接受就绪” ,获取客户链接 SocketChannel sChannel = ssChannel.accept(); //11. 切换非阻塞模式 sChannel.configureBlocking(false); //12. 将该通道注册到选择器上 sChannel.register(selector, SelectionKey.OP_READ); }else if(sk.isReadable()){ //13. 获取当前选择器上“读就绪”状态的通道 SocketChannel sChannel = (SocketChannel) sk.channel(); //14. 读取数据 ByteBuffer buf= ByteBuffer.allocate(1024); int len =0; while((len =sChannel.read(buf))>0){ buf.flip(); System.out.println(new String(buf.array(),0,len)); buf.clear(); } } //15. 取消选择键 SelectionKey it.remove(); } } } }
时间: 2024-11-09 00:15:57