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

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

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

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

客户端代码：

import java.io.DataOutputStream; 
import java.io.File; 
import java.io.FileInputStream; 
import java.net.Socket; 
 
/** 
 * 文件传输Client端<br> 
 * 功能说明： 
 * 
 * @author 大智若愚的小懂 
 * @Date 2016年09月01日 
 * @version 1.0 
 */ 
public class FileTransferClient extends Socket { 
 
  private static final String SERVER_IP = "127.0.0.1"; // 服务端IP 
  private static final int SERVER_PORT = 8899; // 服务端端口 
 
  private Socket client; 
 
  private FileInputStream fis; 
 
  private DataOutputStream dos; 
 
  /** 
   * 构造函数<br/> 
   * 与服务器建立连接 
   * @throws Exception 
   */ 
  public FileTransferClient() throws Exception { 
    super(SERVER_IP, SERVER_PORT); 
    this.client = this; 
    System.out.println("Cliect[port:" + client.getLocalPort() + "] 成功连接服务端"); 
  } 
 
  /** 
   * 向服务端传输文件 
   * @throws Exception 
   */ 
  public void sendFile() throws Exception { 
    try { 
      File file = new File("E:\\JDK1.6中文参考手册(JDK_API_1_6_zh_CN).CHM"); 
      if(file.exists()) { 
        fis = new FileInputStream(file); 
        dos = new DataOutputStream(client.getOutputStream()); 
 
        // 文件名和长度 
        dos.writeUTF(file.getName()); 
        dos.flush(); 
        dos.writeLong(file.length()); 
        dos.flush(); 
 
        // 开始传输文件 
        System.out.println("======== 开始传输文件 ========"); 
        byte[] bytes = new byte[1024]; 
        int length = 0; 
        long progress = 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 
   */ 
  public static void main(String[] args) { 
    try { 
      FileTransferClient client = new FileTransferClient(); // 启动客户端连接 
      client.sendFile(); // 传输文件 
    } catch (Exception e) { 
      e.printStackTrace(); 
    } 
  } 
 
} 

服务端代码：

import java.io.DataInputStream; 
import java.io.File; 
import java.io.FileOutputStream; 
import java.math.RoundingMode; 
import java.net.ServerSocket; 
import java.net.Socket; 
import java.text.DecimalFormat; 
 
/** 
 * 文件传输Server端<br> 
 * 功能说明： 
 * 
 * @author 大智若愚的小懂 
 * @Date 2016年09月01日 
 * @version 1.0 
 */ 
public class FileTransferServer extends ServerSocket { 
 
  private static final int SERVER_PORT = 8899; // 服务端端口 
 
  private static DecimalFormat df = null; 
 
  static { 
    // 设置数字格式，保留一位有效小数 
    df = new DecimalFormat("#0.0"); 
    df.setRoundingMode(RoundingMode.HALF_UP); 
    df.setMinimumFractionDigits(1); 
    df.setMaximumFractionDigits(1); 
  } 
 
  public FileTransferServer() throws Exception { 
    super(SERVER_PORT); 
  } 
 
  /** 
   * 使用线程处理每个客户端传输的文件 
   * @throws Exception 
   */ 
  public void load() throws Exception { 
    while (true) { 
      // server尝试接收其他Socket的连接请求，server的accept方法是阻塞式的 
      Socket socket = this.accept(); 
      /** 
       * 我们的服务端处理客户端的连接请求是同步进行的， 每次接收到来自客户端的连接请求后， 
       * 都要先跟当前的客户端通信完之后才能再处理下一个连接请求。 这在并发比较多的情况下会严重影响程序的性能， 
       * 为此，我们可以把它改为如下这种异步处理与客户端通信的方式 
       */ 
      // 每接收到一个Socket就建立一个新的线程来处理它 
      new Thread(new Task(socket)).start(); 
    } 
  } 
 
  /** 
   * 处理客户端传输过来的文件线程类 
   */ 
  class Task implements Runnable { 
 
    private Socket socket; 
 
    private DataInputStream dis; 
 
    private FileOutputStream fos; 
 
    public Task(Socket socket) { 
      this.socket = socket; 
    } 
 
    @Override 
    public void run() { 
      try { 
        dis = new DataInputStream(socket.getInputStream()); 
 
        // 文件名和长度 
        String fileName = dis.readUTF(); 
        long fileLength = dis.readLong(); 
        File directory = new File("D:\\FTCache"); 
        if(!directory.exists()) { 
          directory.mkdir(); 
        } 
        File file = new File(directory.getAbsolutePath() + File.separatorChar + fileName); 
        fos = new FileOutputStream(file); 
 
        // 开始接收文件 
        byte[] bytes = new byte[1024]; 
        int length = 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 
   */ 
  private String getFormatFileSize(long length) { 
    double size = ((double) length) / (1 << 30); 
    if(size >= 1) { 
      return df.format(size) + "GB"; 
    } 
    size = ((double) length) / (1 << 20); 
    if(size >= 1) { 
      return df.format(size) + "MB"; 
    } 
    size = ((double) length) / (1 << 10); 
    if(size >= 1) { 
      return df.format(size) + "KB"; 
    } 
    return length + "B"; 
  } 
 
  /** 
   * 入口 
   * @param args 
   */ 
  public static void main(String[] args) { 
    try { 
      FileTransferServer server = new FileTransferServer(); // 启动服务端 
      server.load(); 
    } catch (Exception e) { 
      e.printStackTrace(); 
    } 
  } 
} 

测试的结果（客户端）：

测试的结果（服务端）：

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持小牛知识库。