经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置：技术经验 » JS/JS库/框架 » JavaScript » 查看文章
每日一题：吃透大文件上传问题（附可运行的前后端源码）
来源：cnblogs　　作者：菜鸟小何　　时间：2023/9/13 15:36:30　　对本文有异议

大文件上传

前言

在日常开发中，文件上传是常见的操作之一。文件上传技术使得用户可以方便地将本地文件上传到Web服务器上，这在许多场景下都是必需的，比如网盘上传、头像上传等。

但是当我们需要上传比较大的文件的时候，容易碰到以下问题：

  1. 上传时间比较久
  2. 中间一旦出错就需要重新上传
  3. 一般服务端会对文件的大小进行限制

这两个问题会导致上传时候的用户体验是很不好的，针对存在的这些问题，我们可以通过分片上传来解决，这节课我们就在学习下什么是切片上传，以及怎么实现切片上传。

原理介绍

分片上传的原理就像是把一个大蛋糕切成小块一样。

首先，我们将要上传的大文件分成许多小块，每个小块大小相同，比如每块大小为2MB。然后，我们逐个上传这些小块到服务器。上传的时候，可以同时上传多个小块，也可以一个一个地上传。上传每个小块后，服务器会保存这些小块，并记录它们的顺序和位置信息。

所有小块上传完成后，服务器会把这些小块按照正确的顺序拼接起来，还原成完整的大文件。最后，我们就成功地上传了整个大文件。

image

分片上传的好处在于它可以减少上传失败的风险。如果在上传过程中出现了问题，只需要重新上传出错的那个小块，而不需要重新上传整个大文件。

此外，分片上传还可以加快上传速度。因为我们可以同时上传多个小块，充分利用网络的带宽。这样就能够更快地完成文件的上传过程。

实现

项目搭建

要实现大文件上传，还需要后端的支持，所以我们就用nodejs来写后端代码。

前端：vue3 + vite

后端：express 框架，用到的工具包：multiparty、fs-extra、cors、body-parser、nodemon

读取文件

通过监听 input 的 change 事件，当选取了本地文件后，可以在回调函数中拿到对应的文件：

  1. const handleUpload = (e: Event) => {
  2.   const files = (e.target as HTMLInputElement).files
  3.   if (!files) {
  4.     return
  5.   }
  6.   // 读取选择的文件
  7.   console.log(files[0]);
  8. }

文件分片

文件分片的核心是用Blob对象的slice方法，我们在上一步获取到选择的文件是一个File对象，它是继承于Blob，所以我们就可以用slice方法对文件进行分片，用法如下：

  1. let blob = instanceOfBlob.slice([start [, end [, contentType]]]};

start 和 end 代表 Blob 里的下标，表示被拷贝进新的 Blob 的字节的起始位置和结束位置。contentType 会给新的 Blob 赋予一个新的文档类型，在这里我们用不到。接下来就来使用slice方法来实现下对文件的分片。

  1. const createFileChunks = (file: File) => {
  2.   const fileChunkList = []
  3.   let cur = 0
  4.   while (cur < file.size) {
  5.     fileChunkList.push({
  6.       file: file.slice(cur, cur + CHUNK_SIZE),
  7.     })
  8.     cur += CHUNK_SIZE // CHUNK_SIZE为分片的大小
  9.   }
  10.   return fileChunkList
  11. }

hash计算

先来思考一个问题，在向服务器上传文件时，怎么去区分不同的文件呢？如果根据文件名去区分的话可以吗？

答案是不可以，因为文件名我们可以是随便修改的，所以不能根据文件名去区分。但是每一份文件的文件内容都不一样，我们可以根据文件的内容去区分，具体怎么做呢？

