java nio例子_java NIO经典实例

packagenet.chatroom.server;importjava.io.IOException;importjava.nio.ByteBuffer;importjava.nio.channels.Channel;importjava.nio.channels.SelectionKey;importjava.nio.channels.Selector;importjava.nio.channels.ServerSocketChannel;importjava.nio.channels.SocketChannel;importjava.rmi.server.Skeleton;importjava.util.List;importjava.util.Scanner;public classDealer {public staticSocketChannel accept(Selector selector,

ServerSocketChannel serverChannel) {

SocketChannel channel= null;try{

channel=serverChannel.accept();

channel.configureBlocking(false);

channel.register(selector, SelectionKey.OP_READ);

channel.write(Util.charset.encode("Please input your name."));

}catch(Exception e) {

System.out.println("Error while configure socket channel :" +e);if (channel != null) {try{

channel.close();

}catch(IOException e1) {

e1.printStackTrace();

}

}

}returnchannel;

}public static voidread(Selector selector, SelectionKey selectionkey,

List clientChannels) throwsIOException {

SocketChannel socketClientChannel=(SocketChannel) selectionkey

.channel();

ByteBuffer buffer= ByteBuffer.allocateDirect(6 * 1024);

StringBuilder content= newStringBuilder();int num = 0;try{//将客户端发上来的消息读到buffer//循环将通道中数据读入buffer

while (socketClientChannel.read(buffer) > 0) {

buffer.flip();//切换成读

content.append(Util.charset.decode(buffer));

}

System.out.println("num:" +num);

System.out.println("Server is listening from client :"

+ socketClientChannel.getRemoteAddress() + " data rev is: "

+content);

}catch(IOException e) {/*** 如果出现异常，则需要关闭连接。故把num设置为-1，用下面的关闭逻辑来关闭channel*/num= -1;

}if (num >= 0) {if (content.length() > 0) {

String[] arrayContent=content.toString().split(

Util.USER_CONTENT_SPILIT);//注册用户

if (arrayContent != null && arrayContent.length == 1) {

String name= arrayContent[0];if(Util.users.contains(name)) {

socketClientChannel.write(Util.charset

.encode(Util.USER_EXIST));

}else{

Util.users.add(name);int onlineNum =clientChannels.size();

String message= "welcome " +name+ " to chat room! Online numbers:" +onlineNum;

BroadCast2(clientChannels,null, message);

}

}//注册完了，发送消息

else if (arrayContent != null && arrayContent.length > 1) {

String name= arrayContent[0];

String message=content.substring(name.length()+Util.USER_CONTENT_SPILIT.length());

message= name + " say: " +message;if(Util.users.contains(name)) {//不回发给发送此内容的客户端

BroadCast2(clientChannels, socketClientChannel, message);

}

}///**//* 把读到的数据原封不动的下发给客户端//*///int length = clientChannels.size();//for (int index = 0; index < length; index++) {// //循环所有的客户端连接，下发数据//buffer.flip();//try {// //将buffer里的数据再下发给客户端的通道//clientChannels.get(index).write(buffer);//} catch (IOException e) {//e.printStackTrace();//}//}

}

}else{/*** 如果未读到数据，对方关闭了SocketChannel 所以服务器这边也要关闭*/

try{

socketClientChannel.close();int length =clientChannels.size();for (int index = 0; index < length; index++) {if(clientChannels.get(index).equals(socketClientChannel)) {//移除当前接受的通道

clientChannels.remove(index);break;

}

}

}catch(IOException e) {

e.printStackTrace();

}

}

}//TODO 要是能检测下线，就不用这么统计了

public static intOnlineNum(Selector selector) {int res = 0;for(SelectionKey key : selector.keys()) {

Channel targetchannel=key.channel();if (targetchannel instanceofSocketChannel) {

res++;

}

}returnres;

}public voidBroadCast(Selector selector, SocketChannel except,

String content)throwsIOException {//广播数据到所有的SocketChannel中

for(SelectionKey key : selector.keys()) {

Channel targetchannel=key.channel();//如果except不为空，不回发给发送此内容的客户端

if (targetchannel instanceofSocketChannel&& targetchannel !=except) {

SocketChannel dest=(SocketChannel) targetchannel;

dest.write(Util.charset.encode(content));

}

}

}public static void BroadCast2(ListsocketChannels,

SocketChannel except, String content)throwsIOException {for(SocketChannel socketChannel : socketChannels) {if (!socketChannel.equals(except)) {//除了自己，其它通道都通知

socketChannel.write(Util.charset.encode(content));

}

}

}

}