JavaScript實現大文件分片上傳處理
很多時候我們在處理文件上傳時，如視頻文件，小則幾十M，大則 1G+，以一般的HTTP請求發送數據的方式的話，會遇到的問題:
1、文件過大，超出服務端的請求大小限制；
2、請求時間過長，請求超時；
3、傳輸中斷，必須重新上傳導致前功盡棄
這些問題很影響用戶的體驗感，所以下面介紹一種基於原生JavaScript進行文件分片處理上傳的方案，具體實現過程如下：
1、通過dom獲取文件對象，並且對文件進行MD5加密(文件內容+文件標題形式)，采用SparkMD5進行文件加密；
2、進行分片設置，文件File基於Blob, 繼承瞭Blob的功能，可以把File當成Blob的子類，利於Blob的slice方法進行文件分片處理，並且依次進行上傳
3、分片文件上傳完成後，請求合並接口後端進行文件合並處理即可
1. 上傳文件頁面
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<meta http-equiv="X-UA-Compatible" content="ie=edge">
<title>文件上傳</title>
<script src="https://cdn.bootcss.com/axios/0.18.0/axios.min.js"></script>
<script src="https://code.jquery.com/jquery-3.4.1.js"></script>
<script src="https://cdnjs.cloudflare.com/ajax/libs/spark-md5/3.0.0/spark-md5.js"></script>
<style>
/* 自定義進度條樣式 */
.precent input[type=range] {
-webkit-appearance: none;
/*清除系統默認樣式*/
width: 7.8rem;
/* background: -webkit-linear-gradient(#ddd, #ddd) no-repeat, #ddd; */
/*設置左邊顏色為#61bd12，右邊顏色為#ddd*/
background-size: 75% 100%;
/*設置左右寬度比例*/
height: 0.6rem;
/*橫條的高度*/
border-radius: 0.4rem;
border: 1px solid #ddd;
box-shadow: 0 0 10px rgba(0,0,0,.125) inset ;
}
/*拖動塊的樣式*/
.precent input[type=range]::-webkit-slider-thumb {
-webkit-appearance: none;
/*清除系統默認樣式*/
height: .9rem;
/*拖動塊高度*/
width: .9rem;
/*拖動塊寬度*/
background: #fff;
/*拖動塊背景*/
border-radius: 50%;
/*外觀設置為圓形*/
border: solid 1px #ddd;
/*設置邊框*/
}
</style>
</head>
<body>
<h1>大文件分片上傳測試</h1>
<div>
<input id="file" type="file" name="avatar" />
<div style="padding: 10px 0;">
<input id="submitBtn" type="button" value="提交" />
<input id="pauseBtn" type="button" value="暫停" />
</div>
<div class="precent">
<input type="range" value="0" /><span id="precentVal">0%</span>
</div>
</div>
<script type="text/javascript" src="./js/index.js"></script>
</body>
</html>
2. 大文件分片上傳處理
$(document).ready(() => {
const submitBtn = $('#submitBtn'); //提交按鈕
const precentDom = $(".precent input")[0]; // 進度條
const precentVal = $("#precentVal"); // 進度條值對應dom
const pauseBtn = $('#pauseBtn'); // 暫停按鈕
// 每個chunk的大小，設置為1兆
const chunkSize = 1 * 1024 * 1024;
// 獲取slice方法，做兼容處理
const blobSlice = File.prototype.slice || File.prototype.mozSlice || File.prototype.webkitSlice;
// 對文件進行MD5加密(文件內容+文件標題形式)
const hashFile = (file) => {
return new Promise((resolve, reject) => {
const chunks = Math.ceil(file.size / chunkSize);
let currentChunk = 0;
const spark = new SparkMD5.ArrayBuffer();
const fileReader = new FileReader();
function loadNext() {
const start = currentChunk * chunkSize;
const end = start + chunkSize >= file.size ? file.size : start + chunkSize;
fileReader.readAsArrayBuffer(blobSlice.call(file, start, end));
}
fileReader.onload = e => {
spark.append(e.target.result); // Append array buffer
currentChunk += 1;
if (currentChunk < chunks) {
loadNext();
} else {
console.log('finished loading');
const result = spark.end();
// 通過內容和文件名稱進行md5加密
const sparkMd5 = new SparkMD5();
sparkMd5.append(result);
sparkMd5.append(file.name);
const hexHash = sparkMd5.end();
resolve(hexHash);
}
};
fileReader.onerror = () => {
console.warn('文件讀取失敗！');
};
loadNext();
}).catch(err => {
console.log(err);
});
}
// 提交
submitBtn.on('click', async () => {
var pauseStatus = false;
var nowUploadNums = 0
// 1.讀取文件
const fileDom = $('#file')[0];
const files = fileDom.files;
const file = files[0];
if (!file) {
alert('沒有獲取文件');
return;
}
// 2.設置分片參數屬性、獲取文件MD5值
const hash = await hashFile(file); //文件 hash
const blockCount = Math.ceil(file.size / chunkSize); // 分片總數
const axiosPromiseArray = []; // axiosPromise數組
// 文件上傳
const uploadFile = () => {
const start = nowUploadNums * chunkSize;
const end = Math.min(file.size, start + chunkSize);
// 構建表單
const form = new FormData();
// blobSlice.call(file, start, end)方法是用於進行文件分片
form.append('file', blobSlice.call(file, start, end));
form.append('index', nowUploadNums);
form.append('hash', hash);
// ajax提交 分片，此時 content-type 為 multipart/form-data
const axiosOptions = {
onUploadProgress: e => {
nowUploadNums++;
// 判斷分片是否上傳完成
if (nowUploadNums < blockCount) {
setPrecent(nowUploadNums, blockCount);
uploadFile(nowUploadNums)
} else {
// 4.所有分片上傳後，請求合並分片文件
axios.all(axiosPromiseArray).then(() => {
setPrecent(blockCount, blockCount); // 全部上傳完成
axios.post('/file/merge_chunks', {
name: file.name,
total: blockCount,
hash
}).then(res => {
console.log(res.data, file);
pauseStatus = false;
alert('上傳成功');
}).catch(err => {
console.log(err);
});
});
}
},
};
// 加入到 Promise 數組中
if (!pauseStatus) {
axiosPromiseArray.push(axios.post('/file/upload', form, axiosOptions));
}
}
// 設置進度條
function setPrecent(now, total) {
var prencentValue = ((now / total) * 100).toFixed(2)
precentDom.value = prencentValue
precentVal.text(prencentValue + '%')
precentDom.style.cssText = `background:-webkit-linear-gradient(top, #059CFA, #059CFA) 0% 0% / ${prencentValue}% 100% no-repeat`
}
// 暫停
pauseBtn.on('click', (e) => {
pauseStatus = !pauseStatus;
e.currentTarget.value = pauseStatus ? '開始' : '暫停'
if (!pauseStatus) {
uploadFile(nowUploadNums)
}
})
uploadFile();
});
})
3. 文件上傳和合並分片文件接口（node）
const Router = require('koa-router');
const multer = require('koa-multer');
const fs = require('fs-extra');
const path = require('path');
const router = new Router();
const { mkdirsSync } = require('../utils/dir');
const uploadPath = path.join(__dirname, 'upload');
const chunkUploadPath = path.join(uploadPath, 'temp');
const upload = multer({ dest: chunkUploadPath });
// 文件上傳接口
router.post('/file/upload', upload.single('file'), async (ctx, next) => {
const { index, hash } = ctx.req.body;
const chunksPath = path.join(chunkUploadPath, hash, '/');
if(!fs.existsSync(chunksPath)) mkdirsSync(chunksPath);
fs.renameSync(ctx.req.file.path, chunksPath + hash + '-' + index);
ctx.status = 200;
ctx.res.end('Success');
})
// 合並分片文件接口
router.post('/file/merge_chunks', async (ctx, next) => {
const { name, total, hash } = ctx.request.body;
const chunksPath = path.join(chunkUploadPath, hash, '/');
const filePath = path.join(uploadPath, name);
// 讀取所有的chunks
const chunks = fs.readdirSync(chunksPath);
// 創建存儲文件
fs.writeFileSync(filePath, '');
if(chunks.length !== total || chunks.length === 0) {
ctx.status = 200;
ctx.res.end('切片文件數量不符合');
return;
}
for (let i = 0; i < total; i++) {
// 追加寫入到文件中
fs.appendFileSync(filePath, fs.readFileSync(chunksPath + hash + '-' +i));
// 刪除本次使用的chunk
fs.unlinkSync(chunksPath + hash + '-' +i);
}
fs.rmdirSync(chunksPath);
// 文件合並成功，可以把文件信息進行入庫。
ctx.status = 200;
ctx.res.end('Success');
})
以上就是文件分片上傳的基本過程，過程中加入瞭上傳進度條、暫停和開始上傳操作，見詳細代碼
以上就是本文的全部內容，希望對大傢的學習有所幫助，也希望大傢多多支持WalkonNet。
推薦閱讀：
- wasm+js實現文件獲取md5示例詳解
- Vue+NodeJS實現大文件上傳的示例代碼
- 客戶端JavaScript的線程池設計詳解
- 基於React-Dropzone開發上傳組件功能(實例演示)
- 如何基於js管理大文件上傳及斷點續傳詳析