这块还是挺复杂的,挺难理解,但是多练几遍,多看看研究研究其实也就那样,就是一个Selector轮询的过程,这里想要双向通信,客户端和服务端都需要一个Selector,并一直轮询,
直接贴代码:
Server:服务端:
package cn.hou.socket01._03nio01; 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; //nio 服务端
public class Server implements Runnable {
//1 多路复用器
private Selector selector;
//2 建立缓冲区
private ByteBuffer readBuf=ByteBuffer.allocate(1024);
private ByteBuffer writeBuf=ByteBuffer.allocate(1024);
//构造函数
public Server(int port){
try {
//1 打开多路复用器
this.selector=Selector.open();
//2 打开服务器通道
ServerSocketChannel ssc = ServerSocketChannel.open();
//3 设置服务器通道为非阻塞方式
ssc.configureBlocking(false);
//4 绑定ip
ssc.bind(new InetSocketAddress(port));
//5 把服务器通道注册到多路复用器上,只有非阻塞信道才可以注册选择器.并在注册过程中指出该信道可以进行Accept操作
ssc.register(this.selector, SelectionKey.OP_ACCEPT);
System.out.println("服务器已经启动.....");
} catch (IOException e) {
e.printStackTrace();
}
}
@Override
public void run() {
while(true){//一直循环
try {
this.selector.select();//多路复用器开始监听
//获取已经注册在多了复用器上的key通道集
Iterator<SelectionKey> keys = this.selector.selectedKeys().iterator();
//遍历
while (keys.hasNext()) {
SelectionKey key = keys.next();//获取key
//如果是有效的
if(key.isValid()){
// 如果为阻塞状态,一般是服务端通道
if(key.isAcceptable()){
this.accept(key);
}
// 如果为可读状态,一般是客户端通道
if(key.isReadable()){
this.read(key);
}
}
//从容器中移除处理过的key
keys.remove();
}
} catch (IOException e) {
e.printStackTrace();
}
} }
//从客户端通道获取数据并进行处理
private void read(SelectionKey key) {
try {
//1 清空缓冲区旧的数据
this.readBuf.clear();
//2 获取之前注册的socket通道对象
SocketChannel sc = (SocketChannel) key.channel();
//3 读取数据
int count = sc.read(this.readBuf);
//4 如果没有数据
if(count == -1){
key.channel().close();
key.cancel();
return;
}
//5 有数据则进行读取 读取之前需要进行复位方法(把position 和limit进行复位)
this.readBuf.flip();
//6 根据缓冲区的数据长度创建相应大小的byte数组,接收缓冲区的数据
byte[] bytes = new byte[this.readBuf.remaining()];
//7 接收缓冲区数据
this.readBuf.get(bytes);
//8 打印结果
String body = new String(bytes).trim();
System.out.println("服务端接受到客户端请求的数据: " + body);
//9 告诉客户端已收到数据
writeBuf.put("你好,客户端,我已收到数据".getBytes());
//对缓冲区进行复位
writeBuf.flip();
//写出数据到服务端
sc.write(writeBuf);
//清空缓冲区数据
writeBuf.clear();
} catch (IOException e) {
e.printStackTrace();
}
}
//接受一个客户端socket进行处理
private void accept(SelectionKey key) {
try {
//1 获取服务通道
ServerSocketChannel ssc = (ServerSocketChannel) key.channel();
//2 执行阻塞方法,当有客户端请求时,返回客户端通信通道
SocketChannel sc = ssc.accept();
//3 设置阻塞模式
sc.configureBlocking(false);
//4 注册到多路复用器上,并设置可读标识
sc.register(this.selector, SelectionKey.OP_READ);
} catch (IOException e) {
e.printStackTrace();
} } public static void main(String[] args) {
//启动服务器
new Thread(new Server(9527)).start();
} }
Client客户端:
package cn.hou.socket01._03nio01; 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.SocketChannel;
import java.util.Iterator; //nio 客户端
public class Client{
//客户端信道选择器,轮询读取服务端返回数据
private Selector selector;
//连接信道
private SocketChannel sc;
public Client(){
try {
this.sc=SocketChannel.open();//打开信道
sc.connect(new InetSocketAddress("127.0.0.1",9527));////连接服务端
sc.configureBlocking(false);//设置非阻塞
selector = Selector.open();//必须打开
//将当前客户端注册到多路复用器上,并设置为可读状态
sc.register(this.selector, SelectionKey.OP_READ);
//开启线程,一直轮询
new Thread(()->{
while(true){//一直循环
try {
this.selector.select();//多路复用器开始监听
//获取已经注册在多了复用器上的key通道集
Iterator<SelectionKey> keys = this.selector.selectedKeys().iterator();
//遍历
while (keys.hasNext()) {
SelectionKey key = keys.next();//获取key
//如果是有效的
if(key.isValid()){
// 如果为可读状态,读取服务端返回的数据
if(key.isReadable()){
this.read(key);
}
}
//从容器中移除处理过的key
keys.remove();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}).start();
} catch (IOException e) {
e.printStackTrace();
}
} //客户端获取服务端返回的数据
private void read(SelectionKey key) {
try {
//建立写缓冲区
ByteBuffer readBuf = ByteBuffer.allocate(1024);
//2 获取之前注册的socket通道对象
SocketChannel sc = (SocketChannel) key.channel();
//3 读取数据
int count = sc.read(readBuf);
//4 如果没有数据
if(count == -1){
key.channel().close();
key.cancel();
return;
}
//5 有数据则进行读取 读取之前需要进行复位方法(把position 和limit进行复位)
readBuf.flip();
//6 根据缓冲区的数据长度创建相应大小的byte数组,接收缓冲区的数据
byte[] bytes = new byte[readBuf.remaining()];
//7 接收缓冲区数据
readBuf.get(bytes);
//8 打印结果
String body = new String(bytes).trim();
System.out.println("客户端已接受到服务端返回的数据: " + body);
} catch (IOException e) {
e.printStackTrace();
}
} public static void main(String[] args) {
//建立写缓冲区
ByteBuffer writebuf = ByteBuffer.allocate(1024);
Client client = new Client();
try {
while(true){
//定义一个字节数组,然后使用系统录入功能:
byte[] bytes = new byte[1024];
System.in.read(bytes);
//把数据放到缓冲区中
writebuf.put(bytes);
//对缓冲区进行复位
writebuf.flip();
//写出数据到服务端
client.sc.write(writebuf);
//清空缓冲区数据
writebuf.clear();
}
} catch (IOException e) {
e.printStackTrace();
} finally {
if(client.sc != null){
try {
client.sc.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
}
,先启动服务端,然后再启动客户端:
效果如下:
Server:
Client: