| commit | author | age | ||
| 9bcb19 | 1 | /* |
| I | 2 | Copyright [2020] [https://www.xiaonuo.vip] |
| 3 | ||
| 4 | Licensed under the Apache License, Version 2.0 (the "License"); | |
| 5 | you may not use this file except in compliance with the License. | |
| 6 | You may obtain a copy of the License at | |
| 7 | ||
| 8 | http://www.apache.org/licenses/LICENSE-2.0 | |
| 9 | ||
| 10 | Unless required by applicable law or agreed to in writing, software | |
| 11 | distributed under the License is distributed on an "AS IS" BASIS, | |
| 12 | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 13 | See the License for the specific language governing permissions and | |
| 14 | limitations under the License. | |
| 15 | ||
| 16 | Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点： | |
| 17 | ||
| 18 | 1.请不要删除和修改根目录下的LICENSE文件。 | |
| 19 | 2.请不要删除和修改Snowy源码头部的版权声明。 | |
| 20 | 3.请保留源码和相关描述文件的项目出处，作者声明等。 | |
| 21 | 4.分发源码时候，请注明软件出处 https://gitee.com/xiaonuobase/snowy | |
| 22 | 5.在修改包名，模块名称，项目代码等时，请注明软件出处 https://gitee.com/xiaonuobase/snowy | |
| 23 | 6.若您的项目无法满足以上几点，可申请商业授权，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip | |
| 24 | */ | |
| 25 | package vip.xiaonuo.sys.core.error; | |
| 26 | ||
| 27 | import cn.hutool.core.bean.BeanUtil; | |
| 28 | import cn.hutool.http.HttpStatus; | |
| 29 | import org.springframework.boot.web.error.ErrorAttributeOptions; | |
| 30 | import org.springframework.boot.web.servlet.error.DefaultErrorAttributes; | |
| 31 | import org.springframework.web.context.request.WebRequest; | |
| 32 | import vip.xiaonuo.core.exception.ServiceException; | |
| 33 | import vip.xiaonuo.core.exception.enums.PermissionExceptionEnum; | |
| 34 | import vip.xiaonuo.core.exception.enums.ServerExceptionEnum; | |
| 35 | import vip.xiaonuo.core.pojo.response.ErrorResponseData; | |
| 36 | ||
| 37 | import java.util.Map; | |
| 38 | ||
| 39 | /** | |
| 40 | * 将系统管理未知错误异常，输出格式重写为我们熟悉的响应格式 | |
| 41 | * | |
| 42 | * @author yubaoshan | |
| 43 | * @date 2020/4/14 22:21 | |
| 44 | */ | |
| 45 | public class SnowyErrorAttributes extends DefaultErrorAttributes { | |
| 46 | ||
| 47 | @Override | |
| 48 | public Map<String, Object> getErrorAttributes(WebRequest webRequest, ErrorAttributeOptions attributeOptions) { | |
| 49 | ||
| 50 | // 1.先获取spring默认的返回内容 | |
| 51 | Map<String, Object> defaultErrorAttributes = super.getErrorAttributes(webRequest, attributeOptions); | |
| 52 | ||
| 53 | // 2.如果返回的异常是ServiceException，则按ServiceException响应的内容进行返回 | |
| 54 | Throwable throwable = this.getError(webRequest); | |
| 55 | if (throwable instanceof ServiceException) { | |
| 56 | ServiceException serviceException = (ServiceException) throwable; | |
| 57 | return BeanUtil.beanToMap(new ErrorResponseData(serviceException.getCode(), serviceException.getErrorMessage())); | |
| 58 | } | |
| 59 | ||
| 60 | // 3.如果返回的是404 http状态码 | |
| 61 | Integer status = (Integer) defaultErrorAttributes.get("status"); | |
| 62 | if (status.equals(HttpStatus.HTTP_NOT_FOUND)) { | |
| 63 | return BeanUtil.beanToMap(new ErrorResponseData(PermissionExceptionEnum.URL_NOT_EXIST.getCode(), PermissionExceptionEnum.URL_NOT_EXIST.getMessage())); | |
| 64 | } | |
| 65 | ||
| 66 | // 4.无法确定的返回服务器异常 | |
| 67 | return BeanUtil.beanToMap(new ErrorResponseData(ServerExceptionEnum.SERVER_ERROR.getCode(), ServerExceptionEnum.SERVER_ERROR.getMessage())); | |
| 68 | } | |
| 69 | ||
| 70 | } | |
