前端文件上传及大文件上传如何实现

嚣张农民 2024-06-26 16:03:05 阅读 84

文件上传

文件上传是个非常普遍的场景,特别是在一些资源管理相关的业务中(比如网盘)。在文件比较大的时候,普通的上传方式可能会遇到以下四个问题。

对于前两点,虽说可以通过一定的配置来解决,但有时候也不会那么顺利,毕竟调大这些参数会对后台造成一定的压力,需要兼顾实际场景。只是上传慢的话忍一忍是可以接受的,但是失败后重头开始,在网络环境差的时候简直就是灾难。

针对遇到的这些问题,有比较成熟的解决方案。该方案可以简答的概括为切片上传 + 秒传

切片上传是指将一个大文件切割为若干个小文件,分为多个请求依次上传,后台再将文件碎片拼接为一个完整的文件,即使某个碎片上传失败,也不会影响其它文件碎片,只需要重新上传失败的部分就可以了。而且多个请求一起发送文件,提高了传输速度的上限

秒传指的是文件在传输之前计算其内容的散列值,也就是 Hash 值,将该值传到后台,如果后台存在 Hash 值一致的文件,认为该文件上传完成

该方案很巧妙的解决了上述提出的一系列问题,也是目前资源管理类系统的通用解决方案。

本文会梳理大文件上传中的一些知识点,并根据上述方案,实现一个切片上传 + 秒传的前后端,前端我用 react 实现,后台用的 java。

文件上传原理

最开始 XMLHttpRequest 是不支持传输二进制文件的。文件只能使用表单的方式上传,我们需要写一个 Form,然后将 enctype 设置为 multipart/form-data。此时的 Content-Type 为 multipart/form-data,并且会自动跟一个 boundary 字符串,该字符串用于隔离不同的字段。

POST /file/uploadSingle HTTP/1.1

Host: localhost:8080

Connection: keep-alive

Content-Length: 9253791

Content-Type: multipart/form-data; boundary=----WebKitFormBoundaryZ6BFzaoozLLGdTBE

所以 multipart/form-data 既可以上传文件,也可以上传键值对,每个元素由 boundary 分隔放在请求的 body 中。

------WebKitFormBoundaryZ6BFzaoozLLGdTBE

Content-Disposition: form-data; name="file"; filename="BCompare.zip"

Content-Type: application/zip

------WebKitFormBoundaryZ6BFzaoozLLGdTBE--

后来 XMLHttpRequest 升级为 Level 2 之后,新增了 FormData 对象,用于模拟表单数据,并且支持发送和接收二进制数据。我们目前使用的文件上传基本都是基于 XMLHttpRequest Level 2。使用 XMLHttpRequest 后文件上传的报文和上述的一致。写法如下。

let xhr = new XMLHttpRequest();

xhr.upload.onprogress = onProgress;

xhr.onload = () => {};

xhr.onabort = () => {};

xhr.onerror = err => {};

xhr.open(method, url, true);

xhr.send(data);

需要注意的是,xhr.send(data)中 data 参数的数据类型会影响请求头部 content-type 的值。我们上传文件,data 的类型是 FormData,此时 content-type 默认值为 multipart/form-data; boundary=[xxx]。当然,但如果用 xhr.setRequestHeader() 手动设置了中 content-type 的值,以用户设定的为准。因此,在上传文件场景下,不必设置 content-type 的值,浏览器会根据文件类型自动配置

文件切片

文件切片和核心是使用 Blob 对象的 slice 方法。

我们使用 \ 的方式获得一个 File 对象。File 继承于 Blob。所以我们也可以使用 slice 方法对文件进行切割。Blob 对象的 slice 方法会返回一个新的 Blob 对象,包含了源 Blob 对象中制定范围内的数据。

start 和 end 代表 Blob 里的下标,表示被拷贝进新的 Blob 的字节的起始位置和结束位置。contentType 会给新的 Blob 赋予一个新的文档类型,很少使用。

在分片上传场景中,我们一般会规定一个切边大小,根据这个大小对文件进行分割。除了这种固定大小的方案外,还有的文章中会根据当前的网络情况动态的调整切片的大小,类似于 TCP 的拥塞控制。本文为了简单,实用固定大小的切片。代码如下。我定义了 FileChunk 对象,这个对象中除了包含切片本身外,还额外存放了一些数据,比如切边在源文件中的起止位置,这么做是为了方便后台拿到切片数据后做文件合并的。

