package com.vci.starter.web.autoconfigure;
|
|
import com.vci.starter.web.annotation.controller.VciUnUseResponseAdvice;
|
import com.vci.starter.web.constant.TokenKeyConstant;
|
import com.vci.starter.web.pagemodel.BaseResult;
|
import com.vci.starter.web.pagemodel.DataGrid;
|
import com.vci.starter.web.pagemodel.Tree;
|
import com.vci.starter.web.util.MessageUtils;
|
import org.apache.commons.lang3.StringUtils;
|
import org.slf4j.MDC;
|
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.lang.Nullable;
|
import org.springframework.stereotype.Controller;
|
import org.springframework.web.bind.annotation.ControllerAdvice;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
|
|
|
import java.util.ArrayList;
|
import java.util.Collection;
|
import java.util.List;
|
|
/**
|
* 在返回结果前，封装统一的数据对象
|
* @author weidy
|
* @date 2019/11/7 4:05 PM
|
*/
|
@ControllerAdvice(annotations = {RestController.class, Controller.class})
|
public class ResponseModifyAdvice implements ResponseBodyAdvice<Object>{
|
/**
|
* 过滤哪些需要拦截处理
|
* Whether this component supports the given controller method return type
|
* and the selected {@code HttpMessageConverter} type.
|
*
|
* @param returnType the return type
|
* @param converterType the selected converter type
|
* @return {@code true} if {@link #beforeBodyWrite} should be invoked;
|
* {@code false} otherwise
|
*/
|
@Override
|
public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
|
//全部都支持
|
if(returnType.getMethod().isAnnotationPresent(VciUnUseResponseAdvice.class)){
|
return false;
|
}
|
return true;
|
}
|
|
/**
|
* Invoked after an {@code HttpMessageConverter} is selected and just before
|
* its write method is invoked.
|
*
|
* @param body the body to be written
|
* @param returnType the return type of the controller method
|
* @param selectedContentType the content type selected through content negotiation
|
* @param selectedConverterType the converter type selected to write to the response
|
* @param request the current request
|
* @param response the current response
|
* @return the body that was passed in or a modified (possibly new) instance
|
*/
|
@Nullable
|
@Override
|
public Object beforeBodyWrite(@Nullable Object body, MethodParameter returnType, MediaType selectedContentType, Class<? extends HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
|
String traceId = MDC.get(TokenKeyConstant.TRACE_ID);
|
if(returnType.getMethod().isAnnotationPresent(VciUnUseResponseAdvice.class)){
|
return body;
|
}
|
String returnTypeName = returnType.getParameterType().getName();
|
if("void".equalsIgnoreCase(returnTypeName)){
|
BaseResult baseResult = BaseResult.success();
|
baseResult.setFinishTime(System.currentTimeMillis());
|
baseResult.setTraceId(traceId);
|
formateMsg(baseResult);
|
return baseResult;
|
}
|
if("com.vci.starter.web.pagemodel.BaseResult".equals(returnTypeName)){
|
BaseResult baseResult = ((BaseResult)body);
|
baseResult.setTraceId(traceId);
|
baseResult.setFinishTime(System.currentTimeMillis());
|
formateMsg(baseResult);
|
return baseResult;
|
}
|
if("com.vci.starter.web.pagemodel.DataGrid".equals(returnTypeName)){
|
DataGrid dataGrid = ((DataGrid)body);
|
BaseResult baseResult = BaseResult.dataGrid(dataGrid);
|
baseResult.setTraceId(traceId);
|
baseResult.setFinishTime(System.currentTimeMillis());
|
formateMsg(baseResult);
|
return baseResult;
|
}
|
if("com.vci.starter.web.pagemodel.Tree".equals(returnTypeName)){
|
//returnType.getP
|
Tree tree = ((Tree)body);
|
List<Tree> treeList = new ArrayList<>();
|
treeList.add(tree);
|
BaseResult baseResult = BaseResult.tree(treeList);
|
baseResult.setTraceId(traceId);
|
baseResult.setFinishTime(System.currentTimeMillis());
|
formateMsg(baseResult);
|
return baseResult;
|
}
|
if(body instanceof Collection){
|
//说明是集合
|
String genericParamterTypeName = returnType.getGenericParameterType().getTypeName();
|
if(genericParamterTypeName.contains("<")
|
&& genericParamterTypeName.contains(">")){
|
String elementTypeName = genericParamterTypeName.substring(genericParamterTypeName.indexOf("<")+ 1,genericParamterTypeName.indexOf(">"));
|
if("com.vci.starter.web.pagemodel.Tree".equalsIgnoreCase(elementTypeName)){
|
Collection<Tree> treeCollection = ((Collection<Tree>)body);
|
BaseResult baseResult = BaseResult.tree(treeCollection);
|
baseResult.setTraceId(traceId);
|
baseResult.setFinishTime(System.currentTimeMillis());
|
formateMsg(baseResult);
|
return baseResult;
|
}else{
|
//放到Obj属性里
|
Collection<Object> objectCollection = ((Collection<Object>)body);
|
BaseResult baseResult = BaseResult.success(objectCollection);
|
baseResult.setTraceId(traceId);
|
baseResult.setFinishTime(System.currentTimeMillis());
|
formateMsg(baseResult);
|
return baseResult;
|
}
|
}
|
}
|
if(body instanceof Object){
|
BaseResult baseResult = BaseResult.success((Object)body);
|
baseResult.setTraceId(traceId);
|
baseResult.setFinishTime(System.currentTimeMillis());
|
formateMsg(baseResult);
|
return baseResult;
|
}
|
return body;
|
}
|
|
/**
|
* 多语格式化
|
* @param baseResult 结果对象
|
*/
|
private void formateMsg(BaseResult baseResult ){
|
if(baseResult!=null && StringUtils.isNotBlank(baseResult.getMsg())){
|
baseResult.setMsg(MessageUtils.get(baseResult.getMsg(),baseResult.getMsgObjs()));
|
}
|
}
|
}
|
