解决 Express.js 中文件上传的 IO 问题

在 Web 开发中，文件上传是常见的需求之一。而 Express.js 作为流行的 Node.js Web 框架，也提供了文件上传的中间件 multer。然而，在上传大文件时，可能会遇到 IO 问题，导致服务器响应变慢或者崩溃。本文将介绍如何解决 Express.js 中文件上传的 IO 问题。

IO 问题的原因

IO 问题通常是由于服务器在上传文件时，需要将文件内容全部读入内存，然后再写入磁盘。当上传大文件时，这将导致内存占用过高，甚至可能导致服务器崩溃。因此，解决 IO 问题的关键在于减少内存占用。

解决方案

使用流式上传

解决 IO 问题的一种常见方法是使用流式上传。流式上传是指在上传文件时，将文件内容分块读入内存，然后分块写入磁盘，以减少内存占用。

在 Express.js 中，可以使用 multer 的 stream() 方法实现流式上传。下面是一个示例代码：

const express = require('express');
const multer = require('multer');
const app = express();

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

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

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

app.listen(3000, () => {
  console.log('Server is running on port 3000');
});

在上面的示例代码中，我们使用了 multer 的 diskStorage() 方法将上传的文件保存到本地磁盘中。然后，使用 multer 的 single() 方法指定上传的文件字段名为 file。最后，在路由处理函数中，我们可以通过 req.file 属性访问上传的文件信息。

使用流式读写

除了使用 multer 提供的流式上传方法外，我们还可以使用 Node.js 的流式读写来实现流式上传。在 Node.js 中，可以使用 fs.createReadStream() 和 fs.createWriteStream() 方法来创建可读和可写流。下面是一个示例代码：

const express = require('express');
const fs = require('fs');
const app = express();

app.post('/upload', (req, res) => {
  const filePath = './uploads/' + Date.now() + '-' + req.headers['x-file-name'];
  const fileStream = fs.createWriteStream(filePath);

  req.pipe(fileStream);

  fileStream.on('close', () => {
    res.send('File uploaded successfully!');
  });
});

app.listen(3000, () => {
  console.log('Server is running on port 3000');
});

在上面的示例代码中，我们使用了 fs.createWriteStream() 方法创建了一个可写流，并将其与 req 对象相连接。这样，每当有数据写入 req 对象时，就会自动写入到可写流中。最后，当可写流关闭时，我们可以通过 res.send() 方法向客户端发送上传成功的消息。

总结

在本文中，我们介绍了如何解决 Express.js 中文件上传的 IO 问题。通过使用流式上传和流式读写，我们可以减少内存占用，避免服务器响应变慢或者崩溃。希望本文对你有所帮助。

来源：JavaScript中文网 ，转载请注明来源 本文地址：https://www.javascriptcn.com/post/658ab418eb4cecbf2dff5744