实战篇:断点续传?文件秒传?手撸大文件上传

各位看官大家好,今天给大家分享的又是一篇实战文章,希望大家能够喜欢。

开味菜

最近接到一个新的需求,需要上传2G左右的视频文件,用测试环境的OSS试了一下,上传需要十几分钟,再考虑到公司的资源问题,果断放弃该方案。

一提到大文件上传,我最先想到的就是各种网盘了,现在大家都喜欢将自己收藏的「***」上传到网盘进行保存。网盘一般都支持断点续传和文件秒传功能,减少了网络波动和网络带宽对文件的限制,大大提高了用户体验,让人爱不释手。

说到这,大家先来了解一下这几个概念:

  • 「文件分块」:将大文件拆分成小文件,将小文件上传\下载,最后再将小文件组装成大文件;
  • 「断点续传」:在文件分块的基础上,将每个小文件采用单独的线程进行上传\下载,如果碰到网络故障,可以从已经上传\下载的部分开始继续上传\下载未完成的部分,而没有必要从头开始上传\下载;
  • 「文件秒传」:资源服务器中已经存在该文件,其他人上传时直接返回该文件的URI。

RandomAccessFile

平时我们都会使用FileInputStream,FileOutputStream,FileReader以及FileWriter等IO流来读取文件,今天我们来了解一下RandomAccessFile。

它是一个直接继承Object的独立的类,底层实现中它实现的是DataInput和DataOutput接口。该类支持随机读取文件,随机访问文件类似于文件系统中存储的大字节数组。

它的实现基于「文件指针」(一种游标或者指向隐含数组的索引),文件指针可以通过getFilePointer方法读取,也可以通过seek方法设置。

输入时从文件指针开始读取字节,并使文件指针超过读取的字节,如果写入超过隐含数组当前结尾的输出操作会导致扩展数组。该类有四种模式可供选择:

  • r: 以只读方式打开文件,如果执行写入操作会抛出IOException;
  • rw: 以读、写方式打开文件,如果文件不存在,则尝试创建文件;
  • rws: 以读、写方式打开文件,要求对文件内容或元数据的每次更新都同步写入底层存储设备;
  • rwd: 以读、写方式打开文件,要求对文件内容的每次更新都同步写入底层存储设备;

在rw模式下,默认是使用buffer的,只有cache满的或者使用RandomAccessFile.close()关闭流的时候才真正的写到文件。

API

1、void seek(long pos):设置下一次读取或写入时的文件指针偏移量,通俗点说就是指定下次读文件数据的位置。

偏移量可以设置在文件末尾之外,只有在偏移量设置超出文件末尾后,才能通过写入更改文件长度;

2、native long getFilePointer():返回当前文件的光标位置;

3、native long length():返回当前文件的长度;

4、「读」方法

5、「写」方法

6、readFully(byte[] b):这个方法的作用就是将文本中的内容填满这个缓冲区b。如果缓冲b不能被填满,那么读取流的过程将被阻塞,如果发现是流的结尾,那么会抛出异常;

7、FileChannel getChannel():返回与此文件关联的唯一FileChannel对象;

8、int skipBytes(int n):试图跳过n个字节的输入,丢弃跳过的字节;

RandomAccessFile的绝大多数功能,已经被JDK1.4的NIO的「内存映射」文件取代了,即把文件映射到内存后再操作,省去了频繁磁盘io。

主菜

总结经验,砥砺前行:之前的实战文章中过多的粘贴了源码,影响了各位小伙伴的阅读感受。经过大佬的点拨,以后将展示部分关键代码,供各位赏析,源码可在「后台」获取。

文件分块

文件分块需要在前端进行处理,可以利用强大的js库或者现成的组件进行分块处理。需要确定分块的大小和分块的数量,然后为每一个分块指定一个索引值。

为了防止上传文件的分块与其它文件混淆,采用文件的md5值来进行区分,该值也可以用来校验服务器上是否存在该文件以及文件的上传状态。

  • 如果文件存在,直接返回文件地址;
  • 如果文件不存在,但是有上传状态,即部分分块上传成功,则返回未上传的分块索引数组;
  • 如果文件不存在,且上传状态为空,则所有分块均需要上传。
fileRederInstance.readAsBinaryString(file);
fileRederInstance.addEventListener("load", (e) => {
    let fileBolb = e.target.result;
    fileMD5 = md5(fileBolb);
    const formData = new FormData();
    formData.append("md5", fileMD5);
    axios
        .post(http + "/fileUpload/checkFileMd5", formData)
        .then((res) => {
            if (res.data.message == "文件已存在") {
                //文件已存在不走后面分片了,直接返回文件地址到前台页面                 success && success(res);
            } else {
                //文件不存在存在两种情况,一种是返回data:null代表未上传过 一种是data:[xx,xx] 还有哪几片未上传                 if (!res.data.data) {
                    //还有几片未上传情况,断点续传                     chunkArr = res.data.data;
                }
                readChunkMD5();
            }
        })
        .catch((e) => {});
});

