Java NIO socket编程实例

晚上学习了下Java 的 NIO Socket编程,写了下面这个小程序,包括服务器端与客户端。实现的功能为客户端向服务器端发送随即数目的消息,服务器端一条一条的回应。消息内容保存在talks.properties文件中,内容为:


Hi=Hi
Bye=Bye
床前明月光=疑是地上霜
举头望明月=低头思故乡
少小离家老大回=乡音无改鬓毛衰
天王盖地虎=宝塔镇河妖
我是甲=我是乙
我是客户端=我是服务器
我是周星驰=我是周润发


客户端会随即发送“=”左边的消息,服务器端会回应客户端“=”右边的消息。如果客户端想断开连接,会向服务器发送一个"Bye",然后服务器会回应一个"Bye“。收到服务器端的"Bye"后,客户端会断开连接。


当然,java的properties文件不接受中文内容,你需要native2ascii一下。talks.properties 的实际文件内容为:


Hi=Hi
Bye=Bye
\u5E8A\u524D\u660E\u6708\u5149=\u7591\u662F\u5730\u4E0A\u971C
\u4E3E\u5934\u671B\u660E\u6708=\u4F4E\u5934\u601D\u6545\u4E61
\u5C11\u5C0F\u79BB\u5BB6\u8001\u5927\u56DE=\u4E61\u97F3\u65E0\u6539\u9B13\u6BDB\u8870
\u5929\u738B\u76D6\u5730\u864E=\u5B9D\u5854\u9547\u6CB3\u5996
\u6211\u662F\u7532=\u6211\u662F\u4E59
\u6211\u662F\u5BA2\u6237\u7AEF=\u6211\u662F\u670D\u52A1\u5668
\u6211\u662F\u5468\u661F\u9A70=\u6211\u662F\u5468\u6DA6\u53D1




看下服务器端的代码。此例中的服务器端只有一个主线程,用于selector操作,并处理多个客户端的消息。在常规的socket编程中,每个客户端都需要单独开一个线程,效率比较低。代码为:


package helloweenpad;


import java.io.FileInputStream;
import java.net.InetSocketAddress;
import java.net.Socket;
import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import java.nio.charset.CharsetEncoder;
import java.util.Iterator;
import java.util.Properties;


public class MyFirstNIOServer {


public static final int PORT = 12315;


protected Selector selector;
protected Charset charset = Charset.forName("UTF-8");
protected CharsetEncoder charsetEncoder = charset.newEncoder();
protected CharsetDecoder charsetDecoder = charset.newDecoder();


protected Properties talks = new Properties();


int clientCount;


public MyFirstNIOServer() throws Exception {


talks.load(new FileInputStream("E:\\talk.properties"));


selector = Selector.open();


ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
serverSocketChannel.socket().bind(new InetSocketAddress(PORT)); // port
serverSocketChannel.configureBlocking(false);
serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);// register


p("Server localhost:" + PORT + " started. waiting for clients. ");


while (true) {
   // selector 线程。select() 会阻塞,直到有客户端连接,或者有消息读入
   selector.select();
   Iterator<SelectionKey> iterator = selector.selectedKeys().iterator();
   while (iterator.hasNext()) {


    SelectionKey selectionKey = iterator.next();
    iterator.remove(); // 删除此消息


    // 并在当前线程内处理。(为了高效,一般会在另一个线程中处理此消息,例如使用线程池等)
    handleSelectionKey(selectionKey);
   }
}


}


