文件上传之大文件分块上传进度控制处理

在分块上传内容结束以后的事件监听，我们会实现 unlinkSync 删除临时文件操作，那么试想一下，在这个事件监听中，我们是否可以通过totalChunks以及currentChunk获取当前上传的进度情况呢？

后端

upload上传接口：

app.post('/upload',upload.single('file'),(req,res) => {
    const file = req.file; 
    const filename = req.body.filename;
    const totalChunks = parseInt(req.body.totalChunks);    // 获取总块数
    const currentChunk = parseInt(req.body.currentChunk);  // 获取当前块数
    const chunkPath = path.join(
        "uploads/",
        `${filename}-chunk-${currentChunk}`
    ); 
 
    const chunkStream = fs.createReadStream(file.path);
    const writeStream = fs.createWriteStream(chunkPath);
 
    chunkStream.pipe(writeStream);
 
    // 对分块上传内容结束以后的事件监听
    chunkStream.on("end", () => {
        fs.unlinkSync(file.path); //读取文件块的流结束后，删除临时文件

        // 打印 progress 我们就可以获取到文件进度情况
        const progress = ((currentChunk + 1) / totalChunks) * 100;  // 计算上传进度情况
        res.json({ progress }); // 服务器端向客户端进行返回操作
    });
});

前端

html：

<input type="file" id="fileInput">
<progress value="0" max="100" id="progress"></progress>  //设置一个进度条
<button onclick="upload()">上传文件</button>

script：

async function upload() {
    const fileInput = document.getElementById('fileInput'); //获取input框
    const file = fileInput.files[0]; // 对input的文件进行获取
 
    const chunkSize = 1*1024*1024;  //初始化分块的尺寸 每块分块文件大小为1MB（1兆）
    const totalChunks = Math.ceil(file.size / chunkSize); //通过文件尺寸计算出所有的块数
    let currentChunk = 0; //设置块的初始值
 
    // 通过while循环处理
    while (currentChunk < totalChunks){
        const start = currentChunk * chunkSize;  // 计算当前块的起始位置
        const end = Math.min(start + chunkSize, file.size); // 计算当前块的结束  Math.min：返回一组数值中的最小值
        const chunk = file.slice(start, end); // 切割文件获取当前块
 
        const formData = new FormData();
        formData.append('file', chunk); // 添加当前块到 FormData 对象
        formData.append('filename', file.name); // 添加文件名到 FormData 对象
        formData.append('totalChunks', totalChunks); // 添加总块数到 FormData 对象
        formData.append('currentChunk', currentChunk); // 添加当前块数到 FormData 对象
    
        try{
            // 获取返回的内容（进度）
            const res = await axios.post(
                'http://localhost:3000/upload',
                formData,
                {
                    headers:{
                        'Content-Type':'multipart/form-data',
                    },
                }
            ); //发送当前块的上传请求

            const { progress } = res.data;  // 获取当前块的上传进度
            document.getElementById('progress').value = progress; // 更新进度
    
        }catch(error){
            console.error(error);
            return;
        }
 
        currentChunk++; //增加当前块数，继续下一块的上传（实现循环操作）
    }
 
    // 当所有分块文件发送完毕，发起合并请求操作
    try{
        const postData = { filename:file.name,totalChunks:totalChunks }; //构造合并请求的数据
        await http.post('http://localhost:3000/merge', postData,{
            headers: {
            'Content-Type': 'application/json'
            }
        }); //发送合并请求
    }catch(error){
        console.error(error);
    }
}

再次发送文件，进度条已经正常显示：