自己用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类型的字节流可以根据信息类型分别进行处理。

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。