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

相关文章
|
3月前
|
JavaScript API
NodeJs——使用axios下载上传文件
NodeJs——使用axios下载上传文件
129 4
|
2月前
|
JavaScript
NodeJs的安装
文章介绍了Node.js的安装步骤和如何创建第一个Node.js应用。包括从官网下载安装包、安装过程、验证安装是否成功,以及使用Node.js监听端口构建简单服务器的示例代码。
NodeJs的安装
|
1月前
|
JavaScript 开发工具 git
已安装nodejs但是安装hexo报错
已安装nodejs但是安装hexo报错
25 2
|
2月前
|
存储 JavaScript 前端开发
Node 版本控制工具 NVM 的安装和使用(Windows)
本文介绍了NVM(Node Version Manager)的Windows版本——NVM for Windows的安装和使用方法,包括如何安装Node.js的特定版本、列出已安装版本、切换使用不同版本的Node.js,以及其他常用命令,以实现在Windows系统上对Node.js版本的便捷管理。
Node 版本控制工具 NVM 的安装和使用(Windows)
|
1月前
|
Web App开发 JavaScript 前端开发
JavaWeb 22.Node.js_简介和安装
JavaWeb 22.Node.js_简介和安装
|
2月前
|
SQL JavaScript 数据库
sqlite在Windows环境下安装、使用、node.js连接
sqlite在Windows环境下安装、使用、node.js连接
|
2月前
|
JavaScript Linux 开发者
一个用于管理多个 Node.js 版本的安装和切换开源工具
【9月更文挑战第14天】nvm(Node Version Manager)是一个开源工具,用于便捷地管理多个 Node.js 版本。其特点包括:版本安装便捷,支持 LTS 和最新版本;版本切换简单,不影响开发流程;多平台支持,包括 Windows、macOS 和 Linux;社区活跃,持续更新。通过 nvm,开发者可以轻松安装、切换和管理不同项目的 Node.js 版本,提高开发效率。
|
1月前
|
JavaScript 算法 内存技术
如何降低node.js版本(nvm下载安装与使用)
如何降低node.js版本(nvm下载安装与使用)
|
3月前
|
缓存 JavaScript 安全
2022年最新最详细的安装Node.js以及cnpm(详细图解过程、绝对成功)
这篇文章提供了2022年最新最详细的Node.js和cnpm安装教程,包括步骤图解、全局配置路径、cnpm安装命令、nrm的安装与使用,以及如何管理npm源和测试速度。
2022年最新最详细的安装Node.js以及cnpm(详细图解过程、绝对成功)
|
2月前
|
JavaScript 前端开发 Linux