SpringBoot大文件上传

// try 自动资源管理
try (RandomAccessFile randomAccessFile = new RandomAccessFile(fullFileName, "rw")) {
    // 分片大小必须和前端匹配，否则上传会导致文件损坏
    long chunkSize = dto.getChunkSize() == 0L ? defaultChunkSize : dto.getChunkSize().longValue();
    // 偏移量, 意思是从一个位置开始往文件写入，每一片的大小 * 已经存的块数
    long offset = chunkSize * (dto.getChunkNumber() - 1);
    // 定位到该分片的偏移量
    randomAccessFile.seek(offset);
    // 写入该分片数据
    randomAccessFile.write(dto.getFile().getBytes());
} catch (IOException e) {
    log.error("文件分片上传失败：" + e);
    return false;
}

/**
 * 是否上传完成
 *
 * @param dto      文件参数对象
 * @param confFile .conf 文件
 * @return true 文件上传完成
 */
public synchronized boolean checkUploadStatus(FileChunkDTO dto, File confFile) {
    try (RandomAccessFile confAccessFile = new RandomAccessFile(confFile, "rw")) {
        // 设置文件长度
        confAccessFile.setLength(dto.getTotalChunks());
        // 设置起始偏移量
        confAccessFile.seek(dto.getChunkNumber() - 1);
        // 将指定的一个字节写入文件中 127，
        confAccessFile.write(Byte.MAX_VALUE);

        byte[] completeStatusList = Files.readAllBytes(confFile.toPath());
        // 创建conf文件文件长度为总分片数，每上传一个分块即向conf文件中写入一个127，那么没上传的位置就是默认的0,已上传的就是127
        for (byte b : completeStatusList) {
            if (b != Byte.MAX_VALUE) {
                return false;
            }
        }
    } catch (IOException e) {
        log.error(e.getMessage(), e);
        return false;
    }
    // 上传成功记得删除conf文件
    confFile.delete();
    return true;
}

/**
 * 检查md5是否一致
 */
public boolean checkFileMd5(File tmpFile, String md5) {
    try (FileInputStream fis = new FileInputStream(tmpFile)) {
        String checkMd5 = DigestUtils.md5DigestAsHex(fis);
        return checkMd5.equals(md5);
    } catch (Exception e) {
        log.error("check file md5 error:" + e);
        throw new RuntimeException("校验md5值时发生错误");
    }
}

public void cleanUp(File tmpFile) {
    if (tmpFile.exists()) {
        tmpFile.delete();
    }
}

public void renameFile(File tmpFile, String fileName) {
    // 检查要重命名的文件是否存在，是否是文件
    if (!tmpFile.exists() || tmpFile.isDirectory()) {
        log.info("File does not exist: " + tmpFile.getName());
        return;
    }
    String parent = tmpFile.getParent();
    File newFile = new File(parent + File.separator + fileName);
    // 如果存在, 先删除
    if (newFile.exists()) {
        newFile.delete();
    }
    tmpFile.renameTo(newFile);
}

/**
 * 合并分片
 *
 * @param chunkFileFolderPath 分片保存目录
 * @param fileSavePath        文件合并后保存的目录
 * @param totalChunks         分片数量
 */
private boolean mergeChunks(String chunkFileFolderPath, String fileSavePath, Integer totalChunks) {
    // 检查分片是否都存在
    if (this.checkChunks(chunkFileFolderPath, totalChunks)) {
        File chunkFileFolder = new File(chunkFileFolderPath);
        File mergeFile = new File(fileSavePath);
        File[] chunks = chunkFileFolder.listFiles();
        // 排序
        List fileList = Arrays.asList(chunks);
        Collections.sort(fileList, (Comparator<File>) (o1, o2) -> {
            return Integer.parseInt(o1.getName()) - (Integer.parseInt(o2.getName()));
        });
        try {
            RandomAccessFile randomAccessFileWriter = new RandomAccessFile(mergeFile, "rw");
            byte[] bytes = new byte[1024];
            for (File chunk : chunks) {
                RandomAccessFile randomAccessFileReader = new RandomAccessFile(chunk, "r");
                int len;
                while ((len = randomAccessFileReader.read(bytes)) != -1) {
                    randomAccessFileWriter.write(bytes, 0, len);
                }
                randomAccessFileReader.close();
            }
            randomAccessFileWriter.close();
        } catch (Exception e) {
            return false;
        }
        return true;
    }
    return false;
}

/**
 * 检查分片是否都存在
 */
private boolean checkChunks(String chunkFileFolderPath, Integer totalChunks) {
    try {
        for (int i = 1; i <= totalChunks + 1; i++) {
            // tmp文件命名为 1.tmp、2.tmp
            File file = new File(chunkFileFolderPath + File.separator + i);
            if (!file.exists()) {
                return false;
            }
        }
    } catch (Exception e) {
        return false;
    }
    return true;
}