在调用上传接口前,通过slice方法来取出索引在文件中对应位置的分块。

const getChunkInfo = (file, currentChunk, chunkSize) => {
       //获取对应下标下的文件片段        let start = currentChunk * chunkSize;
       let end = Math.min(file.size, start + chunkSize);
       //对文件分块        let chunk = file.slice(start, end);
       return { start, end, chunk };
   };

之后调用上传接口完成上传。

断点续传、文件秒传

后端基于spring boot开发,使用redis来存储上传文件的状态和上传文件的地址。

如果文件完整上传,返回文件路径;部分上传则返回未上传的分块数组;如果未上传过返回提示信息。

在上传分块时会产生两个文件,一个是文件主体,一个是临时文件。临时文件可以看做是一个数组文件,为每一个分块分配一个值为127的字节。

校验MD5值时会用到两个值:

  • 文件上传状态:只要该文件上传过就不为空,如果完整上传则为true,部分上传返回false;
  • 文件上传地址:如果文件完整上传,返回文件路径;部分上传返回临时文件路径。
/**
 * 校验文件的MD5
 **/ public Result checkFileMd5(String md5) throws IOException {
    //文件是否上传状态:只要该文件上传过该值一定存在     Object processingObj = stringRedisTemplate.opsForHash().get(UploadConstants.FILE_UPLOAD_STATUS, md5);
    if (processingObj == null) {
        return Result.ok("该文件没有上传过");
    }
    boolean processing = Boolean.parseBoolean(processingObj.toString());
    //完整文件上传完成时为文件的路径,如果未完成返回临时文件路径(临时文件相当于数组,为每个分块分配一个值为127的字节)     String value = stringRedisTemplate.opsForValue().get(UploadConstants.FILE_MD5_KEY + md5);
    //完整文件上传完成是true,未完成返回false     if (processing) {
        return Result.ok(value,"文件已存在");
    } else {
        File confFile = new File(value);
        byte[] completeList = FileUtils.readFileToByteArray(confFile);
        List<Integer> missChunkList = new LinkedList<>();
        for (int i = 0; i < completeList.length; i++) {
            if (completeList[i] != Byte.MAX_VALUE) {
                //用空格补齐                 missChunkList.add(i);
            }
        }
        return Result.ok(missChunkList,"该文件上传了一部分");
    }
}

说到这,你肯定会问:当这个文件的所有分块上传完成之后,该怎么得到完整的文件呢?接下来我们就说一下分块合并的问题。

分块上传、文件合并

上边我们提到了利用文件的md5值来维护分块和文件的关系,因此我们会将具有相同md5值的分块进行合并,由于每个分块都有自己的索引值,所以我们会将分块按索引像插入数组一样分别插入文件中,形成完整的文件。

分块上传时,要和前端的分块大小、分块数量、当前分块索引等对应好,以备文件合并时使用,此处我们采用的是「磁盘映射」的方式来合并文件。

 //读操作和写操作都是允许的 RandomAccessFile tempRaf = new RandomAccessFile(tmpFile, "rw"); //它返回的就是nio通信中的file的唯一channel FileChannel fileChannel = tempRaf.getChannel(); //写入该分片数据   分片大小 * 第几块分片获取偏移量 long offset = CHUNK_SIZE * multipartFileDTO.getChunk(); //分片文件大小 byte[] fileData = multipartFileDTO.getFile().getBytes(); //将文件的区域直接映射到内存 MappedByteBuffer mappedByteBuffer = fileChannel.map(FileChannel.MapMode.READ_WRITE, offset, fileData.length);
mappedByteBuffer.put(fileData); // 释放 FileMD5Util.freedMappedByteBuffer(mappedByteBuffer);
fileChannel.close();

每当完成一次分块的上传,还需要去检查文件的上传进度,看文件是否上传完成。

RandomAccessFile accessConfFile = new RandomAccessFile(confFile, "rw"); //把该分段标记为 true 表示完成 accessConfFile.setLength(multipartFileDTO.getChunks());
accessConfFile.seek(multipartFileDTO.getChunk());
accessConfFile.write(Byte.MAX_VALUE); //completeList 检查是否全部完成,如果数组里是否全部都是(全部分片都成功上传) byte[] completeList = FileUtils.readFileToByteArray(confFile); byte isComplete = Byte.MAX_VALUE; for (int i = 0; i < completeList.length && isComplete == Byte.MAX_VALUE; i++) {
    //与运算, 如果有部分没有完成则 isComplete 不是 Byte.MAX_VALUE     isComplete = (byte) (isComplete & completeList[i]);
}
accessConfFile.close();

然后更新文件的上传进度到Redis中。

