用Node.js实现断点续传：彻底掌控文件传输

人工智能

2024-02-10 22:04:48

Node.js 断点续传指南：轻松实现高效文件传输

摘要：

断点续传 是一种技术，它允许在文件传输过程中遇到中断（如网络故障或服务器异常）时从中断处继续传输文件。在本指南中，我们将深入探讨如何使用 Node.js 实现断点续传，并通过逐步指南和示例代码帮助您轻松掌握这项技术。

Node.js 中的断点续传

断点续传在 Node.js 中实现非常简单，因为它利用了 HTTP 协议提供的 Range 请求头 。Range 请求头允许客户端指定要从服务器获取文件的一部分，而不是整个文件。

客户端实现

在客户端（即发送请求的文件上传/下载应用程序）中，需要向请求添加 Range 请求头，指定要下载/上传的字节范围。以下是使用 Axios 库发送带有 Range 请求头的 HTTP 请求的示例代码：

const axios = require('axios');

//假设文件大小为 100 字节，我们想要从第 51 个字节开始下载
const range = 'bytes=51-';

axios.get(url, {
  headers: {
    Range: range
  }
}).then((response) => {
  //处理响应
});

服务器端实现

在服务器端（即处理请求的文件服务器）中，需要检查请求是否包含 Range 请求头。如果包含，则需要从指定的字节范围开始传输文件。

以下是使用 Express 框架提供的一个服务器端示例：

const express = require('express');

const app = express();

app.get('/file', (req, res) => {
  const range = req.headers['range'];
  if (range) {
    //解析范围并设置响应头
    const [start, end] = range.replace(/bytes=/, '').split('-');
    res.writeHead(206, {
      'Content-Range': `bytes ${start}-${end}/${file.size}`,
      'Content-Length': end - start + 1,
    });
    //从指定的字节范围传输文件
    res.sendFile(file, { start, end });
  } else {
    //发送整个文件
    res.sendFile(file);
  }
});

逐步指南

以下是如何在 Node.js 中实现断点续传的分步指南：

在客户端，使用带有 Range 请求头的 HTTP 库（例如 Axios）发送请求。
在服务器端，使用 Express 框架或类似框架检查请求是否包含 Range 请求头。
如果包含 Range 请求头，则解析范围并设置适当的响应头。
从指定的字节范围传输文件。

示例代码

以下示例代码展示了如何使用 Node.js 实现文件上传和下载的断点续传：

客户端（上传）：

const axios = require('axios');

const file = 'file.txt';
const start = 50; //假设文件从第 50 个字节开始上传

const range = `bytes ${start}-`;

axios.put(url, file, {
  headers: {
    Range: range,
    'Content-Type': 'text/plain'
  }
}).then((response) => {
  //处理响应
});

客户端（下载）：

const axios = require('axios');

const range = 'bytes=50-';

axios.get(url, {
  headers: {
    Range: range
  }
}).then((response) => {
  //处理响应
});

服务器端：

const express = require('express');

const app = express();

app.put('/file', (req, res) => {
  const range = req.headers['range'];
  if (range) {
    //解析范围并设置响应头
    const [start, end] = range.replace(/bytes=/, '').split('-');
    res.writeHead(206, {
      'Content-Range': `bytes ${start}-${end}/${file.size}`,
      'Content-Length': end - start + 1,
    });
    //从指定的字节范围接受文件
    req.pipe(fs.createWriteStream(file, { start, end }));
  } else {
    //接收整个文件
    req.pipe(fs.createWriteStream(file));
  }
});

app.get('/file', (req, res) => {
  const range = req.headers['range'];
  if (range) {
    //解析范围并设置响应头
    const [start, end] = range.replace(/bytes=/, '').split('-');
    res.writeHead(206, {
      'Content-Range': `bytes ${start}-${end}/${file.size}`,
      'Content-Length': end - start + 1,
    });
    //从指定的字节范围传输文件
    fs.createReadStream(file, { start, end }).pipe(res);
  } else {
    //发送整个文件
    fs.createReadStream(file).pipe(res);
  }
});