Node.js 文件下载三问：深入剖析文件下载机制

后端

2023-09-24 06:39:18

前言

在实际开发中，我们经常会遇到文件下载和上传的需求。在 Node.js 中，我们可以使用各种内置模块和第三方库来实现这些功能。本文将深入剖析 Node.js 中的文件下载机制，解答三个常见问题：

如何控制浏览器下载或打开文件？
如何实现文件分块下载？
如何使用 Node.js 处理文件上传？

如何控制浏览器下载或打开文件？

当我们在浏览器中点击文件的链接时，有时会直接在浏览器中打开文件，有时会下载到本地。作为服务提供方，我们如何控制响应？

浏览器会根据响应头部来判断如何处理文件。我们可以通过设置 Content-Disposition 响应头来控制浏览器的行为。Content-Disposition 头部的值可以是 inline 或 attachment。

inline 表示浏览器将直接在页面中打开文件。
attachment 表示浏览器将下载文件到本地。

例如，以下代码将导致浏览器下载文件：

res.setHeader('Content-Disposition', 'attachment; filename="file.txt"');

如何实现文件分块下载？

在某些情况下，我们需要将大文件分块下载。这可以减少内存的使用并提高下载速度。

在 Node.js 中，我们可以使用 stream 模块来实现文件分块下载。stream 模块提供了 createReadStream() 函数，可以将文件内容作为流输出。我们可以将文件内容分成多个块，然后将每个块作为单独的流输出。

以下代码演示了如何实现文件分块下载：

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

const file = 'large-file.txt';
const blockSize = 1024 * 1024; // 1MB

const readableStream = fs.createReadStream(file);
const chunks = [];

readableStream.on('data', (chunk) => {
  chunks.push(chunk);
});

readableStream.on('end', () => {
  const totalChunks = Math.ceil(chunks.length / blockSize);

  for (let i = 0; i < totalChunks; i++) {
    const start = i * blockSize;
    const end = start + blockSize;

    const chunkStream = new stream.PassThrough();
    chunkStream.end(Buffer.concat(chunks.slice(start, end)));

    // 发送文件块给客户端
  }
});

如何使用 Node.js 处理文件上传？

在 Node.js 中，我们可以使用 multer 库来处理文件上传。multer 库提供了多种方法来处理文件上传，包括：

单个文件上传
多个文件上传
文件大小限制
文件类型限制
文件存储位置

以下代码演示了如何使用 multer 库处理文件上传：

const multer = require('multer');

const storage = multer.diskStorage({
  destination: './uploads/',
  filename: (req, file, cb) => {
    cb(null, Date.now() + '-' + file.originalname);
  },
});

const upload = multer({ storage: storage });

app.post('/upload', upload.single('file'), (req, res) => {
  res.send('File uploaded successfully.');
});