NIO传输文件

 1 package com.gs.practice.nio;
 2 
 3 import java.io.FileInputStream;
 4 import java.io.IOException;
 5 import java.net.InetSocketAddress;
 6 import java.nio.ByteBuffer;
 7 import java.nio.channels.ClosedChannelException;
 8 import java.nio.channels.FileChannel;
 9 import java.nio.channels.SelectionKey;
10 import java.nio.channels.Selector;
11 import java.nio.channels.SocketChannel;
12 import java.util.Set;
13 
14 /**
15  * Copyright (C),HTF<br>
16  * 客户端上传图片
17  * 
18  * @author muzhongjiang
19  * @date 2014年9月5日
20  */
21 public class FileClient {
22     private int port = 8000;
23     private static ByteBuffer sendBuffer = ByteBuffer.allocate(1024);// 发送数据缓冲区
24     private InetSocketAddress SERVER;
25     private static Selector selector;
26     private static SocketChannel client;
27 
28     public FileClient() throws Exception {
29         SERVER = new InetSocketAddress("localhost", port);
30         init();
31     }
32 
33     /**
34      * 初始化
35      */
36     private void init() throws Exception {
37         SocketChannel socketChannel = SocketChannel.open();
38         socketChannel.configureBlocking(false);
39         selector = Selector.open();
40         socketChannel.register(selector, SelectionKey.OP_CONNECT);
41         socketChannel.connect(SERVER);
42         this.listener();
43     }
44 
45     /**
46      * 监听器
47      */
48     private void listener() throws IOException, ClosedChannelException, Exception {
49         while (true) {
50             selector.select();
51             Set<SelectionKey> keySet = selector.selectedKeys();
52             for (final SelectionKey key : keySet) {
53                 if (key.isConnectable()) {
54                     client = (SocketChannel) key.channel();
55                     client.finishConnect();
56                     client.register(selector, SelectionKey.OP_WRITE);
57 
58                 } else if (key.isWritable()) {
59                     this.sendFile(client);
60                 }
61             }
62             keySet.clear();
63         }
64     }
65 
66     private void sendFile(SocketChannel client) throws Exception {
67         FileInputStream fis = null;
68         FileChannel channel = null;// 传输不同的数据选择对应的“通道”
69         fis = new FileInputStream("E:\\1.txt");
70         channel = fis.getChannel();
71         int i = 1;
72         int count = 0;
73         while ((count = channel.read(sendBuffer)) != -1) {
74             sendBuffer.flip();
75             int send = client.write(sendBuffer);
76             System.out.println("i===========" + (i++) + "   count:" + count + " send:" + send);
77             while (send == 0) {// 传输失败（ 服务器端可能因为缓存区满，而导致数据传输失败，需要重新发送）
78                 Thread.sleep(10);
79                 send = client.write(sendBuffer);
80                 System.out.println("i重新传输====" + i + "   count:" + count + " send:" + send);
81             }
82             sendBuffer.clear();
83         }
84         channel.close();
85         fis.close();
86         client.close();
87     }
88 
89     public static void main(String[] args) throws Exception {
90         new FileClient();
91     }
92 }

  1 package com.gs.practice.nio;
  2 
  3 import java.io.FileNotFoundException;
  4 import java.io.FileOutputStream;
  5 import java.io.IOException;
  6 import java.net.InetSocketAddress;
  7 import java.net.ServerSocket;
  8 import java.nio.ByteBuffer;
  9 import java.nio.channels.ClosedChannelException;
 10 import java.nio.channels.FileChannel;
 11 import java.nio.channels.SelectionKey;
 12 import java.nio.channels.Selector;
 13 import java.nio.channels.ServerSocketChannel;
 14 import java.nio.channels.SocketChannel;
 15 import java.util.Set;
 16 /**
 17  * Copyright (C),HTF<br>
 18  * 图片服务器端
 19  * 
 20  * @author muzhongjiang
 21  * @date 2014年9月5日
 22  */
 23 public class FileServer {
 24     private int port = 8000;
 25     private static ByteBuffer revBuffer = ByteBuffer.allocate(1024);// 接收数据缓冲区
 26     private static Selector selector;
 27     private static FileOutputStream fout;
 28     private static FileChannel ch;
 29 
 30     public FileServer() throws Exception {
 31         init();
 32     }
 33 
 34     /**
 35      * 初始化
 36      */
 37     private void init() throws Exception {
 38         ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();
 39         serverSocketChannel.configureBlocking(false);
 40         ServerSocket serverSocket = serverSocketChannel.socket();
 41         serverSocket.bind(new InetSocketAddress(port));
 42         selector = Selector.open();
 43         serverSocketChannel.register(selector, SelectionKey.OP_ACCEPT);
 44         System.out.println("server start on port:" + port);
 45         this.listener();
 46     }
 47 
 48     /**
 49      * 监听器
 50      */
 51     private void listener() throws IOException, ClosedChannelException, FileNotFoundException {
 52         while (true) {
 53             selector.select();// 返回值为本次触发的事件数
 54             Set<SelectionKey> selectionKeys = selector.selectedKeys();
 55 
 56             for (SelectionKey key : selectionKeys) {
 57                 ServerSocketChannel server = null;
 58                 SocketChannel client = null;
 59                 int count = 0;
 60                 if (key.isAcceptable()) {
 61                     server = (ServerSocketChannel) key.channel();
 62                     System.out.println("有客户端连接进入=============)");
 63                     client = server.accept();
 64                     client.configureBlocking(false);
 65                     client.register(selector, SelectionKey.OP_READ);
 66                     fout = new FileOutputStream("E:\\" + client.hashCode() + ".txt");
 67                     ch = fout.getChannel();
 68                 } else if (key.isReadable()) {
 69                     client = (SocketChannel) key.channel();
 70                     revBuffer.clear();
 71                     count = client.read(revBuffer);
 72                     int k = 0;
 73                     // 循环读取缓存区的数据，
 74                     while (count > 0) {
 75                         System.out.println("k=" + (k++) + " 读取到数据量:" + count);
 76                         revBuffer.flip();
 77                         ch.write(revBuffer);
 78                         fout.flush();
 79                         revBuffer.clear();
 80                         count = client.read(revBuffer);
 81                     }
 82                     if (count == -1) {
 83                         client.close();
 84                         ch.close();
 85                         fout.close();
 86                     }
 87                 } else if (key.isWritable()) {
 88                     System.out.println("selectionKey.isWritable()");
 89                 }
 90             }
 91             System.out.println("=======selectionKeys.clear()");
 92             selectionKeys.clear();
 93 
 94         }
 95     }
 96 
 97     public static void main(String[] args) throws Exception {
 98         new FileServer();
 99     }
100 }