如此代码,服务器端读取客户端信息时,如果客户端正常关闭,出现循环读问题,CPU被占满。
package com.timer.netty.select;
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;
/**
* Created by zhubo on 2018/1/16.
*/
public class NIOServer {
private Selector selector;
public void initServer(int port) throws Exception{
//获得一个ServerSocket通道
ServerSocketChannel serverChannel = ServerSocketChannel.open();
//设置通道为非阻塞
serverChannel.configureBlocking(false);
// 该通道对于serverSocket绑定到port端口
serverChannel.socket().bind(new InetSocketAddress(port));
// 获得一选择器
this.selector = Selector.open();
//将通道管理器和该通道绑定,并未该通道注册selectionKey.OP_ACCEPT事件
//注册该事件后,当事件到达的时候,selector.select()会返回,
// 如果事件没有到达selector.select()会一直阻塞
serverChannel.register(selector, SelectionKey.OP_ACCEPT);
}
public void listen() throws Exception{
System.out.println("start server ...");
// 轮训访问seletor
while(true){
//当注册事件到达时,方法返回,否则该方法会一直阻塞
selector.select();
// 获得selector中选中项的地带器,选中项为注册事件
Iterator<SelectionKey> iterator = this.selector.selectedKeys().iterator();
if(iterator.hasNext()){
SelectionKey key = iterator.next();
//删除已选的key 以防止重复处理
iterator.remove();
//客户端请求链接事件
if(key.isValid() && key.isAcceptable()){
System.out.println("acceptable....");
ServerSocketChannel server = (ServerSocketChannel) key.channel();
//获取客户端链接的通道
SocketChannel channel = server.accept();
// 非阻塞模式
channel.configureBlocking(false);
//channel.write(ByteBuffer.wrap(new String("Hello Client").getBytes()));
// 在客户端链接成功后,为了可以连接到客户端的信息,需要给通道设置读的权限
channel.register(this.selector,SelectionKey.OP_READ);
}else if(key.isValid() && key.isReadable()){
read(key);
}
}
}
}
private void read(SelectionKey key) throws Exception {
SocketChannel channel = (SocketChannel)key.channel();
ByteBuffer buffer = ByteBuffer.allocate(100);
channel.read(buffer);
byte[] data = buffer.array();
String msg = new String(data).trim();
System.out.println("server receive from client : "+msg);
ByteBuffer outBuffer = ByteBuffer.wrap(msg.getBytes());
channel.write(outBuffer);
//key.cancel();
//channel.close();
}
public static void main(String[] args) throws Exception{
NIOServer server = new NIOServer();
server.initServer(8989);
server.listen();
}
}
[root@centos1 ~]# telnet 192.168.1.114 8989
Trying 192.168.1.114...
Connected to 192.168.1.114.
Escape character is '^]'.
^]
telnet> send ao
锟斤拷
kdf
kdfdd
ddquit
quit^]
telnet> quit
Connection closed.
[root@centos1 ~]# telnet 192.168.1.114 8989
Trying 192.168.1.114...
Connected to 192.168.1.114.
Escape character is '^]'.
^]
telnet> send ao
锟斤拷
abc
abcdd
dd^]
telnet> quit
Connection closed.
[root@centos1 ~]#
当我们telnet切断客户端链接时,会一直出发SocketChannel读就绪事件。会出现循环度问题。
需要我们手动对读取通道数据代码进行改写:
private void read(SelectionKey key) throws Exception {
SocketChannel channel = (SocketChannel)key.channel();
ByteBuffer buffer = ByteBuffer.allocate(100);
try{
int read = channel.read(buffer);
if(read > 0){
byte[] data = buffer.array();
String msg = new String(data).trim();
System.out.println("server receive from client : "+msg);
ByteBuffer outBuffer = ByteBuffer.wrap(msg.getBytes());
channel.write(outBuffer);
}else if(read < 0){
System.out.println("channel closed");
channel.close();
key.cancel();
}else
;
}catch (Exception e){
System.out.println("channel exception closed");
channel.close();
key.cancel();
}
//key.cancel();
//channel.close();
}
这样不管,客户端异常关闭导致的服务器异常退出,还是 客户端正常关闭导致的循环读问题都不会出现