Koa2 之文件上傳下載的示例代碼

時間 2019-11-08

標籤 koa2 koa 文件上傳下載示例代碼简体版

原文原文鏈接

上傳下載在 web 應用中仍是比較常見的，不管是圖片仍是其餘文件等。在 Koa 中，有不少中間件能夠幫助咱們快速的實現功能。前端

文件上傳node

在前端中上傳文件，咱們都是經過表單來上傳，而上傳的文件，在服務器端並不能像普通參數同樣經過 ctx.request.body 獲取。咱們能夠用 koa-body 中間件來處理文件上傳，它能夠將請求體拼到 ctx.request 中。web

?服務器

1app

2dom

3koa

4async

5post

6ui

// app.js

const koa = require('koa');

const app = new koa();

const koaBody = require('koa-body');

app.use(koaBody({

multipart: true,

formidable: {

maxFileSize: 200*1024*1024 // 設置上傳文件大小最大限制，默認2M

}

}));

app.listen(3001, ()=>{

console.log('koa is listening in 3001');

})

使用中間件後，就能夠在 ctx.request.body.files 中獲取上傳的文件內容。須要注意的就是設置 maxFileSize，否則上傳文件一超過默認限制就會報錯。

接收到文件以後，咱們須要把文件保存到目錄中，返回一個 url 給前端。在 node 中的流程爲

建立可讀流 const reader = fs.createReadStream(file.path)
建立可寫流 const writer = fs.createWriteStream('upload/newpath.txt')
可讀流經過管道寫入可寫流 reader.pipe(writer)

const router = require('koa-router')();

const fs = require('fs');

router.post('/upload', async (ctx){

const file = ctx.request.body.files.file; // 獲取上傳文件

const reader = fs.createReadStream(file.path); // 建立可讀流

const ext = file.name.split('.').pop(); // 獲取上傳文件擴展名

const upStream = fs.createWriteStream(`upload/${Math.random().toString()}.${ext}`); // 建立可寫流

reader.pipe(upStream); // 可讀流經過管道寫入可寫流

return ctx.body = '上傳成功';

})

該方法適用於上傳圖片、文本文件、壓縮文件等。

文件下載

koa-send 是一個靜態文件服務的中間件，可用來實現文件下載功能。

const router = require('koa-router')();

const send = require('koa-send');

router.post('/download/:name', async (ctx){

const name = ctx.params.name;

const path = `upload/${name}`;

ctx.attachment(path);

await send(ctx, path);

})

在前端進行下載，有兩個方法： window.open 和表單提交。這裏使用簡單一點的 window.open 。

<button onclick="handleClick()">當即下載</button>

<script>

const handleClick = () => {

window.open('/download/1.png');

}

</script>

這裏 window.open 默認是開啓一個新的窗口，一閃而後關閉，給用戶的體驗並很差，能夠加上第二個參數 window.open('/download/1.png', '_self'); ，這樣就會在當前窗口直接下載了。然而這樣是將 url 替換當前的頁面，則會觸發 beforeunload 等頁面事件，若是你的頁面監聽了該事件作一些操做的話，那就有影響了。那麼還可使用一個隱藏的 iframe 窗口來達到一樣的效果。