Node.js上传文件的方式有多种，包括使用Express框架、Multer中间件、以及原生Node.js的方式。 其中，Express框架结合Multer中间件是最常用且简便的方法。Multer是一个Node.js中间件，用于处理multipart/form-data类型的表单数据，主要用于上传文件。下面，我们将详细介绍这两种方式，并提供相应的代码示例。
一、使用Express和Multer上传文件
1、安装Express和Multer
首先，我们需要安装Express和Multer。你可以使用npm来进行安装：
npm install express multer
2、设置Express应用
接下来，我们需要在Express应用中设置Multer中间件，以处理文件上传。
const express = require('express');
const multer = require('multer');
const path = require('path');
const app = express();
const port = 3000;
// 设置文件存储位置和文件命名方式
const storage = multer.diskStorage({
destination: function (req, file, cb) {
cb(null, 'uploads/');
},
filename: function (req, file, cb) {
cb(null, file.fieldname + '-' + Date.now() + path.extname(file.originalname));
}
});
// 创建Multer实例并设置限制
const upload = multer({
storage: storage,
limits: { fileSize: 1000000 } // 限制文件大小为1MB
});
// 设置上传文件的路由
app.post('/upload', upload.single('file'), (req, res) => {
try {
res.send('文件上传成功');
} catch (err) {
res.sendStatus(400);
}
});
app.listen(port, () => {
console.log(`服务器在http://localhost:${port}上运行`);
});
在上述代码中，我们设置了文件上传的存储位置为uploads/文件夹，并对文件进行了重命名，以防止文件名冲突。同时，我们限制了文件的大小为1MB。
3、前端HTML表单
我们还需要一个前端HTML表单来上传文件。你可以创建一个简单的HTML文件，如下所示：
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>文件上传</title>
</head>
<body>
<form action="http://localhost:3000/upload" method="POST" enctype="multipart/form-data">
<input type="file" name="file" />
<button type="submit">上传文件</button>
</form>
</body>
</html>
二、使用原生Node.js上传文件
尽管使用Express和Multer是最常见的方法，但在某些情况下，你可能会希望使用原生的Node.js来上传文件。下面是一个简单的示例，展示如何使用原生Node.js处理文件上传。
1、创建HTTP服务器
首先，我们需要创建一个HTTP服务器来处理文件上传请求。
const http = require('http');
const formidable = require('formidable');
const fs = require('fs');
const path = require('path');
const server = http.createServer((req, res) => {
if (req.url == '/upload' && req.method.toLowerCase() == 'post') {
// 创建一个Formidable表单解析实例
const form = new formidable.IncomingForm();
form.uploadDir = 'uploads/';
form.keepExtensions = true;
form.parse(req, (err, fields, files) => {
if (err) {
res.writeHead(400, { 'Content-Type': 'text/plain' });
res.end('文件上传出错');
return;
}
const oldPath = files.file.path;
const newPath = path.join(form.uploadDir, files.file.name);
fs.rename(oldPath, newPath, (err) => {
if (err) {
res.writeHead(500, { 'Content-Type': 'text/plain' });
res.end('文件重命名出错');
return;
}
res.writeHead(200, { 'Content-Type': 'text/plain' });
res.end('文件上传成功');
});
});
} else {
// 显示文件上传表单
res.writeHead(200, { 'Content-Type': 'text/html' });
res.end(`
<form action="/upload" enctype="multipart/form-data" method="post">
<input type="file" name="file" />
<button type="submit">上传文件</button>
</form>
`);
}
});
server.listen(3000, () => {
console.log('服务器在http://localhost:3000上运行');
});
在上述代码中，我们使用了formidable模块来处理文件上传。Formidable是一个Node.js模块，用于解析表单数据，尤其是文件上传。
三、文件上传后的处理
1、文件验证
在上传文件之前，我们可以对文件进行验证，比如检查文件类型和大小。这可以防止用户上传不合法的文件。
const upload = multer({
storage: storage,
fileFilter: function (req, file, cb) {
// 检查文件类型
const filetypes = /jpeg|jpg|png/;
const mimetype = filetypes.test(file.mimetype);
const extname = filetypes.test(path.extname(file.originalname).toLowerCase());
if (mimetype && extname) {
return cb(null, true);
} else {
cb('Error: 文件类型不支持');
}
},
limits: { fileSize: 1000000 } // 限制文件大小为1MB
});
2、存储到数据库
在某些情况下，你可能需要将文件元数据存储到数据库中。以下是一个使用MongoDB的示例。
const mongoose = require('mongoose');
mongoose.connect('mongodb://localhost:27017/fileuploads', { useNewUrlParser: true, useUnifiedTopology: true });
const fileSchema = new mongoose.Schema({
filename: String,
path: String,
size: Number,
mimetype: String,
uploadedAt: { type: Date, default: Date.now }
});
const File = mongoose.model('File', fileSchema);
app.post('/upload', upload.single('file'), (req, res) => {
const fileData = new File({
filename: req.file.filename,
path: req.file.path,
size: req.file.size,
mimetype: req.file.mimetype
});
fileData.save((err) => {
if (err) {
res.status(500).send('文件存储到数据库出错');
return;
}
res.send('文件上传并存储到数据库成功');
});
});
四、文件下载与删除
1、文件下载
我们可以通过文件路径来提供文件下载服务。
app.get('/download/:filename', (req, res) => {
const filePath = path.join(__dirname, 'uploads', req.params.filename);
res.download(filePath, (err) => {
if (err) {
res.status(404).send('文件未找到');
}
});
});
2、文件删除
我们可以通过文件路径来删除文件。
app.delete('/delete/:filename', (req, res) => {
const filePath = path.join(__dirname, 'uploads', req.params.filename);
fs.unlink(filePath, (err) => {
if (err) {
res.status(404).send('文件未找到');
return;
}
res.send('文件删除成功');
});
});
五、文件上传的安全性
1、避免路径遍历攻击
我们需要确保用户上传的文件不会覆盖服务器上的重要文件。使用path.join和path.normalize可以防止路径遍历攻击。
const safePath = path.join(__dirname, 'uploads', path.normalize(req.params.filename));
2、文件类型验证
验证文件类型可以防止用户上传恶意文件。上文中已经展示了如何在Multer中验证文件类型。
3、文件大小限制
限制文件大小可以防止用户上传过大的文件，消耗服务器资源。上文中已经展示了如何在Multer中限制文件大小。
通过以上的方法，你可以在Node.js应用中实现文件上传、验证、存储、下载和删除功能。同时，要注意文件上传的安全性，防止路径遍历攻击和恶意文件上传。推荐使用研发项目管理系统PingCode和通用项目协作软件Worktile来管理你的项目和文件，这些工具可以帮助你更好地组织和协作。
相关问答FAQs：
1. 如何在Node.js中实现文件上传？
Node.js提供了多种方法来实现文件上传。你可以使用内置的http模块或者第三方模块，比如express来处理文件上传。以下是一个基本的示例：
const http = require('http');
const fs = require('fs');
http.createServer((req, res) => {
if (req.method === 'POST') {
// 处理文件上传
const file = fs.createWriteStream('uploads/uploaded_file.txt');
req.pipe(file);
req.on('end', () => {
res.end('文件上传成功！');
});
} else {
res.end('请使用POST请求上传文件。');
}
}).listen(3000, () => {
console.log('服务器已启动，监听端口3000');
});
2. 如何限制文件上传的大小和类型？
为了限制文件上传的大小和类型，你可以在文件上传之前进行验证。以下是一个示例：
const express = require('express');
const multer = require('multer');
const upload = multer({
dest: 'uploads/', // 指定文件上传的目录
limits: {
fileSize: 1024 * 1024, // 限制文件大小为1MB
},
fileFilter: (req, file, cb) => {
if (file.mimetype === 'image/jpeg' || file.mimetype === 'image/png') {
cb(null, true); // 允许上传jpeg和png格式的图片
} else {
cb(new Error('只允许上传jpeg和png格式的图片。'));
}
},
});
const app = express();
app.post('/upload', upload.single('file'), (req, res) => {
res.send('文件上传成功！');
});
app.listen(3000, () => {
console.log('服务器已启动，监听端口3000');
});
3. 如何处理多个文件的同时上传？
如果你需要处理多个文件的同时上传，可以使用multer模块的array方法或者fields方法。以下是一个示例：
const express = require('express');
const multer = require('multer');
const upload = multer({ dest: 'uploads/' });
const app = express();
app.post('/upload', upload.array('files', 5), (req, res) => {
res.send('文件上传成功！');
});
app.listen(3000, () => {
console.log('服务器已启动，监听端口3000');
});
在上述示例中，upload.array('files', 5)表示接受名为files的文件数组，最多允许上传5个文件。你也可以使用upload.fields方法来指定多个字段以及每个字段允许上传的文件数量。
文章包含AI辅助创作，作者：Edit1，如若转载，请注明出处：https://docs.pingcode.com/baike/3795248
