java nio实现非阻塞Socket通信实例

服务器

1. package com.java.xiong.Net17;
2. import java.io.IOException;
3. import java.net.InetSocketAddress;
4. import java.nio.ByteBuffer;
5. import java.nio.channels.Channel;
6. import java.nio.channels.SelectionKey;
7. import java.nio.channels.Selector;
8. import java.nio.channels.ServerSocketChannel;
9. import java.nio.channels.SocketChannel;
10. import java.nio.charset.Charset;
11. public class NServer {
12. // 用于检测所有的Channel状态的selector
13. private Selector selector = null;
14. static final int PORT = 30000;
15. // 定义实现编码、解码的字符串集对象
16. private Charset charse = Charset.forName("GBK");
17. public void init() throws IOException {
18. selector = Selector.open();
19. // 通过open方法来打开一个未绑定的ServerSocketChannel是咧
20. ServerSocketChannel server = ServerSocketChannel.open();
21. InetSocketAddress isa = new InetSocketAddress("127.0.0.1", PORT);
22. // 将该ServerSocketChannel绑定到指定的IP地址
23. server.bind(isa);
24. // 设置serverSocket已非阻塞方式工作
25. server.configureBlocking(false);
26. // 将server注册到指定的selector对象
27. server.register(selector, SelectionKey.OP_ACCEPT);
28. while (selector.select() > 0) {
29. // 一次处理selector上的每个选择的SelectionKey
30. for (SelectionKey sk : selector.selectedKeys()) {
31. // 从selector上已选择的Kye集中删除正在处理的SelectionKey
32. selector.selectedKeys().remove(sk);
33. // 如果sk对应的Channel包含客户端的连接请求
34. if (sk.isAcceptable()) {
35. // 调用accept方法接收连接，产生服务器段的SocketChennal
36. SocketChannel sc = server.accept();
37. // 设置采用非阻塞模式
38. sc.configureBlocking(false);
39. // 将该SocketChannel注册到selector
40. sc.register(selector, SelectionKey.OP_READ);
41. }
42. // 如果sk对应的Channel有数据需要读取
43. if (sk.isReadable()) {
44. // 获取该SelectionKey对银行的Channel，该Channel中有刻度的数据
45. SocketChannel sc = (SocketChannel) sk.channel();
46. // 定义备注执行读取数据源的ByteBuffer
47. ByteBuffer buff = ByteBuffer.allocate(1024);
48. String content = "";
49. // 开始读取数据
50. try {
51. while (sc.read(buff) > 0) {
52. buff.flip();
53. content += charse.decode(buff);
54. }
55. System.out.println("读取的数据：" + content);
56. // 将sk对应的Channel设置成准备下一次读取
57. sk.interestOps(SelectionKey.OP_READ);
58. }
59. // 如果捕获到该sk对银行的Channel出现了异常，表明
60. // Channel对应的Client出现了问题，所以从Selector中取消
61. catch (IOException io) {
62. // 从Selector中删除指定的SelectionKey
63. sk.cancel();
64. if (sk.channel() != null) {
65. sk.channel().close();
66. }
67. }
68. // 如果content的长度大于0,则连天信息不为空
69. if (content.length() > 0) {
70. // 遍历selector里注册的所有SelectionKey
71. for (SelectionKey key : selector.keys()) {
72. // 获取该key对应的Channel
73. Channel targerChannel = key.channel();
74. // 如果该Channel是SocketChannel对象
75. if (targerChannel instanceof SocketChannel) {
76. // 将读取到的内容写入该Channel中
77. SocketChannel dest = (SocketChannel) targerChannel;
78. dest.write(charse.encode(content));
79. }
80. }
81. }
82. }
83. }
84. }
85. }
86. public static void main(String [] args) throws IOException{
87. new NServer().init();
88. }
89. }

客户端

1. package com.java.xiong.Net17;
2. import java.io.IOException;
3. import java.net.InetSocketAddress;
4. import java.nio.ByteBuffer;
5. import java.nio.channels.SelectionKey;
6. import java.nio.channels.Selector;
7. import java.nio.channels.SocketChannel;
8. import java.nio.charset.Charset;
9. import java.util.Scanner;
10. public class NClient {
11. //定义检测Sockethannel的Selector对象
12. private Selector  selector=null;
13. static final int PORT=30000;
14. //定义处理编码的字符集
15. private Charset charset=Charset.forName("GBK");
16. //客户端SocketChannel
17. private SocketChannel sc=null;
18. public void init() throws IOException{
19. selector=Selector.open();
20. InetSocketAddress isa=new InetSocketAddress("127.0.0.1", PORT);
21. //调用open的静态方法创建连接指定的主机的SocketChannel
22. sc=SocketChannel.open(isa);
23. //设置该sc已非阻塞的方式工作
24. sc.configureBlocking(false);
25. //将SocketChannel对象注册到指定的Selector
26. sc.register(selector, SelectionKey.OP_READ);
27. //启动读取服务器数据端的线程
28. new ClientThread().start();
29. //创建键盘输入流
30. Scanner scan=new Scanner(System.in);
31. while(scan.hasNextLine()){
32. //读取键盘的输入
33. String line=scan.nextLine();
34. //将键盘的内容输出到SocketChanenel中
35. sc.write(charset.encode(line));
36. }
37. }
38. //定义读取服务器端的数据的线程
39. private class ClientThread extends Thread{
40. @Override
41. public void run() {
42. try{
43. while(selector.select()>0){
44. //遍历每个有可能的IO操作的Channel对银行的SelectionKey
45. for(SelectionKey sk:selector.selectedKeys()){
46. //删除正在处理的SelectionKey
47. selector.selectedKeys().remove(sk);
48. //如果该SelectionKey对应的Channel中有可读的数据
49. if(sk.isReadable()){
50. //使用NIO读取Channel中的数据
51. SocketChannel sc=(SocketChannel)sk.channel();
52. String content="";
53. ByteBuffer bff=ByteBuffer.allocate(1024);
54. while(sc.read(bff)>0){
55. sc.read(bff);
56. bff.flip();
57. content+=charset.decode(bff);
58. }
59. //打印读取的内容
60. System.out.println("聊天信息:"+content);
61. sk.interestOps(SelectionKey.OP_READ);
62. }
63. }
64. }
65. }catch(IOException io){
66. io.printStackTrace();
67. }
68. }
69. }
70. public static void main(String [] args) throws IOException{
71. new NClient().init();
72. }
73. }