public void handleSelectionKey(SelectionKey selectionKey) throws Exception {


if (selectionKey.isAcceptable()) {


   // 有客户端进来
   clientCount++;


   ServerSocketChannel serverSocketChannel = (ServerSocketChannel) selectionKey.channel();
   SocketChannel socketChannel = serverSocketChannel.accept();
   socketChannel.configureBlocking(false);
   Socket socket = socketChannel.socket();


   // 立即注册一个 OP_READ 的SelectionKey, 接收客户端的消息
   SelectionKey key = socketChannel.register(selector, SelectionKey.OP_READ);
   key.attach("第 " + clientCount + " 个客户端 [" + socket.getRemoteSocketAddress() + "]: ");


   p(key.attachment() + "\t[connected] =========================================");


} else if (selectionKey.isReadable()) {


   // 有消息进来


   ByteBuffer byteBuffer = ByteBuffer.allocate(100);
   SocketChannel socketChannel = (SocketChannel) selectionKey.channel();


   try {
    int len = socketChannel.read(byteBuffer);


    // 如果len>0,表示有输入。如果len==0, 表示输入结束。需要关闭 socketChannel
    if (len > 0) {


     byteBuffer.flip();
     String msg = charsetDecoder.decode(byteBuffer).toString();


     // 根据客户端的消息,查找到对应的输出
     String newMsg = talks.getProperty(msg);
     if (newMsg == null)
      newMsg = "Sorry? I don't understand your message. ";


     // UTF-8 格式输出到客户端,并输出一个'n'


     socketChannel.write(charsetEncoder.encode(CharBuffer.wrap(newMsg + "\n")));
     p(selectionKey.attachment() + "\t[recieved]: " + msg + " ----->\t[send]: " + newMsg);


    } else {
     // 输入结束,关闭 socketChannel
     p(selectionKey.attachment() + "read finished. close socketChannel. ");
     socketChannel.close();
    }


   } catch (Exception e) {


    // 如果read抛出异常,表示连接异常中断,需要关闭 socketChannel
    e.printStackTrace();


    p(selectionKey.attachment() + "socket closed? ");
    socketChannel.close();
   }


} else if (selectionKey.isWritable()) {
   p(selectionKey.attachment() + "TODO: isWritable() ???????????????????????????? ");
} else if (selectionKey.isConnectable()) {
   p(selectionKey.attachment() + "TODO: isConnectable() ????????????????????????? ");
} else {
   p(selectionKey.attachment() + "TODO: else. ");
}


}


public static void p(Object object) {
System.out.println(object);
}


public static void main(String[] args) throws Exception {
new MyFirstNIOServer();
}


}


再看下客户端代码。这个客户端使用了常规的socket编程,没有使用NIO。是否使用NIO对另一方是透明的,对方看不见,也不关心。无论使用NIO还是使用常规socket,效果都是一样的,只是NIO的效率要高一些。代码为:


package helloweenpad;


import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.Socket;
import java.util.Properties;
import java.util.Random;


public class MyFirstNIOClientTest extends Thread {


public static final String HOST = "localhost";
public static final int PORT = 12315;


boolean exist = false;


Properties talks = new Properties();
Random random = new Random();
String[] keys;


int messageCount = 0;


public void run() {


try {


   // 对话内容
   talks.load(new FileInputStream("E:\\talk.properties"));


   // 客户端发送 "=" 左边的内容
   keys = new String[talks.size()];
   talks.keySet().toArray(keys);


   Socket socket = new Socket(HOST, PORT);


   OutputStream ous = socket.getOutputStream();
   InputStream ins = socket.getInputStream();


   // 接收线程,接收服务器的回应
   RecieverThread reciever = new RecieverThread(ins);
   reciever.start();


   while (!exist) {


    messageCount++;


    // 选择一个随机消息
    String msg = chooseMessage();


    synchronized (ins) {


     // 发送给服务器端
     ous.write(msg.getBytes("UTF-8"));


     System.out.println("[send]\t" + messageCount + ": " + msg);


     // 然后等待接收线程
     ins.wait();
    }


    if (msg.equals("Bye")) {
     break;
    }
   }


   ins.close();
   ous.close();
   socket.close();


} catch (Exception e) {
   e.printStackTrace();
}


}


public String chooseMessage() {


int index = random.nextInt(keys.length);
String msg = keys[index];


// 如果 10 次就选中 Bye,则重新选择,为了使对话内容多一些
if (messageCount < 10 && msg.equalsIgnoreCase("Bye")) {
   return chooseMessage();
}


return msg;
}


// 接收线程
class RecieverThread extends Thread {
private InputStream ins;


public RecieverThread(InputStream ins) {
   this.ins = ins;
}


@Override
public void run() {


   try {
    String line = null;


    BufferedReader r = new BufferedReader(new InputStreamReader(
      ins, "UTF-8"));


    // readLine()会阻塞,直到服务器输出一个 '\n'
    while ((line = r.readLine()) != null) {


     System.out.println("[Recieved]: " + line);


     synchronized (ins) {
      // 接收到服务器的消息,通知下主线程
      ins.notify();
     }
     if (line.trim().equals("Bye")) {
      exist = true;
      break;
     }
    }
   } catch (Exception e) {
    e.printStackTrace();
   }
}


}


public static void main(String[] args) throws Exception {


// 开三个客户端线程
for (int i = 0; i < 3; i++) {
   try {
    new MyFirstNIOClientTest().start();
   } catch (Exception e) {
    e.printStackTrace();
   }
}


}
}


