@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());
}
}
@RestControllerAdvice(basePackages = {"com.test.controller"})
public class ControllerResponseAdvice implements ResponseBodyAdvice
@Override
public boolean supports(MethodParameter methodParameter, Class>aClass) {
// 过滤不包装的类
return !(methodParameter.getParameterType().isAssignableFrom(ResultVo.class)
|| methodParameter.hasMethodAnnotation(NotControllerResponseAdvice.class));
}
//beforeBodyWrite对supports进行拦截的response进行处理，封装你需要的类型参数，加密等等
@Override
public Object beforeBodyWrite(Object data, MethodParameter returnType, MediaType mediaType, Class>aClass, ServerHttpRequest request, ServerHttpResponse response) {
// String类型不能直接包装
if (returnType.getGenericParameterType().equals(String.class)) {
ObjectMapper objectMapper = new ObjectMapper();
try {
// 将数据包装在ResultTO里后转换为json串进行返回
return objectMapper.writeValueAsString(ResultTO.OK(data));
} catch (JsonProcessingException e) {
throw new APIException(AppCode.VALIDATE_ERROR, e.getMessage());
}
}
// 否则直接包装成ResultVo返回
return ResultTO.OK(data);
}
你是否还在寻找稳定的海外服务器提供商？创新互联www.cdcxhl.cn海外机房具备T级流量清洗系统配攻击溯源，准确流量调度确保服务器高可用性，企业级服务器适合批量采购，新人活动首月15元起，快前往官网查看详情吧
