在返回数据到前端时,通常会使用结果机来嵌套着数据,所以我们可以把它提取出来变成一个工具类.

建立结果集工具类

//使用枚举类
public enum StatustypeEnum {
  //成功时的返回消息
    SUCCESS(200,"success"),
//失败时返回的消息
    ERROR(404,"error");

//连接状态码
    private  int status;l
//返回消息
    private String msg;
//有参构造
    StatustypeEnum(int status, String msg) {
        this.status = status;
        this.msg = msg;
    }
}

建立返回成功类

@Data
@AllArgsConstructor
@NoArgsConstructor
public class SuccessResult<T> implements Result<T> {
//结果集
    private StatustypeEnum  status;
//返回的数据
    private T data;
//成功的方法
    public static SuccessResult success(){return new SuccessResult(StatustypeEnum.SUCCESS,null);}

    public static <T> SuccessResult success(T data){return new SuccessResult(StatustypeEnum.SUCCESS,data);}}
//三个注解是用来代替GET SET 有参构造以及无参构造

建立返回失败类

@Data
@NoArgsConstructor
@AllArgsConstructor
public class ErrorResult implements Result {
//状态码
    private int status;
//返回消息
    private String msg;
//返回异常消息
    public static ErrorResult error(int status, Exception ex) {
        return new ErrorResult(status, ex.getMessage());
    }
    public static ErrorResult error() {
        return new ErrorResult(500, "网络异常!!! 请联系运维小哥哥");
    }

}
//这里是因为失败后无需返回数据所有没有泛型的数据

核心,统一处理返回值

//建立Controller拦截
@RestControllerAdvice
//使用@RestContollerAdivice注解拦截Contoller的所有请求
@Slf4j
//实现ResponseBodyAdvice接口
public class GlobalResultAdvice  implements ResponseBodyAdvice {
//下面两个方法必须重写
    @Override
    public boolean supports(MethodParameter returnType, Class converterType) {
        log.error("supports");
      //这里若是返回False则不会执行下面的方法
        return true;
    }
//这个方法是用来判断如果类型
  @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
        Object resp = null;
//判断如果传入值的类型是符合条件的则输出
        if (body instanceof SuccessResult){
            resp = body;
        }else{
            resp =  SuccessResult.succes(body);
        }

        return resp;
    }
}