NIO
NIO-Snipaste_2019-11-30_19-05-41.png
代码实现
服务端代码启动服务,并处理连接与消息
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.Channel;
import java.nio.channels.ClosedChannelException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.Iterator;
import java.util.Set;
public class NioServer {
public static void main(String[] args) {
try {
new NioServer().start();
} catch (IOException e) {
e.printStackTrace();
}
}
public void start() throws IOException {
/**
* 创建一个selector
*/
Selector selector = Selector.open();
/**
* 通过serversocketchannel 创建channel通道
*/
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
/**
* 为channel 通道绑定监听端口
*/
serverSocketChannel.bind(new InetSocketAddress(8000));
/**
* 设置channel为非阻塞模式
*/
serverSocketChannel.configureBlocking(false);
/**
* 将channel 注册到selector上,监听链接事件
*/
serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
System.out.println("server started seccessful");
/**
* 循环等待新连接
*/
while (true) {
/**
* TODO: 获取可用channel数量
*/
int readyChannels = selector.select();
/**
* TODO: 为什么要这样?
*/
if (readyChannels == 0)
continue;
/**
* 获取可用channel的集合
*/
Set selectionKeys = selector.selectedKeys();
Iterator iterator = selectionKeys.iterator();
while (iterator.hasNext()) {
/**
* selectionKey实例
*/
SelectionKey selectionKey = (SelectionKey) iterator.next();
/**
* [!] 移除set中的当前的selectionKey
*/
iterator.remove();
/**
* 根据就绪状态来判断相应的逻辑
*/
/**
* 如果是接入事件
*/
if (selectionKey.isAcceptable()) {
acceptHandler(serverSocketChannel, selector);
}
/**
* 如果是可读事件
*/
if (selectionKey.isReadable()) {
readHandler(selectionKey, selector);
}
}
}
/**
*
*/
}
public void acceptHandler(ServerSocketChannel serverSocketChannel, Selector selector) throws IOException {
/**
* 如果是接入事件,创建socketchannel
*/
SocketChannel socketChannel = serverSocketChannel.accept();
/**
* 将socketChannel设置为非阻塞工作模式
*/
socketChannel.configureBlocking(false);
/**
* 将channel注册到selector上,监听可读事件
*/
socketChannel.register(selector, SelectionKey.OP_READ);
/**
* 回复客户端提示信息
*/
socketChannel.write(Charset.forName("UTF-8").encode("请注意隐私安全"));
}
public void readHandler(SelectionKey selectionKey, Selector selector) {
/**
* 要从selectionKey中获取到已经就绪的channel
*/
SocketChannel socketChannel = (SocketChannel) selectionKey.channel();
/**
* 创建buffer
*/
ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
/**
* 循环读取客户端请求信息
*/
String request = "";
try {
while (socketChannel.read(byteBuffer) > 0) {
/**
* 切换buffer为读模式
*/
byteBuffer.flip();
/**
* 读取buffer中的内容
*/
request += Charset.forName("UTF-8").decode(byteBuffer);
}
} catch (IOException ioe) {
ioe.printStackTrace();
return;
}
/**
* 将channel再次注册到selector上,监听它的可读事件
*/
try {
socketChannel.register(selector, SelectionKey.OP_READ);
} catch (ClosedChannelException cce) {
cce.printStackTrace();
return;
}
/**
* 将客户端发送的请求信息,广播给其他客户端
*/
if (request.length() > 0) {
// 广播给其他客户端
System.out.println("::" + request);
this.broadcast(selector, socketChannel, request);
}
}
private void broadcast(Selector selector, SocketChannel socketChannel, String request) {
/**
* 获取所有已介入的客户端channel
*/
Set selectionKeys = selector.keys();
selectionKeys.forEach(selectionKey -> {
Channel targetChannel = selectionKey.channel();
// 剔除发消息的客户端
if (targetChannel instanceof SocketChannel && targetChannel != socketChannel) {
try {
((SocketChannel) targetChannel).write(Charset.forName("UTF-8").encode(request));
} catch (IOException ioe) {
ioe.printStackTrace();
}
}
});
}
}
客户端响应处理线程类
import java.nio.channels.Selector;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.ClosedChannelException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.Iterator;
import java.util.Set;
public class ResponseHandler implements Runnable {
private Selector selector;
public ResponseHandler(Selector selector) {
this.selector = selector;
}
@Override
public void run() {
try {
/**
* 循环等待新连接
*/
while (true) {
/**
* TODO: 获取可用channel数量
*/
int readyChannels = selector.select();
/**
* TODO: 为什么要这样?
*/
if (readyChannels == 0)
continue;
/**
* 获取可用channel的集合
*/
Set selectionKeys = selector.selectedKeys();
Iterator iterator = selectionKeys.iterator();
while (iterator.hasNext()) {
/**
* selectionKey实例
*/
SelectionKey selectionKey = (SelectionKey) iterator.next();
/**
* [!] 移除set中的当前的selectionKey
*/
iterator.remove();
/**
* 根据就绪状态来判断相应的逻辑
*/
/**
* 如果是可读事件
*/
if (selectionKey.isReadable()) {
readHandler(selectionKey, selector);
}
}
}
} catch (Exception e) {
// TODO: handle exception
}
}
public void readHandler(SelectionKey selectionKey, Selector selector) {
/**
* 要从selectionKey中获取到已经就绪的channel
*/
SocketChannel socketChannel = (SocketChannel) selectionKey.channel();
/**
* 创建buffer
*/
ByteBuffer byteBuffer = ByteBuffer.allocate(1024);
/**
* 循环读取客户端请求信息
*/
String response = "";
try {
while (socketChannel.read(byteBuffer) > 0) {
/**
* 切换buffer为读模式
*/
byteBuffer.flip();
/**
* 读取buffer中的内容
*/
response += Charset.forName("UTF-8").decode(byteBuffer);
}
} catch (IOException ioe) {
ioe.printStackTrace();
return;
}
/**
* 将channel再次注册到selector上,监听它的可读事件
*/
try {
socketChannel.register(selector, SelectionKey.OP_READ);
} catch (ClosedChannelException cce) {
cce.printStackTrace();
return;
}
/**
* 将客户端发送的请求信息,广播给其他客户端
*/
if (response.length() > 0) {
// 广播给其他客户端
System.out.println(response);
}
}
}
客户端,接收并展示消息
import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.util.Scanner;
public class NioClient {
public void start() throws IOException {
/**
* 连接服务器端
*/
SocketChannel socketChannel = SocketChannel.open(new InetSocketAddress("127.0.0.1", 8000));
/**
* 向服务器端发送数据
*/
Selector selector = Selector.open();
socketChannel.configureBlocking(false);
socketChannel.register(selector, SelectionKey.OP_READ);
new Thread(new ResponseHandler(selector)).start();
Scanner scanner = new Scanner(System.in);
while (scanner.hasNextLine()) {
String request = scanner.nextLine();
if (request != null && request.length() > 0) {
socketChannel.write(Charset.forName("UTF-8").encode(request));
}
}
scanner.close();
/**
* 接受服务器响应
*/
}
public static void main(String[] args) {
try {
new NioClient().start();
} catch (IOException e) {
e.printStackTrace();
}
}
}
运行演示
消息群发demo演示
启动一个服务端,可以启动多个客户端连接到同一服务,实现消息群发。