在前端开发中，文件上传和下载是一个非常常见的需求。本文将介绍如何使用 Koa2 实现基本的文件上传和下载功能，包括如何设置路由、处理文件上传、保存文件和提供文件下载等。
环境准备
在开始之前，需要确保已经安装了 Node.js 和 Koa2。如果还没有安装，可以在官网下载安装包并按照步骤进行安装。
设置路由
首先，我们需要设置一个上传文件和下载文件的路由。可以使用 Koa2 的 koa-router 模块来设置路由，示例代码如下所示：
-- -------------------- ---- ------- ----- ------ - ---------------------- ----- ------ - --- --------- ---------------------- ----- ----- -- - -- ------ --- --------------------------------- ----- ----- -- - -- ------ --- -------------- - -------
这里创建了两个路由：一个是上传路由，使用 POST 方法，另一个是下载路由，使用 GET 方法。其中，下载路由使用了动态路由，将文件名作为路由参数传递。
处理文件上传
接下来，我们需要处理文件上传。可以使用 koa-body 模块来解析上传的表单数据，示例代码如下所示：
-- -------------------- ---- ------- ----- --- - --------------- ----- ------ - ---------------------- ----- ------- - -------------------- ----- --- - --- ------ ----- ------ - --- --------- ----------------- ---------- ---- ---- ------------------------- ---------------------- ----- ----- -- - ----- ---- - ----------------------- -------- - - --------- --------- -- ---
在这个示例中，我们使用了 koa-body 中间件来解析包含文件的 multipart/form-data 表单数据。当用户上传文件时，文件被包装在 request.files 对象中。我们可以从该对象中访问文件名、文件类型和文件内容等信息。
在这个示例中，我们只是简单地返回了接收到的文件名，但是实际上，我们需要将文件保存到服务器或者数据库中。
保存文件
在处理完文件上传之后，我们需要将文件保存到服务器或者数据库中。在这个示例中，我们将文件保存到服务器中的一个特定目录下，并为文件取一个唯一的文件名，示例代码如下所示：
-- -------------------- ---- ------- ----- -- - -------------- ----- ---- - ---------------- -------- -------------- - ----- ------ - ------------------------------- ----- ------------- - ------------------------ ----- -------------- - -------------------------------- ----- ------ - ----------------------------------------- ---------- ----------------- -------------------- ------ --------------- - ---------------------- ---------- -- - ----- ---- - ----------------------- ----- -------------- - --------------- -------- - - --------- -------------- -- ---
在这个示例中，我们使用了 Node.js 内置模块 fs 来读写文件。将上传的文件流保存到服务器的一个特定目录下（这个目录必须存在），并在文件名前为文件添加一个时间戳，以确保文件名的唯一性。
提供文件下载
最后，我们需要设置一个路由来提供下载功能。在这个示例中，我们将从服务器中的一个目录中读取文件流，并将其发送给客户端，示例代码如下所示：
-- -------------------- ---- ------- -------- ----------------------- - ------ ---------------------------------------- ---------- ----------- - --------------------------------- ----- ----- -- - ----- -------- - -------------------- ------------------------- -------- - ----- ------------------------ ---
在这个示例中，我们使用了 Koa2 的 attachment 方法来将文件下载的响应头设置为附件，这将提示用户保存文件而不是在浏览器中打开文件。我们使用 getFileStream 函数获取文件流并将其放入响应正文中。
总结
本文介绍了如何使用 Koa2 实现基本的文件上传和下载功能，包括如何设置路由、处理文件上传、保存文件和提供文件下载等。可以在实际开发中使用这种方法来上传和下载各种文件，提高开发效率。如果您已经学会了这种方法，那么您可以在开发中体验更多的技术，并提高您的技能水平。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/647d985c968c7c53b0862e89