diff --git a/src/main/java/com/objecteye/exception/GeneralExceptionHandler.java b/src/main/java/com/objecteye/exception/GeneralExceptionHandler.java index 2cfa827..c967567 100644 --- a/src/main/java/com/objecteye/exception/GeneralExceptionHandler.java +++ b/src/main/java/com/objecteye/exception/GeneralExceptionHandler.java @@ -17,6 +17,9 @@ public class GeneralExceptionHandler { @ResponseBody public CommonResult exception(Exception e) { log.error(e.getMessage(), e); + if (e instanceof CustomXException) { + return CommonResult.success(Integer.parseInt(((CustomXException) e).getCode()), ((CustomXException) e).getMsg(), null); + } return CommonResult.success(504, e.getMessage(), null); } }