Spring Boot 实现大文件分片上传

蔚落 2023-10-10 20:20 64阅读 0赞

### 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());
        }
    }