在SpringBoot项目中优雅地进行参数校验（二）

前言：

在上一篇文章中介绍了参数验证的简单使用，本文将介绍级联校验，分组校验，以及Springboot支持的在Service进行参数校验

一.级联校验

1.实体类

@Data
@JsonInclude(value= JsonInclude.Include.NON_NULL)
public class MyUser {

    @NotEmpty
    @NotNull
    private String id;

    @NotNull
    @NotEmpty
    private String name;

    @NotNull
    @NotEmpty
    @Email
    private String email;

    private String message;

    @Valid
    @NotNull
    private Department department;

}

@Data
@JsonInclude(value= JsonInclude.Include.NON_NULL)
public class Department {

    @NotEmpty
    @NotNull
    private String id;

    @NotEmpty
    @NotNull
    private String name;
}

这里新建了两个实体类，其中Department类对象作为MyUser类的成员变量，在上一篇文章我们知道了数据校验功能注解的基本使用，而在此稍微有些不同，当我们校验MyUser对象时需要对该对象的成员变量Deparment对象也进行数据校验（即级联校验），故需要在MyUser成员变量Department department上添加@Vail注解，即能开启对department对象的数据校验。

2.控制层

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;

    @PostMapping
    public ResultVo addUser(@RequestBody @Vail MyUser myuser ){
        //...
    }
}

3.异常拦截器

@ControllerAdvice
@ResponseBody
public class GlobalExceptionHandler {

    @ExceptionHandler(MethodArgumentNotValidException.class)
    public ResultVo MethodArgumentNotValidExceptionHandler(MethodArgumentNotValidException e){
       //创建了一个map，key为变量名称，value为错误信息
        Map<String,String> map=new HashMap<>();
        List<FieldError> list=e.getBindingResult().getFieldErrors();
        for (FieldError error : list) {
           //参数的名称和错误信息
            map.put(error.getField(),error.getDefaultMessage());
        }
       //封装了更加友好的数据返回给前端
        return ResultVo.fail(ErrorCode.PARAM_ERROR,map);
    }

4.测试

此处使用Postman进行测试


image.png

二.分组校验

1.实体类

@Data
@JsonInclude(value= JsonInclude.Include.NON_NULL)
public class MyUser {

    public interface add{
    }

    public interface update{
    }

    @NotEmpty(groups = update.class)
    @NotNull(groups = update.class)
    @Null(groups = add.class)
    private String id;

    @NotNull
    @NotEmpty
    private String name;

    @NotNull
    @NotEmpty
    @Email
    private String email;

    private String message;

    @Valid
    @NotNull
    private Department department;

}

在实际项目中会遇到这么一种类型的情况：用户注册时，前端返回的用户的数据是不具有用户id的，所以此时需要保证用户id为空，而当用户修改自己的信息时，此时是具有用户id的，故需保证用户id不为空。对于id在不同时期需要有不同的验证策略，故使用分组验证。
分组验证需要在实体类中添加内部接口，每一个接口即是一种类别，之后再往需要进行分组验证的成员变量的注解属性值groups添加接口字节码文件即可，共同拥有一种接口字节码文件的注解就是一组验证策略。

2.控制层

@PostMapping
    public ResultVo addUser(@RequestBody @Validated({MyUser.add.class, Default.class}) MyUser myUser){
        
        //...
    }

此处需要使用Spring提供的@Validated注解以支持分组校验。
注意：在@Validated添加分组时，需添加Defaut.class，表示其他无分组的验证注解一并生效，若无添加就只是add.class组别的验证注解生效而已

3.异常拦截器

同上文一致，此处不再赘述

4.测试

image.png

三.Service层验证

项目中有时无法在控制层就完成数据校验工作，就需要在业务层完成，SpringBoot也对业务层的数据校验提供了支持

1.实体类

@Data
@JsonInclude(value= JsonInclude.Include.NON_NULL)
public class MyUser {

    @NotEmpty
    @NotNull
    private String id;

    @NotNull
    @NotEmpty
    private String name;

    @NotNull
    @NotEmpty
    @Email
    private String email;

    private String message;

    @Valid
    @NotNull
    private Department department;

}

2.控制层

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;

    @PostMapping
    public ResultVo addUser(@RequestBody String userMessage){
        JSONObject jsonObject= JSON.parseObject(userMessage);

        String code=(String)jsonObject.get("code");
        //提取出验证码后可对其进行验证
        //这里假装进行了验证（手动滑稽）
        if (code!=null&&code.equals("12345")) {
            MyUser myUser=userService.addUser(JSONObject.parseObject(userMessage,MyUser.class));
            return ResultVo.success(myUser);
        }
        return ResultVo.fail(ErrorCode.CODE_ERROR);
    }

此处前端的数据除了用户信息外还有验证码code，在控制层对验证码进行检验，若验证码符合，再将json中的数据封装进对象中传递给业务层

3.业务层

@Service
@Validated
public class UserService {

