如何实现上传文件到 nodejs 和文件下载

简介: 最近拿 next.js 做个全栈项目,需要文件上传和下载,这里记录下实现方式,也写一下使用原生 node 代码如何实现。

最近拿 next.js 做个全栈项目,需要文件上传和下载,这里记录下实现方式,也写一下使用原生 node 代码如何实现。

文件上传

要实现文件上传,首先需要在页面中拿到文件,demo 里我们简单的写一下:

<input type="file" id="file" />
复制代码
document.querySelector('#file').addEventListener('change', e => {
    if (e.target.files?.[0]) doUpload(e.target.files[0]);
});
复制代码

为了简化一下我们这里例子中使用单文件上传。再来实现下 doUpload,上传文件我们需要使用到 FormData

const doUpload = async file => {
    const body = new FormData();
    body.append('file', file);
    const response = await fetch('/api/file', {
        method: 'POST',
        body
    });
};
复制代码

我们需要先实例化一个 FormData,然后使用 append 将文件作为 form 字段添加进去,然后可以直接通过 fetch 发送 formDatanode 端,当然也可以使用 axios 等库或 xhr 来实现。

axios.post('/api/file', formData, {
    headers: {
        'Content-Type': 'multipart/form-data'
    }
});
复制代码

需要注意在使用 xhr 时要主动配置下请求头中的 Content-Type

这样在选择文件后就会将文件 post 发送到 node/api/file 中,然后我们再在做下 node 端处理下发送过来的请求:

import formidable from 'formidable';
import fs from 'fs';
export const config = {
    api: {
        bodyParser: false
    }
};
const saveFile = async file => {
    const data = fs.readFileSync(file.path);
    fs.writeFileSync(`./public/${file.name}`, data);
    await fs.unlinkSync(file.path);
    return;
};
export default (req, res) => {
    if (req.method === 'POST') {
        const form = new formidable.IncomingForm();
        form.parse(req, async function (err, fields, files) {
            await saveFile(files.file);
            return res.status(201).send('');
        });
        post(req, res);
    }
};
复制代码

由于篇幅有限,删除了一些异常处理代码。这里我们使用 formidable 来解析 FormData,注意由于 next.js 默认会启用 bodyParser,所以需要 config 中显示的禁用 bodyParser

formidable 会将文件放到一个临时目录中,我们直接从临时目录读取该文件然后存到指定目录即可,注意转存完成后使用 unlink 释放缓存文件,也可以直接使用 rename 移动文件位置。

await fs.renameAsync(file.path, `./public/${file.name}`);
复制代码

这样一个简单的文件上传功能就完成了,而如果使用原生 node 的话差别不大:

http.createServer(function (req, res) {
    if (req.url == '/api/file' && req.method === 'POST') {
        const form = new formidable.IncomingForm();
        form.parse(req, async function (err, fields, files) {
            await saveFile(files.file);
            return res.status(201).send('');
        });
        post(req, res);
    }
}).listen(3000);
复制代码

只是需要自己判断路由地址,实现代码并无差别。

文件下载

实现完了文件上传,我们再看下文件下载,文件下载其实只需要实现一个静态文件服务器即可,最简单的办法是通过流管道了:

export default (req, res) => {
    if (req.method === 'GET') {
        const filePath = path.join(process.env.FILE_PATH, req.url);
        const readStream = fs.createReadStream(filePath);
        readStream.pipe(res);
        readStream.on('finish', () => {
            readStream.close();
        });
    }
};
复制代码

我们直接将文件读取流通过管道接到相应中,即可完成文件下载。node 端代码和上面上传相同,只是做路径上的识别,别的并无差别。

此外我们还可以通过直接发送文件内容的方式来返回文件,下面为原生 node 代码示例:

const fileContent = fs.readFileSync(filePath);
res.writeHead(200, { 'Content-Type': contentType });
res.end(content, 'utf-8');
复制代码

而在页面上我们只需要使用 a 标签的 download,即可实现下载,也可以通过 window.open 等其它方式,不过要注意做好防拦截处理:

<a download="fileName.png" href="/node/static/fileName.png">点击下载</a>
复制代码

总结

node 端实现文件上传和下载的难度并不大,但是要额外注意的是安全隐患等问题,避免被恶意上传木马,这块后面会单独写一篇。

相关文章
|
30天前
|
JavaScript API
NodeJs——使用axios下载上传文件
NodeJs——使用axios下载上传文件
51 4
|
12月前
|
JavaScript 大数据 数据处理
在 Node.js 里使用 multiparty 上传文件
在 Node.js 里使用 multiparty 上传文件
|
27天前
|
缓存 JavaScript 安全
2022年最新最详细的安装Node.js以及cnpm(详细图解过程、绝对成功)
这篇文章提供了2022年最新最详细的Node.js和cnpm安装教程,包括步骤图解、全局配置路径、cnpm安装命令、nrm的安装与使用,以及如何管理npm源和测试速度。
2022年最新最详细的安装Node.js以及cnpm(详细图解过程、绝对成功)
|
16天前
|
JavaScript NoSQL 前端开发
|
24天前
|
JavaScript Linux
2022年超详细在CentOS 7上安装Node.js方法(源码安装)
这篇文章介绍了在CentOS 7系统上通过源码安装Node.js的详细步骤,包括从官网下载Node.js源码包、将安装包上传至虚拟机、解压安装包、删除压缩文件、编译安装Node.js、检查Node.js和npm版本,以及切换npm源到淘宝镜像以加速下载。此外,还提供了一个获取Linux下Node.js离线安装包的微信公众号搜索方式。
|
24天前
|
JavaScript
成功解决node、node-sass和sass-loader版本冲突问题、不需要降低node版本。如何在vue项目中安装node-sass,以及安装node-sass可能遇到的版本冲突问题
这篇文章介绍了在Vue项目中安装node-sass和sass-loader时遇到的版本冲突问题,并提供了解决这些问题的方法,包括在不降低node版本的情况下成功安装node-sass。
成功解决node、node-sass和sass-loader版本冲突问题、不需要降低node版本。如何在vue项目中安装node-sass,以及安装node-sass可能遇到的版本冲突问题
|
27天前
Mac 安装 Node Error: Could not symlink include/node/common.gypi
Mac 安装 Node Error: Could not symlink include/node/common.gypi
18 3
|
28天前
|
JavaScript
Node.js的安装
这篇文章提供了Node.js的安装指南,包括从官网下载、安装步骤、验证安装是否成功,以及如何安装淘宝镜像加速器cnpm或使用淘宝npm镜像来加速npm包的安装过程。
Node.js的安装
|
30天前
|
JavaScript 内存技术
NodeJs——nvm安装和使用
NodeJs——nvm安装和使用
54 0
NodeJs——nvm安装和使用
|
21天前
|
JavaScript 前端开发 Shell
mac和windows上安装nvm管理node版本
NVM(Node Version Manager)是前端开发者常用的命令行工具,用于管理计算机上的不同Node.js版本。通过NVM,开发者可以轻松地在多个项目间切换所需的Node.js版本。在Mac上,可以通过cURL或Wget下载安装脚本,或使用包管理工具brew安装。安装后需配置环境变量以识别NVM命令。Windows用户则可通过专用的nvm-windows安装程序完成安装。常用命令包括安装、卸载特定版本、列出已安装版本等。
57 0