import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
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.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;
public class EchoServer {
private static int SOCKET_NUM = 55555;
private DateFormat dateFormatter = new SimpleDateFormat("yyyyMMdd HH:mm:ss");
/**
* @param args
*/
public static void main(String[] args) {
new EchoServer().start();
}
public void start() {
try {
Selector selector = bindServer(); // 綁定服務(wù)端口，并定義一個事件選擇器對象記錄套接字通道的事件
/* 通過此循環(huán)來遍例事件 */
while (true) {
log("Waiting events.");
int n = selector.select(); // 查詢事件如果一個事件都沒有，這里就會阻塞
log("Got events: " + n);
ByteBuffer echoBuffer = ByteBuffer.allocate(50); // 定義一個byte緩沖區(qū)來存儲收發(fā)的數(shù)據(jù)
/* 循環(huán)遍例所有產(chǎn)生的事件 */
for (SelectionKey key : selector.selectedKeys()) {
SocketChannel sc;
selector.selectedKeys().remove(key); // 將本此事件從迭帶器中刪除
/* 如果產(chǎn)生的事件為接受客戶端連接(當(dāng)有客戶端連接服務(wù)器的時(shí)候產(chǎn)生) */
if ((key.readyOps() & SelectionKey.OP_ACCEPT) == SelectionKey.OP_ACCEPT) {
ServerSocketChannel subssc = (ServerSocketChannel) key.channel(); // 定義一個服務(wù)器socket通道
sc = subssc.accept(); // 將臨時(shí)socket對象實(shí)例化為接收到的客戶端的socket
sc.configureBlocking(false); // 將客戶端的socket設(shè)置為異步
sc.register(selector, SelectionKey.OP_READ); // 將客戶端的socket的讀取事件注冊到事件選擇器中
System.out.println("Got new client:" + sc);
}
/* 如果產(chǎn)生的事件為讀取數(shù)據(jù)(當(dāng)已連接的客戶端向服務(wù)器發(fā)送數(shù)據(jù)的時(shí)候產(chǎn)生) */
else if ((key.readyOps() & SelectionKey.OP_READ) == SelectionKey.OP_READ) {
sc = (SocketChannel) key.channel(); // 臨時(shí)socket對象實(shí)例化為產(chǎn)生本事件的socket
ByteArrayOutputStream bos = new ByteArrayOutputStream(); // 定義一個用于存儲byte數(shù)據(jù)的流對象，存儲全部信息
echoBuffer.clear(); // 先將客戶端的數(shù)據(jù)清空
try {
// 循環(huán)讀取所有客戶端數(shù)據(jù)到byte緩沖區(qū)中,當(dāng)有數(shù)據(jù)的時(shí)候read函數(shù)返回?cái)?shù)據(jù)長度
// NIO會自動的將緩沖區(qū)一次容納不下的自動分段
int readInt = 0; // 為讀取到數(shù)據(jù)的長度
while ((readInt = sc.read(echoBuffer)) > 0) {
// 如果獲得數(shù)據(jù)長度比緩沖區(qū)大小小的話
if (readInt < echoBuffer.capacity()) {
byte[] readByte = new byte[readInt]; // 建立一個臨時(shí)byte數(shù)組,將齊長度設(shè)為獲取的數(shù)據(jù)的長度
// 循環(huán)向此臨時(shí)數(shù)組中添加數(shù)據(jù)
for (int i = 0; i < readInt; i++) {
readByte[i] = echoBuffer.get(i);
}
bos.write(readByte); // 將此數(shù)據(jù)存入byte流中
}
// 否則就是獲得數(shù)據(jù)長度等于緩沖區(qū)大小
else {
bos.write(echoBuffer.array()); // 將讀取到的數(shù)據(jù)寫入到byte流對象中
}
}
// 當(dāng)循環(huán)結(jié)束時(shí)byte流中已經(jīng)存儲了客戶端發(fā)送的所有byte數(shù)據(jù)
log("Recive msg: " + new String(bos.toByteArray()));
} catch (Exception e) {
e.printStackTrace(); // 當(dāng)客戶端在讀取數(shù)據(jù)操作執(zhí)行之前斷開連接會產(chǎn)生異常信息
key.cancel(); // 將本socket的事件在選擇器中刪除
break;
}
writeBack(sc, bos.toByteArray()); // 向客戶端寫入收到的數(shù)據(jù)
}
}
}
} catch (Exception e) {
e.printStackTrace();
}
}
/**
* 綁定服務(wù)端口，初始化整個服務(wù)
* @throws IOException
*/
private Selector bindServer() throws IOException {
log("Start binding server socket：" + SOCKET_NUM);
Selector selector = Selector.open(); // 定義一個事件選擇器對象記錄套接字通道的事件
ServerSocketChannel ssc = ServerSocketChannel.open(); // 定義一個異步服務(wù)器socket對象
ssc.configureBlocking(false);// 將此socket對象設(shè)置為異步
ServerSocket ss = ssc.socket(); // 定義服務(wù)器socket對象-用來指定異步socket的監(jiān)聽端口等信息
InetSocketAddress address = new InetSocketAddress(SOCKET_NUM); // 定義存放監(jiān)聽端口的對象
ss.bind(address); // 將服務(wù)器與這個端口綁定
ssc.register(selector, SelectionKey.OP_ACCEPT); // 將異步的服務(wù)器socket對象的接受客戶端連接事件注冊到selector對象內(nèi)
log("Binded socket at：" + SOCKET_NUM);
return selector;
}
private boolean writeBack(SocketChannel sc, byte[] b) {
ByteBuffer echoBuffer = ByteBuffer.allocate(b.length); // 建立這個byte對象的ByteBuffer
echoBuffer.put(b); // 將數(shù)據(jù)存入
echoBuffer.flip(); // 將緩沖區(qū)復(fù)位以便于進(jìn)行其他讀寫操作
try {
// 向客戶端寫入數(shù)據(jù),數(shù)據(jù)為接受到數(shù)據(jù)
sc.write(echoBuffer);
} catch (IOException e) {
e.printStackTrace();
return false;
}
System.out.println("Msg echo back: " + new String(echoBuffer.array()));
return true;
}
private void log(Object msg) {
System.out.println("SERVER [" + dateFormatter.format(new Date()) + "]: " + msg);
}
}
@import url(http://www.shnenglu.com/cutesoft_client/cuteeditor/Load.ashx?type=style&file=SyntaxHighlighter.css);@import url(/css/cuteeditor.css);
From: http://topic.csdn.net/u/20120719/16/a5d0c2ca-d669-4902-9f3e-d4b8f4ad90b3.html?32104