SpringMVC中没有文件验证注解怎么办?自定义验证注解@ValidFile解决问题

本文主要是介绍SpringMVC中没有文件验证注解怎么办?自定义验证注解@ValidFile解决问题,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

在web项目开发过程中,文件上传是一个很常见的功能,但是我们常常需要对上传文件的大小,文件的格式,上传的文件是否空等进行判断,SpringMVC中我们只需要简单的配置就可以实现上传文件功并对上传的文件限制,Springboot中默认开启文件上传配置,默认采用Servlet3.0文件上传的API实现,单文件限制是1MB,单次请求允许总上传文件为10MB, 我们可以通过如下属性修改默认配置

spring.http.multipart.max-file-size: 1MB
spring.http.multipart.max-request-size: 10MB

或通过配置 Bean  MultipartConfigElement 来覆盖默认配置,但是Spring中文件上传配置是针对全局的,那么如果项目中不同的文件上传接口对文件大小限制不一样,改怎么办呢? 有人会说: 读取文件信息,然后用if条件判断啊,那对于多个接口每个接口都要读文件信息,都要写好几个if 去判断代码是不是很冗余,时间都用在了写if上了,代码也显得很low。又有人说我们可以将文件验证抽象成一个公共的方法,对于文件的大小,文件格式要求作为参数,就可以减少冗余代码了。这是一个办法,但是最终还是至少要写一个 if 去判断最终的验证结果并做处理。那么还有什么优雅的方式吗?Spring中确实是没有提供针对单个文件上传接口限制的配置,但是Spring中可以很容易集成JSR-303标准验证注解,帮助我们对方法参数的验证,SpringBoot中也是默认开启了支持,然而这些标准注解中也没有提供对文件验证的实现。该怎么办呢,Spring中我们可以很容易的实现自定义的验证注解逻辑。下面就开始实现

首先自定义验证注解 @ValidFile 验证注解中必须要有如下三个方法,这是验证注解必须的有的否则会抛出异常。
String message() default "The uploaded file is not verified.";
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
其余为自定义属性
 //支持的文件后缀类型,默认全部,AliasFor("value")
String[] endsWith() default {};
 //文件后缀是否区分大小写
 boolean ignoreCase() default true;
 //上传的文件是否允许为空
 boolean allowEmpty() default false;
 //文件上传最大值,默认不限制但必须小于等于SpringMVC中文件上传配置,如果大于则根本不会走到自定义注解验证实现逻辑tomcat内部已经抛出异常,可以定义全局异常处理提示前端
 String maxSize() default DEFAULT_MAXSIZE;
  //文件上传最小值,默认不限制
 String minSize() default "0";

其次编写验证逻辑实现类 MultipartFileValidator,该类必须实现 ConstraintValidator接口,该类有两个泛型参数,第一个表示处理哪个验证注解类,第二参数为被注解标记的类型

具体代码如下:

/*** 默认:<br/>* 上传文件不能为空<br/>* 允许上传所有后缀格式的文件<br/>* 文件后缀名不区分大小写<br/>* 文件最大不超过springmvc配置的文件大小<br/>* 文件最小不小于0 kb <br/>* Created by bruce on 2018/8/30 16:34*/
@Target({ElementType.FIELD, ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = MultipartFileValidator.class)
public @interface ValidFile {String DEFAULT_MAXSIZE = "-1";/*** AliasFor("endsWith")*/String[] value() default {};/*** 支持的文件后缀类型,默认全部,AliasFor("value")*/String[] endsWith() default {};/*** 文件后缀是否区分大小写*/boolean ignoreCase() default true;/*** 上传的文件是否允许为空*/boolean allowEmpty() default false;/*** Max file size. Values can use the suffixes "MB" or "KB" to indicate megabytes or* kilobytes respectively.<br/>* 默认不限制但必须小于等于SpringMVC中文件上传配置*/String maxSize() default DEFAULT_MAXSIZE;/*** Min file size. Values can use the suffixes "MB" or "KB" to indicate megabytes or* kilobytes respectively. default byte*/String minSize() default "0";String message() default "The uploaded file is not verified.";Class<?>[] groups() default {};Class<? extends Payload>[] payload() default {};}

验证逻辑代码

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.web.MultipartProperties;
import org.springframework.util.Assert;
import org.springframework.web.multipart.MultipartFile;import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.ArrayList;
import java.util.Collections;/*** 校验器只有在真正使用的时候才会被实例化,不需要自己加实例化注解* Created by bruce on 2018/8/31 0:23*/
public class MultipartFileValidator implements ConstraintValidator<ValidFile, MultipartFile> {@Autowiredprivate MultipartProperties multipartProperties;private long maxSize = -1;private long minSize = 0;private ValidFile fileValid;private ArrayList<String> extension = new ArrayList<>();@Overridepublic void initialize(ValidFile constraintAnnotation) {this.fileValid = constraintAnnotation;//支持的文件扩展名集合Collections.addAll(extension, fileValid.value());Collections.addAll(extension, fileValid.endsWith());//文件上传的最大值if (constraintAnnotation.maxSize().equals(ValidFile.DEFAULT_MAXSIZE)) {//默认最大值采用Spring中配置的单文件大小this.maxSize = parseSize(multipartProperties.getMaxFileSize());} else {this.maxSize = parseSize(constraintAnnotation.maxSize());}//文件上传的最小值this.minSize = parseSize(constraintAnnotation.minSize());}private void validMessage(String message, ConstraintValidatorContext cvc) {cvc.disableDefaultConstraintViolation();cvc.buildConstraintViolationWithTemplate(message).addConstraintViolation();}private long parseSize(String size) {Assert.hasLength(size, "Size must not be empty");size = size.toUpperCase();if (size.endsWith("KB")) {return Long.valueOf(size.substring(0, size.length() - 2)) * 1024;}if (size.endsWith("MB")) {return Long.valueOf(size.substring(0, size.length() - 2)) * 1024 * 1024;}return Long.valueOf(size);}@Overridepublic boolean isValid(MultipartFile multipartFile, ConstraintValidatorContext cvc) {String fieldName = multipartFile.getName();//上传的文件是空的情况if (multipartFile.isEmpty()) {if (fileValid.allowEmpty()) {return true;}validMessage("上传文件不能为空" + ",参数名:" + fieldName, cvc);return false;}//上传的文件不是空的情况,验证其他条件是否成立//获取文件名,如果上传文件后缀名不区分大小写则统一转成小写String originalFilename = multipartFile.getOriginalFilename();if (fileValid.ignoreCase()) {originalFilename = originalFilename.toLowerCase();}//TODO 可以添加读取文件头信息if (extension.size() > 0 && extension.stream().noneMatch(originalFilename::endsWith)) {validMessage("上传文件类型不符合要求" + ",参数名:" + fieldName, cvc);return false;}//上传文件字节数long size = multipartFile.getSize();if (size < this.minSize) {validMessage("上传文件不能小于指定最小值" + ",参数名:" + fieldName, cvc);return false;}if (size > this.maxSize) {validMessage("上传文件不能大于指定最大值" + ",参数名:" + fieldName, cvc);return false;}return true;}
}

使用Demo

@Validated
@RestController
@RequestMapping("/import")
public class ImportController {@PostMapping("/test1")public String test1(@RequestParam("file")@ValidFile(value = {".xlsx", ".xls"}, maxSize = "2MB") MultipartFile multipartFile,HttpServletRequest request) throws IOException {Map<String, Object> result = new HashMap<>();result.put("getName", multipartFile.getName());result.put("getSize", multipartFile.getSize());result.put("length", multipartFile.getBytes().length);result.put("getContentType", multipartFile.getContentType());result.put("getOriginalFilename", multipartFile.getOriginalFilename());result.put("isEmpty", multipartFile.isEmpty());result.put("request", request.getContentType());return JSON.toJSONString(result, SerializerFeature.PrettyFormat);}}

总结:通过自定义验证注解,可以很简单实现针对单个文件上传接口文件大小,格式的限制,我们可以直接关注业务代码的编写避免各种使用if对文件的验证,该验证注解的缺点是只是简单的验证文件的后缀,这是不全的,稍微安全的方式是读取文件的文件头进行验证。这个就大家自己实现吧,还有一点需要注意的是注解中允许上传文件的最大值设置须小于等于SpringMVC中文件上传配置,因为如果大于则根本不会走到自定义注解验证实现逻辑tomcat内部已经抛出异常,可以定义全局异常处理提示前端。

这篇关于SpringMVC中没有文件验证注解怎么办?自定义验证注解@ValidFile解决问题的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/1124643

相关文章

Java中StopWatch的使用示例详解

《Java中StopWatch的使用示例详解》stopWatch是org.springframework.util包下的一个工具类,使用它可直观的输出代码执行耗时,以及执行时间百分比,这篇文章主要介绍... 目录stopWatch 是org.springframework.util 包下的一个工具类,使用它

Java进行文件格式校验的方案详解

《Java进行文件格式校验的方案详解》这篇文章主要为大家详细介绍了Java中进行文件格式校验的相关方案,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录一、背景异常现象原因排查用户的无心之过二、解决方案Magandroidic Number判断主流检测库对比Tika的使用区分zip

Java实现时间与字符串互相转换详解

《Java实现时间与字符串互相转换详解》这篇文章主要为大家详细介绍了Java中实现时间与字符串互相转换的相关方法,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录一、日期格式化为字符串(一)使用预定义格式(二)自定义格式二、字符串解析为日期(一)解析ISO格式字符串(二)解析自定义

opencv图像处理之指纹验证的实现

《opencv图像处理之指纹验证的实现》本文主要介绍了opencv图像处理之指纹验证的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学... 目录一、简介二、具体案例实现1. 图像显示函数2. 指纹验证函数3. 主函数4、运行结果三、总结一、

Java使用Curator进行ZooKeeper操作的详细教程

《Java使用Curator进行ZooKeeper操作的详细教程》ApacheCurator是一个基于ZooKeeper的Java客户端库,它极大地简化了使用ZooKeeper的开发工作,在分布式系统... 目录1、简述2、核心功能2.1 CuratorFramework2.2 Recipes3、示例实践3

Springboot处理跨域的实现方式(附Demo)

《Springboot处理跨域的实现方式(附Demo)》:本文主要介绍Springboot处理跨域的实现方式(附Demo),具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不... 目录Springboot处理跨域的方式1. 基本知识2. @CrossOrigin3. 全局跨域设置4.

springboot security使用jwt认证方式

《springbootsecurity使用jwt认证方式》:本文主要介绍springbootsecurity使用jwt认证方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地... 目录前言代码示例依赖定义mapper定义用户信息的实体beansecurity相关的类提供登录接口测试提供一

Spring Boot 3.4.3 基于 Spring WebFlux 实现 SSE 功能(代码示例)

《SpringBoot3.4.3基于SpringWebFlux实现SSE功能(代码示例)》SpringBoot3.4.3结合SpringWebFlux实现SSE功能,为实时数据推送提供... 目录1. SSE 简介1.1 什么是 SSE?1.2 SSE 的优点1.3 适用场景2. Spring WebFlu

基于SpringBoot实现文件秒传功能

《基于SpringBoot实现文件秒传功能》在开发Web应用时,文件上传是一个常见需求,然而,当用户需要上传大文件或相同文件多次时,会造成带宽浪费和服务器存储冗余,此时可以使用文件秒传技术通过识别重复... 目录前言文件秒传原理代码实现1. 创建项目基础结构2. 创建上传存储代码3. 创建Result类4.

Java利用JSONPath操作JSON数据的技术指南

《Java利用JSONPath操作JSON数据的技术指南》JSONPath是一种强大的工具,用于查询和操作JSON数据,类似于SQL的语法,它为处理复杂的JSON数据结构提供了简单且高效... 目录1、简述2、什么是 jsONPath?3、Java 示例3.1 基本查询3.2 过滤查询3.3 递归搜索3.4