大家好，我是磊哥。

本篇主要要介绍的就是controller层的处理，一个完整的后端请求由4部分组成：

1、 接口地址(也就是URL地址)
2、 请求方式(一般就是get、set、put、delete)
3、 请求数据(request，有head跟body)
4、 响应数据(response)

本篇将解决以下3个问题：

1、当接收到请求时，如何优雅的校验参数；
2、返回响应数据该如何统一的进行处理；
3、接收到请求，处理业务逻辑时抛出了异常又该如何处理；

一、Controller层参数接收（太基础了，可以跳过）

常见的请求就分为get跟post2种

@RestController
@RequestMapping('/product/product-info')
public class ProductInfoController {

    @Autowired
    ProductInfoService productInfoService;

    @GetMapping('/findById')
    public ProductInfoQueryVo findById(Integer id) {

     ...
    }

    @PostMapping('/page')
    public IPage findPage(Page page, ProductInfoQueryVo vo) {

     ...
    }
}

1、 @RestController：之前解释过，@RestController=@Controller+ResponseBody加上这个注解，springboot就会吧这个类当成controller进行处理，然后把所有返回的参数放到ResponseBody中；
2、 @RequestMapping：请求的前缀，也就是所有该Controller下的请求都需要加上/product/product-info的前缀；
3、 @GetMapping('/findById')：标志这是一个get请求，并且需要通过/findById地址才可以访问到；
4、 @PostMapping('/page')：同理，表示是个post请求；
5、 参数：至于参数部分，只需要写上ProductInfoQueryVo，前端过来的json请求便会通过映射赋值到对应的对象中，例如请求这么写，productId就会自动被映射到vo对应的属性当中；

注意

文末有：7701页互联网大厂面试题

二、统一状态码

1. 返回格式

为了跟前端妹妹打好关系，我们通常需要对后端返回的数据进行包装一下，增加一下状态码，状态信息，这样前端妹妹接收到数据就可以根据不同的状态码，判断响应数据状态，是否成功是否异常进行不同的显示。当然这让你拥有了更多跟前端妹妹的交流机会，假设我们约定了1000就是成功的意思
如果你不封装，那么返回的数据是这样子的

经过封装以后时这样子的

2. 封装ResultVo

这些状态码肯定都是要预先编好的，怎么编呢？写个常量1000？还是直接写死1000？要这么写就真的书白读的了，写状态码当然是用枚举拉

1、首先先定义一个状态码的接口，所有状态码都需要实现它，有了标准才好做事；

public interface StatusCode {
     
    public int getCode();
    public String getMsg();
}

1、然后去找前端妹妹，跟他约定好状态码（这可能是你们唯一的约定了）枚举类嘛，当然不能有setter方法了，因此我们不能在用@Data注解了，我们要用@Getter；

@Getter
public enum ResultCode implements StatusCode{
   
    SUCCESS(1000, '请求成功'),
    FAILED(1001, '请求失败'),
    VALIDATE_ERROR(1002, '参数校验失败'),
    RESPONSE_PACK_ERROR(1003, 'response返回包装失败');

    private int code;
    private String msg;

    ResultCode(int code, String msg) {
     
        this.code = code;
        this.msg = msg;
    }
}

1、写好枚举类，就开始写ResultVo包装类了，我们预设了几种默认的方法，比如成功的话就默认传入object就可以了，我们自动包装成success；

@Data
public class ResultVo {
   
    // 状态码
    private int code;

    // 状态信息
    private String msg;

    // 返回对象
    private Object data;

    // 手动设置返回vo
    public ResultVo(int code, String msg, Object data) {
   
     
        this.code = code;
        this.msg = msg;
        this.data = data;
    }

    // 默认返回成功状态码，数据对象
    public ResultVo(Object data) {
   
     
        this.code = ResultCode.SUCCESS.getCode();
        this.msg = ResultCode.SUCCESS.getMsg();
        this.data = data;
    }

    // 返回指定状态码，数据对象
    public ResultVo(StatusCode statusCode, Object data) {
   
     
        this.code = statusCode.getCode();
        this.msg = statusCode.getMsg();
        this.data = data;
    }

    // 只返回状态码
    public ResultVo(StatusCode statusCode) {
   
     
        this.code = statusCode.getCode();
        this.msg = statusCode.getMsg();
        this.data = null;
    }
}

