这是一个相当精辟的一个程序,值得继续学习这个程序
package nonblock;
import java.io.*;
import java.nio.*;
import java.nio.channels.*;
import java.nio.charset.*;
import java.net.*;
import java.util.*;
public class EchoServer {
private Selector selector = null;
private ServerSocketChannel serverSocketChannel = null;
private int port = 8000;
private Charset charset = Charset.forName("GBK");
public EchoServer()throws IOException{
selector = Selector.open();
serverSocketChannel = ServerSocketChannel.open();
//使得在同一个主机上关闭了服务器程序,紧接着再启动该服务器程序时,
//可以顺利绑定到同步端口
serverSocketChannel.socket().setReuseAddress(true);
serverSocketChannel.configureBlocking(false);
serverSocketChannel.socket().bind(new InetSocketAddress(port));
System.out.println("服务器启动");
}
public void service()throws IOException{
serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
while(selector.select()>0){
Set readyKeys = selector.selectedKeys();
Iterator it = readyKeys.iterator();
while(it.hasNext()){
SelectionKey key = null;
try{
key = (SelectionKey)it.next();
it.remove();
if(key.isAcceptable()){
ServerSocketChannel ssc = (ServerSocketChannel)key.channel();
SocketChannel socketChannel = (SocketChannel)ssc.accept();
System.out.println("接收到客户连接,来自:"+
socketChannel.socket().getInetAddress()+
";"+socketChannel.socket().getPort());
socketChannel.configureBlocking(false);
ByteBuffer buffer = ByteBuffer.allocate(1024);
socketChannel.register(selector, SelectionKey.OP_READ|
SelectionKey.OP_WRITE);
}
if(key.isReadable()){
receive(key);
}
if(key.isWritable()){
send(key);
}
}catch(IOException e){
e.printStackTrace();
try{
if(key != null){
key.cancel();
key.channel().close();
}
}catch(Exception ex){
ex.printStackTrace();
}
}
}//#while
}//#while
}
public void send(SelectionKey key)throws IOException{
ByteBuffer buffer = (ByteBuffer)key.attachment();
SocketChannel socketChannel = (SocketChannel)key.channel();
buffer.flip();
String data = decode(buffer);
if(data.indexOf("/r/n") == -1) return;
String outputData = data.substring(0, data.indexOf("/n")+1);
System.out.println(outputData);
ByteBuffer outputBuffer = encode("echo:"+outputData);
while(outputBuffer.hasRemaining())
socketChannel.write(outputBuffer);
ByteBuffer temp = encode(outputData);
buffer.position(temp.limit());
buffer.compact();
if(outputData.equals("bye/r/n")){
key.cancel();
socketChannel.close();
System.out.println("关闭与客户的连接");
}
}
public void receive(SelectionKey key)throws IOException{
ByteBuffer buffer = (ByteBuffer)key.attachment();
SocketChannel socketChannel = (SocketChannel)key.channel();
ByteBuffer readBuffer = ByteBuffer.allocate(32);
socketChannel.read(readBuffer);
readBuffer.flip();
buffer.limit(buffer.capacity());
buffer.put(readBuffer);
}
private ByteBuffer encode(String outputData) {
return charset.encode(outputData);
}
private String decode(ByteBuffer buffer) {
CharBuffer charBuffer = charset.decode(buffer);
return charBuffer.toString();
}
public static void main(String[] args)throws Exception{
EchoServer server = new EchoServer();
server.service();
}
}