    public MyUser addUser(@Vail MyUser myUser){
        //...
        return myUser;
    }

在业务层需要对其开启SpringBoot对业务层参数校验的支持，即添加@Validated注解和在需要参数校验的变量前添加@Vail注解即可

4.异常拦截

此处抛出的异常与在控制层进行参数校验抛出的异常不是同一个，故需再写一个异常拦截器

@ControllerAdvice
@ResponseBody
@JsonInclude(value= JsonInclude.Include.NON_NULL)
public class GlobalExceptionHandler {

    @ExceptionHandler(ConstraintViolationException.class)
    public ResultVo ConstraintViolationExceptionHandler(ConstraintViolationException e){
        Map<String,String> map=new HashMap<>();
        Set<ConstraintViolation<?>> set=e.getConstraintViolations();

        for (ConstraintViolation cv : set){
            String[] path=cv.getPropertyPath().toString().split("\\.");
            String message=cv.getMessage();
            map.put(path[path.length-1],message);
        }

        return ResultVo.fail(ErrorCode.PARAM_ERROR,map);
    }

5.测试

image.png

三点一.Service的分组校验加级联验证

第三部分介绍了Service层的参数校验，是SpringBoot提供支持的，但是SpringBoot貌似并没有提供对于Service层的分组校验，楼主搜遍了各大博客网站没有找到一篇Service层分组校验的技术博客，这就非常的蛋疼了。

那咋办嘛，只能自己写个参数校验器咯

1.实体类

@Data
@JsonInclude(value= JsonInclude.Include.NON_NULL)
public class MyUser {

    public interface add{
    }

    public interface update{
    }

    @NotEmpty(groups = update.class)
    @NotNull(groups = update.class)
    @Null(groups = add.class)
    private String id;

    @NotNull
    @NotEmpty
    private String name;

    @NotNull
    @NotEmpty
    @Email
    private String email;

    private String message;

    @Valid
    @NotNull
    private Department department;

}


@Data
@JsonInclude(value= JsonInclude.Include.NON_NULL)
public class Department {

    @Null(groups = MyUser.add.class)
    @NotEmpty(groups = MyUser.update.class)
    @NotNull(groups = MyUser.update.class)
    private String d_id;

    @NotEmpty
    @NotNull
    private String d_name;
}

2.控制层

@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    private UserService userService;

    @PostMapping
    public ResultVo addUser(@RequestBody String userMessage){
        JSONObject jsonObject= JSON.parseObject(userMessage);

        String code=(String)jsonObject.get("code");
        //提取出验证码后可对其进行验证
        if (code!=null&&code.equals("12345")) {
            MyUser myUser=userService.addUser(JSONObject.parseObject(userMessage,MyUser.class));
            return ResultVo.success(myUser);
        }
        return ResultVo.fail(ErrorCode.CODE_ERROR);
    }

3.参数校验器

@Component
public class ValidationUtil {

    //SpringBoot能将Validator直接注入
    @Autowired
    private Validator validator;

    public void validateTest(Object object,Class[] classes){
        Set<ConstraintViolation<Object>> set = validator.validate(object, classes);
        //set.size大于0则表明参数出错
        if (set.size()>0){
            Map<String,String> map=new HashMap<>();
            for (ConstraintViolation<Object> cv : set){
                String[] param=cv.getPropertyPath().toString().split("\\.");
                String message=cv.getMessage();
                map.put(param[param.length-1],message);
            }
            //抛出业务异常，并且需要根据业务异常再写一个异常拦截器，此处不再赘述
            throw new MyException(map);
        }
    }
}

4.业务层

@Service
public class UserService {

    //将参数校验工具注入
    @Autowired
    ValidationUtil validationUtil;

    public MyUser addUser( MyUser myUser){
        //调用参数校验
        validationUtil.validateTest(myUser,new Class[]{Default.class,MyUser.add.class});

        //...
        return myUser;
    }

5.测试

image.png

总结

如有错误，还劳烦大佬指正，感激不尽

2020.2.14
午夜 11：44

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 212,029评论 6赞 492
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 90,395评论 3赞 385
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 157,570评论 0赞 348
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 56,535评论 1赞 284
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 65,650评论 6赞 386
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 49,850评论 1赞 290
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 39,006评论 3赞 408
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 37,747评论 0赞 268
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 44,207评论 1赞 303
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,536评论 2赞 327
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,683评论 1赞 341
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 34,342评论 4赞 330
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 39,964评论 3赞 315
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,772评论 0赞 21
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,004评论 1赞 266
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 46,401评论 2赞 360
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 43,566评论 2赞 349