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

在前端开发中，文件上传是一个经常会遇到的需求。Node.js 提供了处理文件上传的强大工具，让文件上传变得轻松简单。本文将介绍如何在 Node.js 中处理文件上传。

准备工作

在开始处理文件上传之前，需要安装 multer 模块，它是一个用于处理 multipart/form-data 类型数据的中间件。可以使用 npm 包管理器进行安装：

npm install --save multer

基本用法

在 Express 中使用 multer 中间件非常简单。以下是一个基本的文件上传示例：

// javascriptcn.com 代码示例
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, Date.now() + '-' + file.originalname)
    }
})

const upload = multer({ storage: storage })

app.post('/uploadfile', upload.single('file'), (req, res, next) => {
    const file = req.file
    if (!file) {
        const error = new Error('Please upload a file')
        error.httpStatusCode = 400
        return next(error)
    }
    res.send(file)
})

上述示例使用 multer 中间件处理文件上传。首先定义了文件上传的目标路径和文件名生成规则。在 app.post 方法中，使用 upload.single('file') 处理单个文件的上传。最后，处理完上传之后，将上传的文件内容返回给请求方。

处理多个文件

如果需要处理多个文件的上传，可以使用 upload.array() 方法。以下是一个处理多个文件上传的示例：

// javascriptcn.com 代码示例
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, Date.now() + '-' + file.originalname)
    }
})

const upload = multer({ storage: storage })

app.post('/upload', upload.array('files', 10), (req, res, next) => {
    const files = req.files
    if (!files) {
        const error = new Error('Please upload a file')
        error.httpStatusCode = 400
        return next(error)
    }
    res.send(files)
})

上述示例使用 upload.array('files', 10) 处理多个文件的上传。'files' 是表单字段的名称，10 是最多可上传的文件数量。在最后将上传的文件内容返回给请求方。

处理文件类型

在上传文件的过程中，有时需要限制文件类型。可以使用 fileFilter 方法进行限制。

// javascriptcn.com 代码示例
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, Date.now() + '-' + file.originalname)
    }
})

const upload = multer({ 
    storage: storage,
    fileFilter: function (req, file, cb) {
        const allowedTypes = ['image/jpeg', 'image/png'];
        if (!allowedTypes.includes(file.mimetype)) {
            const error = new Error('Only JPEG and PNG images are allowed')
            error.httpStatusCode = 400
            return cb(error, false)
        }
        cb(null, true)
    }
})

app.post('/uploadfile', upload.single('file'), (req, res, next) => {
    const file = req.file
    if (!file) {
        const error = new Error('Please upload a file')
        error.httpStatusCode = 400
        return next(error)
    }
    res.send(file)
})

上述示例使用 fileFilter 方法限制上传文件的类型，在上传时，如果上传的文件类型不在指定的类型中，将返回错误信息。

总结

以上是在 Node.js 中处理文件上传的基本方法和技巧。使用 multer 模块可以轻松处理文件上传的功能。需要注意的是，在处理文件上传时，需要考虑多种情况，并对用户上传的内容进行校验，保证系统的安全性。

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