=============================================================================


服务器端的输出:


Server localhost:12315 started. waiting for clients.
第 1 个客户端 [/127.0.0.1:1890]: [connected] =========================================
第 2 个客户端 [/127.0.0.1:1865]: [connected] =========================================
第 3 个客户端 [/127.0.0.1:1866]: [connected] =========================================
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 床前明月光 -----> [send]: 疑是地上霜
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 我是周星驰 -----> [send]: 我是周润发
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 少小离家老大回 -----> [send]: 乡音无改鬓毛衰
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 床前明月光 -----> [send]: 疑是地上霜
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 我是客户端 -----> [send]: 我是服务器
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 举头望明月 -----> [send]: 低头思故乡
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 床前明月光 -----> [send]: 疑是地上霜
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 举头望明月 -----> [send]: 低头思故乡
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 我是甲 -----> [send]: 我是乙
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 我是周星驰 -----> [send]: 我是周润发
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 举头望明月 -----> [send]: 低头思故乡
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 床前明月光 -----> [send]: 疑是地上霜
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 床前明月光 -----> [send]: 疑是地上霜
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 床前明月光 -----> [send]: 疑是地上霜
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: Hi -----> [send]: Hi
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 举头望明月 -----> [send]: 低头思故乡
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 举头望明月 -----> [send]: 低头思故乡
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 少小离家老大回 -----> [send]: 乡音无改鬓毛衰
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 我是甲 -----> [send]: 我是乙
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 床前明月光 -----> [send]: 疑是地上霜
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 我是客户端 -----> [send]: 我是服务器
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: Hi -----> [send]: Hi
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: Hi -----> [send]: Hi
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: 我是周星驰 -----> [send]: 我是周润发
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: 我是周星驰 -----> [send]: 我是周润发
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 举头望明月 -----> [send]: 低头思故乡
第 2 个客户端 [/127.0.0.1:1865]: [recieved]: Bye -----> [send]: Bye
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: Hi -----> [send]: Hi
第 2 个客户端 [/127.0.0.1:1865]: read finished. close socketChannel.
第 1 个客户端 [/127.0.0.1:1890]: [recieved]: Bye -----> [send]: Bye
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 我是客户端 -----> [send]: 我是服务器
第 1 个客户端 [/127.0.0.1:1890]: read finished. close socketChannel.
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: 我是客户端 -----> [send]: 我是服务器
第 3 个客户端 [/127.0.0.1:1866]: [recieved]: Bye -----> [send]: Bye
第 3 个客户端 [/127.0.0.1:1866]: read finished. close socketChannel.


客户端的输出:


[send] 1: 我是周星驰
[send] 1: 少小离家老大回
[Recieved]: 乡音无改鬓毛衰
[send] 2: 床前明月光
[Recieved]: 疑是地上霜
[send] 3: 我是客户端
[Recieved]: 我是服务器
[Recieved]: 疑是地上霜
[Recieved]: 我是周润发
[Recieved]: 低头思故乡
[send] 2: 举头望明月
[Recieved]: 疑是地上霜
[send] 3: 床前明月光
[send] 1: 床前明月光
[send] 2: 举头望明月
[Recieved]: 低头思故乡
[Recieved]: 我是乙
[send] 4: 我是甲
[Recieved]: 我是周润发
[send] 5: 我是周星驰
[send] 3: 床前明月光
[send] 6: 床前明月光
[send] 4: 举头望明月
[Recieved]: 低头思故乡
[send] 5: 床前明月光
[Recieved]: 疑是地上霜
[Recieved]: 疑是地上霜
[Recieved]: 疑是地上霜
[Recieved]: Hi
[send] 4: Hi


[Recieved]: 低头思故乡
[send] 5: 举头望明月
[Recieved]: 低头思故乡




[Recieved]: 低头思故乡
[send] 6: 举头望明月




[send] 7: 少小离家老大回


[s


http://yangguangfu.iteye.com/blog/774194

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值