Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.github.wzc789376152.springboot.config.CommonResponseAdvice Maven / Gradle / Ivy
package com.github.wzc789376152.springboot.config;
import com.alibaba.fastjson.JSONObject;
import com.github.wzc789376152.exception.BizRuntimeException;
import com.github.wzc789376152.exception.SystemException;
import com.github.wzc789376152.service.IResponseService;
import com.github.wzc789376152.springboot.annotation.NoResultFormatter;
import com.github.wzc789376152.utils.JSONUtils;
import com.github.wzc789376152.vo.RetResult;
import feign.FeignException;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.poi.ss.formula.functions.T;
import org.slf4j.MDC;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.MethodParameter;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.validation.BindException;
import org.springframework.validation.BindingResult;
import org.springframework.validation.FieldError;
import org.springframework.web.bind.MissingServletRequestParameterException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseStatus;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
import javax.servlet.http.HttpServletResponse;
import javax.validation.ConstraintViolationException;
import java.text.MessageFormat;
import java.util.List;
import java.util.Objects;
@Slf4j
public abstract class CommonResponseAdvice implements ResponseBodyAdvice {
@Autowired
IResponseService responseService;
public IResponseService getResponseService() {
return this.responseService;
}
public String getTraceId() {
return MDC.get("traceId");
}
@Override
public boolean supports(MethodParameter methodParameter, Class aClass) {
return true;
}
@Override
public Object beforeBodyWrite(Object o, MethodParameter methodParameter, MediaType mediaType, Class aClass, ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
List headerList = serverHttpRequest.getHeaders().get("FeignResultFormat");
String traceId = getTraceId();
if (o != null) {
if (o instanceof String) {
if (((String) o).length() >= 2 && ((String) o).startsWith("\"") && ((String) o).endsWith("\"")) {
o = JSONObject.parse((String) o);
}
}
}
if (headerList == null || !headerList.get(0).equals("true")) {
NoResultFormatter noResultFormatter = AnnotationUtils.findAnnotation(Objects.requireNonNull(methodParameter.getMethod()), NoResultFormatter.class);
if (noResultFormatter == null) {
if (o == null || !o.getClass().getName().equals(responseService.getResultClass().getName())) {
ApiOperation apiOperation = AnnotationUtils.findAnnotation(Objects.requireNonNull(methodParameter.getMethod()), ApiOperation.class);
if (apiOperation != null) {
o = responseService.success(traceId, o, apiOperation.value() + "成功");
} else {
o = responseService.success(traceId, o, "");
}
}
}
}
if (mediaType.equals(MediaType.APPLICATION_JSON)) {
return o;
}
if (o != null && o.getClass().getName().equals(responseService.getResultClass().getName())) {
return JSONUtils.parse(o, String.class);
}
return o;
}
@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
@ExceptionHandler(value = DuplicateKeyException.class)
public Object exception(DuplicateKeyException e) {
log.error("唯一索引异常", e);
return error(500, "记录已存在，无法重复添加", e);
}
@ResponseStatus(HttpStatus.BAD_REQUEST)
@ExceptionHandler(value = MissingServletRequestParameterException.class)
public Object exception(MissingServletRequestParameterException e) {
String format = MessageFormat.format("参数 {0} 类型 {1} 解析失败", e.getParameterName(), e.getParameterType());
log.error(format, e);
return error(400, format, e);
}
@ResponseStatus(HttpStatus.BAD_REQUEST)
@ExceptionHandler(value = BindException.class)
public Object exception(BindException exception, HttpServletResponse response) {
BindingResult result = exception.getBindingResult();
final List fieldErrors = result.getFieldErrors();
StringBuilder builder = new StringBuilder();
for (FieldError error : fieldErrors) {
String message = error.getDefaultMessage();
if (message != null && message.contains("IllegalArgumentException")) {
message = message.substring(message.lastIndexOf("IllegalArgumentException") + 1);
message = message.substring(message.indexOf(":") + 1);
message = message.substring(0, message.contains(";") ? message.indexOf(";") : message.length() - 1);
}
builder.append(message).append("\n");
}
log.error(builder.toString(), exception);
response.setStatus(400);
return error(400, builder.toString(), exception);
}
/**
* 处理请求单个参数不满足校验规则的异常信息
*
* @param exception 异常
* @return RetResult
*/
@ResponseStatus(HttpStatus.BAD_REQUEST)
@ExceptionHandler(value = ConstraintViolationException.class)
public Object exception(ConstraintViolationException exception, HttpServletResponse response) {
log.error(exception.getMessage(), exception);
response.setStatus(400);
return error(400, exception.getMessage(), exception);
}
/**
* 处理自定义异常
*
* @param exception 异常
* @return RetResult
*/
@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
@ExceptionHandler(value = BizRuntimeException.class)
public Object exception(BizRuntimeException exception, HttpServletResponse response) {
log.error(exception.getMessage(), exception);
response.setStatus(500);
return error(ObjectUtils.isNotEmpty(exception.getErrorCode()) ? exception.getErrorCode() : 500, exception.getMessage(), exception);
}
@ResponseStatus(HttpStatus.BAD_REQUEST)
@ExceptionHandler(value = HttpMessageNotReadableException.class)
public Object exception(HttpMessageNotReadableException exception, HttpServletResponse response) {
log.error(exception.getMessage(), exception);
response.setStatus(400);
return error(400, "请输入正确数据格式", exception);
}
@ExceptionHandler({FeignException.class})
public Object exception(FeignException exception, HttpServletResponse response) {
log.error(exception.getMessage(), exception);
response.setStatus(exception.status());
String message = exception.getMessage();
if (message.contains("[{")) {
message = message.substring(message.lastIndexOf("[{") + 1, message.lastIndexOf("}]") + 1);
return JSONUtils.parse(message, responseService.getResultClass());
} else if (message.contains("\"]")) {
message = message.substring(message.lastIndexOf("[\"") + 2, message.lastIndexOf("\"]"));
}
return error(exception.status(), message, exception);
}
@ResponseStatus(HttpStatus.INTERNAL_SERVER_ERROR)
@ExceptionHandler(value = {Exception.class})
public Object exception(Exception exception) {
log.error(exception.getMessage(), exception);
if (exception.getCause() != null && exception.getCause() instanceof BizRuntimeException) {
return exception((BizRuntimeException) exception.getCause());
}
return error(500, SystemException.SYSTEM_ERROR_MSG, exception);
}
protected abstract void afterError(Integer code, String msg, Exception exception);
protected Object error(Integer code, String msg, Exception exception) {
afterError(code, msg, exception);
return responseService.error(getTraceId(), code, msg);
}
}