const CHUNK_SIZE = 10 * 1024 * 1024;

// 生成文件切片

function createFileChunk(file, blockSize = CHUNK_SIZE) {

const fileChunkList = [];

const { name, size } = file;

let cur = 0;

while (cur < size) {

let end = cur + blockSize;

if (end > size) {

end = size;

}

// 调用 slice 方法进行文件切割

fileChunkList.push(

new FileChunk(file.slice(cur, end), name, cur, end, size),

);

cur += blockSize;

}

...

return fileChunkList;

}

class FileChunk {

constructor(chunk, fileName, start, end, total) {

// 切片对象

this.chunk = chunk;

// 文件名称

this.fileName = fileName;

// 切片起始位置

this.start = start;

// 切片结束位置

this.end = end;

// 文件总大小

this.total = total;

// 切片名称

this.chunkName = '';

// 整体文件Hash

this.fileHash = '';

// 索引

this.index = 0;

// 文件切片总数

this.chunkNum = 0;

// 文件状态 'READY', 'UPLOADING', 'SUCCESS', 'ERROR'

this.status = 'READY';

}

}

文件合并

文件合并方案有这么几种。

前端发送切片完成后,发送一个合并请求,后端收到请求后,将之前上传的切片文件合并。后台记录切片文件上传数据,当后台检测到切片上传完成后,自动完成合并。创建一个和源文件大小相同的文件,根据切片文件的起止位置直接将切片写入对应位置。

这三种方案中,前两种都是比较通用的方案,且都是可行的,方案一的代价在于多发了一次请求,极小的概率会出现文件上传成功,但是合并请求发送失败的情况,好处就是流程比较清晰。方案二比方案一少了一次请求,代价是每次上传结束后需要判断当前切片是否是最后一个切片,需要在数据库中维护切片的状态。

方案三比较好的,相当于直接省略了文件合并的步骤,速度比较快。但是不用语言的实现难度不同。如果没有合适的 API 的话,自己实现的难度很大。由于我后台是用 java 编写的。我们可以充分利用 java 中的 RandomAccessFile 这个类。

RandomAccessFile 既可以读取文件内容,也可以向文件输出数据。它最大的特点就是支持 “随机访问” 的方式,程序快可以直接跳转到文件的任意地方来读写数据。在切片上传场景下,由于请求是并行发送的,后台会一次性收到大量的切片文件。每个切片文件都携带了当前切片在总文件中的位置信息,我们使用 RandomAccessFile 的 seek 方法定位到切片的起始位置,然后将切片从这个位置开始写入。这也是 RandomAccessFile 的一个重要使用场景。

/**

* 上传切片文件

* @param chunk 切片文件

* @param fileChunk 切片文件的信息

* @return true | false

*/

@PostMapping("/uploadChunk")

@ResponseBody

public Boolean upload(@RequestParam("chunk") MultipartFile chunk,

FileChunk fileChunk) throws IOException {

String fullPath = filePath + fileChunk.getFileName();

// 模块写入对应的位置

try(RandomAccessFile rf = new RandomAccessFile(fullPath,

"rw")) {

rf.seek(fileChunk.getStart());

rf.write(chunk.getBytes());

} catch (Exception e) {

LOGGER.error(e.getMessage());

return false;

}

...

return true;

}

显示进度

旧版的 XMLHttpRequest 是不支持显示进度的,升级为 Level 2 之后,有一个 progress 事件,用来返回进度信息。这也是为什么该场景下推荐使用 xhr ,而不使用 fetch 的原因。fetch 不提供相关的接口,我们无法获得文件的上传进度

我们可以通过 onprogress 事件来实时显示进度,默认情况下这个事件每 50ms 触发一次。需要注意的是,上传过程和下载过程触发的是不同对象的 onprogress 事件:上传触发的是 xhr.upload 对象的 onprogress 事件,下载触发的是 xhr对象的 onprogress 事件。

xhr.onprogress = updateProgress;

xhr.upload.onprogress = updateProgress;

这个事件有一些属性。event.total 是需要传输的总字节,event.loaded 是已经传输的字节。

function updateProgress(event) {

if (event.lengthComputable) {

var completedPercent = event.loaded / event.total;

}

}

由于在切片上传场景下,我们获得的是单个切片的上传进度,所以一般需要将单个的进度进行累加,用于计算总的进度,具体代码就不贴了。当然,我们也可以兼顾两种方式。我在大圣老师的文章中找到一种很好的显示进度的方式。将每个切片算作是一个小方块,通过颜色表示进度,非常直观。