1、使用，现在的返回肯定就不是returndata;这么简单了，而是需要newResultVo(data);；

    @PostMapping('/findByVo')
    public ResultVo findByVo(@Validated ProductInfoVo vo) {
   
     
        ProductInfo productInfo = new ProductInfo();
        BeanUtils.copyProperties(vo, productInfo);
        return new ResultVo(productInfoService.getOne(new QueryWrapper(productInfo)));
    }

最后返回就会是上面带了状态码的数据了

三、统一校验

1. 原始做法

假设有一个添加ProductInfo的接口，在没有统一校验时，我们需要这么做

@Data
public class ProductInfoVo {
   
     
    // 商品名称
    private String productName;
    // 商品价格
    private BigDecimal productPrice;
    // 上架状态
    private Integer productStatus;
}

    @PostMapping('/findByVo')
    public ProductInfo findByVo(ProductInfoVo vo) {
   
     
        if (StringUtils.isNotBlank(vo.getProductName())) {
   
     
            throw new APIException('商品名称不能为空');
        }
        if (null != vo.getProductPrice() && vo.getProductPrice().compareTo(new BigDecimal(0)) < 0) {
   
     
            throw new APIException('商品价格不能为负数');
        }
        ...
        
        ProductInfo productInfo = new ProductInfo();
        BeanUtils.copyProperties(vo, productInfo);
        return new ResultVo(productInfoService.getOne(new QueryWrapper(productInfo)));
    }

这if写的人都傻了，能忍吗？肯定不能忍啊

2. @Validated参数校验

好在有@Validated，又是一个校验参数必备良药了。有了@Validated我们只需要再vo上面加一点小小的注解，便可以完成校验功能

@Data
public class ProductInfoVo {
   
     
    @NotNull(message = '商品名称不允许为空')
    private String productName;

    @Min(value = 0, message = '商品价格不允许为负数')
    private BigDecimal productPrice;

    private Integer productStatus;
}

    @PostMapping('/findByVo')
    public ProductInfo findByVo(@Validated ProductInfoVo vo) {
   
     
        ProductInfo productInfo = new ProductInfo();
        BeanUtils.copyProperties(vo, productInfo);
        return new ResultVo(productInfoService.getOne(new QueryWrapper(productInfo)));
    }

运行看看，如果参数不对会发生什么？
我们故意传一个价格为-1的参数过去

productName : 泡脚
productPrice : -1
productStatus : 1

{
   
     
  'timestamp': '2020-04-19T03:06:37.268+0000',
  'status': 400,
  'error': 'Bad Request',
  'errors': [
    {
   
     
      'codes': [
        'Min.productInfoVo.productPrice',
        'Min.productPrice',
        'Min.java.math.BigDecimal',
        'Min'
      ],
      'arguments': [
        {
   
     
          'codes': [
            'productInfoVo.productPrice',
            'productPrice'
          ],
          'defaultMessage': 'productPrice',
          'code': 'productPrice'
        },
        0
      ],
      'defaultMessage': '商品价格不允许为负数',
      'objectName': 'productInfoVo',
      'field': 'productPrice',
      'rejectedValue': -1,
      'bindingFailure': false,
      'code': 'Min'
    }
  ],
  'message': 'Validation failed for object\u003d\u0027productInfoVo\u0027. Error count: 1',
  'trace': '',
  'path': '/leilema/product/product-info/findByVo'
}

大功告成了吗？虽然成功校验了参数，也返回了异常，并且带上'商品价格不允许为负数'的信息。但是你要是这样返回给前端，前端妹妹就提刀过来了，当年约定好的状态码，你个负心人说忘就忘？用户体验小于等于0啊！所以我们要进行优化一下，每次出现异常的时候，自动把状态码写好，不负妹妹之约！

3. 优化异常处理

首先我们先看看校验参数抛出了什么异常

