自己用java实现飞鸽传书 2 - 实现文件传输

第二步：实现文件传递。

上一步只是从服务端传递了一个字符串到客户端，这次需要对代码进行调整，实现从服务端获取文件，在客户端将文件存入目标地址。

调整后的代码：

服务端：

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.net.ServerSocket;
import java.net.Socket;

public class Server {
    
    public static void main(String[] args){
        //创建服务端socket，用以网络信息的接收
        ServerSocket serverSocket = null;
        try {
            //服务端socket使用本地IP，端口号使用36000
            serverSocket = new ServerSocket(36000);
            //保证服务端持续运行
            while(true){
                //接收网络信息，未接收到则阻塞当前线程
                Socket socket = serverSocket.accept();
                //将文件信息写入socket通信
                writeFile(socket);
                //关闭socket，防止客户端长时间等待
                socket.close();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    
    /**
     * 将文件信息写入socket通信流
     * @param socket
     *             socket
     */
    private static void writeFile(Socket socket){
        try{
            //获取本地文件
            String filePath ="D:/test.pdf";
            //创建文件对象
            File file = new File(filePath);
            //使用DataOutputStream传送文件信息
            DataOutputStream dos = new DataOutputStream(socket.getOutputStream());
            //写入文件名
            dos.writeUTF(file.getName());
            //写入文件大小
            dos.writeLong(file.length());
            //清空缓冲区
            dos.flush();
            //使用DataInputStream将文件写入
            DataInputStream dis = new DataInputStream( new FileInputStream(file) ); 
            byte[] buffer = new byte[1024];
            int head = 0;
            while((head=dis.read(buffer)) > 0){
                dos.write(buffer, 0, head);
            }
            dis.close();
            dos.close();
        }catch(Exception e){
            e.printStackTrace();
        }
    }
}

客户端：

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.FileOutputStream;
import java.net.Socket;

public class Client {

    public static void main(String[] args){
        DataInputStream input = null;
        Socket clientSocket = null;
        try {
            //创建客户端socket，用以连接服务端
            clientSocket = new Socket("127.0.0.1", 36000);
            //读取服务端发送的信息流
            input = new DataInputStream(clientSocket.getInputStream());
            String fileName = input.readUTF();
            long fileSize = input.readLong();
            System.out.println("收到服务端发过来的文件，文件名称为：" 
                                + fileName + "，文件大小为：" + fileSize);
            //将文件读入本地
            String savePath = "E:\\just a test.pdf";
            DataOutputStream fos = new DataOutputStream(new FileOutputStream(savePath));
            byte[] buffer = new byte[1024];
            int head = 0;
            while((head=input.read(buffer)) > 0){
                fos.write(buffer, 0, head);
            }
            
            fos.close();
            input.close();
            clientSocket.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

本次文件流的传送使用了DataOutputStream和DataInputStream，使用Data类型的字节流可以根据信息类型分别进行处理。