栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Java

SpringBoot统一返回值和统一异常处理

Java 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

SpringBoot统一返回值和统一异常处理

在项目开发中,我们通常会约定一个接口返回的数据格式,其中包含code,message,data等信息,像下面这样:

{
  "code": 0,
  "message": "success",
  "data": ""
}
没有异常的情况,正常返回数据 接口数据对应的实体类
@Data
@AllArgsConstructor
@NoArgsConstructor
public class ResultVo implements Serializable {
    
    private Integer code ;
    private String message ;
    private Object data ;
    
}
用于生成Result实体的工具类
public class ResultUtils {
    
    
    public static ResultVo success(Object o) {
        return new ResultVo(ResultEnum.SUCCESS.getCode(), ResultEnum.SUCCESS.getMessage(), o);
    }
    
    public static ResultVo success() {
        return success(null);
    }
    
    public static ResultVo error(Integer code, String message) {
        return new ResultVo(code, message,null);
    }
    
}
封装code和message的枚举类

可以约定更多的code和对应的message封装成枚举类,以便不同情况下灵活处理

public enum ResultEnum {
    
    UNKUOW_ERROR(-1,"未知错误"),
    SUCCESS(0,"success"),
    ERROR(1,"error");
    
    private int code;
    private String message;
    
    ResultEnum(int code ,String message) {
        this.code = code;
        this.message = message;
    }
    //………省略getter and setter
}
在Controller中返回

在Controller中就可以调用ResultUtils的方法进行返回了

@RequestMapping("/test")
public ResultVo test(){
    User user = new User();
    user.setName("lisi");
    user.setId(1);
    return ResultUtils.success(user);
}

但是这样在每个请求方法代码中都要对返回数据进行处理,请求返回的返回值也都是ResultVo,不能方便的看到每个方法的返回值。我们使用AOP切面对Controller中的方法进行封装来解决这个问题

统一返回值切面
@ControllerAdvice
public class MyResponseAdvice implements ResponseBodyAdvice {

    @Autowired
    ObjectMapper objectMapper;

    //判断是否要执行beforeBodyWrite方法,true为执行,false不执行. 通过该方法可以选择哪些类或那些方法的response要进行处理, 其他的不进行处理.
    @Override
    public boolean supports(MethodParameter returnType, Class converterType) {
        return true;
    }

    //对response方法进行具体操作处理
    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
        //返回类型是否已经封装
        if (body instanceof ResultVo){
            return body;
        }
        // String特殊处理
        if (body instanceof String) {
            try {
                return objectMapper.writeValueAsString(ResultUtils.success(body));
            } catch (JsonProcessingException e) {
                return ResultUtils.error(500,"失败");
            }
        }
        return ResultUtils.success(body);
    }
}
 
有异常的情况,统一异常处理 
自定义异常类 

自定义异常类,继承RuntimeException。为方便使用可以使用枚举类列举出常见的异常情况,这里直接使用了上面的枚举类。

@Data
@NoArgsConstructor
@AllArgsConstructor
public class BusinessException extends RuntimeException{

    private int code;

    private String msg;

    public BusinessException(ResultEnum resultEnum) {
        this.code = resultEnum.getCode();
        this.msg = resultEnum.getMessage();
    }
}
自定统一异常处理器
@ControllerAdvice
public class BusinessExceptionHandler {

    @ResponseBody
    @ExceptionHandler({BusinessException.class})
    public ResultVo exceptionHandler(BusinessException exception){
        // 省略记录日志
        return new ResultVo(exception.getCode(), exception.getMsg(), null);
    }

    @ResponseBody
    @ExceptionHandler({Exception.class})
    public ResultVo exceptionHandler(Exception exception){
        // 省略记录日志
        return new ResultVo(ResultEnum.UNKUOW_ERROR.getCode(), ResultEnum.UNKUOW_ERROR.getMessage(), exception.getMessage());
    }
    
}
在Service层直接抛出异常
@Override
public User test(String str) {

    if (Objects.isNull(str)){
        throw new BusinessException(ResultEnum.ERROR);
    }

    return new User();

}
@RequestMapping("/test")
public User test(){
    User user = userService.test(null);
    return user;
}

转载请注明:文章转载自 www.mshxw.com
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号