可以根据文件内容生产一个唯一的 hash 值，大家应该都见过用 webpack 打包出来的文件的文件名都有一串不一样的字符串，这个字符串就是根据文件的内容生成的 hash 值，文件内容变化，hash 值就会跟着发生变化。我们在这里，也可以用这个办法来区分不同的文件。而且通过这个办法，我们还可以实现秒传的功能，怎么做呢？

就是服务器在处理上传文件的请求的时候，要先判断下对应文件的 hash 值有没有记录，如果A和B先后上传一份内容相同的文件，所以这两份文件的 hash 值是一样的。当A上传的时候会根据文件内容生成一个对应的 hash 值，然后在服务器上就会有一个对应的文件，B再上传的时候，服务器就会发现这个文件的 hash 值之前已经有记录了，说明之前已经上传过相同内容的文件了，所以就不用处理B的这个上传请求了，给用户的感觉就像是实现了秒传。

那么怎么计算文件的hash值呢？可以通过一个工具：spark-md5，所以我们得先安装它。

在上一步获取到了文件的所有切片，我们就可以用这些切片来算该文件的 hash 值，但是如果一个文件特别大，每个切片的所有内容都参与计算的话会很耗时间，所有我们可以采取以下策略：

  1. 第一个和最后一个切片的内容全部参与计算

  2. 中间剩余的切片我们分别在前面、后面和中间取2个字节参与计算

这样就既能保证所有的切片参与了计算，也能保证不耗费很长的时间

  1. /**
  2.  * 计算文件的hash值，计算的时候并不是根据所用的切片的内容去计算的，那样会很耗时间，我们采取下面的策略去计算：
  3.  * 1. 第一个和最后一个切片的内容全部参与计算
  4.  * 2. 中间剩余的切片我们分别在前面、后面和中间取2个字节参与计算
  5.  * 这样做会节省计算hash的时间
  6.  */
  7. const calculateHash = async (fileChunks: Array<{file: Blob}>) => {
  8.   return new Promise(resolve => {
  9.     const spark = new sparkMD5.ArrayBuffer()
  10.     const chunks: Blob[] = []
  12.     fileChunks.forEach((chunk, index) => {
  13.       if (index === 0 || index === fileChunks.length - 1) {
  14.         // 1. 第一个和最后一个切片的内容全部参与计算
  15.         chunks.push(chunk.file)
  16.       } else {
  17.         // 2. 中间剩余的切片我们分别在前面、后面和中间取2个字节参与计算
  18.         // 前面的2字节
  19.         chunks.push(chunk.file.slice(0, 2))
  20.         // 中间的2字节
  21.         chunks.push(chunk.file.slice(CHUNK_SIZE / 2, CHUNK_SIZE / 2 + 2))
  22.         // 后面的2字节
  23.         chunks.push(chunk.file.slice(CHUNK_SIZE - 2, CHUNK_SIZE))
  24.       }
  25.     })
  27.     const reader = new FileReader()
  28.     reader.readAsArrayBuffer(new Blob(chunks))
  29.     reader.onload = (e: Event) => {
  30.       spark.append(e?.target?.result as ArrayBuffer)
  31.       resolve(spark.end())
  32.     }
  33.   })
  34. }

文件上传

前端实现

前面已经完成了上传的前置操作，接下来就来看下如何去上传这些切片。

我们以1G的文件来分析，假如每个分片的大小为1M，那么总的分片数将会是1024个，如果我们同时发送这1024个分片，浏览器肯定处理不了，原因是切片文件过多，浏览器一次性创建了太多的请求。这是没有必要的，拿 chrome 浏览器来说，默认的并发数量只有 6，过多的请求并不会提升上传速度，反而是给浏览器带来了巨大的负担。因此，我们有必要限制前端请求个数。

怎么做呢，我们要创建最大并发数的请求，比如6个，那么同一时刻我们就允许浏览器只发送6个请求，其中一个请求有了返回的结果后我们再发起一个新的请求，依此类推，直至所有的请求发送完毕。

