package com.zy.common.config;
|
|
import com.core.common.R;
|
import com.core.exception.CoolException;
|
import org.springframework.web.HttpRequestMethodNotSupportedException;
|
import org.springframework.web.bind.annotation.ExceptionHandler;
|
import org.springframework.web.bind.annotation.RestControllerAdvice;
|
import org.springframework.web.method.HandlerMethod;
|
import javax.servlet.http.HttpServletRequest;
|
import java.util.HashMap;
|
import java.util.Map;
|
|
/**
|
* Created by vincent on 2019-06-09
|
*/
|
@RestControllerAdvice
|
public class CoolExceptionHandler {
|
|
@ExceptionHandler(Exception.class)
|
public Object handlerException(HandlerMethod handler, Exception e, HttpServletRequest request) {
|
e.printStackTrace();
|
if (useAlt(request)) {
|
return altError("500", "服务器内部错误");
|
}
|
return R.error();
|
}
|
|
@ExceptionHandler(HttpRequestMethodNotSupportedException.class)
|
public Object handleHttpRequestMethodNotSupportedException(HttpRequestMethodNotSupportedException e,
|
HttpServletRequest request) {
|
if (useAlt(request)) {
|
return altError("405", "HTTP方法不支持");
|
}
|
return R.error();
|
}
|
|
@ExceptionHandler(CoolException.class)
|
public Object handleRRException(CoolException e, HttpServletRequest request) {
|
if (useAlt(request)) {
|
String[] split = e.getMessage().split("-");
|
String code = null;
|
String msg = e.getMessage();
|
if (split.length == 2 && String.valueOf(split[0]).length() < 3) {
|
code = split[0];
|
msg = split[1];
|
}
|
return altError(code, msg);
|
}
|
String[] split = e.getMessage().split("-");
|
if (split.length == 2) {
|
if (String.valueOf(split[0]).length() < 3) {
|
return R.parse(e.getMessage());
|
}
|
}
|
return R.error(e.getMessage());
|
}
|
|
private boolean useAlt(HttpServletRequest request) {
|
if (null == request) {
|
return false;
|
}
|
String header = request.getHeader("X-Response-Format");
|
if ("alt".equalsIgnoreCase(header)) {
|
return true;
|
}
|
String param = request.getParameter("responseFormat");
|
return "alt".equalsIgnoreCase(param);
|
}
|
|
private Map<String, Object> altError(String code, String message) {
|
Map<String, Object> map = new HashMap<>();
|
map.put("success", false);
|
if (null != code) {
|
map.put("code", code);
|
} else {
|
map.put("code", "200");
|
}
|
map.put("message", "失败");
|
map.put("returnMessage", message);
|
map.put("timestamp", System.currentTimeMillis());
|
return map;
|
}
|
}
|
