Node 程序怎么实现断点续传

Node程序实现断点续传的机制通常依靠文件切片、HTTP协议中的Range头部、本地记录已上传文件信息等元素。具体而言，文件在传输前会被分割成多个小的数据块，上传时通过记录每个数据块的上传状态，即可在中断后重新开始未完成部分的传输。另外，使用HTTP协议中的Range头部，可以指定请求获取资源的某个部分，这样即使连接断开，下次请求可以从上次的终点开始，而非从头开始上传或下载。

一、文件切片与数据记录

为了实现断点续传，需要将大文件切割成若干小文件片段。每个片段上传前都会在本地记录其相关信息，例如偏移量、已上传的大小等，如此一来，即便上传过程中发生了中断，也能通过这些记录找到中断点，并从该点继续上传。

在Node.js中实现文件切片通常采用fs模块，通过该模块提供的文件读写接口，可以控制文件读取的起止位置。

const fs = require('fs');
const path = require('path');
const FILE_PATH = 'path/to/your/large/file';
const CHUNK_SIZE = 1024 * 1024; // 比如1MB大小
let transferStatus = []; // 用于记录每个文件片段的传输状态
// 文件切片
function splitFile(filePath) {
  const fileSize = fs.statSync(filePath).size;
  const chunks = [];
  for(let i = 0; i < fileSize; i += CHUNK_SIZE) {
    chunks.push({
      filePath,
      start: i,
      end: Math.min(fileSize, i + CHUNK_SIZE),
      chunkId: i / CHUNK_SIZE
    });
  }
  return chunks;
}
// 模拟上传过程中的记录
function uploadChunks(chunks) {
  chunks.forEach(chunk => {
    // 上传每个chunk，并更新transferStatus
    // 该部分通常涉及到网络请求
    // ...
  });
}
// 从记录继续上传
function resumeUpload() {
  const incompleteChunks = transferStatus.filter(status => !status.completed);
  uploadChunks(incompleteChunks);
}

二、使用HTTP Range请求

HTTP中的Range头部使得可以仅请求资源的一部分，这对于断点续传是至关重要的。客户端可以通过发送带有Range头部的HTTP请求，告诉服务器它需要文件的哪一部分。

在Node.js的http模块或是像Axios这样的第三方库中，可以轻松地设置这些头部信息，以支持断点续传功能。

const http = require('http');
const fs = require('fs');
// 服务器端实现支持Range请求
http.createServer((req, res) => {
  // 获取Range头部信息
  const range = req.headers.range;
  if (range) {
    const positions = range.replace(/bytes=/, '').split('-');
    const start = parseInt(positions[0], 10);
    // 设置响应状态码为206，表示部分内容
    res.writeHead(206, {
      'Content-Range': 'bytes ' + start + '-' + (fileSize - 1) + '/' + fileSize,
      // 其他响应头部...
    });
    // 创建文件读取流，并通过管道返回客户端
    const stream = fs.createReadStream(FILE_PATH, { start, end: fileSize - 1 });
    stream.pipe(res);
  } else {
    res.writeHead(200, {
      // 响应头部...
    });
    fs.createReadStream(FILE_PATH).pipe(res);
  }
})
.listen(8000);
// 客户端发送带Range头部的请求
const axios = require('axios');
const URL = 'http://localhost:8000/';
const startByte = getLastSuccessfulByte(); // 从记录位置获取上次成功传输的最后一个字节
axios({
  method: 'get',
  url: URL,
  headers: {
    'Range': `bytes=${startByte}-`
  },
  // 其他配置...
}).then(response => {
  // 处理收到的文件片段
});

上述代码展示了如何在服务端处理带有Range头的请求，并码示客户端如何发送一个带有Range头的请求。服务端将返回请求的文件片段。客户端发送请求时，根据已有的传输记录设定开始字节。这样，即使在传输过程中发生断开，也可以通过该机制继续传输，而无需重新开始。

三、断点续传的数据管理

客户端的本地存储或数据库通常用来跟踪每次上传的状态。这样，在上传过程中断后，理论上可以从中断处恢复上传过程，而不是从头开始。一些常用的Node.js数据库模块，例如MongoDB、低级JSON数据库如lowdb、或者简单的文件系统，可用于存储这类信息。

const lowdb = require('lowdb');
const FileSync = require('lowdb/adapters/FileSync');
// 创建一个本地数据库文件来存储上传状态
const adapter = new FileSync('db.json');
const db = lowdb(adapter);
db.defaults({ uploadedChunks: [] }).write();
// 更新数据库中的上传状态
function updateUploadStatus(chunkId, completed) {
  db.get('uploadedChunks')
    .push({ chunkId, completed })
    .write();
}
// 获取上传状态
function getUploadedChunks() {
  return db.get('uploadedChunks').value();
}

上述代码片段展示了如何使用lowdb（一个非常轻量级的本地JSON数据库）来存储每块文件的上传状态。每次上传成功后，会更新数据库，以便未来可能发生的断点续传。

四、综合实例与错误处理

实现断点续传的完整示例需要综合上述各个部分。此外，实现中还应包括异常处理、网络波动应对、重试逻辑处理等。

错误处理至关重要，例如若网络出现问题，应有策略重试失败的传输。此外，要考虑服务器端验证（如校验上传的片段的完整性）和客户端的文件完整性验证（比如通过文件的MD5等哈希值比较）。

// 简化版的综合示例 - 客户端上传逻辑
async function uploadFileWithResume(filePath) {
  const chunks = splitFile(filePath); // 1. 文件切片
  for (let i = 0; i < chunks.length; i++) {
    const chunk = chunks[i];
    try {
      // 2. 尝试上传片段，并更新数据库状态
      awAIt uploadChunk(chunk);
      updateUploadStatus(chunk.chunkId, true);
    } catch (error) {
      // 3. 出错时记录失败状态，根据重试策略可能重新尝试
      updateUploadStatus(chunk.chunkId, false);
      handleUploadError(error);
      break; // 根据逻辑决定是中断还是进行重试
    }
  }
}
function handleUploadError(error) {
  // 4. 处理上传错误，根据需求实现重试逻辑、通知用户等
  console.error('Upload error:', error);
}
// 启动断点续传
resumeUpload();