上传文件时一般还要用到 FormData 对象，需要将我们要传递的文件还有额外信息放到这个 FormData 对象里面。

  1. const uploadChunks = async (fileChunks: Array<{file: Blob}>) => {
  2.   const data = fileChunks.map(({ file }, index) => ({
  3.     fileHash: fileHash.value,
  4.     index,
  5.     chunkHash: `${fileHash.value}-${index}`,
  6.     chunk: file,
  7.     size: file.size,
  8.   }))
  9.  
  10.   const formDatas = data
  11.     .map(({ chunk, chunkHash }) => {      
  12.       const formData = new FormData()
  13.       // 切片文件
  14.       formData.append('chunk', chunk)
  15.       // 切片文件hash
  16.       formData.append('chunkHash', chunkHash)
  17.       // 大文件的文件名
  18.       formData.append('fileName', fileName.value)
  19.       // 大文件hash
  20.       formData.append('fileHash', fileHash.value)
  21.       return formData
  22.     })
  24.   let index = 0;
  25.   const max = 6; // 并发请求数量
  26.   const taskPool: any = [] // 请求队列 
  27.   
  28.   while(index < formDatas.length) {
  29.     const task = fetch('http://127.0.0.1:3000/upload', {
  30.       method: 'POST',
  31.       body: formDatas[index],
  32.     })
  34.     task.then(() => {
  35.       taskPool.splice(taskPool.findIndex((item: any) => item === task))
  36.     })
  37.     taskPool.push(task);
  38.     if (taskPool.length === max) {
  39.       // 当请求队列中的请求数达到最大并行请求数的时候，得等之前的请求完成再循环下一个
  40.       await Promise.race(taskPool)
  41.     }
  42.     index ++
  43.     percentage.value = (index / formDatas.length * 100).toFixed(0)
  44.   }  
  46.   await Promise.all(taskPool)
  47. }
后端实现

后端我们处理文件时需要用到 multiparty 这个工具，所以也是得先安装，然后再引入它。

我们在处理每个上传的分片的时候，应该先将它们临时存放到服务器的一个地方，方便我们合并的时候再去读取。为了区分不同文件的分片，我们就用文件对应的那个hash为文件夹的名称，将这个文件的所有分片放到这个文件夹中。

  1. // 所有上传的文件存放到该目录下
  2. const UPLOAD_DIR = path.resolve(__dirname, 'uploads');
  4. // 处理上传的分片
  5. app.post('/upload', async (req, res) => {
  6.   const form = new multiparty.Form();
  7.   
  8.   form.parse(req, async function (err, fields, files) {
  9.     if (err) {
  10.       res.status(401).json({ 
  11.         ok: false,
  12.         msg: '上传失败'
  13.       });
  14.     }
  15.     const chunkHash = fields['chunkHash'][0]
  16.     const fileName = fields['fileName'][0]
  17.     const fileHash = fields['fileHash'][0]
  19.     // 存储切片的临时文件夹
  20.     const chunkDir = path.resolve(UPLOAD_DIR, fileHash)
  22.     // 切片目录不存在，则创建切片目录
  23.     if (!fse.existsSync(chunkDir)) {
  24.       await fse.mkdirs(chunkDir)
  25.     }
  27.     const oldPath = files.chunk[0].path;
  28.     // 把文件切片移动到我们的切片文件夹中
  29.     await fse.move(oldPath, path.resolve(chunkDir, chunkHash))
  31.     res.status(200).json({ 
  32.       ok: true,
  33.       msg: 'received file chunk'
  34.     });
  35.   });
  36. });

写完前后端代码后就可以来试下看看文件能不能实现切片的上传，如果没有错误的话，我们的 uploads 文件夹下应该就会多一个文件夹，这个文件夹里面就是存储的所有文件的分片了。

文件合并

上一步我们已经实现了将所有切片上传到服务器了，上传完成之后，我们就可以将所有的切片合并成一个完整的文件了，下面就一块来实现下。

