Node.js断点续传的实现

下面就是“Node.js断点续传的实现”的完整攻略。

一、什么是断点续传

断点续传顾名思义就是在文件下载中断时，一定时间段后通过已下载的数据点开始接着上次的下载进行下载，从而达到不用重头下载的效果，实现了文件下载的高效性。

二、实现断点续传的关键点

• 获取已下载的数据断点
• 根据数据断点设置请求头 Range
• 保存数据断点

三、实现思路

我们可以通过读取已下载的文件获取已下载数据长度，通过在http请求头中设置 Range 参数来请求剩余数据。

以下是实现代码：

const fs = require('fs');
const http = require('http');

const url = 'http://example.com';
const file = 'test.mp4';
const FILE_PATH = `./${file}`;

const download = (url, path, cb) => {
  let start = 0; // 断点续传的起始点
  if (fs.existsSync(FILE_PATH)) {
    const stat = fs.statSync(FILE_PATH);
    start = stat.size;
  }

  const options = {
    headers: {
      Range: `bytes=${start}-`,
    },
  };

  const fileStream = fs.createWriteStream(path, { flags: 'a' });
  http.get(url, options, (res) => {
    const total = parseInt(res.headers['content-length'], 10) + start;

    res.on('data', (chunk) => {
      fileStream.write(chunk);
    });

    res.on('end', () => {
      fileStream.end();
      cb(null, `下载完成，共下载 ${total} 字节数据`);
    });

    res.on('error', (err) => {
      cb(err.message);
    });
  });
};

download(url, FILE_PATH, (err, message) => {
  if (err) {
    console.error(`下载出错：${err}`);
  } else {
    console.log(message);
  }
});

四、示例说明

1. 下载文件

下载一个test.mp4文件

curl http://videos.example.com/test.mp4 --output test.mp4

2. 启动服务

npx http-server

3. 示例1：下载文件

下载完整文件

const url = 'http://127.0.0.1:8080/test.mp4';
const file = 'test.mp4';
const FILE_PATH = `./${file}`;
download(url, FILE_PATH, (err, message) => {
  if (err) {
    console.error(`下载出错：${err}`);
  } else {
    console.log(message);
  }
});

4. 示例2：断点续传下载文件

模拟100字节下载，已经下载了30字节，接着从30字节开始下载

const url = 'http://127.0.0.1:8080/test.mp4';
const file = 'test.mp4';
const FILE_PATH = `./${file}`;

fs.truncateSync(FILE_PATH, 30); // 截取文件，保留前30个字节
download(url, FILE_PATH, (err, message) => {
  if (err) {
    console.error(`下载出错：${err}`);
  } else {
    console.log(message);
  }
});

五、总结

以上就是 Node.js 断点续传的实现攻略。基本思路就是通过在 http 请求头中设置 Range 参数来实现断点续传。在代码中，我们通过 fs 文件系统模块来获取已经下载的数据长度，
并截取已经下载的文件，实现了文件的断点续传下载。