ResponseAdvisor.java 1.0 KB

12345678910111213141516171819202122232425262728293031
  1. package com.suyang;
  2. import org.springframework.core.MethodParameter;
  3. import org.springframework.http.MediaType;
  4. import org.springframework.http.converter.HttpMessageConverter;
  5. import org.springframework.http.server.ServerHttpRequest;
  6. import org.springframework.http.server.ServerHttpResponse;
  7. import org.springframework.web.bind.annotation.ControllerAdvice;
  8. import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
  9. import com.suyang.domain.ResponseMessage;
  10. @ControllerAdvice
  11. public class ResponseAdvisor implements ResponseBodyAdvice<Object> {
  12. @Override
  13. public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
  14. return true;
  15. }
  16. @Override
  17. public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType,
  18. Class<? extends HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request,
  19. ServerHttpResponse response) {
  20. if(body instanceof ResponseMessage) {
  21. return body;
  22. }
  23. return ResponseMessage.success(body);
  24. }
  25. }