Node.js 中如何处理大文件上传？

在前端开发中，文件上传是非常常见的操作之一。但是，当需要上传大文件时，就需要考虑如何处理这个问题。Node.js 提供了一些解决方案，本文将介绍如何使用 Node.js 处理大文件上传。

什么是大文件上传？

大文件上传通常指的是文件大小超过几百 MB 的文件。这种情况下，直接使用前端的默认上传方式会导致上传速度慢、占用大量内存等问题，甚至可能导致浏览器崩溃。

如何处理大文件上传？

Node.js 中可以使用以下方案处理大文件上传：

1. 使用第三方库

Node.js 中有很多第三方库可以用来处理大文件上传，如 multer、formidable 等。这些库可以帮助我们轻松地处理大文件上传，并提供了一些方便的选项，如限制上传文件大小、支持多文件上传等。

下面是使用 multer 处理大文件上传的示例代码：

// javascriptcn.com 代码示例
const multer = require('multer');

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

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

app.post('/upload', upload.single('file'), function (req, res, next) {
  // 文件上传成功
});

2. 分块上传

另一种处理大文件上传的方法是分块上传。将大文件分成多个小块，分别上传，上传完成后再将这些小块合并成一个完整的文件。这种方法可以避免上传速度慢、占用大量内存等问题。

下面是使用分块上传处理大文件上传的示例代码：

// javascriptcn.com 代码示例
const fs = require('fs');

// 分块上传
function uploadChunk(filePath, data, callback) {
  fs.open(filePath, 'a', function(err, fd) {
    if (err) {
      callback(err);
    } else {
      fs.write(fd, data, 0, data.length, null, function(err) {
        if (err) {
          callback(err);
        } else {
          fs.close(fd, function() {
            callback(null);
          });
        }
      });
    }
  });
}

// 合并文件
function mergeFiles(filePath, chunkSize, callback) {
  const chunks = Math.ceil(fs.statSync(filePath).size / chunkSize);
  const chunkPaths = Array.from({ length: chunks }, (v, i) => `${filePath}-${i}`);

  const writeStream = fs.createWriteStream(filePath);

  function next(i) {
    if (i < chunks) {
      const readStream = fs.createReadStream(chunkPaths[i]);
      readStream.pipe(writeStream, { end: false });
      readStream.on('end', function() {
        next(i + 1);
      });
    } else {
      writeStream.end();
      callback(null);
    }
  }

  next(0);
}

// 文件上传
app.post('/upload', function(req, res, next) {
  const filePath = 'uploads/' + req.body.filename;
  const chunkSize = 1024 * 1024;

  uploadChunk(filePath + '-' + req.body.index, req.body.data, function(err) {
    if (err) {
      next(err);
    } else {
      if (req.body.index == req.body.total - 1) {
        mergeFiles(filePath, chunkSize, function(err) {
          if (err) {
            next(err);
          } else {
            res.send('上传成功');
          }
        });
      } else {
        res.send('上传中');
      }
    }
  });
});

总结

本文介绍了 Node.js 中如何处理大文件上传的两种方法：使用第三方库和分块上传。分块上传相比直接上传可以避免上传速度慢、占用大量内存等问题，但需要额外处理文件合并的逻辑。选择哪种方法取决于具体情况，开发者需要根据自己的需求选择合适的方法。

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