java nio网络编程_Java NIO网络编程开发入门教程

构建网络应用服务端：EchoServer.java

package hxb.echo;

import java.io.IOException;

import java.io.UnsupportedEncodingException;

import java.net.InetSocketAddress;

import java.net.SocketAddress;

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;

import java.util.Set;

public class EchoServer {

private Selector selector =null;

public EchoServer(){

}

/**

* 写消息到客户端

*/

private void doWrite(SocketChannel channel,String response) throws IOException{

byte[] bytes = response.getBytes();

ByteBuffer writeBuffer = ByteBuffer.allocate(bytes.length);

writeBuffer.put(bytes);

writeBuffer.flip();

channel.write(writeBuffer);

}

/**

* 处理客户端信息

*/

private void handleForReadKey(SelectionKey key) throws IOException, UnsupportedEncodingException {

SocketChannel sc = (SocketChannel) key.channel();

ByteBuffer buffer = ByteBuffer.allocate(1024);

int readBytes = sc.read(buffer);

if(readBytes>0){

buffer.flip();

byte[] bytes = new byte[buffer.remaining()];

buffer.get(bytes);

String msg = new String(bytes);

System.out.println("Message send by Client：" + msg);

//响应消息

String rspMsg = "Send by Server:"+msg;

doWrite(sc,rspMsg);

}

else if(readBytes<0){

key.cancel();

sc.close();

}

}

/**

* 启动服务器

* @param port 服务器端口

*/

public void start(int port)

{

ServerSocketChannel serverChannel = null;

try {

selector = Selector.open();

serverChannel = ServerSocketChannel.open();

serverChannel.configureBlocking(false);//配置为非阻塞模式

SocketAddress addr = new InetSocketAddress(port);

serverChannel.bind(addr );

System.out.println("Start Echo server and listen on the port "+port);

//监听

serverChannel.register(selector, SelectionKey.OP_ACCEPT);

while(true)

{

//必须：唤醒selector

selector.select();

Set keys = selector.selectedKeys();

Iterator iterator = keys.iterator();

SelectionKey key = null;

while(iterator.hasNext())

{

key=iterator.next();

iterator.remove();

handle(key);

}

}

} catch (IOException e) {

e.printStackTrace();

}

finally

{

if(serverChannel!=null)

{

try {

serverChannel.close();

} catch (IOException e) {

e.printStackTrace();

}

}

}

}

/**

* 处理消息事件

* @param key

*/

private void handle(SelectionKey key) {

if(key.isValid())

{

if(key.isAcceptable())

{

ServerSocketChannel ssc = (ServerSocketChannel) key.channel();

try {

SocketChannel ss = ssc.accept();

ss.configureBlocking(false);

//Selection 从Accept 切换到read数据

ss.register(selector, SelectionKey.OP_READ);

} catch (IOException e) {

e.printStackTrace();

}

}

else if(key.isReadable())

{

try {

handleForReadKey(key);

} catch (IOException e) {

e.printStackTrace();

}

}

}

}

public static void main(String[] args) {

EchoServer server = new EchoServer();

server.start(9527);

}

}