Spring Boot 实现大文件分片上传
在 Maven 中添加依赖项
需要在项目中添加以下依赖项：
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>commons-fileupload</groupId>
<artifactId>commons-fileupload</artifactId>
<version>1.4</version>
</dependency>
创建控制器类
创建一个控制器类，用于处理上传请求。在该类中，需要实现以下操作：
- 接收上传的文件
- 将文件分成多个部分
- 将每个部分保存到磁盘上的临时文件中
- 当所有部分都上传完成后，将它们合并成一个完整的文件
下面是一个示例代码：
@RestController
public class FileUploadController {
private static final String UPLOAD_DIRECTORY = "/tmp/uploads";
@PostMapping("/upload")
public ResponseEntity<String> upload(@RequestParam("file") MultipartFile file,
@RequestParam("fileName") String fileName,
@RequestParam("chunkNumber") int chunkNumber,
@RequestParam("totalChunks") int totalChunks) throws IOException {
File uploadDirectory = new File(UPLOAD_DIRECTORY);
if (!uploadDirectory.exists()) {
uploadDirectory.mkdirs();
}
File destFile = new File(UPLOAD_DIRECTORY + File.separator + fileName + ".part" + chunkNumber);
FileUtils.copyInputStreamToFile(file.getInputStream(), destFile);
if (chunkNumber == totalChunks) {
// 如果所有部分都已上传，则将它们组合成一个完整的文件
String targetFilePath = UPLOAD_DIRECTORY + File.separator + fileName;
for (int i = 1; i <= totalChunks; i++) {
File partFile = new File(UPLOAD_DIRECTORY + File.separator + fileName + ".part" + i);
try (FileOutputStream fos = new FileOutputStream(targetFilePath, true)) {
FileUtils.copyFile(partFile, fos);
partFile.delete();
}
}
}
return ResponseEntity.ok("Upload successful");
}
}
前端实现
在前端，需要使用 JavaScript 将文件分割成多个部分，并向后端发送分块数据。以下是一个示例代码：
javascript复制代码function uploadFile(file) {
const chunkSize = 1024 * 1024; // 每个部分的大小（1MB）
const totalChunks = Math.ceil(file.size / chunkSize); // 总部分数
let currentChunk = 1;
let startByte = 0;
while (startByte < file.size) {
// 分割文件为多个部分，并上传每个部分
const endByte = Math.min(startByte + chunkSize, file.size);
const chunk = file.slice(startByte, endByte);
const formData = new FormData();
formData.append('file', chunk);
formData.append('fileName', file.name);
formData.append('chunkNumber', currentChunk);
formData.append('totalChunks', totalChunks);
axios.post('/upload', formData);
startByte += chunkSize;
currentChunk++;
}
}
通过以上步骤，就可以使用 Spring Boot 实现大文件分片上传了。
第二种：
package cn.js.Controller;
import org.apache.commons.io.FileUtils;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
/**
*@description:
* 1.前端实现：
*
* 使用JavaScript实现前端上传代码。从前端发送请求时，将文件切成若干个小块（每个块大小在1MB到10MB之间），
* 并发送给后端。当所有的块都上传完成后，再向后端发送“合并”请求。
*
* 2.后端实现：
*
* 在后端控制器中，首先接收上传的文件，并按照指定的大小进行分块。然后将每一个块存储为临时文件。
* 当所有块都上传完成后，再将这些临时文件合并成一个完整的文件。
*
* 具体实现方法如下：
**/
/**
*@description:
* 在上述代码中，我们定义了一个接口“/file/upload”，用于接收上传的块。在接收到一个块时，
* 首先判断是否为第一个块，如果是，则新建一个空文件。然后将当前块写入文件。
* 如果是最后一个块，则调用 mergeFile() 方法将所有块合并成一个完整的文件。
*
* mergeFile() 方法中，我们首先获取所有以 guid 为前缀的临时文件，
* 并按照文件名排序。然后从第一个文件开始，将每个文件的内容写入目标文件中，
* 并删除临时文件。最后输出合并完成的消息。
*
* 这样，就可以使用Spring Boot实现大文件分片上传了。需要注意的是，
* 为了防止文件名冲突，可以为每个上传任务生成一个唯一的 guid，并使用该 guid
* 作为文件名的前缀。此外，如果上传任务长时间未完成，还需要定期清理临时文件。
**/
@RestController
@RequestMapping("/file")
public class FileUploadController {
private final String UPLOAD_PATH = "D:/upload/";
@PostMapping("/upload")
public String upload(HttpServletRequest request) throws Exception {
String fileName = request.getHeader("fileName");
String guid = request.getHeader("guid");
int chunkIndex = Integer.parseInt(request.getHeader("chunkIndex"));//分片文件的索引从0开始
int totalChunks = Integer.parseInt(request.getHeader("totalChunks"));//分片文件的索引总数
// 获取上传的文件
File file = new File(UPLOAD_PATH + fileName);
// 如果是第一个块，则新建一个文件
if (chunkIndex == 0) {
file.createNewFile();
}
// 将当前块写入文件
FileOutputStream fos = new FileOutputStream(file, true);
InputStream is = request.getInputStream();
byte[] buf = new byte[1024];
int len;
while ((len = is.read(buf)) != -1) {
fos.write(buf, 0, len);
}
// 如果是最后一个块，则合并文件
if (chunkIndex == totalChunks - 1) {
mergeFile(file, guid);
}
return "success";
}
private void mergeFile(File file, String guid) throws Exception {
String fileName = file.getName();
String ext = fileName.substring(fileName.lastIndexOf("."));
File newFile = new File(UPLOAD_PATH + guid + "." + ext);
List<File> files = Arrays.asList(file.getParentFile().listFiles((dir, name) -> name.startsWith(guid)));
Collections.sort(files, Comparator.comparing(File::getName));
FileOutputStream fos = new FileOutputStream(newFile);
byte[] buf = new byte[1024];
int len;
for (File f : files) {
FileInputStream fis = new FileInputStream(f);
while ((len = fis.read(buf)) != -1) {
fos.write(buf, 0, len);
}
fis.close();
f.delete();
}
fos.close();
System.out.println("文件合并完成：" + newFile.getAbsolutePath());
}
}