ctripcorp/zeus
1
1
Fork 0
mirror of https://github.com/ctripcorp/zeus.git synced 2024-11-11 09:46:31 +08:00
Code Issues Projects Releases Packages Wiki Activity

hide stacktrace

Browse source
This commit is contained in:
Mengyi Zhou 2015-11-09 15:23:15 +08:00
parent 724ed397a7
commit 07aa454cb5
3 changed files with 23 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/main/java/com/ctrip/zeus/restful/message/impl/ErrorResponseHandler.java
View file

@ -18,8 +18,8 @@ import javax.ws.rs.core.Response;
public class ErrorResponseHandler implements ResponseHandler {
private static final MediaType defaultMediaType = MediaType.APPLICATION_JSON_TYPE;
public Message generateMessage(Throwable object, String type) throws Exception {
ErrorMessage em = ExceptionUtils.getErrorMessage(object);
public Message generateMessage(Throwable object, String type, boolean printStackTrace) throws Exception {
ErrorMessage em = ExceptionUtils.getErrorMessage(object, printStackTrace);
ErrorResponse err = new ErrorResponse();
if (type.equals(MediaType.APPLICATION_XML)) {
@ -45,14 +45,17 @@ public class ErrorResponseHandler implements ResponseHandler {
@Override
public Response handle(Object object, MediaType mediaType) throws Exception {
return handle(object, mediaType, false);
}
public Response handle(Object object, MediaType mediaType, boolean printStackTrace) throws Exception {
if (object == null || !(object instanceof Throwable)) {
throw new ValidationException("ErrorResponseHandler only accepts Throwable object");
}
if (mediaType == null || !MediaType.APPLICATION_XML_TYPE.equals(mediaType)) {
mediaType = defaultMediaType;
}
Message message = generateMessage((Throwable) object, mediaType.toString());
Message message = generateMessage((Throwable) object, mediaType.toString(), printStackTrace);
return Response.status(message.getStatus()).entity(message.getResponse())
.type(mediaType).build();
}

8
src/main/java/com/ctrip/zeus/service/aop/ExceptionAspect.java
View file

@ -43,17 +43,23 @@ public class ExceptionAspect implements Ordered{
logger.error(builder.toString());
MediaType mediaType = null;
boolean printStackTrace = false;
for (Object arg : point.getArgs()) {
if (arg instanceof ContainerRequest) {
ContainerRequest cr = (ContainerRequest) arg;
mediaType = cr.getMediaType();
try {
printStackTrace = Boolean.parseBoolean(cr.getHeaderString("slb-stack-trace"));
} catch (Exception ex) {
printStackTrace = false;
}
break;
}
}
if (mediaType == null) {
logger.warn("Request media type cannot be found - use json by default.");
}
return errorResponseHandler.handle(cause, mediaType);
return errorResponseHandler.handle(cause, mediaType, printStackTrace);
} catch (Exception e) {
logger.error("Error response handler doesn't work.");
return null;

4
src/main/java/com/ctrip/zeus/util/ExceptionUtils.java
View file

@ -22,9 +22,9 @@ public class ExceptionUtils {
return builder.toString();
}
public static ErrorMessage getErrorMessage(Throwable throwable) {
public static ErrorMessage getErrorMessage(Throwable throwable, boolean printStackTrace) {
return new ErrorMessage().setCode(ExceptionUtils.getErrorCode(throwable))
.setMessage(ExceptionUtils.getMessage(throwable))
.setStackTrace(ExceptionUtils.getStackTrace(throwable));
.setStackTrace(printStackTrace ? ExceptionUtils.getStackTrace(throwable) : null);
}
}