@UploadMode(mode = UploadModeEnum.MAPPED_BYTEBUFFER)@Slf4jpublic class MappedByteBufferUploadStrategy extends SliceUploadTemplate {@Autowiredprivate FilePathUtil filePathUtil;@Value("${upload.chunkSize}")private long defaultChunkSize;@Overridepublic boolean upload(FileUploadRequestDTO param) {RandomAccessFile tempRaf = null;FileChannel fileChannel = null;MappedByteBuffer mappedByteBuffer = null;try {String uploadDirPath = filePathUtil.getPath(param);File tmpFile = super.createTmpFile(param);tempRaf = new RandomAccessFile(tmpFile, "rw");fileChannel = tempRaf.getChannel();long chunkSize = Objects.isNull(param.getChunkSize()) ? defaultChunkSize * 1024 * 1024: param.getChunkSize();//写入该分片数据long offset = chunkSize * param.getChunk();byte[] fileData = http://www.kingceram.com/post/param.getFile().getBytes();mappedByteBuffer = fileChannel.map(FileChannel.MapMode.READ_WRITE, offset, fileData.length);mappedByteBuffer.put(fileData);boolean isOk = super.checkAndSetUploadProgress(param, uploadDirPath);return isOk;} catch (IOException e) {log.error(e.getMessage(), e);} finally {FileUtil.freedMappedByteBuffer(mappedByteBuffer);FileUtil.close(fileChannel);FileUtil.close(tempRaf);}return false;}}
@Slf4jpublic abstract class SliceUploadTemplate implements SliceUploadStrategy {public abstract boolean upload(FileUploadRequestDTO param);protected File createTmpFile(FileUploadRequestDTO param) {FilePathUtil filePathUtil = SpringContextHolder.getBean(FilePathUtil.class);param.setPath(FileUtil.withoutHeadAndTailDiagonal(param.getPath()));String fileName = param.getFile().getOriginalFilename();String uploadDirPath = filePathUtil.getPath(param);String tempFileName = fileName + "_tmp";File tmpDir = new File(uploadDirPath);File tmpFile = new File(uploadDirPath, tempFileName);if (!tmpDir.exists()) {tmpDir.mkdirs();}return tmpFile;}@Overridepublic FileUploadDTO sliceUpload(FileUploadRequestDTO param) {boolean isOk = this.upload(param);if (isOk) {File tmpFile = this.createTmpFile(param);FileUploadDTO fileUploadDTO = this.saveAndFileUploadDTO(param.getFile().getOriginalFilename(), tmpFile);return fileUploadDTO;}String md5 = FileMD5Util.getFileMD5(param.getFile());Map map = new HashMap<>();map.put(param.getChunk(), md5);return FileUploadDTO.builder().chunkMd5Info(map).build();}/*** 检查并修改文件上传进度*/public boolean checkAndSetUploadProgress(FileUploadRequestDTO param, String uploadDirPath) {String fileName = param.getFile().getOriginalFilename();File confFile = new File(uploadDirPath, fileName + ".conf");byte isComplete = 0;RandomAccessFile accessConfFile = null;try {accessConfFile = new RandomAccessFile(confFile, "rw");//把该分段标记为 true 表示完成System.out.println("set part " + param.getChunk() + " complete");//创建conf文件文件长度为总分片数 ， 每上传一个分块即向conf文件中写入一个127 ， 那么没上传的位置就是默认0,已上传的就是Byte.MAX_VALUE 127accessConfFile.setLength(param.getChunks());accessConfFile.seek(param.getChunk());accessConfFile.write(Byte.MAX_VALUE);//completeList 检查是否全部完成,如果数组里是否全部都是127(全部分片都成功上传)byte[] completeList = FileUtils.readFileToByteArray(confFile);isComplete = Byte.MAX_VALUE;for (int i = 0; i < completeList.length && isComplete == Byte.MAX_VALUE; i++) {//与运算, 如果有部分没有完成则 isComplete 不是 Byte.MAX_VALUEisComplete = (byte) (isComplete & completeList[i]);System.out.println("check part " + i + " complete?:" + completeList[i]);}} catch (IOException e) {log.error(e.getMessage(), e);} finally {FileUtil.close(accessConfFile);}boolean isOk = setUploadProgress2Redis(param, uploadDirPath, fileName, confFile, isComplete);return isOk;}/*** 把上传进度信息存进redis*/private boolean setUploadProgress2Redis(FileUploadRequestDTO param, String uploadDirPath,String fileName, File confFile, byte isComplete) {RedisUtil redisUtil = SpringContextHolder.getBean(RedisUtil.class);if (isComplete == Byte.MAX_VALUE) {redisUtil.hset(FileConstant.FILE_UPLOAD_STATUS, param.getMd5(), "true");redisUtil.del(FileConstant.FILE_MD5_KEY + param.getMd5());confFile.delete();return true;} else {if (!redisUtil.hHasKey(FileConstant.FILE_UPLOAD_STATUS, param.getMd5())) {redisUtil.hset(FileConstant.FILE_UPLOAD_STATUS, param.getMd5(), "false");redisUtil.set(FileConstant.FILE_MD5_KEY + param.getMd5(),uploadDirPath + FileConstant.FILE_SEPARATORCHAR + fileName + ".conf");}return false;}}/*** 保存文件操作*/public FileUploadDTO saveAndFileUploadDTO(String fileName, File tmpFile) {FileUploadDTO fileUploadDTO = null;try {fileUploadDTO = renameFile(tmpFile, fileName);if (fileUploadDTO.isUploadComplete()) {System.out.println("upload complete !!" + fileUploadDTO.isUploadComplete() + " name=" + fileName);//TODO 保存文件信息到数据库}} catch (Exception e) {log.error(e.getMessage(), e);} finally {}return fileUploadDTO;}/*** 文件重命名** @param toBeRenamed 将要修改名字的文件* @param toFileNewName 新的名字*/private FileUploadDTO renameFile(File toBeRenamed, String toFileNewName) {//检查要重命名的文件是否存在 ， 是否是文件FileUploadDTO fileUploadDTO = new FileUploadDTO();if (!toBeRenamed.exists() || toBeRenamed.isDirectory()) {log.info("File does not exist: {}", toBeRenamed.getName());fileUploadDTO.setUploadComplete(false);return fileUploadDTO;}String ext = FileUtil.getExtension(toFileNewName);String p = toBeRenamed.getParent();String filePath = p + FileConstant.FILE_SEPARATORCHAR + toFileNewName;File newFile = new File(filePath);//修改文件名boolean uploadFlag = toBeRenamed.renameTo(newFile);fileUploadDTO.setMtime(DateUtil.getCurrentTimeStamp());fileUploadDTO.setUploadComplete(uploadFlag);fileUploadDTO.setPath(filePath);fileUploadDTO.setSize(newFile.length());fileUploadDTO.setFileExt(ext);fileUploadDTO.setFileId(toFileNewName);return fileUploadDTO;}}
