SpringBoot文件上传/下载：从零开始构建文件处理系统

2023-06-27 02:23:59

解锁文件上传与下载的力量：使用 SpringBoot 构建完整系统

简介

在当今数据驱动的世界中，文件上传和下载已成为众多在线应用程序的关键功能。无论你是管理复杂的企业级系统还是维护个人博客，都需要一种简单而高效的方式来处理用户上传的文件。SpringBoot，一种流行的 Java 框架，提供了丰富的工具和组件，可让你轻松构建全面的文件上传和下载系统。

一、构建文件上传系统

1. 添加必需的依赖项

开始构建文件上传系统的第一步是将必要的依赖项添加到你的 SpringBoot 项目中。在 pom.xml 文件中，添加以下依赖项：

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-jpa</artifactId>
</dependency>

2. 定义文件上传控制器

接下来，创建一个控制器类来处理上传的文件：

@Controller
public class FileUploadController {

    @PostMapping("/upload")
    public String handleFileUpload(@RequestParam("file") MultipartFile file) {
        // 将文件存储到服务器
        String filePath = storageService.storeFile(file);

        // 将文件路径存储到数据库
        FileEntity fileEntity = new FileEntity();
        fileEntity.setFilePath(filePath);
        fileEntityRepository.save(fileEntity);

        // 重定向到文件列表页面
        return "redirect:/files";
    }
}

3. 定义文件下载控制器

同样，创建一个控制器类来处理文件下载请求：

@Controller
public class FileDownloadController {

    @GetMapping("/download/{id}")
    public ResponseEntity<Resource> handleFileDownload(@PathVariable Long id) {
        // 从数据库中获取文件路径
        FileEntity fileEntity = fileEntityRepository.findById(id).orElseThrow(() -> new NotFoundException("File not found"));

        // 从服务器获取文件
        Resource file = storageService.loadFileAsResource(fileEntity.getFilePath());

        // 设置文件下载头
        String fileName = fileEntity.getFileName();
        HttpHeaders headers = new HttpHeaders();
        headers.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\"" + fileName + "\"");

        // 返回文件下载响应
        return ResponseEntity.ok().headers(headers).body(file);
    }
}

4. 定义文件存储服务

文件存储服务负责处理文件的存储和持久化：

@Service
public class StorageService {

    public String storeFile(MultipartFile file) {
        // 将文件存储到服务器
        String filePath = "/path/to/file/" + file.getOriginalFilename();
        try {
            file.transferTo(new File(filePath));
        } catch (IOException e) {
            throw new RuntimeException("Failed to store file", e);
        }

        // 返回文件路径
        return filePath;
    }

    public Resource loadFileAsResource(String filePath) {
        // 从服务器获取文件
        try {
            Path file = Paths.get(filePath);
            Resource resource = new UrlResource(file.toUri());

            // 如果文件不存在，抛出异常
            if (resource.exists() || resource.isReadable()) {
                return resource;
            }
        } catch (MalformedURLException e) {
            throw new RuntimeException("Failed to load file", e);
        }

        // 如果文件不存在，抛出异常
        throw new RuntimeException("Failed to load file");
    }
}

5. 定义文件实体类

文件实体类表示数据库中的文件信息：

@Entity
public class FileEntity {

    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long id;

    private String fileName;

    private String filePath;

    // 省略getter和setter方法
}

6. 定义文件仓库接口

文件仓库接口用于操作数据库中的文件信息：

public interface FileEntityRepository extends JpaRepository<FileEntity, Long> {

}

7. 创建数据库表

SpringBoot 应用程序启动时会自动创建数据库表。在应用程序启动时运行以下代码来创建表：

@SpringBootApplication
public class Application {

    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }

    @Bean
    public CommandLineRunner init(FileEntityRepository fileEntityRepository) {
        return (args) -> {
            // 创建一些初始数据
            fileEntityRepository.save(new FileEntity("file1.txt", "/path/to/file1.txt"));
            fileEntityRepository.save(new FileEntity("file2.txt", "/path/to/file2.txt"));
        };
    }
}