package com.mes.exception;
|
|
import com.mes.result.Result;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.http.HttpStatus;
|
import org.springframework.http.MediaType;
|
import org.springframework.web.HttpMediaTypeNotAcceptableException;
|
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 javax.servlet.http.HttpServletRequest;
|
|
/**
|
* @author zhoush
|
* @Date 2024/1/26 15:31
|
*/
|
@ControllerAdvice
|
@Slf4j
|
public class GlobalExceptionHandler {
|
/**
|
* 如果是serviceExcaption，则调用该方法
|
*/
|
@ExceptionHandler(ServiceException.class)
|
@ResponseBody
|
public Result handle(ServiceException se) {
|
return Result.error(se.getCode(), se.getMessage());
|
}
|
|
/**
|
* 处理媒体类型不匹配异常（通常发生在SSE连接已关闭时）
|
*/
|
@ExceptionHandler(HttpMediaTypeNotAcceptableException.class)
|
@ResponseStatus(HttpStatus.NOT_ACCEPTABLE)
|
@ResponseBody
|
public Result<Object> handleMediaTypeNotAcceptable(HttpMediaTypeNotAcceptableException e, HttpServletRequest request) {
|
// 如果是SSE请求，静默处理，不记录错误
|
String acceptHeader = request.getHeader("Accept");
|
if (acceptHeader != null && acceptHeader.contains("text/event-stream")) {
|
log.debug("SSE连接已关闭，忽略媒体类型异常: {}", e.getMessage());
|
return null; // 返回null，不写入响应
|
}
|
log.warn("媒体类型不匹配: {}", e.getMessage());
|
return Result.error();
|
}
|
|
@ExceptionHandler(Exception.class)
|
@ResponseBody
|
public Result<Object> error(Exception e, HttpServletRequest request) {
|
// 如果是SSE请求且是媒体类型异常，静默处理
|
if (e instanceof HttpMediaTypeNotAcceptableException) {
|
String acceptHeader = request.getHeader("Accept");
|
if (acceptHeader != null && acceptHeader.contains("text/event-stream")) {
|
log.debug("SSE连接已关闭，忽略异常: {}", e.getMessage());
|
return null;
|
}
|
}
|
log.error("全局异常处理: ", e);
|
return Result.error();
|
}
|
|
}
|
