le.kuang
2 years ago
parent
34a3d04f1d
commit
9de258af0d
@ -0,0 +1,42 @@
|
|||||||
|
package com.java3y.austin.web.advice;
|
||||||
|
|
||||||
|
import com.java3y.austin.common.vo.BasicResultVO;
|
||||||
|
import com.java3y.austin.web.annotation.AustinResult;
|
||||||
|
import org.jetbrains.annotations.NotNull;
|
||||||
|
import org.springframework.core.MethodParameter;
|
||||||
|
import org.springframework.http.MediaType;
|
||||||
|
import org.springframework.http.server.ServerHttpRequest;
|
||||||
|
import org.springframework.http.server.ServerHttpResponse;
|
||||||
|
import org.springframework.web.bind.annotation.ControllerAdvice;
|
||||||
|
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
|
||||||
|
|
||||||
|
import java.util.Objects;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author kl
|
||||||
|
* @version 1.0.0
|
||||||
|
* @description 统一返回结构
|
||||||
|
* @date 2023/2/9 19:00
|
||||||
|
*/
|
||||||
|
@ControllerAdvice(basePackages = "com.java3y.austin.web.controller")
|
||||||
|
public class AustinResponseBodyAdvice implements ResponseBodyAdvice<Object> {
|
||||||
|
|
||||||
|
private static final String RETURN_CLASS = "BasicResultVO";
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean supports(MethodParameter methodParameter, Class aClass) {
|
||||||
|
return methodParameter.getContainingClass().isAnnotationPresent(AustinResult.class) || methodParameter.hasMethodAnnotation(AustinResult.class);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Object beforeBodyWrite(Object data, MethodParameter methodParameter, MediaType mediaType, Class aClass,
|
||||||
|
ServerHttpRequest serverHttpRequest, ServerHttpResponse serverHttpResponse) {
|
||||||
|
if (Objects.nonNull(data) && Objects.nonNull(data.getClass())) {
|
||||||
|
String simpleName = data.getClass().getSimpleName();
|
||||||
|
if (RETURN_CLASS.equalsIgnoreCase(simpleName)) {
|
||||||
|
return data;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return BasicResultVO.success(data);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,15 @@
|
|||||||
|
package com.java3y.austin.web.annotation;
|
||||||
|
|
||||||
|
import java.lang.annotation.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author kl
|
||||||
|
* @version 1.0.0
|
||||||
|
* @description 统一返回注解
|
||||||
|
* @date 2023/2/9 19:00
|
||||||
|
*/
|
||||||
|
@Target({ElementType.TYPE, ElementType.METHOD})
|
||||||
|
@Retention(RetentionPolicy.RUNTIME)
|
||||||
|
@Documented
|
||||||
|
public @interface AustinResult {
|
||||||
|
}
|
||||||
@ -0,0 +1,40 @@
|
|||||||
|
package com.java3y.austin.web.exception;
|
||||||
|
|
||||||
|
import com.java3y.austin.common.enums.RespStatusEnum;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author kl
|
||||||
|
* @version 1.0.0
|
||||||
|
* @description 通用异常
|
||||||
|
* @date 2023/2/9 19:00
|
||||||
|
*/
|
||||||
|
public class CommonException extends RuntimeException {
|
||||||
|
private String code = RespStatusEnum.ERROR_400.getCode();
|
||||||
|
|
||||||
|
public CommonException(String message) {
|
||||||
|
super(message);
|
||||||
|
}
|
||||||
|
|
||||||
|
public CommonException(RespStatusEnum respStatusEnum) {
|
||||||
|
super(respStatusEnum.getMsg());
|
||||||
|
this.code = respStatusEnum.getCode();
|
||||||
|
}
|
||||||
|
|
||||||
|
public CommonException(String code, String message) {
|
||||||
|
super(message);
|
||||||
|
this.code = code;
|
||||||
|
}
|
||||||
|
|
||||||
|
public CommonException(String message, Exception e) {
|
||||||
|
super(message, e);
|
||||||
|
}
|
||||||
|
|
||||||
|
public CommonException(String code, String message, Exception e) {
|
||||||
|
super(message, e);
|
||||||
|
this.code = code;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getCode() {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,49 @@
|
|||||||
|
package com.java3y.austin.web.exception;
|
||||||
|
|
||||||
|
import com.java3y.austin.common.enums.RespStatusEnum;
|
||||||
|
import com.java3y.austin.common.vo.BasicResultVO;
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
import org.springframework.http.HttpStatus;
|
||||||
|
import org.springframework.web.bind.annotation.ControllerAdvice;
|
||||||
|
import org.springframework.web.bind.annotation.ExceptionHandler;
|
||||||
|
import org.springframework.web.bind.annotation.ResponseBody;
|
||||||
|
import org.springframework.web.bind.annotation.ResponseStatus;
|
||||||
|
|
||||||
|
import java.io.PrintWriter;
|
||||||
|
import java.io.StringWriter;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author kl
|
||||||
|
* @version 1.0.0
|
||||||
|
* @description 拦截异常统一返回
|
||||||
|
* @date 2023/2/9 19:03
|
||||||
|
*/
|
||||||
|
@ControllerAdvice
|
||||||
|
@ResponseBody
|
||||||
|
public class ExceptionHandlerAdvice {
|
||||||
|
private static final Logger log = LoggerFactory.getLogger(ExceptionHandlerAdvice.class);
|
||||||
|
|
||||||
|
public ExceptionHandlerAdvice() {
|
||||||
|
}
|
||||||
|
|
||||||
|
@ExceptionHandler({Exception.class})
|
||||||
|
@ResponseStatus(HttpStatus.OK)
|
||||||
|
public BasicResultVO exceptionResponse(Exception e) {
|
||||||
|
StringWriter sw = new StringWriter();
|
||||||
|
PrintWriter pw = new PrintWriter(sw);
|
||||||
|
e.printStackTrace(pw);
|
||||||
|
BasicResultVO result = BasicResultVO.fail(RespStatusEnum.ERROR_500,"\r\n" + sw + "\r\n");
|
||||||
|
log.error(e.getMessage(), e);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
@ExceptionHandler({CommonException.class})
|
||||||
|
@ResponseStatus(HttpStatus.OK)
|
||||||
|
public BasicResultVO commonResponse(CommonException ce) {
|
||||||
|
BasicResultVO result = BasicResultVO.fail(RespStatusEnum.ERROR_400, ce.getMessage());
|
||||||
|
log.error(ce.getMessage(), ce);
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Loading…
Reference in new issue