我们使用SpringBoot来实现文件的上传,本文中的SpringBoot环境和SpringBoot+MyBatis+通用Mapper中的环境基本一致,我们就不再重复造轮子了,而是直接上手这个功能。
依赖
在SpringBoot+MyBatis+通用Mapper中我们已经引入了Web开发相关的依赖,所以这里就不需要再次引入了。
配置
我们需要对SpringMVC的文件上传功能进行一些配置,比如MaxFileSize等等属性。我们使用JavaConfig来进行配置,写一个配置类UploadConfig,代码如下:
@ConfigurationpublicclassUploadConfig{@BeanpublicMultipartConfigElementmultipartConfigElement(){MultipartConfigFactoryfactory=newMultipartConfigFactory();//单个数据大小factory.setMaxFileSize("10240KB");///总上传数据大小factory.setMaxRequestSize("102400KB");returnfactory.createMultipartConfig();}
然后我们在application.yml文件中自定义一些配置:
file:upload:path:G:\temp\images\#文件上传目标路径allowTypes:#文件上传允许的类型-image/jpeg-image/png-image/bmp
使用ConfigurationProperties将配置读取到Java文件中:
@Data@Component@ConfigurationProperties(prefix="file.upload")publicclassUploadProperties{privateStringpath;privateList<String>allowTypes;}
代码
在写文件上传的代码之前,我们还需要做一些准备工作,比如准备一些工具类,可以方便我们实现文件的上传功能。
编写工具类
publicclassIDUtils{/***唯一ID生成器,可以生成唯一ID*@return*/publicstaticStringgenerateUniqueId(){returnUUID.randomUUID().toString()+System.currentTimeMillis();}}
- 文件名称替换工具,用来替换文件名称,避免文件名称重复而导致名称相同的文件被覆盖掉。
publicclassUploadUtils{/***文件名称替换工具,将文件名称替换为随机名称*@paramoldName*@return*/publicstaticStringgenerateFileName(StringoldName){Stringsuffix=oldName.substring(oldName.lastIndexOf("."));returnIDUtils.generateUniqueId()+suffix;}}
编写Web层
@RestController@RequestMapping("upload")publicclassUploadController{@AutowiredprivateUploadServiceuploadService;@PostMapping("image")publicResponseEntity<String>upload(@RequestParam("file")MultipartFilefile)throwsException{returnResponseEntity.ok(uploadService.uploadImage(file));}}
编写Service层
publicinterfaceUploadService{/***上传图片*@paramfile*@return*/StringuploadImage(MultipartFilefile)throwsException;}
@ServicepublicclassUploadServiceImplimplementsUploadService{@AutowiredprivateUploadPropertiesuploadProperties;@OverridepublicStringuploadImage(MultipartFilefile)throwsIOException{if(!uploadProperties.getAllowTypes().contains(file.getContentType())){thrownewIOException("文件上传类型错误!");}StringfileName=UploadUtils.generateFileName(file.getOriginalFilename());file.transferTo(newFile(uploadProperties.getPath()+fileName));returnfileName;}}
测试
由于我们也没有写html文件,所以我们直接使用Postman这个软件来测试。
第一步:我们使用POST请求,请求路径为/upload/image。
【注意,我们需要将请求头中的内容设置为空】
第二步:我们在请求体中将文件携带过去。
第三步,发出请求。