Java Socket实现文件传输示例代码

作者:简简单单 2017-03-06

最近学Socket学上瘾了,就写了一个简单的文件传输程序。

客户端设计思路:客户端与服务端建立连接,选择客户端本地文件,先将文件名及大小等属性发送给服务端,再将文件通过流的方式传输给服务端。传输的进度打印到控制台中,直到传输完成。

服务端设计思路:服务端接收客户端的请求(阻塞式),每接收到一个客户端请求连接后,就新开一个处理文件的线程,开始写入流,将文件到服务器的指定目录下,并与传输过来的文件同名。
下面是客户端和服务端的代码实现:

客户端代码:

 代码如下复制代码

importjava.io.DataOutputStream;

importjava.io.File;

importjava.io.FileInputStream;

importjava.net.Socket;

  

/**

 * 文件传输Client端

 * 功能说明:

 *

 * @author 大智若愚的小懂

 * @Date 2016年09月01日

 * @version 1.0

 */

publicclassFileTransferClientextendsSocket {

  

  privatestaticfinalString SERVER_IP ="127.0.0.1";// 服务端IP

  privatestaticfinalintSERVER_PORT =8899;// 服务端端口

  

  privateSocket client;

  

  privateFileInputStream fis;

  

  privateDataOutputStream dos;

  

  /**

   * 构造函数

   * 与服务器建立连接

   * @throws Exception

   */

  publicFileTransferClient()throwsException {

    super(SERVER_IP, SERVER_PORT);

    this.client =this;

    System.out.println("Cliect[port:"+ client.getLocalPort() +"] 成功连接服务端");

  }

  

  /**

   * 向服务端传输文件

   * @throws Exception

   */

  publicvoidsendFile()throwsException {

    try{

      File file =newFile("E:\\JDK1.6中文参考手册(JDK_API_1_6_zh_CN).CHM");

      if(file.exists()) {

        fis =newFileInputStream(file);

        dos =newDataOutputStream(client.getOutputStream());

  

        // 文件名和长度

        dos.writeUTF(file.getName());

        dos.flush();

        dos.writeLong(file.length());

        dos.flush();

  

        // 开始传输文件

        System.out.println("======== 开始传输文件 ========");

        byte[] bytes =newbyte[1024];

        intlength =0;

        longprogress =0;

        while((length = fis.read(bytes,0, bytes.length)) != -1) {

          dos.write(bytes,0, length);

          dos.flush();

          progress += length;

          System.out.print("| "+ (100*progress/file.length()) +"% |");

        }

        System.out.println();

        System.out.println("======== 文件传输成功 ========");

      }

    }catch(Exception e) {

      e.printStackTrace();

    }finally{

      if(fis !=null)

        fis.close();

      if(dos !=null)

        dos.close();

      client.close();

    }

  }

  

  /**

   * 入口

   * @param args

   */

  publicstaticvoidmain(String[] args) {

    try{

      FileTransferClient client =newFileTransferClient();// 启动客户端连接

      client.sendFile();// 传输文件

    }catch(Exception e) {

      e.printStackTrace();

    }

  }

  

}

服务端代码:

 代码如下复制代码

importjava.io.DataInputStream;

importjava.io.File;

importjava.io.FileOutputStream;

importjava.math.RoundingMode;

importjava.net.ServerSocket;

importjava.net.Socket;

importjava.text.DecimalFormat;

  

/**

 * 文件传输Server端

 * 功能说明:

 *

 * @author 大智若愚的小懂

 * @Date 2016年09月01日

 * @version 1.0

 */

publicclassFileTransferServerextendsServerSocket {

  

  privatestaticfinalintSERVER_PORT =8899;// 服务端端口

  

  privatestaticDecimalFormat df =null;

  

  static{

    // 设置数字格式,保留一位有效小数

    df =newDecimalFormat("#0.0");

    df.setRoundingMode(RoundingMode.HALF_UP);

    df.setMinimumFractionDigits(1);

    df.setMaximumFractionDigits(1);

  }

  

  publicFileTransferServer()throwsException {

    super(SERVER_PORT);

  }

  

  /**

   * 使用线程处理每个客户端传输的文件

   * @throws Exception

   */

  publicvoidload()throwsException {

    while(true) {

      // server尝试接收其他Socket的连接请求,server的accept方法是阻塞式的

      Socket socket =this.accept();

      /**

       * 我们的服务端处理客户端的连接请求是同步进行的, 每次接收到来自客户端的连接请求后,

       * 都要先跟当前的客户端通信完之后才能再处理下一个连接请求。 这在并发比较多的情况下会严重影响程序的性能,

       * 为此,我们可以把它改为如下这种异步处理与客户端通信的方式

       */

      // 每接收到一个Socket就建立一个新的线程来处理它

      newThread(newTask(socket)).start();

    }

  }

  

  /**

   * 处理客户端传输过来的文件线程类

   */

  classTaskimplementsRunnable {

  

    privateSocket socket;

  

    privateDataInputStream dis;

  

    privateFileOutputStream fos;

  

    publicTask(Socket socket) {

      this.socket = socket;

    }

  

    @Override

    publicvoidrun() {

      try{

        dis =newDataInputStream(socket.getInputStream());

  

        // 文件名和长度

        String fileName = dis.readUTF();

        longfileLength = dis.readLong();

        File directory =newFile("D:\\FTCache");

        if(!directory.exists()) {

          directory.mkdir();

        }

        File file =newFile(directory.getAbsolutePath() + File.separatorChar + fileName);

        fos =newFileOutputStream(file);

  

        // 开始接收文件

        byte[] bytes =newbyte[1024];

        intlength =0;

        while((length = dis.read(bytes,0, bytes.length)) != -1) {

          fos.write(bytes,0, length);

          fos.flush();

        }

        System.out.println("======== 文件接收成功 [File Name:"+ fileName +"] [Size:"+ getFormatFileSize(fileLength) +"] ========");

      }catch(Exception e) {

        e.printStackTrace();

      }finally{

        try{

          if(fos !=null)

            fos.close();

          if(dis !=null)

            dis.close();

          socket.close();

        }catch(Exception e) {}

      }

    }

  }

  

  /**

   * 格式化文件大小

   * @param length

   * @return

   */

  privateString getFormatFileSize(longlength) {

    doublesize = ((double) length) / (1<<30);

    if(size >=1) {

      returndf.format(size) +"GB";

    }

    size = ((double) length) / (1<<20);

    if(size >=1) {

      returndf.format(size) +"MB";

    }

    size = ((double) length) / (1<<10);

    if(size >=1) {

      returndf.format(size) +"KB";

    }

    returnlength +"B";

  }

  

  /**

   * 入口

   * @param args

   */

  publicstaticvoidmain(String[] args) {

    try{

      FileTransferServer server =newFileTransferServer();// 启动服务端

      server.load();

    }catch(Exception e) {

      e.printStackTrace();

    }

  }

}

测试的结果(客户端):

测试的结果(服务端):

相关文章

精彩推荐