断点续传

切片上传有一个很好的特性就是上传过程可以中断,不论是人为的暂停还是由于网络环境导致的链接的中断,都只会影响到当前的切片,而不会导致整体文件的失败,下次开始上传的时候可以从失败的切片继续上传。

我们为当前的上传操作增加一个停止按钮,用于模拟网络错误导致的上传失败。一个请求能被取消的前提是,我们需要将未收到响应的请求保存在一个列表中,然后依次调用每个 xhr 对象的 abort 方法。调用这个方法后,xhr 对象会停止触发事件,将请求的 status 置为 0,并且无法访问任何与响应有关的属性。

// 取消上传操作

pause = () => {

this.requestList.forEach(xhr => xhr?.abort());

this.requestList = [];

};

从后端的角度看,一个上传请求被取消,意味着当前浏览器不会再向后端传输数据流,后端此时会报错,如下,错误信息也很清楚,就是文件还没到末尾就被客户端中断。当前文件切片写入失败。

java.io.EOFException: Unexpected EOF read on the socket

接下来就是如何实现断点续传,关键点是后端需要记录文件文件切片的信息。用户在上传一个文件之前,先询问服务器,当前文件是否存在已经上传完毕的切片,如果存在的话,需要返回切片信息。前端根据返回的信息,调整当前的进度,上传未完成的切片。

具体的做法是,切片上传完成后,后端记录当前切片的详细信息。

@PostMapping("/uploadChunk")

@ResponseBody

public Boolean upload(@RequestParam("chunk") MultipartFile chunk,

FileChunk fileChunk) {

String fullPath = filePath + fileChunk.getFileName();

// 存储文件

...

// chunk 记录到数据库

uploadService.addChunk(fileChunk);

...

}

前端在文件上传之前,多加一个步骤,那就是从后端获得已经存在的切片文件。这里我们先用文件名来查询,这只是临时方案,文件名并不能作为文件的唯一标志,后续我们会改为使用文件 Hash 的方式来查询。

upload = file => {

// 1. 文件切片

const chunkList = createFileChunk(file);

// 2. 判断后端是或已经存在该文件

this.getExistFileChunk(file.name).then(res => {

// 标记已经完成上传的

this.chunkList.forEach(chunk => {

uploadedChunkList.forEach(uploadedChunk => {

if (uploadedChunk.chunkName === chunk.chunkName) {

this.markAsSuccess(chunk);

}

});

});

// 上传切片

this.uploadChunks(chunkList);

}

});

return this;

};

如果存在已经上传好的切片,将这些切片的状态更新为成功,修改进度为 100,后续发送请求的时候会过滤掉状态为成功的切片。

// 标记为上传完成

markAsSuccess = fileChunk => {

fileChunk.status = 'SUCCESS';

const chunkSize = fileChunk.chunk.size;

fileChunk.progress = {

percentage: 100,

loaded: chunkSize,

total: chunkSize,

};

};

限制请求个数

我在尝试将一个 5G 大小的文件上传的时候,发现前端浏览器出现卡死现象,原因是切片文件过多,浏览器一次性创建了太多了 xhr 请求。这是没有必要的,拿 chrome 浏览器来说,默认的并发数量只有 6,过多的请求并不会提升上传速度,反而是给浏览器带来了巨大的负担。因此,我们有必要限制前端请求个数。

思路比较简单,先创建最大并发数的请求,然后在请求的回调函数中再次创建请求,直到全部请求都发出为止

const MAX_REQUEST_NUM = 4;

// 限制请求并发数

requestWithLimit = (

fileChunkList,

max = MAX_REQUEST_NUM,

) => {

return new Promise((resolve, reject) => {

// 切片数量

const requestNum = fileChunkList.filter(fileChunk => {

return fileChunk.status === 'READY';

}).length;

// 发送成功数量

let counter = 0;

const request = () => {

// max 限制了最大并发数

while (counter < requestNum && max > 0) {

max--;

// 等待发送的切片

const fileChunk = fileChunkList.find(chunk => {

return chunk.status === 'READY';

});

const formData = fileChunk.toFormData();

fileChunk.status = 'UPLOADING';

ajax4Upload({

method: 'POST',

url: this.uploadUrl,

data: formData

})

.then(() => {

fileChunk.status = 'SUCCESS';

// 释放通道

max++;

counter++;

if (counter === requestNum) {

resolve();

} else {

request();

}

})

.catch(e => {

reject(e);

});

}

};

request();

});

};

