SpringBoot中这样编写 Controller 层代码，优雅

前言

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

• 接口地址（也就是 URL 地址）
• 请求方式（一般就是 get、set，当然还有 put、delete）
• 请求数据（request，有 head 跟 body）
• 响应数据（response）

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

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

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

常见的请求就分为 get 跟 post 两种：

@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) {
        ...
    }
}

RestController： @RestController=@Controller+ResponseBody。

加上这个注解，Springboot 就会把这个类当成 controller 进行处理，然后把所有返回的参数放到 ResponseBody 中。

@RequestMapping： 请求的前缀，也就是所有该 Controller 下的请求都需要加上 /product/product-info 的前缀。

@GetMapping("/findById")： 标志这是一个 get 请求，并且需要通过 /findById 地址才可以访问到。

@PostMapping("/page")： 同理，表示是个 post 请求。

参数：至于参数部分，只需要写上 ProductInfoQueryVo，前端过来的 json 请求便会通过映射赋值到对应的对象中，例如请求这么写，productId 就会自动被映射到 vo 对应的属性当中。

size : 1
current : 1

productId : 1
productName : 泡脚

统一状态码

返回格式

为了跟前端妹妹打好关系，我们通常需要对后端返回的数据进行包装一下，增加一下状态码，状态信息，这样前端妹妹接收到数据就可以根据不同的状态码，判断响应数据状态，是否成功是否异常进行不同的显示。

当然这让你拥有了更多跟前端妹妹的交流机会，假设我们约定了 1000 就是成功的意思。

如果你不封装，那么返回的数据是这样子的：

{
  "productId": 1,
  "productName": "泡脚",
  "productPrice": 100.00,
  "productDescription": "中药泡脚加按摩",
  "productStatus": 0,
}

经过封装以后时这样子的：

{
  "code": 1000,
  "msg": "请求成功",
  "data": {
    "productId": 1,
    "productName": "泡脚",
    "productPrice": 100.00,
    "productDescription": "中药泡脚加按摩",
    "productStatus": 0,
  }
}

封装 ResultVo

这些状态码肯定都是要预先编好的，怎么编呢？写个常量 1000？还是直接写死 1000？

要这么写就真的书白读的了，写状态码当然是用枚举拉：

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

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

② 然后去找前端妹妹，跟他约定好状态码（这可能是你们唯一的约定了）枚举类嘛，当然不能有 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;
    }
}

③ 写好枚举类，就开始写 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;
    }
}

使用，现在的返回肯定就不是 return data；这么简单了，而是需要 new ResultVo(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)));
}

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

统一校验

原始做法

假设有一个添加 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 写的人都傻了，能忍吗？肯定不能忍啊。

@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": "org.springframework.validation.BindException: org.springframework.validation.BeanPropertyBindingResult: 1 errors\nField error in object \u0027productInfoVo\u0027 on field \u0027productPrice\u0027: rejected value [-1]; ......",
  "path": "/leilema/product/product-info/findByVo"
}

大功告成了吗？虽然成功校验了参数，也返回了异常，并且带上"商品价格不允许为负数"的信息。

但是你要是这样返回给前端，前端妹妹就提刀过来了，当年约定好的状态码，你个负心人说忘就忘？

用户体验小于等于 0 啊！所以我们要进行优化一下，每次出现异常的时候，自动把状态码写好，不负妹妹之约！

优化异常处理

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

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 与前端妹妹约定好的状态码：

{
  "code": 1002,
  "msg": "参数校验失败",
  "data": "商品价格不允许为负数"
}

统一响应

统一包装响应

再回头看一下 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