86 lines
2.7 KiB
Java
86 lines
2.7 KiB
Java
|
package com.thinker.common.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.Iterator;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* nio nio different from bio nio
|
|||
|
|
*
|
|||
|
|
* */
|
|||
|
|
public class NioServer{
|
|||
|
|
|
|||
|
|
private Selector selector;
|
|||
|
|
/**
|
|||
|
|
*
|
|||
|
|
* @param port 监听端口
|
|||
|
|
* 初始化服务器
|
|||
|
|
*
|
|||
|
|
* */
|
|||
|
|
private NioServer init(int port) throws IOException{
|
|||
|
|
ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
|
|||
|
|
serverSocketChannel.configureBlocking(false);
|
|||
|
|
serverSocketChannel.bind(new InetSocketAddress(port));
|
|||
|
|
selector = Selector.open();
|
|||
|
|
//将通道管理器与通道绑定，并为该通道注册SelectionKey.OP_ACCEPT事件
|
|||
|
|
//只有当该事件到达时Select.select()会返回，否则一直阻塞。
|
|||
|
|
serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
|
|||
|
|
return this;
|
|||
|
|
}
|
|||
|
|
public void listen() throws IOException{
|
|||
|
|
System.out.println("service starting.....");
|
|||
|
|
while(true){
|
|||
|
|
//当有注册事件到达时方法返回，否则阻塞
|
|||
|
|
selector.select();
|
|||
|
|
//获取selector中的迭代器，选中项为注册的事件
|
|||
|
|
Iterator<SelectionKey> ite = selector.selectedKeys().iterator();
|
|||
|
|
while(ite.hasNext()){
|
|||
|
|
SelectionKey key = ite.next();
|
|||
|
|
//删除已选key，防止重复处理
|
|||
|
|
ite.remove();
|
|||
|
|
//客户端连接事件
|
|||
|
|
if(key.isAcceptable()){
|
|||
|
|
ServerSocketChannel server = (ServerSocketChannel)key.channel();
|
|||
|
|
//获取客户端连接通道
|
|||
|
|
SocketChannel channel = server.accept();
|
|||
|
|
channel.configureBlocking(false);
|
|||
|
|
channel.write(ByteBuffer.wrap(new String("send to client").getBytes()));
|
|||
|
|
channel.register(selector, SelectionKey.OP_READ);
|
|||
|
|
System.out.println("client connect");
|
|||
|
|
}else if(key.isReadable()){//有可读数据事件
|
|||
|
|
//获取客户端传输数据可读消息通道。
|
|||
|
|
SocketChannel channel = (SocketChannel)key.channel();
|
|||
|
|
//创建读取数据缓存器
|
|||
|
|
ByteBuffer buffer = ByteBuffer.allocate(1024);
|
|||
|
|
int read = channel.read(buffer);
|
|||
|
|
byte[] data = buffer.array();
|
|||
|
|
String message = new String(data);
|
|||
|
|
|
|||
|
|
System.out.println("receive message from client, size:"+buffer.position()+"\tmessage:"+message);
|
|||
|
|
try{
|
|||
|
|
Thread.sleep(1000);
|
|||
|
|
}catch(InterruptedException e){
|
|||
|
|
e.printStackTrace();
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
ByteBuffer out = ByteBuffer.wrap(new String("hello world").getBytes());
|
|||
|
|
channel.write(out);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
public static void main(String[] args) {
|
|||
|
|
try{
|
|||
|
|
new NioServer().init(1994).listen();
|
|||
|
|
}catch(IOException e){
|
|||
|
|
e.printStackTrace();
|
|||
|
|
System.exit(0);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|