前端实现

前端只需要向服务器发送一个合并的请求，并且为了区分要合并的文件，需要将文件的hash值给传过去

  1. /**
  2.  * 发请求通知服务器，合并切片
  3.  */
  4. const mergeRequest = () => {  
  5.   // 发送合并请求
  6.   fetch('http://127.0.0.1:3000/merge', {
  7.     method: 'POST',
  8.     headers: {
  9.       'Content-Type': 'application/json'
  10.     },
  11.     body: JSON.stringify({
  12.       size: CHUNK_SIZE,
  13.       fileHash: fileHash.value,
  14.       fileName: fileName.value,
  15.     }),
  16.   })
  17.     .then((response) => response.json())
  18.     .then(() => {
  19.       alert('上传成功')
  20.     })
  21. }
后端实现

在之前已经可以将所有的切片上传到服务器并存储到对应的目录里面去了，合并的时候需要从对应的文件夹中获取所有的切片，然后利用文件的读写操作，就可以实现文件的合并了。合并完成之后，我们将生成的文件以hash值命名存放到对应的位置就可以了。

  1. // 提取文件后缀名
  2. const extractExt = filename => {
  3. 	return filename.slice(filename.lastIndexOf('.'), filename.length)
  4. }
  6. /**
  7.  * 读的内容写到writeStream中
  8.  */
  9. const pipeStream = (path, writeStream) => {
  10. 	return new Promise((resolve, reject) => {
  11. 		// 创建可读流
  12. 		const readStream = fse.createReadStream(path)
  13. 		readStream.on('end', async () => {
  14. 			fse.unlinkSync(path)
  15. 			resolve()
  16. 		})
  17. 		readStream.pipe(writeStream)
  18. 	})
  19. }
  21. /**
  22.  * 合并文件夹中的切片，生成一个完整的文件
  23.  */
  24. async function mergeFileChunk(filePath, fileHash, size) {
  25.   const chunkDir = path.resolve(UPLOAD_DIR, fileHash)
  26.   const chunkPaths = await fse.readdir(chunkDir)
  27.   // 根据切片下标进行排序
  28.   // 否则直接读取目录的获得的顺序可能会错乱
  29.   chunkPaths.sort((a, b) => {
  30.     return a.split('-')[1] - b.split('-')[1]
  31.   })
  33.   const list = chunkPaths.map((chunkPath, index) => {
  34.     return pipeStream(
  35.       path.resolve(chunkDir, chunkPath),
  36.       fse.createWriteStream(filePath, {
  37.         start: index * size,
  38.         end: (index + 1) * size
  39.       })
  40.     )
  41.   })
  43.   await Promise.all(list)
  44. 	// 文件合并后删除保存切片的目录
  45. 	fse.rmdirSync(chunkDir)
  46. }
  49. // 合并文件
  50. app.post('/merge', async (req, res) => {
  51.   const { fileHash, fileName, size } = req.body
  52.   const filePath = path.resolve(UPLOAD_DIR, `${fileHash}${extractExt(fileName)}`)
  53.   // 如果大文件已经存在，则直接返回
  54.   if (fse.existsSync(filePath)) {
  55.     res.status(200).json({ 
  56.       ok: true,
  57.       msg: '合并成功'
  58.     });
  59.     return
  60.   }
  61.   const chunkDir = path.resolve(UPLOAD_DIR, fileHash)
  62.   // 切片目录不存在，则无法合并切片，报异常
  63.   if (!fse.existsSync(chunkDir)) {
  64.     res.status(200).json({ 
  65.       ok: false,
  66.       msg: '合并失败，请重新上传'
  67.     });
  68.     return
  69.   }
  70.   await mergeFileChunk(filePath, fileHash, size)
  71.   res.status(200).json({ 
  72.     ok: true,
  73.     msg: '合并成功'
  74.   });
  75. });

