欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Spring Boot:使用 Spring Boot 上传文件

程序员文章站 2022-05-04 09:56:29
...

1、pom 包配置

我们使用 Spring Boot 版本 2.1.0、jdk 1.8、tomcat 8.0。

<parent>    <groupId>org.springframework.boot</groupId>    <artifactId>spring-boot-starter-parent</artifactId>    <version>2.1.0.RELEASE</version></parent>
<properties>    <java.version>1.8</java.version></properties>
<dependencies>    <dependency>        <groupId>org.springframework.boot</groupId>        <artifactId>spring-boot-starter-web</artifactId>    </dependency>    <dependency>        <groupId>org.springframework.boot</groupId>        <artifactId>spring-boot-starter-thymeleaf</artifactId>    </dependency>    <dependency>        <groupId>org.springframework.boot</groupId>        <artifactId>spring-boot-devtools</artifactId>        <optional>true</optional>    </dependency></dependencies>

引入了spring-boot-starter-thymeleaf做页面模板引擎,写一些简单的上传示例。

2、启动类设置

@SpringBootApplicationpublic class FileUploadWebApplication {
    public static void main(String[] args) throws Exception {        SpringApplication.run(FileUploadWebApplication.class, args);    }
    @Bean    public TomcatServletWebServerFactory tomcatEmbedded() {        TomcatServletWebServerFactory tomcat = new TomcatServletWebServerFactory();        tomcat.addConnectorCustomizers((TomcatConnectorCustomizer) connector -> {            if ((connector.getProtocolHandler() instanceof AbstractHttp11Protocol<?>)) {                //-1 means unlimited                ((AbstractHttp11Protocol<?>) connector.getProtocolHandler()).setMaxSwallowSize(-1);            }        });        return tomcat;    }
}

tomcatEmbedded 这段代码是为了解决,上传文件大于10M出现连接重置的问题。此异常内容 GlobalException 也捕获不到。

Spring Boot:使用 Spring Boot 上传文件

详细内容参考:Tomcat large file upload connection reset[1]

3、编写前端页面

上传页面

<!DOCTYPE html><html xmlns:th="http://www.thymeleaf.org"><body><h1>Spring Boot file upload example</h1><form method="POST" action="/upload" enctype="multipart/form-data">    <input type="file" name="file" /><br/><br/>    <input type="submit" value="Submit" /></form></body></html>

非常简单的一个 Post 请求,一个选择框选择文件,一个提交按钮,效果如下:

Spring Boot:使用 Spring Boot 上传文件

上传结果展示页面:

<!DOCTYPE html><html lang="en" xmlns:th="http://www.thymeleaf.org"><body><h1>Spring Boot - Upload Status</h1><div th:if="${message}">    <h2 th:text="${message}"/></div></body></html>

效果图如下:

Spring Boot:使用 Spring Boot 上传文件

4、编写上传控制类

访问 localhost 自动跳转到上传页面:

@GetMapping("/")public String index() {    return "upload";}

上传业务处理

@PostMapping("/upload") public String singleFileUpload(@RequestParam("file") MultipartFile file,                               RedirectAttributes redirectAttributes) {    if (file.isEmpty()) {        redirectAttributes.addFlashAttribute("message", "Please select a file to upload");        return "redirect:uploadStatus";    }
    try {        // Get the file and save it somewhere        byte[] bytes = file.getBytes();        Path path = Paths.get(UPLOADED_FOLDER + file.getOriginalFilename());        Files.write(path, bytes);
        redirectAttributes.addFlashAttribute("message",                "You successfully uploaded '" + file.getOriginalFilename() + "'");
    } catch (IOException e) {        e.printStackTrace();    }
    return "redirect:/uploadStatus";}

上面代码的意思就是,通过MultipartFile读取文件信息,如果文件为空跳转到结果页并给出提示;如果不为空读取文件流并写入到指定目录,最后将结果展示到页面。

MultipartFile是Spring上传文件的封装类,包含了文件的二进制流和文件属性等信息,在配置文件中也可对相关属性进行配置,基本的配置信息如下:

  • spring.http.multipart.enabled=true #默认支持文件上传.

  • spring.http.multipart.file-size-threshold=0 #支持文件写入磁盘.

  • spring.http.multipart.location=# 上传文件的临时目录

  • spring.http.multipart.max-file-size=1Mb # 最大支持文件大小

  • spring.http.multipart.max-request-size=10Mb # 最大支持请求大小

最常用的是最后两个配置内容,限制文件上传大小,上传时超过大小会抛出异常:

Spring Boot:使用 Spring Boot 上传文件

更多配置信息参考这里:Common application properties[2]

5、异常处理

@ControllerAdvicepublic class GlobalExceptionHandler {
    @ExceptionHandler(MultipartException.class)    public String handleError1(MultipartException e, RedirectAttributes redirectAttributes) {        redirectAttributes.addFlashAttribute("message", e.getCause().getMessage());        return "redirect:/uploadStatus";    }}

设置一个@ControllerAdvice用来监控Multipart上传的文件大小是否受限,当出现此异常时在前端页面给出提示。利用@ControllerAdvice可以做很多东西,比如全局的统一异常处理等,感兴趣的同学可以下来了解。