并发重试

切片上传的过程中,我们有可能因为各种原因导致某个切片上传失败,比如网络抖动、后端文件进程占用等等。对于这种情况,最好的方案就是为切片上传增加一个失败重试机制。由于切片不大,重试的代价很小,我们设定一个最大重试次数,如果在次数内依然没有上传成功,认为上传失败。

具体的做法就是改造 requestWithLimit 方法。

定义一个 retryArr 数组,用于记录文件上传失败的次数。改造 catch 方法,一个文件切片上传报错时候,先判断 retryArr 中给切片的错误次数是否达到最大值,如果没有的话,清空当前切片上传进度,重新请求。相应的,我们之前只上传处于 READY 状态的切片,现在要稍微调整下,处于 ERROR 状态的切片也获得上传资格。

const MAX_RETRY_NUM = 3;

requestWithLimit = (

fileChunkList,

max = MAX_REQUEST_NUM,

retry = MAX_RETRY_NUM,

) => {

return new Promise((resolve, reject) => {

...

// 记录文件上传失败的次数

const retryArr = [];

const request = () => {

while (counter < requestNum && max > 0) {

max--;

// READY 或者 ERROR

const fileChunk = fileChunkList.find(chunk => {

return chunk.status === 'ERROR' || chunk.status === 'READY';

});

...

ajax4Upload({

method: 'POST',

url: this.uploadUrl,

data: formData,

onProgress: this.onProgressHandler.bind(this, fileChunk),

requestList: this.requestList,

})

.then(() => {

...

})

.catch(e => {

fileChunk.status = 'ERROR';

// 触发重试机制

if (typeof retryArr[fileChunk.index] !== 'number') {

retryArr[fileChunk.index] = 0;

}

// 次数累加

retryArr[fileChunk.index]++;

// 一个请求报错超过最大重试次数

if (retryArr[fileChunk.index] >= retry) {

return reject();

}

// 清空进度条

fileChunk.progress = {};

// 释放当前占用的通道,但是counter不累加

max++;

request();

});

}

};

request();

});

};

后台坐下设置,每个切片第一次上传一定失败,会触发重传机制,如下所示。每个切片都会上传两次,发现进度显示的有点魔性,抽空优化吧。

秒传指的是文件如果在后台已经存了一份,就没必要再次上传了,直接返回上传成功。在体量比较大的应用场景下,秒传是个必要的功能,既能提高用户上传体验,又能节约自己的硬盘资源。

秒传的关键在于计算文件的唯一性标识

文件的不同不是命名的差异,而是内容的差异,所以我们将整个文件的二进制码作为入参,计算 Hash 值,将其作为文件的唯一性标识。一般而言,这样做就够了,但是摘要算法是存在碰撞概率的,我们如果想要再严谨点的话,可以将文件大小也作为衡量指标,只有文件摘要和文件大小同时相等,才认为是相同的文件

文件 Hash 值的计算是 CPU 密集型任务,线程在计算 Hash 值的过程中,页面处于假死状态。所以,该任务一定不能在当前线程进行,我们使用 Web Worker 执行计算任务

Web Worker 是 HTML5 标准的一部分,它允许一段 JavaScript 程序运行在主线程之外的另外一个线程中。这样计算任务就不会影响到当前线程的渲染任务。

目前网上有很多 Web Worker 使用方案,我使用的前端框架是 umi,直接配置下就好了。

export default defineConfig({

workerLoader: {

worker: 'Worker',

esModule: true,

},

});

Web Worker 是一段单独的 JS 程序,它和当前线程间使用 postMessage 的方式进行通讯。

import Worker from './hash.worker.js';

// 生成文件 hash(web-worker)

function calculateFileHash(fileChunkList) {

return new Promise(resolve => {

const worker = new Worker();

worker.postMessage({ fileChunkList, type: 'HASH' });

worker.onmessage = e => {

const { hash } = e.data;

if (hash) {

resolve(hash);

}

};

});

}

如何快速计算文件的 md5 值呢? 我们使用 js-spark-md5 这个库

js-spark-md5 是号称全宇宙最快的前端类包。我在本机测下来,计算 1G 文件大概 15 秒,确实很快。由于是在 Web Worker 中计算,需要将 spark-md5.min.js 放到静态资源目录下,方便引用。