Resolved [org.springframework.validation.BindException: org.springframework.validation.BeanPropertyBindingResult: 1 errors

我们看到代码抛出了org.springframework.validation.BindException的绑定异常，因此我们的思路就是AOP拦截所有controller，然后异常的时候统一拦截起来，进行封装！完美！

玩你个头啊完美，这么呆瓜的操作springboot不知道吗？spring mvc当然知道拉，所以给我们提供了一个@RestControllerAdvice来增强所有@RestController，然后使用@ExceptionHandler注解，就可以拦截到对应的异常。

这里我们就拦截BindException.class就好了。最后在返回之前，我们对异常信息进行包装一下，包装成ResultVo，当然要跟上ResultCode.VALIDATE_ERROR的异常状态码。这样前端妹妹看到VALIDATE_ERROR的状态码，就会调用数据校验异常的弹窗提示用户哪里没填好

@RestControllerAdvice
public class ControllerExceptionAdvice {
   
     

    @ExceptionHandler({
   
     BindException.class})
    public ResultVo MethodArgumentNotValidExceptionHandler(BindException e) {
   
     
        // 从异常对象中拿到ObjectError对象
        ObjectError objectError = e.getBindingResult().getAllErrors().get(0);
        return new ResultVo(ResultCode.VALIDATE_ERROR, objectError.getDefaultMessage());
    }
}

来康康效果，完美。1002与前端妹妹约定好的状态码

四、统一响应

1. 统一包装响应

再回头看一下controller层的返回

return new ResultVo(productInfoService.getOne(new QueryWrapper(productInfo)));

开发小哥肯定不乐意了，谁有空天天写new ResultVo(data)啊，我就想返回一个实体！怎么实现我不管！好把，那就是AOP拦截所有Controller，再@After的时候统一帮你封装一下咯

怕是上一次脸打的不够疼，springboot能不知道这么个操作吗？

@RestControllerAdvice(basePackages = {
   
     'com.bugpool.leilema'})
public class ControllerResponseAdvice implements ResponseBodyAdvice<Object> {
   
     
    @Override
    public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
   
     
        // response是ResultVo类型，或者注释了NotControllerResponseAdvice都不进行包装
        return !methodParameter.getParameterType().isAssignableFrom(ResultVo.class);
    }

    @Override
    public Object beforeBodyWrite(Object data, MethodParameter returnType, MediaType mediaType, Class<? extends HttpMessageConverter<?>> aClass, ServerHttpRequest request, ServerHttpResponse response) {
   
     
        // String类型不能直接包装
        if (returnType.getGenericParameterType().equals(String.class)) {
   
     
            ObjectMapper objectMapper = new ObjectMapper();
            try {
   
     
                // 将数据包装在ResultVo里后转换为json串进行返回
                return objectMapper.writeValueAsString(new ResultVo(data));
            } catch (JsonProcessingException e) {
   
     
                throw new APIException(ResultCode.RESPONSE_PACK_ERROR, e.getMessage());
            }
        }
        // 否则直接包装成ResultVo返回
        return new ResultVo(data);
    }
}

1、 @RestControllerAdvice(basePackages={'com.bugpool.leilema'})自动扫描了所有指定包下的controller，在Response时进行统一处理；
2、重写supports方法，也就是说，当返回类型已经是ResultVo了，那就不需要封装了，当不等与ResultVo时才进行调用beforeBodyWrite方法，跟过滤器的效果是一样的；
3、最后重写我们的封装方法beforeBodyWrite，注意除了String的返回值有点特殊，无法直接封装成json，我们需要进行特殊处理，其他的直接newResultVo(data);就ok了；

打完收工，康康效果

    @PostMapping('/findByVo')
    public ProductInfo findByVo(@Validated ProductInfoVo vo) {
   
     
        ProductInfo productInfo = new ProductInfo();
        BeanUtils.copyProperties(vo, productInfo);
        return productInfoService.getOne(new QueryWrapper(productInfo));
    }

此时就算我们返回的是po，接收到的返回就是标准格式了，开发小哥露出了欣慰的笑容

2. NOT统一响应

不开启统一响应原因

开发小哥是开心了，可是其他系统就不开心了。举个例子：我们项目中集成了一个健康检测的功能，也就是这货

@RestController
public class HealthController {
   
     
    @GetMapping('/health')
    public String health() {
   
     
        return 'success';
    }
}

公司部署了一套校验所有系统存活状态的工具，这工具就定时发送get请求给我们系统

“兄弟，你死了吗？”
“我没死，滚”
“兄弟，你死了吗？”
“我没死，滚”

是的，web项目的本质就是复读机。一旦发送的请求没响应，就会给负责人发信息（企业微信或者短信之类的），你的系统死啦！赶紧回来排查bug吧！让大家感受一下。每次看到我都射射发抖，早上6点！我tm！！！！！

好吧，没办法，人家是老大，人家要的返回不是

人家要的返回只要一个success，人家定的标准不可能因为你一个系统改。俗话说的好，如果你改变不了环境，那你就只能我****

