Spring Boot
上传文件,根据官方上传文件示例修改的,打包成WAR
上传到Tomcat
上,主要步骤是创建异常类,属性类,接口类与控制器类,最后进行少量修改打包部署到服务器上。
Tomcat 9.0.30
Spring boot 2.2.2
选择Spring initializer
:
改一下包名,打包选项JAR
/WAR
均可,选JAR
的话可以在构建的时候再生成WAR
。
这里用的是模板引擎Thymeleaf
,选择Spring Web
与Thymeleaf
。
最后点击完成。
4个包,service
,properties
,controller
,exception
。
处理两个异常,分别是存储异常与存储文件找不到异常。
StorageException
package kr.test.exception;
public class StorageException extends RuntimeException
{
public StorageException(String message)
{
super(message);
}
public StorageException(String message,Throwable cause)
{
super(message,cause);
}
}
StorageFileNotFoundException
package kr.test.exception;
public class StorageFileNotFoundException extends StorageException
{
public StorageFileNotFoundException(String message)
{
super(message);
}
public StorageFileNotFoundException(String message,Throwable cause)
{
super(message,cause);
}
}
Exception(String message,Throwable cause);
这个构造函数中的cause
是引起这个异常的异常,允许空值,如果是空值则表示这个引起这个异常的异常不存在或者未知。
新建StorageProperties.java
,设定存储文件的位置,就是location
的值,可以使用../../
这样的值,什么也不加的话会在项目路径下新建文件夹,若有同名的文件夹会被删除再重新创建。
注意一下权限的问题,后面部署到Tomcat
上面时可能会因为没有写权限而不能写入文件,要确保文件夹拥有写权限。
package kr.test.properties;
import org.springframework.boot.context.properties.ConfigurationProperties;
@ConfigurationProperties("storage")
public class StorageProperties {
private String location = "upload_dir";
public String getLocation()
{
return location;
}
public void setLocation(String location)
{
this.location = location;
}
}
这里使用@ConfigurationProperties
会报红,提示没有@EnableConfigurationProperties
:
可以先不管,后面会在Main
类中添加@EnableConfigurationProperties(StorageProperties.class)
。
先加一个StorageService
接口:
package kr.test.service;
import org.springframework.core.io.Resource;
import org.springframework.web.multipart.MultipartFile;
import java.nio.file.Path;
import java.util.stream.Stream;
public interface StorageService
{
void init();
void store(MultipartFile file);
Stream<Path> loadAll();
Path load(String filename);
Resource loadAsResource(String filename);
void deleteAll();
}
然后新建一个FileSystemStorageService
实现该接口:
package kr.test.service;
import kr.test.exception.StorageException;
import kr.test.exception.StorageFileNotFoundException;
import kr.test.properties.StorageProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.Resource;
import org.springframework.core.io.UrlResource;
import org.springframework.stereotype.Service;
import org.springframework.util.FileSystemUtils;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;
import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;
import java.util.stream.Stream;
@Service
public class FileSystemStroageService implements StorageService
{
private final Path rootLocation;
@Autowired
public FileSystemStroageService(StorageProperties properties)
{
this.rootLocation = Paths.get(properties.getLocation());
}
@Override
public void init()
{
try {
Files.createDirectories(rootLocation);
}
catch (IOException e)
{
throw new StorageException("Could not initialize storage",e);
}
}
@Override
public void deleteAll()
{
FileSystemUtils.deleteRecursively(rootLocation.toFile());
}
@Override
public Path load(String filename)
{
return rootLocation.resolve(filename);
}
@Override
public Stream<Path> loadAll()
{
try
{
return Files.walk(rootLocation,1)
.filter(path -> !path.equals(rootLocation))
.map(rootLocation::relativize);
}
catch (IOException e)
{
throw new StorageException("Failed to read stored file.",e);
}
}
@Override
public Resource loadAsResource(String filename)
{
try {
Path file = load(filename);
Resource resource = new UrlResource(file.toUri());
if(resource.exists() || resource.isReadable())
{
return resource;
}
else {
throw new StorageFileNotFoundException("Could not read file: "+filename);
}
}
catch (MalformedURLException e)
{
throw new StorageFileNotFoundException("Could not read file : "+filename,e);
}
}
@Override
public void store(MultipartFile file)
{
String filename = StringUtils.cleanPath(file.getOriginalFilename());
try {
if(file.isEmpty())
{
throw new StorageException("Failed to store empty file : "+filename);
}
if(filename.contains(".."))
{
throw new StorageException("Cannot store file with relative path outside current directory"+filename);
}
try(InputStream inputStream = file.getInputStream())
{
Files.copy(inputStream,rootLocation.resolve(filename), StandardCopyOption.REPLACE_EXISTING);
}
}
catch (IOException e)
{
throw new StorageException("Failed to store file : "+ filename,e);
}
}
}
init
@Override
public void init()
{
try {
Files.createDirectories(rootLocation);
}
catch (IOException e)
{
throw new StorageException("Could not initialize storage",e);
}
}
使用java.nio.file.Files.createDirectories()
创建存储目录,可以建立多级目录。
deleteAll
@Override
public void deleteAll()
{
FileSystemUtils.deleteRecursively(rootLocation.toFile());
}
使用工具类FileSystemUtils
的方法递归删除文件与文件夹。
参数是一个File
,下面是方法源码:
public static boolean deleteRecursively(File root)
{
if (root != null && root.exists())
{
if (root.isDirectory())
{
File[] children = root.listFiles();
if (children != null)
{
for (File child : children)
{
deleteRecursively(child);
}
}
}
return root.delete();
}
return false;
}
首先判断根是否为空,不为空的话判断是否是目录,不是目录的话直接删除,是目录的话,利用listFiles()
获取所有文件及文件夹,判断是否为空并进行递归删除。
load
@Override
public Path load(String filename) {
return rootLocation.resolve(filename);
}
Path.resolve(String)
返回相对于this
的路径,具体来说,等于执行
cd rootLocation
cd filename
pwd
返回pwd
的值。
loadAll
@Override
public Stream<Path> loadAll()
{
try
{
return Files.walk(rootLocation,1)
.filter(path -> !path.equals(rootLocation))
.map(rootLocation::relativize);
}
catch (IOException e)
{
throw new StorageException("Failed to read stored file.",e);
}
}
Files.walk
遍历目录,返回一个Stream
返回的Stream
包含打开的一个或多个目录的引用,会在Stream
关闭时关闭,第二个参数1表示遍历的最大深度。
然后对这个Stream
进行filter
过滤,这里是把与rootLocation
不相等的Path
留下,注意是不相等,就是留下filter()
中条件为真的Path
,不是把条件为真的Path
给"删去"。
最后进行map
,relativize
返回参数相对于调用者的路径,这里是返回Stream
中的每个Path
相对于rootLocation
的路径。
对于relativize
,无论什么情况下:
Path a = xxxx;
Path b = xxxx;
都有
a.relativize(a.resolve(b)).equals(b)
为真。
loadAsResource
@Override
public Resource loadAsResource(String filename)
{
try {
Path file = load(filename);
Resource resource = new UrlResource(file.toUri());
if(resource.exists() || resource.isReadable())
{
return resource;
}
else {
throw new StorageFileNotFoundException("Could not read file: "+filename);
}
}
catch (MalformedURLException e)
{
throw new StorageFileNotFoundException("Could not read file : "+filename,e);
}
}
这里的Resource
是org.springframework.core.io.Resource
,是一个接口,可以通过它访问各种资源,实现类有UrlResource
,InputStreamResource
等,这里利用Path.toUri()
把file
转换为Resource
后,判断这个源是否存在或者是否可读并返回,否则抛出存储文件找不到异常。
store
@Override
public void store(MultipartFile file)
{
String filename = StringUtils.cleanPath(file.getOriginalFilename());
try {
if(file.isEmpty())
{
throw new StorageException("Failed to store empty file : "+filename);
}
if(filename.contains(".."))
{
throw new StorageException("Cannot store file with relative path outside current directory"+filename);
}
try(InputStream inputStream = file.getInputStream())
{
Files.copy(inputStream,rootLocation.resolve(filename), StandardCopyOption.REPLACE_EXISTING);
}
}
catch (IOException e)
{
throw new StorageException("Failed to store file : "+ filename,e);
}
getOriginalFilename()
获取文件原名字,然后通过StringUtils.cleanPath()
将其标准化,处理掉.
与..
,然后判断文件是否为空与是否包含相对路径,没有的话利用Files.copy()
进行复制,resolve
获取filename
相对于rootLocation
的值,复制选项是REPLACE_EXISTING
。
StandardCopyOption
有三个可选值:
ATOMIC_MOVE
:原子性的移动操作,一般在移动文件或目录时使用COPY_ATTRIBUTES
:复制属性,可以保留源文件或源目录的属性REPLACE_EXISTING
:替换已存在的文件新建FileUploadController
:
package kr.test.controller;
import kr.test.exception.StorageFileNotFoundException;
import kr.test.service.StorageService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.Resource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.servlet.mvc.method.annotation.MvcUriComponentsBuilder;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
import java.util.stream.Collectors;
@Controller
public class FileUploadController {
private final StorageService storageService;
@Autowired
public FileUploadController(StorageService storageService)
{
this.storageService = storageService;
}
@GetMapping("/")
public String listUploadedFiles(Model model)
{
model.addAttribute("files",storageService.loadAll().map(
path -> MvcUriComponentsBuilder.fromMethodName(FileUploadController.class,
"serveFile",path.getFileName().toString()).build().toString())
.collect(Collectors.toList()));
return "uploadForm";
}
@GetMapping("/files/{filename:.+}")
@ResponseBody
public ResponseEntity<Resource> serveFile(@PathVariable String filename)
{
Resource file = storageService.loadAsResource(filename);
return ResponseEntity.ok().header(HttpHeaders.CONTENT_DISPOSITION,"attachment;filename=\""+file.getFilename()+"\"").body(file);
}
@PostMapping("/")
public String handleFileUpload(@RequestParam("file") MultipartFile file, RedirectAttributes redirectAttributes)
{
storageService.store(file);
redirectAttributes.addFlashAttribute("message","You successully uploaded "+file.getOriginalFilename()+"!");
return "redirect:/";
}
@ExceptionHandler(StorageFileNotFoundException.class)
public ResponseEntity<?> handleStorageFileNotFound(StorageFileNotFoundException e)
{
return ResponseEntity.notFound().build();
}
}
listUploadedFiles
@GetMapping("/")
public String listUploadedFiles(Model model)
{
model.addAttribute("files",storageService.loadAll().map(
path -> MvcUriComponentsBuilder.fromMethodName(FileUploadController.class,
"serveFile",path.getFileName().toString()).build().toString())
.collect(Collectors.toList()));
return "uploadForm";
}
@GetMapping
是@RequestMapping(method = RequestMethod.GET)
的简化写法,将HTTP GET
路径映射到特定的处理方法上。
方法的参数是Spring MVC
中的Model
,Model
实质上是一个Map
,添加的key
可以在视图中用${key}
获取值,比如,这里添加了files
作为key
,则在视图中可用${files}
获取值。
MvcUriComponentsBuilder
可以为Controller
指定uri
,fromMethod
简单地说就是会调用FileUploadController
的serveFile()
,参数是path.getFileName().toString()
,由于serveFile()
返回的是Stream
,利用Stream
的collect
将其转换成List
添加到model
中,然后返回uploadForm
,表示这是视图的名称,会到resource/templates
下寻找。
这里说一下RequestMapping
与Model
。
RequestMapping
可以用@RequestMapping()
来映射URL
,可以映射到某个类或某个具体方法。@RequestMapping
常用的有以下属性:
value
:请求的URL路径,支持URL
模板,正则表达式method
:HTTP
请求方法,如GET
、POST
、PUT
、DELTE
等consumes
:允许的媒体类型,如consumes="application/json"
,对应于HTTP
请求的Content-Type
produces
:相应的媒体类型,如produces="application/json"
,对于HTTP
请求的Accept
params
:请求参数,如params="action=update"
headers
:请求头Spring
提供了简化的@RequestMapping
,提供了新的注解来标识HTTP
方法:
@GetMapping
@PostMapping
@PutMapping
所以这里的@GetMapping
是简化了的@RequestMapping
。
Model
可以向Model
添加视图所需要的变量,Model
主要有以下方法:
Model addAttribute(Object value);
Model addAttribute(String name,Object value);
Model addAllAttributes(Map attributes);
Model addAllAttributes(Collection<?> attributes);
Model mergeAttributes(Map attributes);
boolean containAttribute(String name);
addAttribute()
添加一个变量,对于两个参数的,使用name
作为变量名称,后面的是值,对于只有一个Object
的,变量的名字就是类名字首字母小写后转为的Java
变量。
addAttributes()
添加多个变量,如果变量存在则覆盖,其中参数为Collection>
的方法添加变量名时与addAttribute(Object)
的命名规范类似。
mergeAttributes()
也是添加多个变量,不过变量已存在的话会忽略。
containAttributte()
判断是否存在变量。
serveFile
@GetMapping("/files/{filename:.+}")
@ResponseBody
public ResponseEntity<Resource> serveFile(@PathVariable String filename)
{
Resource file = storageService.loadAsResource(filename);
return ResponseEntity.ok().header(HttpHeaders.CONTENT_DISPOSITION,"attachment;filename=\""+file.getFilename()+"\"").body(file);
}
这里的@GetMapping
用来表示显示的用来供下载的文件名,@ResponseBody
表示直接返回内容而不是视图名,因为默认返回的是视图名称,@ResponseBody
对于String
直接返回,否则默认使用Jackson
进行序列化。
@PathVariable
表示这是@GetMapping
中的参数的值,可以省略,默认同名,就是形参的名字与GetMapping
中的名字一样,从中取值赋给形参,通过filename
加载资源后,作为ResponseEntity
的请求体。
ResponseEntity
从HttpEntity
继承而来,ResponseEntity.ok()
是一个静态方法,表示构建一个状态为ok
的ResponseEntity
,然后添加请求头。
HttpHeaders.CONTENT_DISPOSITION,"attachment;filename=\""+file.getFilename()+"\""
content_disposition
表示文件是直接在浏览器打开还是下载,attachment
表示是要下载,文件名为file.getFilename()
。
handleFileUpload
@PostMapping("/")
public String handleFileUpload(@RequestParam("file") MultipartFile file,RedirectAttributes redirectAttributes)
{
storageService.store(file);
redirectAttributes.addFlashAttribute("message","You successully uploaded "+file.getOriginalFilename()+"!");
return "redirect:/";
}
@PostMapping()
与@GetMapping()
类似,只不过方法不是GET
而是POST
。@RequestParam
表示请求参数,里面的是请求参数的名字,使用MultipartFile
来处理文件上传。
RedirectAttributes
是用于重定向使用的,可以附带参数,RedirectAttributes
有两种带参的形式:
addAttribute(String name,Object value);
addFlashAttribute(String name,Object value);
addAttribute()
相当于直接在重定向的地址添
name=value
这样的形式,会将参数暴露在重定向的地址上。
而addFlashAttribute()
隐藏了参数,只能在重定向的页面中获取参数的值,用到了session
,session
跳转到页面后就会删除对象。
handleFileUpload
首先保存文件,然后添加一个保存成功的信息,由于Controller
中重定向可以返回以redirect:
或以forward:
为前缀的URI
,因此返回redirect:/
,重定向到根。
handleStorageFileNotFound
@ExceptionHandler(StorageFileNotFoundException.class)
public ResponseEntity<?> handleStorageFileNotFound(StorageFileNotFoundException e)
{
return ResponseEntity.notFound().build();
}
@ExceptionHandler()
注解会处理Controller
层抛出的所有StorageFileNotFoundException
类及其子类的异常,ResponseEntity.notFound()
相当于返回404
标识码。
main
package kr.test;
import kr.test.properties.StorageProperties;
import kr.test.service.StorageService;
import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Bean;
@SpringBootApplication
@EnableConfigurationProperties(StorageProperties.class)
public class TestApplication {
public static void main(String[] args) {
SpringApplication.run(TestApplication.class, args);
}
@Bean
CommandLineRunner init(StorageService storageService)
{
return (args) ->
{
storageService.deleteAll();
storageService.init();
};
}
}
在原来的基础上添加
@EnableConfigurationProperties(StorageProperties.class)
与
@Bean
CommandLineRunner init(StorageService storageService)
{
return (args) ->
{
storageService.deleteAll();
storageService.init();
};
}
@EnableConfigurationProperties
可以为带有@ConfigurationProperties
注解的Bean
提供有效的支持,将带有@Configuration
注解的类注入为Spring
的Bean
,在这里是使StorageProperties
的@ConfigurationProperties
生效,如果没有这一行会报红:
@Bean
标注在方法上,等价于Spring
的xml
配置文件的
,注册bean
对象。
CommandLineRunner
接口用于应用初始化后去执行一段代码逻辑,这段代码在整个应用周期只执行一次。
这里可以设置一些环境配置属性,Spring Boot
允许准备多个配置文件,在部署时可以指定那个配置文件覆盖默认的application.properties
。这里是有关上传文件的设置:
默认如下:
spring.servlet.multipart.enabled=true
spring.servlet.multipart.file-size-threshold=0
spring.servlet.multipart.location=
spring.servlet.multipart.max-file-size=1MB
spring.servlet.multipart.max-request-size=10MB
spring.servlet.multipart.resolve-lazily=false
enabled
表示允许上传,file-size-threshold
表示上传文件超过一定长度就先写入临时文件,单位MB
或KB
,location
是临时文件存放目录,不设定的话使用Web
服务器提供的临时目录。max-file-size
表示单个文件最大长度,默认1MB
,max-request-size
为单次HTTP
请求上传的最大长度,默认10MB
,resolve-lazily
表示文件和参数被访问的时候再解析成文件。
在这里只需把max-size
调大一点即可。
这是在本地进行的测试,直接在IDE上点击运行应用,然后打开浏览器输入:
localhost:8080
Spring Boot
通常打成JAR
包或WAR
包,这里部署到Tomcat
上的是打成WAR
包。
pom.xml
中,
改成WAR
:
Tomcat
依赖Spring Boot
默认自带了一个嵌入式的Tomcat
,需要把Tomcat
依赖方式改为provided
。
pom.xml
中,在
添加:
<dependency>
<groupId>org.springframework.bootgroupId>
<artifactId>spring-boot-starter-tomcatartifactId>
<scope>providedscope>
dependency>
Main
类修改Main
类,让其继承SpringBootServletInitializer
,重载configure()
,同时main()
保持不变。
@SpringBootApplication
public class MainClass extends SpringBootServletInitializer
{
@Override
protected SpringApplicationBuilder configure(SpringApplicationBuilder application)
{
return application.sources(MainClass.class);
}
//main()不变
}
这个很重要,设置不当的话就无法访问了,主要就是四个路径:
action
:@GetMapping
@PostMapping
redirect
action
这个是绝对路径,要加上/war项目名
。
/war项目名/上传路径名
比如这里WAR
项目名是kr
,上传路径名是upload
。
@GetMapping
这个是相对路径,相对于当前项目的路径,不用加上/war项目名
。
/上传路径名
这里是upload
。
@PostMapping
与@GetMapping
一样,上传路径名。
/上传路径名
redirect
这个是返回的重定向的路径名,相对路径,与上两个一样,也是上传路径名.
/上传路径名
在
中添加
,指定打包出来的WAR
名,注意这个要与上面的WAR
项目名一样,这里设置的是kr。
Maven
打包运行
mvn package
即可打包,对于IDEA,可以在IDEA右侧栏的Maven
中,打开Lifecycle
,选择package
:
打包后的WAR
默认放在target
下,名字默认为
+
。
上传的话笔者用的是密钥认证的scp
:
scp -i xxxx\id_rsa kr.war username@ip:/usr/local/tomcat/webapps
放到服务器的Tomcat
下的webapps
目录。
Tomcat
进入到Tomcat
目录的bin
下:
cd /usr/local/tomcat/bin
./startup.sh
如果正在运行的话就不用启动了,因为会自动检测到webapps
目录的变化,把新的WAR
自动解包。
略过,与本地测试类似,不过要注意的是上传的文件夹是在tomcat/bin
下,想要修改的话可以修改StorageProperties
的location
。