self.importScripts('spark-md5.min.js'); // 导入脚本

// 全量 Hash

postHashMsg = fileChunkList => {

const spark = new self.SparkMD5.ArrayBuffer();

let count = 0;

const loadNext = index => {

const reader = new FileReader();

reader.readAsArrayBuffer(fileChunkList[index].chunk);

reader.onload = e => {

count++;

spark.append(e.target.result);

if (count === fileChunkList.length) {

self.postMessage({

hash: spark.end(),

});

self.close();

} else {

// 递归计算下一个切片

loadNext(count);

}

};

};

loadNext(0);

};

self.onmessage = e => {

const { fileChunkList, type } = e.data;

if (type === 'HASH') {

postHashMsg(fileChunkList);

}

};

此时前端上传文件的流程是这样的。

文件切片。计算全量 Hash。判断文件是否符合秒传条件,如果不满足,判断是否满足断点续传条件。文件上传。

后端也做下调整,上传文件的接口,需要在文件全部上传完毕后,记录下文件的详细信息,包括 md5 值,作为后续文件秒传的依据。

/**

* 上传切片文件

* @param chunk 切片文件

* @param fileChunk 切片文件的源数据

* @return true | false

*/

@PostMapping("/uploadChunk")

@ResponseBody

public Boolean upload(@RequestParam("chunk") MultipartFile chunk,

FileChunk fileChunk) {

...

// 文件全部上传完成

Integer chunkSize = uploadService.getChunkNumByContentHash(fileChunk.getFileHash());

if (chunkSize.equals(fileChunk.getChunkNum())) {

// 删除 chunk 记录

uploadService.removeChunkRecord(fileChunk.getFileHash());

// 增加 file 记录

FileModel fileModel = new FileModel(fileChunk.getFileName(),

fileChunk.getFullPath(),

fileChunk.getFileHash(),

fileChunk.getTotal(),

"SUCCESS");

uploadService.addFile(fileModel);

}

}

判断秒传和断点续传的代码可以合并为一个接口,作为文件上传的前置接口。

秒传功能由于需要计算 Hash 值,会导致整体上传速度变慢,但是和大文件上传需要的耗时以及消耗的流量比起来,是一种性价比很高的选择。

如果觉得文件计算全量 Hash 比较慢的话,还有一种方式就是计算抽样 Hash,减少计算的字节数可以大幅度减少耗时,但是抽样 Hash 的结果不能作为文件的唯一性标识,抽样 Hash 的值如果和后端一致,前端再计算全量 Hash,如果和后端不一致,那么这个文件一定无法秒传,此时可以直接将文件上传,当然,上传之后还是需要计算全量 Hash 值的。这种方案在文件重复率较少的场景下是很好的,它能极大减少了前端的计算量,提高了速度,相当于转移了一部分工作到后端,也是不错的选择。

断点续传的重点是文件的切割与合并,整个上传流程需要前后端配合好,细节较多。秒传的关键是如何快速计算大文件的摘要信息。我们最后再梳理下文件上传的完整流程。获得文件后,使用 Blob 对象的 slice 方法对其进行切割,并封装一些上传需要的数据,文件切割的速度很快,不影响主线程渲染。计算整个文件的 MD5 值,大文件比较耗时,我们将这部分任务放在 Web Worker 中执行。获得文件的 MD5 值之后,我们将 MD5 值以及文件大小发送到后端,后端查询是否存在该文件,如果不存在的话,查询是否存在该文件的切片文件,如果存在,返回切片文件的详细信息。根据后端返回结果,依次判断是否满足“秒传” 或是 “断点续传” 的条件。如果满足,更新文件切片的状态与文件进度。根据文件切片的状态,发送上传请求,由于存在并发限制,我们限制 request 创建个数,避免页面卡死。对于上传失败的文件,设置最大重试次数,将其继续加入到上传任务中,超过最大重试次数的才认为上传失败。后端收到文件后,首先保存文件,保存成功后记录切片信息,判断当前切片是否是最后一个切片,如果是最后一个切片,记录文件信息,认为文件上传成功,清空切片记录。

目前还有一些可以优化的点。

多人上传同一个文件,只要其中一人上传成功即可认为其他人上传成功。拥塞控制,动态计算文件切片大小,大圣老师文章中已经实现。进度条优化,进度条在断点续传和失败重传会出现倒退的情形,

如果您觉得有所收获,就点个赞吧!



声明

本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。