本文主要是介绍原生分片文件上传、合并的具体步骤,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
前端是把文件分片传来的,获得第一片文件的时候要进行下特殊的判断:数据库中是否已经有了该文件(通过文件的MD5值进行查询),如果有的话就重命名,并且实现秒传。
//如果上传的是第一个分片
if (chunkIndex == 0) {
//封装一个查询,查询数据库中是否已经有FileInfoQuery infoQuery = new FileInfoQuery();infoQuery.setFileMd5(fileMd5);//设置md5infoQuery.setStatus(FileStatus.USING.getStatus()); //必须是usingList<FileInfo> dbFileList = fileInfoMapper.selectList(infoQuery); //查询这个分片是否上传了
if (!dbFileList.isEmpty()) { //已经有了,就需要重新命名FileInfo fileInfo = dbFileList.get(0);if (fileInfo.getFileSize() < 0) //TODO:到了后面修改,判断现有空间是否可以装下,此分片throw new BusinessException(ResponseCodeEnum.CODE_904);
/*封装信息*/fileInfo.setFileId(fileId);fileInfo.setFilePid(filePid);fileInfo.setUserId(userInfo.getUserId());fileInfo.setLastUpdateTime(LocalDateTime.now());fileInfo.setDelFlag(FileDelFlagEnums.USING.getFlag());fileInfo.setStatus(FileStatus.USING.getStatus());fileInfo.setFileMd5(fileMd5);
//进行重新命名操作fileName = autoRename(filePid, userInfo.getUserId(), fileName);fileInfo.setFileName(fileName);
//更新数据信息fileInfoMapper.insert(fileInfo);
resultDTO.setStatus(UploadStatus.UPLOAD_SECONDS.getStatus()); //设置秒传
//更新用户使用空间updateUserSpace(userInfo, fileInfo.getFileSize());return resultDTO;}
}
除了最后一个文件片段,其余的片段都要获取chunkIndex, 放到临时的目录下面去,返回信息到前端,以便前端可以继续上传其余片段。
//判断是不是上传最后一次切片
if (chunkIndex < chunks - 1) {resultDTO.setStatus(UploadStatus.UPLOADING.getStatus());//TODO:更新Redis中的信息return resultDTO;
}
到了最后一个片段,就需要获得文件的一些信息:后缀名、文件类型的枚举、不重复的文件名称等,封装好对应的信息,放到数据库中去。
String suffix = StringTools.getFileSuffix(fileName);
//真实的文件名
String realFileName = currentUserFolderName + suffix;
// 文件类型的枚举
FileTypeEnums fileTypeEnums = FileTypeEnums.getFileTypeBySuffix(suffix);
//自动重命名
fileName = autoRename(filePid,userInfo.getUserId(),fileName);
FileInfo fileInfo = new FileInfo(); //Entity
fileInfo.setFileId(fileId);
fileInfo.setUserId(userInfo.getUserId());
fileInfo.setFileName(fileName);
fileInfo.setFileMd5(fileMd5);
fileInfo.setFileName(getYearAndMouth() + "/" + realFileName );
fileInfo.setCreateTime(LocalDateTime.now()); //设置时间
fileInfo.setLastUpdateTime(LocalDateTime.now()); //最后一次更新时间
fileInfo.setFileCategory(fileTypeEnums.getType()); //设置种类
fileInfo.setFileType(fileTypeEnums.getType()); //设置种类
fileInfo.setStatus(FileStatus.TRANSFER.getStatus()); //设置转码中
fileInfo.setFolderType(FileFolderTypeEnums.FILE.getType()); //文件
fileInfo.setDelFlag(FileDelFlagEnums.USING.getFlag()); //使用中
//插入到数据库中去
this.fileInfoMapper.insert(fileInfo);
//TODO:更新redis中容量,Mysql中数据的占用
resultDTO.setStatus(UploadStatus.UPLOAD_FINISH.getStatus()); //设置文件上传完毕
上传完成所有的文件片段后,使用异步的方式,完成文件片段的合成。
TransactionSynchronizationManager.registerSynchronization(new TransactionSynchronization() {@Overridepublic void afterCommit() {fileInfoService.transferFile(fileInfo.getFileId(),userInfo); //异步请求}
});
这里的fileInfoService 也得在fileInfoServiceImpl中注入,而且要加上@Lazy,避免循环引用。
在transferFile中,需要根据传入的信息,进行前面insert数据的查询,如果不存在就直接返回,因为合并的文件不存在。
if(fileInfos == null || fileInfos.size() == 0 ||! FileStatus.TRANSFER.getStatus().equals(fileInfos.get(0).getStatus())){ //文件不是转码中或者文件为空return;
}
拼装临时目录、真实文件路径和真实文件名称。
//临时目录
String tempFolderName = appconfiguration.getProjectFolder() + Constants.FILE_FOLDER_TEMP;
String currentUserFolderName = sessionWebDTO.getUserId() + fileId;
File fileFolder = new File(tempFolderName + currentUserFolderName);
String fileSuffix = StringTools.getFileSuffix(fileInfo.getFileName()); //获得文件后缀
String date = getYearAndMouth(); //就是为了获得目标目录
String targetFolderName = appconfiguration.getProjectFolder() + Constants.FILE_FOLDER_FILE; //真实目录
获得上一步存入的临时文件夹。
File targetFolder = new File(targetFolderName + "/" + date);
调用union方法,合并文件。
union(fileFolder.getPath(),targetFilePath,fileInfo.getFileName(),true);
在union中进行文件的合并,根据文件路径,获得对应的临时文件夹,使用writeFile(RandomAccessFile)写入的最最终的文件,使用readFile(RandomAccessFile)读入每个临时文件,放到最终的文件中去。
try{writeFile = new RandomAccessFile(targetFile,"rw");byte[] b = new byte[1024 * 10];for(int i = 0;i < fileList.length; i++){int len = -1;File chunkFile = new File(dirPath + "/" + i); //分片文件RandomAccessFile readFile = null;try{readFile = new RandomAccessFile(chunkFile,"r"); //读取while((len = readFile.read(b))!= -1){writeFile.write(b,0,len);}}catch (Exception e){System.err.printf("-----------------------文件读取错误:%s :-------------------------",e.getMessage());throw new BusinessException("文件读取失败!");}finally {readFile.close();}}
}catch(Exception e){log.error("合并文件失败 :{}",fileName);throw new BusinessException("上传文件失败 !");
}finally {if(writeFile != null){try{writeFile.close(); //关闭对应的文件流}catch (IOException e){e.printStackTrace();}}if(delSource && dir.exists()){try {FileUtils.deleteDirectory(dir);} catch (IOException e) {e.printStackTrace();}}
}
最后关闭各种流,更新file_info中的文件大小、状态信息。
finally {
//更新转码状态FileInfoUpdateDto updateInfo = new FileInfoUpdateDto();
updateInfo.setFileSize(new File(targetFilePath).length()); //获得目标文件的大小updateInfo.setFileCover(cover);updateInfo.setStatus(transferSuccess ? FileStatus.USING.getStatus() : FileStatus.TRANSFER_FAIL.getStatus() );updateInfo.setFileId(fileId);updateInfo.setUserId(sessionWebDTO.getUserId());updateInfo.setNewValue(updateInfo.getStatus());updateInfo.setOldStatus(FileStatus.TRANSFER.getStatus());
//封装信息fileInfoMapper.updateFileStatusWithOldStatus(updateInfo);
}
这篇关于原生分片文件上传、合并的具体步骤的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!