Springboot中如何优雅的写好controller层代码

Posted 凡夫贩夫

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Springboot中如何优雅的写好controller层代码相关的知识,希望对你有一定的参考价值。

前言

优雅?看到这个词,我第一反应是什么是优雅?怎么写才算优雅?一千个读者有一千个哈姆雷特,每个人的经验、阅历不同,也许理解就不同。我对优雅的理解很简单,就是简洁有效、容易理解,别那么多套路。java中使用Sping的web项目通常会分为三层,分别是controller、service、dao,这似乎已成为了一个既定规则。很少有人去想为什么要这样分?可不可不以不这样分?java属于面向对象的高级编程语言,其实这种分法并不符合面向对象的理念,而实际这是按照一次B-S请求过程从外到内的调用过程划分的,然后根据面向接口编程的理念,外层调用内层接口,内层接口实际为外层提供服务能力的是内层接口的实现类,接口是标准接口,实现类可以根据实际业务更换,按照这种设计实现了层间解耦,提供了程序维护的便利性和开发效率。因此,虽然这种分法不符合面向对象的理念,但是很优雅(简单有效、容易被大多数人理解)。

java项目分层的含义

  • Controller:俗称控制器,用于处理请求映射,在jsp时代,调用service层业务接口,在controll层包装一个视频图对象,返回给页面;现在通常直接返回数据对象,springboot会自动把返回结果格式化为json返回给前端;

  • Service层:通常是系统的具体业务逻辑,供controller层调用;

  • DAO层:操作数据库,供service层调用;

正如前面说的,这种分法不能算是面对对象,倒是有点面向过程的味道,但是这种分法实践了面向接口编程的理念,使层与层之间解耦,提高了程序的可维护性和开发效率,所以还是优雅的。

其中controller层作为前端与后端实际业务接口的连接者,如何优雅写好这一层的代码至关重要。要想优雅写好这一层代码,可以从以下几个方面着手

入口参数统一校验

正例:

  1. 引入spring-boot-starter-validation包;

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-validation</artifactId>
    <version>2.3.9.RELEASE</version>
</dependency>
  1. 在用于接收参数的实体类上,使用@NotNull、@Null等结束注解对参数属性进行标记;

@Data
public class RemindTaskBean implements Serializable 
    private static final long serialVersionUID = 777197918651078049L;
    @NotNull(message = "调度任务名称不能为空")
    private String taskName;
    @Pattern(regexp = "\\\\d/\\\\d+ \\\\*1 \\\\*1 \\\\*1 \\\\*1 \\\\?1",message = "表达式格式错误,请更正")
    private String cron;
    @NotNull(message = "任务状态不能为空")
    @Range(min = 0,max = 1,message = "任务状态错误,请更正")
    private Integer taskStatus;
  1. 在controller层入口参数处,使用@Valid或@Validated注解对@RequstBody类型的参数进行校验;

@RequestMapping("/add")
public void add(@Valid @RequestBody RemindTaskBean remindTaskBean)
    remindTaskService.add(remindTaskBean);

关于参数的统一校验其实不止这点,还有一些更高级的用法,如嵌套校验、分组校验、集合校验、自定义校验,在之前的文章里已经总结的很详细,如有需要可移步这里:优雅的Springboot参数校验(一)优雅的Springboot参数校验(二)

反例:

真的不想看到为了校验参数,在controller层的方法内写了大量的if else判断,(这样不感觉累吗,加班到天亮也是活该)如下:

@RequestMapping("/add2")
public CommRes add2(@Valid @RequestBody RemindTaskBean remindTaskBean)
    if (remindTaskBean.getTaskName() == null) 
        CommRes.fail("调度任务名不能为空");
    
    if (remindTaskBean.getTaskStatus() == null) 
        CommRes.fail("调度任务状态不能为空");
    
    if (remindTaskBean.getTaskStatus()!=0||remindTaskBean.getTaskStatus()!=1) 
        CommRes.fail("调度任务状态错误,请更正");
    
    if (remindTaskBean.getCron() != null) 
        String reg="\\\\d/\\\\d+ \\\\*1 \\\\*1 \\\\*1 \\\\*1 \\\\?1";
        if (!remindTaskBean.getCron().matches(reg)) 
            CommRes.fail("表达式格式错误,请更正");
        
    
    remindTaskService.add(remindTaskBean);
    return CommRes.success(remindTaskBean);

异常信息统一处理

在controller层使用统一的参数校验后,如果入参数与约束注解相违背,框架就会自动抛出异常处理,再使用异常信息的统一处理机制来捕获这些异常,把异常提示信息进行包装返回给前端友好提示用户。