新增不进行封装注解

因为百分之99的请求还是需要包装的，只有个别不需要，写在包装的过滤器吧？又不是很好维护，那就加个注解好了。所有不需要包装的就加上这个注解。

@Target({
   
     ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface NotControllerResponseAdvice {
   
     
}

然后在我们的增强过滤方法上过滤包含这个注解的方法

@RestControllerAdvice(basePackages = {
   
     'com.bugpool.leilema'})
public class ControllerResponseAdvice implements ResponseBodyAdvice<Object> {
   
     
    @Override
    public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
   
     
        // response是ResultVo类型，或者注释了NotControllerResponseAdvice都不进行包装
        return !(methodParameter.getParameterType().isAssignableFrom(ResultVo.class)
                || methodParameter.hasMethodAnnotation(NotControllerResponseAdvice.class));
    }
    ...

最后就在不需要包装的方法上加上注解

@RestController
public class HealthController {
   
     

    @GetMapping('/health')
    @NotControllerResponseAdvice
    public String health() {
   
     
        return 'success';
    }
}

这时候就不会自动封装了，而其他没加注解的则依旧自动包装

五、统一异常

每个系统都会有自己的业务异常，比如库存不能小于0子类的，这种异常并非程序异常，而是业务操作引发的异常，我们也需要进行规范的编排业务异常状态码，并且写一个专门处理的异常类，最后通过刚刚学习过的异常拦截统一进行处理，以及打日志

1、异常状态码枚举，既然是状态码，那就肯定要实现我们的标准接口StatusCode；

@Getter
public enum  AppCode implements StatusCode {
   
     

    APP_ERROR(2000, '业务异常'),
    PRICE_ERROR(2001, '价格异常');

    private int code;
    private String msg;

    AppCode(int code, String msg) {
   
     
        this.code = code;
        this.msg = msg;
    }
}

2、异常类，这里需要强调一下，code代表AppCode的异常状态码，也就是2000；msg代表业务异常，这只是一个大类，一般前端会放到弹窗title上；最后super(message);这才是抛出的详细信息，在前端显示在弹窗体中，在ResultVo则保存在data中；

@Getter
public class APIException extends RuntimeException {
   
     
    private int code;
    private String msg;

    // 手动设置异常
    public APIException(StatusCode statusCode, String message) {
   
     
        // message用于用户设置抛出错误详情，例如：当前价格-5，小于0
        super(message);
        // 状态码
        this.code = statusCode.getCode();
        // 状态码配套的msg
        this.msg = statusCode.getMsg();
    }

    // 默认异常使用APP_ERROR状态码
    public APIException(String message) {
   
     
        super(message);
        this.code = AppCode.APP_ERROR.getCode();
        this.msg = AppCode.APP_ERROR.getMsg();
    }

}

3、最后进行统一异常的拦截，这样无论在service层还是controller层，开发人员只管抛出API异常，不需要关系怎么返回给前端，更不需要关心日志的打印；

@RestControllerAdvice
public class ControllerExceptionAdvice {

    @ExceptionHandler({
   
     BindException.class})
    public ResultVo MethodArgumentNotValidExceptionHandler(BindException e) {
   
     
        // 从异常对象中拿到ObjectError对象
        ObjectError objectError = e.getBindingResult().getAllErrors().get(0);
        return new ResultVo(ResultCode.VALIDATE_ERROR, objectError.getDefaultMessage());
    }

    @ExceptionHandler(APIException.class)
    public ResultVo APIExceptionHandler(APIException e) {
   
     
     // log.error(e.getMessage(), e); 由于还没集成日志框架，暂且放着，写上TODO
        return new ResultVo(e.getCode(), e.getMsg(), e.getMessage());
    }
}

4、最后使用，我们的代码只需要这么写；

if (null == orderMaster) {
    throw new APIException(AppCode.ORDER_NOT_EXIST, '订单号不存在：' + orderId);
 }

就会自动抛出AppCode.ORDER_NOT_EXIST状态码的响应，并且带上异常详细信息订单号不存在：xxxx。后端小哥开发有效率，前端妹妹获取到2003状态码，调用对应警告弹窗，title写上订单不存在，body详细信息记载'订单号不存在：1998'。同时日志还自动打上去了！666！老哥们三连点个赞！

来源：bugpool.blog.csdn.net/

本站仅提供存储服务，所有内容均由用户发布，如发现有害或侵权内容，请点击举报。