到这里，我们就已经实现了大文件的分片上传的基本功能了，但是我们没有考虑到如果上传相同的文件的情况，而且如果中间网络断了，我们就得重新上传所有的分片，这些情况在大文件上传中也都需要考虑到，下面，我们就来解决下这两个问题。

秒传&断点续传

我们在上面有提到，如果内容相同的文件进行hash计算时，对应的hash值应该是一样的，而且我们在服务器上给上传的文件命名的时候就是用对应的hash值命名的，所以在上传之前是不是可以加一个判断，如果有对应的这个文件，就不用再重复上传了，直接告诉用户上传成功，给用户的感觉就像是实现了秒传。接下来，就来看下如何实现的。

前端实现

前端在上传之前，需要将对应文件的hash值告诉服务器，看看服务器上有没有对应的这个文件，如果有，就直接返回，不执行上传分片的操作了。

  1. /**
  2.  * 验证该文件是否需要上传，文件通过hash生成唯一，改名后也是不需要再上传的，也就相当于秒传
  3.  */
  4. const verifyUpload = async () => {
  5.   return fetch('http://127.0.0.1:3000/verify', {
  6.     method: 'POST',
  7.     headers: {
  8.       'Content-Type': 'application/json',
  9.     },
  10.     body: JSON.stringify({
  11.       fileName: fileName.value,
  12.       fileHash: fileHash.value
  13.     })
  14.   })
  15.   .then((response) => response.json())
  16.   .then((data) => {
  17.     return data; // data中包含对应的表示服务器上有没有该文件的查询结果
  18.   });
  19. }
  22. // 点击上传事件
  23. const handleUpload = async (e: Event) => {  
  24.   // ...
  26.   // uploadedList已上传的切片的切片文件名称
  27.   const res = await verifyUpload()
  29.   const { shouldUpload } = res.data
  31.   if (!shouldUpload) {
  32.     // 服务器上已经有该文件，不需要上传
  33.     alert('秒传：上传成功')
  34.     return;
  35.   }
  37.   // 服务器上不存在该文件，继续上传
  38.   uploadChunks(fileChunks)
  39. }
后端实现

因为我们在合并文件时，文件名时根据该文件的hash值命名的，所以只需要看看服务器上有没有对应的这个hash值的那个文件就可以判断了。

  1. // 根据文件hash验证文件有没有上传过
  2. app.post('/verify', async (req, res) => {
  3.   const { fileHash, fileName } = req.body
  4.   const filePath = path.resolve(UPLOAD_DIR, `${fileHash}${extractExt(fileName)}`)
  6.   if (fse.existsSync(filePath)) {
  7.     // 文件存在服务器中，不需要再上传了
  8.     res.status(200).json({ 
  9.       ok: true,
  10.       data: {
  11.         shouldUpload: false,
  12.       }
  13.     });
  14.   } else {
  15.     // 文件不在服务器中，就需要上传
  16.     res.status(200).json({ 
  17.       ok: true,
  18.       data: {
  19.         shouldUpload: true,
  20.       }
  21.     });
  22.   }
  23. });

完成上面的步骤后，当我们再上传相同的文件，即使改了文件名，也会提示我们秒传成功了，因为服务器上已经有对应的那个文件了。

上面我们解决了重复上传的文件，但是对于网络中断需要重新上传的问题没有解决，那该如何解决呢？

如果我们之前已经上传了一部分分片了，我们只需要再上传之前拿到这部分分片，然后再过滤掉是不是就可以避免去重复上传这些分片了，也就是只需要上传那些上传失败的分片，所以，再上传之前还得加一个判断。

前端实现

