如何在Spring Boot中访问RestController方法（endpoint）返回的POJO？

Question

我希望能够在RestController方法（或endpoint）作为数据流序列化为HttpServletResponse之前修改它返回的POJO。但是我希望能够在控制器方法代码之外（作为中间件）完成。

我曾尝试使用HandlerInterceptor来完成此操作，但我无法访问POJO。我也尝试过使用AOP，但从未调用切入点。

@RestController
public class TestController {
  @GetMapping("/test")
  public Resource<User> getTest() {
    Resource<User> resource = new Resource<>();
    resource.setData(new User("test user"));
    return resource;
  }


  @Builder
  @Getter
  @Setter
  @AllArgsConstructor
  static class User {
    private String username;
  }

  @Getter
  @Setter
  @AllArgsConstructor
  @NoArgsConstructor
  static class Resource<T> {
    private T data;
    private Set<String> errors;
  }
}

我希望能够在控制器执行自己的逻辑并返回后，向控制器返回的资源添加错误列表（如果需要）。

Answer 1

要在从@restcontroller方法返回后但在将其写入HTTP响应之前更改该对象，可以实现ResponseBodyAdvise并将其声明为@ControllerAdvisebean：

@ControllerAdvice
public static class Foo implements ResponseBodyAdvice {

    @Override
    public boolean supports(MethodParameter returnType, Class converterType) {
        return true;
    }

    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType,
            Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {

        //body is the object return from the @RestController method
        //Cast it and modify it accordingly.
        if(body instanceof Resource) {
            Resource res = (Resource)body;
            //Modify it .... blablablba
        }
        return body;
    }
}