spring项目自定义全局响应处理器,统一处理响应结果
实现方式
使用@ControllerAdvice注解和实现ResponseBodyAdvice接口自定义全局响应处理器
实现步骤
- 自定义统一返回类
定义统一返回类,包装返回结果
@Data
@JsonInclude(JsonInclude.Include.NON_NULL)//过滤null,提高性能
@ApiModel(value = "返回类")
public class Wrapper<T> implements Serializable {
/**
* 成功码
*/
public static final int SUCCESS_CODE = 200;
/**
* 成功信息
*/
public static final String SUCCESS_MESSAGE = "操作成功";
/**
* 错误码
*/
public static final int ERROR_CODE = 500;
/**
* 错误信息
*/
public static final String ERROR_MESSAGE = "内部异常";
/**
* 编号
*/
@ApiModelProperty("编号")
private int code;
/**
* 信息
*/
@ApiModelProperty("信息")
private String message;
/**
* 结果数据
*/
@ApiModelProperty("结果数据")
private T result;
/** 为什么使用this替代new类名
* 代码简洁:使用 this 可以使得代码更加简洁,因为你不需要每次都重复使用类的名称。
* 明确性:通过使用 this,你清楚地表明你正在调用当前类的另一个构造函数或方法,而不是其他类的构造函数或方法。
* 减少重复:在多构造函数的类中,如果每个构造函数都需要设置相同的属性,那么使用 this 可以避免重复代码。
* 提高可读性:当一个构造函数调用了另一个构造函数时,使用 this 可以让读者更容易理解正在发生的事情,尤其是当构造函数的参数不同,但都包含某些共同参数时。
*/
//默认构造函数,它直接调用第二个构造函数,并传递 SUCCESS_CODE 和 SUCCESS_MESSAGE 作为参数。
Wrapper() {
this(SUCCESS_CODE, SUCCESS_MESSAGE);
}
//不需要返回数据 带有两个参数的构造函数,它调用第三个构造函数,并传递相同的参数。
Wrapper(int code, String message) {
this(code, message, null);
}
//需要返回数据
Wrapper(int code, String message, T result) {
super();
this.code(code).message(message).result(result);
}
private Wrapper<T> code(int code) {
this.setCode(code);
return this;
}
private Wrapper<T> message(String message) {
this.setMessage(message);
return this;
}
public Wrapper<T> result(T result) {
this.setResult(result);
return this;
}
@JsonIgnore
public boolean success() {
return Wrapper.SUCCESS_CODE == this.code;
}
@JsonIgnore
public boolean error() {
return !success();
}
}
- 定义 统一返回包装类
public class WrapMapper {
/**
* 在 `WrapMapper` 类中,私有构造函数的设计是为了防止外部实例化这个类。这是一个常见的做法,有以下几个原因:
* 1. **单一职责原则**:`WrapMapper` 类被设计为一个工具类,只提供一系列静态方法,用于创建和处理 `Wrapper` 对象。它不包含任何实例变量,因此它没有必要被实例化。
* 2. **减少依赖**:如果 `WrapMapper` 类可以被实例化,那么任何使用这个类的代码都需要导入这个类的引用。通过将构造函数设为私有,你可以限制它的使用,只允许通过静态方法来使用这个类。
* 3. **提高效率**:工具类通常不需要创建多个实例,因为它们不依赖于外部状态,而且它们的操作通常可以通过静态方法完成。因此,设计为不可实例化可以提高效率。
* 4. **清晰意图**:通过将构造函数设为私有,明确地表明这个类不应该被实例化,这有助于其他开发者理解这个类的用途和设计意图。
* 总之,将 `WrapMapper` 类的构造函数设为私有是为了遵循单一职责原则,减少依赖,提高效率,并提高代码的可读性和可维护性。这样的设计有助于确保 `WrapMapper` 类只作为工具类被使用,而不被误用或滥用。
*/
private WrapMapper() {
}
public static <E> Wrapper<E> wrap(int code) {
return wrap(code, null);
}
public static <E> Wrapper<E> wrap(int code, String message) {
return wrap(code, message, null);
}
public static <E> Wrapper<E> wrap(int code, String message, E o) {
return new Wrapper<>(code, message, o);
}
public static <E> Wrapper<E> wrap(Exception e) {
return new Wrapper<>(Wrapper.ERROR_CODE, e.getMessage());
}
public static <E> E unWrap(Wrapper<E> wrapper) {
return wrapper.getResult();
}
public static <E> Wrapper<E> error() {
return wrap(Wrapper.ERROR_CODE, Wrapper.ERROR_MESSAGE);
}
public static <E> Wrapper<E> error(int code,String message) {
return wrap(code, message);
}
public static <E> Wrapper<E> error(String message) {
return wrap(Wrapper.ERROR_CODE, StringUtils.isBlank(message) ? Wrapper.ERROR_MESSAGE : message);
}
public static <E> Wrapper<E> ok() {
return new Wrapper<>();
}
public static <E> Wrapper<E> ok(E o) {
return new Wrapper<>(Wrapper.SUCCESS_CODE, Wrapper.SUCCESS_MESSAGE, o);
}
}
- 定义启用全局响应注解
@Target({ElementType.TYPE, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface EnableResponseHandler {
}
- 定义忽略全局响应注解
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface IgnoreResponseHandler {
}
- 定义全局响应处理器
//@Component标记的类,Spring 容器会自动扫描这个类,并将它作为一个 bean 创建和注册到 Spring 容器中。
//可以被其他类通过 @Autowired 注解自动装配,你不需要显式地定义 bean 的依赖关系,Spring 容器会自动处理。
@Component
@ControllerAdvice//全局控制器增强
public class GlobalResponseHandler implements ResponseBodyAdvice<Object> {
@Override
public boolean supports(MethodParameter methodParameter, Class<? extends HttpMessageConverter<?>> aClass) {
if (methodParameter.getMethod().isAnnotationPresent(IgnoreResponseHandler.class)) {
return false;
}
//只有EnableResponseHandler注解标识的类才会去调用beforeBodyWrite方法
if (methodParameter.getMethod().isAnnotationPresent(EnableResponseHandler.class)) {
return true;
}
if (methodParameter.getDeclaringClass().isAnnotationPresent(EnableResponseHandler.class)) {
return true;
}
return false;
}
/**
*beforeBodyWrite方法在 Spring MVC 将返回的对象写入 HTTP 响应之前被调用,它提供了一个机会来增强或修改响应体。
* 作用
* 1.增强响应:在返回对象被写入响应之前,允许你对响应进行增强或修改。
* 自定义格式化:如果你需要对返回的对象进行特殊的格式化,比如添加额外的头信息,或者修改对象的属性,beforeBodyWrite 提供了这种能力。
* 全局应用:你可以定义多个 ResponseBodyAdvice 实现类,并将它们应用于整个应用程序,或者针对特定的 Controller 或请求类型进行过滤。
*/
@Nullable
@Override
public Object beforeBodyWrite(@Nullable Object o,
MethodParameter methodParameter,
MediaType mediaType,
Class<? extends HttpMessageConverter<?>> aClass,
ServerHttpRequest serverHttpRequest,
ServerHttpResponse serverHttpResponse) {
//如果返回的不是自己定义的包装类
if (o instanceof Wrapper) {
return o;
}
//使用WrapMapper包装后返回
return WrapMapper.ok(o);
}
}