java nio Selector的使用－客户端

java nio Selector的使用－客户端

　　接上一篇，客户端的程序就相对于简单了，只需要负责连接，发送下载文件名，再读数据就行了。主要步骤就是注册->连接服务器->发送下载请求->读数据->断开连接。

　　第一步：注册，并注册connect事件。

Java代码 

1. if(selector == null)   
2.     selector = Selector.open();   
3. SocketChannel channel = SocketChannel.open();   
4. channel.configureBlocking(false);   
5. channel.connect(new InetSocketAddress("localhost", 1234));   
6. channel.register(selector, SelectionKey.OP_CONNECT);  

[java]  view plain copy

1. if(selector == null)  
2.     selector = Selector.open();  
3. SocketChannel channel = SocketChannel.open();  
4. channel.configureBlocking(false);  
5. channel.connect(new InetSocketAddress("localhost", 1234));  
6. channel.register(selector, SelectionKey.OP_CONNECT);  

 

　　以上步骤，在实际中发现configure和connet并不能随意交换位置，即不能将connect放在configureBlocking的前面，更不能在open中直接添加InetSocketAddress参数了。在官方doc中，open(InetAddress)的解释是“这种便捷方法的工作方式就像以下过程一样：调用 open() 方法、在得到的套接字通道上调用 connect 方法、向其传递 remote，然后返回该通道。”，但不知道为什么在connect之后，就不能配置block了，导致无法进行数据下载和通信。

 

　　第二步：处理connect事件

Java代码 

1. //连接事件   
2. if(key.isConnectable()) {   
3.     SocketChannel socketChannel = (SocketChannel) key.channel();   
4.     if(socketChannel.isConnectionPending())   
5.         socketChannel.finishConnect();   
6.     socketChannel.write(ByteBuffer.wrap(serverFileName.getBytes()));//向服务器发信息,信息中即服务器上的文件名   
7.     socketChannel.register(selector, SelectionKey.OP_READ);   
8. }  

[java]  view plain copy

1. //连接事件  
2. if(key.isConnectable()) {  
3.     SocketChannel socketChannel = (SocketChannel) key.channel();  
4.     if(socketChannel.isConnectionPending())  
5.         socketChannel.finishConnect();  
6.     socketChannel.write(ByteBuffer.wrap(serverFileName.getBytes()));//向服务器发信息,信息中即服务器上的文件名  
7.     socketChannel.register(selector, SelectionKey.OP_READ);  
8. }  

 

　　在以上步骤中，要完成连接之后，向服务器端发送了下载的文件名的数据信息，并注册read事件。服务器端在接收到相应文件名之后，就开启write事件向客户端进行发送数据了，客户端此时就可以进行数据读取了。

　　

　　第三步：处理read事件

Java代码 

1. if(key.isReadable()) {   
2.     SocketChannel socketChannel = (SocketChannel) key.channel();   
3.     byteBuffer.clear();   
4.     if(!socketChannel.isConnected())   
5.         return null;   
6. //向本机下载文件创建文件channel   
7.     if(fileChannel == null)   
8.         fileChannel = new RandomAccessFile(localFileName, "rw").getChannel();   
9.     int r = socketChannel.read(byteBuffer);   
10. //如果文件下载完毕,则关掉channel,同时关掉socketChannel   
11.     if(r <= 0) {   
12.         if(fileChannel != null)   
13.             fileChannel.close();   
14.         channel.close();   
15.         key.cancel();   
16.         return null;   
17.     }   
18.     byteBuffer.flip();   
19. //写到下载文件中   
20.     fileChannel.write(byteBuffer);   
21. }  

[java]  view plain copy

1. if(key.isReadable()) {  
2.     SocketChannel socketChannel = (SocketChannel) key.channel();  
3.     byteBuffer.clear();  
4.     if(!socketChannel.isConnected())  
5.         return null;  
6. //向本机下载文件创建文件channel  
7.     if(fileChannel == null)  
8.         fileChannel = new RandomAccessFile(localFileName, "rw").getChannel();  
9.     int r = socketChannel.read(byteBuffer);  
10. //如果文件下载完毕,则关掉channel,同时关掉socketChannel  
11.     if(r <= 0) {  
12.         if(fileChannel != null)  
13.             fileChannel.close();  
14.         channel.close();  
15.         key.cancel();  
16.         return null;  
17.     }  
18.     byteBuffer.flip();  
19. //写到下载文件中  
20.     fileChannel.write(byteBuffer);  
21. }  

 

　　就是处理读信息，如果数据已经读取完毕，则完成相应下载保存文件的文件流，并退出程序。

 

　　这样，整个客户端就完成了，在运行时，我同时启用１０个线程来向服务器端读同一个文件，并保存为不同的文件备份，以达到模拟数据传输的功能。如下所示：

Java代码 

1. ExecutorService executorService = Executors.newSingleThreadExecutor();   
2. for(int i = 0; i < 10; i++) {   
3.     executorService.submit(new DownloadClient<Object>("d:/log4j.log", "d:/down" + i + ".log"));   
4. }   
5. executorService.shutdown();  

[java]  view plain copy

1. ExecutorService executorService = Executors.newSingleThreadExecutor();  
2. for(int i = 0; i < 10; i++) {  
3.     executorService.submit(new DownloadClient<Object>("d:/log4j.log", "d:/down" + i + ".log"));  
4. }  
5. executorService.shutdown();  

 

　　整个selector仅是一个作为练习用的小例子，如果用在实际代码中，还需要处理不同的异常和相应的逻辑等。对于学习还是有一定的帮助的。希望对你有用。