socket断点续传文件 java

原创

mob64ca12e95b2b 2024-09-04 05:24:05 ©著作权

文章标签 客户端服务器偏移量 文章分类 Java 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12e95b2b的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用Socket实现文件的断点续传

在网络编程中，文件传输是一个常见的需求，尤其是在需要发送大文件时，可能会遇到网络中断等问题，导致数据传输无法完成。因此，实现文件的断点续传功能就显得尤为重要。本文将通过Java中的Socket编程，展示如何实现文件的断点续传。

什么是断点续传

断点续传是指在文件传输过程中如果因为某种原因中断了，可以在中断的地方继续传输，而不是从头开始。这种技术可以极大地提高网络传输的效率，尤其是对于大文件传输而言。

基本思路

检查本地文件：在开始传输文件之前，检查本地是否存在部分下载的文件，并获取其大小。
通知服务器：将本地文件的大小告诉服务器，以便服务器可以从该位置继续发送文件。
文件传输：使用Socket连接进行传输，接收未传输部分的数据。
保存文件：将接收到的数据追加到本地文件中。

代码示例

服务器端

首先，我们来看服务器端的代码。服务器会接收客户端的请求，并根据客户端的偏移量来发送文件数据。

import java.io.*;
import java.net.*;

public class FileServer {
    public static void main(String[] args) {
        try (ServerSocket serverSocket = new ServerSocket(8888)) {
            System.out.println("服务器启动，等待连接...");
            while (true) {
                Socket socket = serverSocket.accept();
                System.out.println("客户端连接：" + socket.getInetAddress());

                new Thread(() -> {
                    try (DataInputStream dis = new DataInputStream(socket.getInputStream());
                         OutputStream os = new FileOutputStream("received_file", true)) {

                        long offset = dis.readLong(); // 读取客户端的文件偏移量
                        System.out.println("偏移量：" + offset);
                        File file = new File("file_to_send");
                        RandomAccessFile raf = new RandomAccessFile(file, "r");
                        raf.seek(offset); // 移动到偏移量处

                        byte[] buffer = new byte[1024];
                        int bytesRead;
                        while ((bytesRead = raf.read(buffer)) != -1) {
                            os.write(buffer, 0, bytesRead);
                        }
                        raf.close();
                        System.out.println("文件传输完成。");

                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }).start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

客户端

接下来是客户端的代码。客户端在传输文件之前需先确定文件的现有大小。

import java.io.*;
import java.net.*;

public class FileClient {
    public static void main(String[] args) {
        try (Socket socket = new Socket("localhost", 8888);
             DataOutputStream dos = new DataOutputStream(socket.getOutputStream());
             FileOutputStream fos = new FileOutputStream("received_file", true)) {

            File file = new File("file_to_send");
            long offset = fos.getChannel().size(); // 获取本地文件的当前大小
            dos.writeLong(offset); // 将偏移量发送给服务器

            // 如果文件已经完整，无需继续传输
            if (offset >= file.length()) {
                System.out.println("文件已经完全传输。");
                return;
            }

            // 从服务器接收数据
            InputStream is = socket.getInputStream();
            byte[] buffer = new byte[1024];
            int bytesRead;
            while ((bytesRead = is.read(buffer)) != -1) {
                fos.write(buffer, 0, bytesRead);
            }

            System.out.println("文件传输完成。");

        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

工作机制

以下是整个传输过程的工作机制：

journey
    title 文件断点续传过程
    section 文件检查
      客户端检查本地文件 : 5: 客户端
    section 发送偏移量
      客户端发送当前偏移量给服务器 : 5: 客户端 --> 服务器
    section 文件传输
      服务器根据偏移量发送文件数据 : 5: 服务器 --> 客户端
    section 文件保存
      客户端接收数据并保存 : 5: 客户端