SpringBoot秒杀系统：5.利用validation进行参数校验

雨点打透心脏的1/2处 2023-02-18 10:55 111阅读 0赞

1.  引入validation依赖

<dependency>
          <groupId>org.springframework.boot</groupId>
          <artifactId>spring-boot-starter-validation</artifactId>
        </dependency>

1.  在实体类中加入限制注解

@NotNull
        private String mobile;
    
        @NotNull
        private String password;

常用注解如下  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0aGlyZG9uZWw_size_16_color_FFFFFF_t_70]  
3. 编写自定义注解限制手机号格式  
编写注解IsMobile  
知识点：利用@interface 根据@NotNull注解定义如下注解

package com.imooc.miaosha.validator;
    
    import static java.lang.annotation.ElementType.ANNOTATION_TYPE;
    import static java.lang.annotation.ElementType.CONSTRUCTOR;
    import static java.lang.annotation.ElementType.FIELD;
    import static java.lang.annotation.ElementType.METHOD;
    import static java.lang.annotation.ElementType.PARAMETER;
    import static java.lang.annotation.RetentionPolicy.RUNTIME;
    
    import java.lang.annotation.Documented;
    import java.lang.annotation.Retention;
    import java.lang.annotation.Target;
    
    import javax.validation.Constraint;
    import javax.validation.Payload;
    
    @Target({
         METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER })
    @Retention(RUNTIME)
    @Documented
    @Constraint(validatedBy = {
        IsMobileValidator.class })
    public @interface  IsMobile {
        
    	
    	boolean required() default true;
    	
    	String message() default "手机号码格式错误";
    
    	Class<?>[] groups() default {
         };
    
    	Class<? extends Payload>[] payload() default {
         };
    }

注意@Constraint(validatedBy = \{IsMobileValidator.class \})

package com.liuyang.seckill.validator;
    
    import com.liuyang.seckill.util.ValidatorUtil;
    import org.apache.commons.lang3.StringUtils;
    
    import javax.validation.ConstraintValidator;
    import javax.validation.ConstraintValidatorContext;
    public class IsMobileValidator implements ConstraintValidator<IsMobile, String> {
        
    
    	private boolean required = false;
    	
    	public void initialize(IsMobile constraintAnnotation) {
        
    		required = constraintAnnotation.required();
    	}
    
    	public boolean isValid(String value, ConstraintValidatorContext context) {
        
    		if(required) {
        
    			return ValidatorUtil.isMobile(value);
    		}else {
        
    			if(StringUtils.isEmpty(value)) {
        
    				return true;
    			}else {
        
    				return ValidatorUtil.isMobile(value);
    			}
    		}
    	}
    }

其中 ValidatorUtil.isMobile(value)静态方法定义如下

package com.liuyang.seckill.util;
    
    import org.apache.commons.lang3.StringUtils;
    
    import java.util.regex.Matcher;
    import java.util.regex.Pattern;
    
    public class ValidatorUtil {
        
    	
    	private static final Pattern mobile_pattern = Pattern.compile("1\\d{10}");
    	
    	public static boolean isMobile(String src) {
        
    		if(StringUtils.isEmpty(src)) {
        
    			return false;
    		}
    		Matcher m = mobile_pattern.matcher(src);
    		return m.matches();
    	}
    	
    //	public static void main(String[] args) {
        
    //			System.out.println(isMobile("18912341234"));
    //			System.out.println(isMobile("1891234123"));
    //	}
    }

1.  Controller中添加测试

LoginVo表单实体类中添加注解

@NotNull
        @IsMobile
        private String mobile;
    
        @NotNull
        @Length(min=32)
        private String password;

Controller中添加@Valid测试

@ResponseBody
        @RequestMapping("/do_login")
        public Result<CodeMsg> doLogin(@Valid LoginVo loginVo) {
        
            log.info(loginVo.toString());
            CodeMsg cm = seckillUserService.login(loginVo);
    
            if(cm.getCode()!=0){
        
                return Result.error(CodeMsg.SERVER_ERROR);
            }else {
        
                return Result.success(cm);
            }
        }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0aGlyZG9uZWw_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200618093601421.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2x0aGlyZG9uZWw=,size_16,color_FFFFFF,t_70