All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.doyospy.core.advice.RestResponseAdvice Maven / Gradle / Ivy

Go to download

Doyospy是一个小而全的基于SpringBoot的java工具类库，通过静态方法封装，方便开发者使用，降低学习成本并提升工作效率。

The newest version!
package com.doyospy.core.advice;

import com.doyospy.core.annotation.RestResponse;
import com.doyospy.core.result.ResponseResult;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.SneakyThrows;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;

/**
 * Rest响应处理
 * @author  Robin
 */
@RestControllerAdvice
public class RestResponseAdvice implements ResponseBodyAdvice {
    private final ObjectMapper objectMapper;

    public RestResponseAdvice(ObjectMapper objectMapper) {
        this.objectMapper = objectMapper;
    }

    @Override
    public boolean supports(MethodParameter returnType, Class> converterType) {
        return returnType.hasMethodAnnotation(RestResponse.class) ||
            returnType.getDeclaringClass().isAnnotationPresent(RestResponse.class);
    }

    @SneakyThrows
    @Override
    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType,
                                  Class> selectedConverterType,
                                  ServerHttpRequest request, ServerHttpResponse response) {
        if (body instanceof String) {
            return objectMapper.writeValueAsString(ResponseResult.success(body));
        }
        if (body instanceof ResponseResult) {
            return body;
        }
        return ResponseResult.success(body);
    }
}
 















© 2015 - 2024 Weber Informatics LLC | Privacy Policy