SpringBoot统一返回结果问题
作者：阿信爸
目前很多项目都是前后端分离，前后端会事先约定好返回格式，前端如果使用axios的话，会在全局过滤器里进行校验，那么后端如何做，才能优雅的返回固定的统一格式呢，接下来我会带着你一步步实现。
1. 直接返回结果
先看一下最基本的例子，直接将结果原封不动返回：
@Data
@AllArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public class TestVo {
private static final long serialVersionUID = 1L;
@Schema(name = "姓名")
private String name;
@Schema(name = "年龄")
private Integer age;
}@RestController
@RequestMapping(value = "/test")
public class TestApi {
@GetMapping("/simple")
public TestVo simple() {
TestVo testVo = new TestVo("张三", 30);
return testVo;
}
}返回结果：
{
"name": "张三",
"age": 30
}2. 约定返回格式
假如与前端开发妹子约定好了格式，比如：
{
"code": 0,
"msg": "错误信息",
"data": 实际返回结果
}那么我们首先需要编写一个封装结果类Result，为了方便封装，我们再提供一个success方法：
@Data
@JsonInclude(JsonInclude.Include.NON_NULL)
public class Result<T> implements Serializable {
private static final long serialVersionUID = 1L;
/**
* 返回编码
*/
private Integer code;
/**
* 编码描述
*/
private String msg;
/**
* 业务数据
*/
private T data;
/**
* 返回成功结果对象
*
* @param data
* @param <T>
* @return
*/
public static <T> Result<T> success(T data) {
Result result = new Result();
result.setCode(0);
result.setMsg("success");
result.setData(data);
return result;
}
}3. 返回统一格式结果
后台接口代码微调一下即可，返回值改为Result，泛型为原返回值的类型：
@RestController
@RequestMapping(value = "/test")
public class TestApi {
@GetMapping("/withResult")
public Result<TestVo> withResult() {
TestVo testVo = new TestVo("张三", 30);
return Result.success(testVo);
}
}返回结果：
{
"code": 0,
"msg": "success",
"data": {
"name": "张三",
"age": 30
}
}至此，后端返回结果完美符合格式，但是每个接口的返回值都是Result<>，并且return的时候都要用success方法封装一下，不够优雅，怎么办？我们继续看：
4. 切片封装统一格式
编写注解
实际使用场景中，并不是所有接口都能完美的统一格式，使用注解可以按需控制接口返回格式。
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface ApiResult {
String value() default "";
int successCode() default 0;
String successMsg() default "success";
Class<? extends IResult> resultClass() default Result.class;
}编写ControllerAdvice
@Order(0)
@ControllerAdvice
public class MyResponseBodyAdvice implements ResponseBodyAdvice {
protected boolean isStringConverter(Class converterType) {
return converterType.equals(StringHttpMessageConverter.class);
}
protected boolean isApiResult(MethodParameter returnType) {
return returnType.hasMethodAnnotation(ApiResult.class);
}
@Override
public boolean supports(MethodParameter returnType, Class converterType) {
return !isStringConverter(converterType) && isApiResult(returnType);
}
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType,
Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
//关键
return Result.success(body);
}
}这里有一点要注意，这个advice中supports方法中判断返回结果类型必须为非String类型。如果返回结果为String类型，那么result要转为json字符串后再返回，需要再写一个advice。
接下来见证奇迹的时刻到了
@ApiResult
@GetMapping("/withResultHide")
public TestVo withResultHide() {
TestVo testVo = new TestVo("张三", 30);
return testVo;
}这段代码与直接返回结果的一样，仅仅加上了@ApiResult注解，我们看返回结果：
{
"code": 0,
"msg": "success",
"data": {
"name": "张三",
"age": 30
}
}大功告成！
以上只是最精简的例子，实际使用中还结合了 统一异常封装、自定义返回格式 等功能。
我们注意到@ApiResult注解中，有三个参数：successCode、successMsg、resultClass，就是为了自定义返回格式预留的，
下面看个例子：
5. 自定义返回格式
返回成功时code为200
如果个别接口的返回格式与默认格式相同，但是要求code等于200代表成功，那么修改下successCode参数即可：
@ApiResult(successCode = 200, successMsg = "ok")
@GetMapping("/withResultHide")
public TestVo withResultHide() {
TestVo testVo = new TestVo("张三", 30);
return testVo;
}我们发现，返回成功时，结果中的code和msg都变为设置的值：
{
"code": 200,
"msg": "ok",
"data": {
"name": "张三",
"age": 30
}
}自定义返回格式
如果个别的接口返回格式不是默认的返回格式，那么我们需要新增一个返回结果类，比如ReturnResult，里面包含returnCode、returnDesc、data三个属性：
@Data
@JsonInclude(JsonInclude.Include.NON_NULL)
public class ReturnResult<T> implements Serializable {
private static final long serialVersionUID = 1L;
/**
* 返回编码
*/
private String returnCode;
/**
* 编码描述
*/
private String returnDesc;
/**
* 业务数据
*/
private T data;
/**
* 返回成功结果对象
*
* @param data
* @param <T>
* @return
*/
public static <T> ReturnResult<T> success(T data) {
ReturnResult result = new ReturnResult();
result.setReturnCode(0);
result.setReturnDesc("success");
result.setData(data);
return result;
}
}然后修改接口上的@ApiResult注解中的resultClass属性
@ApiResult(resultClass = ReturnResult.class)
@GetMapping("/withResultHide")
public TestVo withResultHide() {
TestVo testVo = new TestVo("张三", 30);
return testVo;
}这时，返回结果变为如下：
{
"returnCode": "0",
"returnDesc": "success",
"data": {
"name": "张三",
"age": 30
}
}注意：以上自定义返回结果的例子需要自行根据实际情况添加，文章中的例子并不能实现
总结
只要按照上面一步步改造，即可实现统一返回结果，既规范、又优雅，也可根据自己所需添加对应的属性和数据。还等什么，搞起来吧~
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。