我们还是在那个 verify 的接口中去获取已经上传成功的分片，然后在上传分片前进行一个过滤

  1. const uploadChunks = async (fileChunks: Array<{file: Blob}>, uploadedList: Array<string>) => {
  2.   const formDatas = fileChunks
  3.   	.filter((chunk, index) => {
  4.       // 过滤服务器上已经有的切片
  5.       return !uploadedList.includes(`${fileHash.value}-${index}`)
  6.     })
  7.   	.map(({ file }, index) => {
  8.       const formData = new FormData()
  9.       // 切片文件
  10.       formData.append('file', file)
  11.       // 切片文件hash
  12.       formData.append('chunkHash', `${fileHash.value}-${index}`)
  13.       // 大文件的文件名
  14.       formData.append('fileName', fileName.value)
  15.       // 大文件hash
  16.       formData.append('fileHash', fileHash.value)
  17.       return formData
  18.     })
  20.   // ...
  21. }
后端实现

只需要在 /verify 这个接口中加上已经上传成功的所有切片的名称就可以，因为所有的切片都存放在以文件的hash值命名的那个文件夹，所以需要读取这个文件夹中所有的切片的名称就可以。

  1. /**
  2.  * 返回已经上传切片名
  3.  * @param {*} fileHash 
  4.  * @returns 
  5.  */
  6. const createUploadedList = async fileHash => {
  7. 	return fse.existsSync(path.resolve(UPLOAD_DIR, fileHash))
  8. 		? await fse.readdir(path.resolve(UPLOAD_DIR, fileHash)) // 读取该文件夹下所有的文件的名称
  9. 		: []
  10. }
  12. // 根据文件hash验证文件有没有上传过
  13. app.post('/verify', async (req, res) => {
  14.   const { fileHash, fileName } = req.body
  15.   const filePath = path.resolve(UPLOAD_DIR, `${fileHash}${extractExt(fileName)}`)
  17.   if (fse.existsSync(filePath)) {
  18.     // 文件存在服务器中，不需要再上传了
  19.     res.status(200).json({ 
  20.       ok: true,
  21.       data: {
  22.         shouldUpload: false,
  23.       }
  24.     });
  25.   } else {
  26.     // 文件不在服务器中，就需要上传，并且返回服务器上已经存在的切片
  27.     res.status(200).json({ 
  28.       ok: true,
  29.       data: {
  30.         shouldUpload: true,
  31.         uploadedList: await createUploadedList(fileHash)
  32.       }
  33.     });
  34.   }
  35. });

完整项目地址

点击查看

原文链接：https://www.cnblogs.com/never404/p/17699440.html

 友情链接：直通硅谷 　点职佳 　北美留学生论坛

JavaScript热门文章

  • echarts 饼图 + 全屏显示
  • js正则匹配以{开头,以}结尾的表达式怎么写？-附源码
  • echarts 实现数据(tooltip)自动轮播插件
  • Js设置博客园主题色(theme-color)
  • elementUI 设置input的只读或禁用的方法
  • 超级好用的流程图js框架 值得收藏
  • layui table+复杂表头+合并单元格
  • 获取地理位置
  • 如何使用JavaScript导入和导出Excel文件
  • WebStorm eslint插件报错解决 - TypeError: this.CliEngine is not a constructor

    • JavaScript推荐文章

  • 一些八股：1.fetch 的理解。2.let、const、var
  • 电子表格转身购物车：三步轻松实现
  • Canvas简历编辑器-图形绘制与状态管理(轻量级DOM)
  • IntersectionObserver + scrollIntoView 实现电梯导航 - xingba-coder
  • 可视化—gojs 超多超实用经验分享(三)
  • PixiJS源码分析系列： 第一章 从最简单的例子入手
  • WebRTC入门
  • 前端 Array.sort() 源码学习
  • 【技巧】JS代码这么写，前端小姐姐都会爱上你
  • 【JS】await异常捕获，这样做才完美

    • 本站QQ群：前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

    W3xue 的所有内容仅供测试，对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
    关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接（目前9元/月）请联系QQ：27243702 沸活量
    皖ICP备17017327号-2　皖公网安备34020702000426号