java nio 编程

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

Java代码  
晚上学习了下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
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值