@RestControllerAdvice
public class CommonExceptionHandler 
    //用于捕获@RequestBody类型参数触发校验规则抛出的异常
    @ExceptionHandler(value = MethodArgumentNotValidException.class)
    public CommRes handleValidException(MethodArgumentNotValidException e) 
        StringBuilder sb = new StringBuilder();
        List<ObjectError> allErrors = e.getBindingResult().getAllErrors();
        if (!CollectionUtils.isEmpty(allErrors)) 
            for (ObjectError error : allErrors) 
                sb.append(error.getDefaultMessage()).append(";");
            
        
        return CommRes.fail(sb.toString());
    
    //用于捕获@RequestParam/@PathVariable参数触发校验规则抛出的异常
    @ExceptionHandler(value = ConstraintViolationException.class)
    public CommRes handleConstraintViolationException(ConstraintViolationException e) 
        StringBuilder sb = new StringBuilder();
        Set<ConstraintViolation<?>> conSet = e.getConstraintViolations();
        for (ConstraintViolation<?> con : conSet) 
            String message = con.getMessage();
            sb.append(message).append(";");
        
        return CommRes.fail(sb.toString());
    
    @ExceptionHandler(value = BindException.class)
    public CommRes handleConstraintViolationException(BindException e) 
        StringBuilder sb =  new StringBuilder();
        List<ObjectError> allErrors = e.getAllErrors();
        for (ObjectError allError : allErrors) 
            String defaultMessage = allError.getDefaultMessage();
            sb.append(defaultMessage).append(";");
        

        return CommRes.fail(sb.toString());
    
    @ExceptionHandler(value = Exception.class)
    public CommRes exception(Exception e) 
        return CommRes.fail(e.getMessage());
    

另外关于异常处理,不要动不动就try catch,除非有必要(有人犟劲上来了,我感觉都很有必要呀,所以他的代码里到处是try catch,我也真是醉了),比如:有人用try catch把增加一个调度任务的dao接口调用包上,他加try catch的理由是如果sql写错了呢不就异常了(锤死他的心都有了,你就不会写对喽!!!),有人还担心万一数据库挂了呢(如果数据库真挂了,抛个异常有什么用,难道能把数据库恢复了不成);最搞笑的是,有的人只try catch,然后就没有然后了;总而言之,有的小朋友真是超可爱。

反例:

public CommRes add2(RemindTaskBean remindTask) 
    RemindTaskBean taskBean = this.remindTaskDao.queryByTaskName("测试任务");
    if (taskBean != null) 
      return  CommRes.fail("调度任务已存在,请勿重复注册");
    
    try 
        this.remindTaskDao.insert(remindTask);
     catch (Exception e) 
        e.printStackTrace();
    
    return CommRes.success("");

那什么叫除非有必要呢?比如增加一个调度任务,但是要求相同的名字不能重复注册,这时可以在插入调度信息前查询是否有相同名字的调度任务,如果有,则抛出异常提示,优雅的写法应该是这样的(ServiceException是自定义的异常):

正例:

@Override
public void add(RemindTaskBean remindTask) 
    RemindTaskBean taskBean  = this.remindTaskDao.queryByTaskName("测试任务");
    if (taskBean != null) 
        throw new ServiceException("调度任务已存在,请勿重复注册");
    
    this.remindTaskDao.insert(remindTask);
public class ServiceException extends RuntimeException    
    public ServiceException(String message) 
        super(message);
    

返回结果统一格式

细心的小伙伴发现了CommRes.java,这个类是把返回结果统一格式的包装类

@Data
public class CommRes 
    private String code;
    private String msg;
    private Object data;
    public static CommRes success(Object data)
        CommRes commRes = new CommRes();
        commRes.setCode("200");
        commRes.setMsg("操作成功");
        commRes.setData(data);
        return commRes;
    
    public static CommRes fail(String msg)
        CommRes commRes = new CommRes();
        commRes.setCode("400");
        commRes.setMsg(msg);
        commRes.setData("");
        return commRes;
    

相信很多人也知道,要封装一个包装类对返回结果统一格式,有的小伙伴是这样用,其实不是,这是一个反例:

@RequestMapping("/add2")
public CommRes add2(@Valid @RequestBody RemindTaskBean remindTaskBean)
    if (remindTaskBean.getTaskName() == null) 
        CommRes.fail("调度任务名不能为空");
    
    remindTaskService.add(remindTaskBean);
    return CommRes.success(remindTaskBean);

优雅的用法应该是这样的,不要手动去调用它,而是使用@RestControllerAdvice或@ControllerAdvice标记一个类并实现ResponseBodyAdvicer接口,作为返回结果统一处理类,然后在controller层方法里得到返回结果直接返回就好,被@RestControllerAdvice标记的ResponseBodyAdvicer接口的实现类可以帮你完成所有返回值的统一格式包装,看下面的正例

正例:

@RestControllerAdvice
public class ResultResponseBoydAdvice implements ResponseBodyAdvice 
    @Override
    public boolean supports(MethodParameter returnType, Class converterType) 
        return true;
    

    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) 
        if (body instanceof CommRes) 
            return body;
        
        return CommRes.success(body);
    
@GetMapping("/list")
public List<RemindTaskBean> list()
    List<RemindTaskBean> remindTasks = dynamicScheduleTask.taskList();
    return remindTasks;

经过这样一处理,返回结果就是这样了

正常情况下的返回结果:


    "code": "200",
    "msg": "操作成功",
    "data": [返回数据在这里面]

异常情况下的返回结果:


    "code": "400",
    "msg": "表达式格式错误,请更正;",
    "data": ""

小结

在controller层,统一进行参数校验、统一处理异常、统一返回结果格式后,是不是感觉controller层的代码清爽很多了,而且效率还高了,终于结束无效的加班了。

示例是所用源代码地址:https://gitcode.net/fox9916/fanfu-web.git优雅的controller分支

以上是关于Springboot中如何优雅的写好controller层代码的主要内容,如果未能解决你的问题,请参考以下文章

Springboot中如何优雅的进行字段校验

Springboot中优雅进行字段校验

Springboot中优雅进行字段校验

SpringBoot:如何优雅地处理全局异常?

如何优雅的写单元测试?

如何优雅的写单元测试?