//更新redis中的状态:如果是true的话证明是已经该大文件全部上传完成 if (isComplete == Byte.MAX_VALUE) {
    stringRedisTemplate.opsForHash().put(UploadConstants.FILE_UPLOAD_STATUS, multipartFileDTO.getMd5(), "true");
    stringRedisTemplate.opsForValue().set(UploadConstants.FILE_MD5_KEY + multipartFileDTO.getMd5(), uploadDirPath + "/" + fileName);
} else {
    if (!stringRedisTemplate.opsForHash().hasKey(UploadConstants.FILE_UPLOAD_STATUS, multipartFileDTO.getMd5())) {
        stringRedisTemplate.opsForHash().put(UploadConstants.FILE_UPLOAD_STATUS, multipartFileDTO.getMd5(), "false");
    }
    if (!stringRedisTemplate.hasKey(UploadConstants.FILE_MD5_KEY + multipartFileDTO.getMd5())) {
        stringRedisTemplate.opsForValue().set(UploadConstants.FILE_MD5_KEY + multipartFileDTO.getMd5(), uploadDirPath + "/" + fileName + ".conf");
    }
}
#计算机网络知识点总结##Go##C#(17531)##Ruby##MySQL##MongoDB##Oracle##SQL Server#
全部评论
已收藏,谢谢大佬分享
点赞 回复 分享
发布于 2022-05-20 20:50
太强了,大佬!
点赞 回复 分享
发布于 2022-05-20 20:42
感谢大佬分享,已收藏
点赞 回复 分享
发布于 2022-05-20 20:37
不明觉厉
点赞 回复 分享
发布于 2022-05-20 20:20
大佬好厉害呀!
点赞 回复 分享
发布于 2022-05-20 20:17

相关推荐

04-09 16:14
门头沟学院 Java
(二月多面的)同样没问实习,首先让手撕一个&nbsp;Redis&nbsp;分布式锁(可能是因为简历上提到有相关内容,但题目是腾讯题库里的),鼠鼠不会,支支吾吾说忘了&nbsp;API,他潇洒的甩了下头发,说可以在本地写。😅装模作样弄了一阵,计上心来装作如梦初醒:“借的电脑面试没找到&nbsp;IDEA”他又潇洒一摆手,那就写伪代码吧😅憋半天,把几个关键点的八股写成注释,他看了后,用不再潇洒的语气问我:“伪代码你全写中文干哈?”😅哈哈,会的话哥们还想写拉丁文呢后面又出了一个多线程打印&nbsp;ABC&nbsp;的手撕,同样不会,这次没再装,直接说不会,于是开始八股,有些鼠鼠背过,但明显感觉他应该没背过,或者干脆他就不是做&nbsp;Java&nbsp;的,吟唱过程没有一点反应,全程也没有追问,属于是那种问完只要你吱声就行,当然也可能是鼠鼠前面手撕表现太骚让他兴致全无,毕竟刚开始他还不停甩头发防止刘海遮住眼睛看不清,后面全程没动过头,估计都是隔着铁窗般的刘海望着鼠鼠我尴尬直播。1、HashMap&nbsp;怎么实现的2、HashMap&nbsp;的&nbsp;Index&nbsp;如何计算3、HashMap&nbsp;和&nbsp;HashTable&nbsp;有什么区别4、HashMap&nbsp;和&nbsp;TreeMap&nbsp;的区别5、HashMap线程安全吗,有什么线程安全的?6、ConcurrentMap&nbsp;的实现原理7、HashMap&nbsp;插入、查找、删除的时间复杂度8、最坏情况的时间复杂度9、高并发场景下除了线程不安全还会存在哪些问题?10、HashMap&nbsp;的扩容11、TCP&nbsp;三次握手和四次挥手12、和&nbsp;UDP&nbsp;的区别13、HTTP&nbsp;中的长链接和短链接的区别14、怎么去关闭链接15、HTTP&nbsp;2&nbsp;相比&nbsp;HTTP1.1做了一些什么改进呢?16、UDP&nbsp;有实践操作过吗?17、Spring&nbsp;中&nbsp;bean&nbsp;的生命周期18、bean&nbsp;有哪些作用域(答了单例,追问还有哪些)19、Spring&nbsp;AOP&nbsp;的实现原理(答了动态代理,追问具体怎么实现)20、Spring&nbsp;AOP&nbsp;有哪些定义,通知类型21、Spring&nbsp;中的事务22、Spring&nbsp;中类的构造机制有哪些(有哪些注入方式)23、IoC&nbsp;容器的原理,如何管理&nbsp;bean&nbsp;的24、如何去解决循环依赖问题25、介绍实习26、具体讲一下乐观锁方案的实现(实习项目里的)27、SQL&nbsp;语句如何写的(实习项目里的)28、kafka、rabbit&nbsp;MQ、rocket&nbsp;MQ如何选择(项目里的)29、list&nbsp;的常见数据结构有哪些30、redis&nbsp;如何保证原子性?31、redis&nbsp;的持久化机制32、redis&nbsp;分布式锁如何删除锁33、删除锁如何保证原子性34、如何实现锁的自动续期35、市面上还有哪些常见的分布式锁?有使用过吗?36、介绍下策略模式37、策略模式的缺点面完隔天挂。
点赞 评论 收藏
分享
评论
17
85
分享

创作者周榜

更多
牛客网
牛客企业服务