[HSEARCH] hsearch-elasticsearch-wikipedia: format REST endpoint errors as JSON

Author: yrodiere

hibernate-search/hsearch-elasticsearch-wikipedia/src/main/java/org/hibernate/search/demos/wikipedia/endpoint/config/JerseyConfig.java

@@ -3,12 +3,16 @@
 import org.glassfish.jersey.server.ResourceConfig;
 import org.hibernate.search.demos.wikipedia.endpoint.AdminEndpoint;
 import org.hibernate.search.demos.wikipedia.endpoint.PageEndpoint;
+import org.hibernate.search.demos.wikipedia.endpoint.error.ToJsonExceptionMapper;
+
 import org.springframework.stereotype.Component;
 
 @Component
 public class JerseyConfig extends ResourceConfig {
 
 	public JerseyConfig() {
+		register( ToJsonExceptionMapper.class );
+
 		registerEndpoints();
 	}
 

hibernate-search/hsearch-elasticsearch-wikipedia/src/main/java/org/hibernate/search/demos/wikipedia/endpoint/error/ErrorEntity.java

@@ -0,0 +1,39 @@
+/*
+ * Hibernate Search, full-text search for your domain model
+ *
+ * License: GNU Lesser General Public License (LGPL), version 2.1 or later
+ * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
+ */
+package org.hibernate.search.demos.wikipedia.endpoint.error;
+
+import javax.ws.rs.core.Response;
+
+public class ErrorEntity {
+	private Response.Status status;
+	private String message;
+	private String stackTrace;
+
+	public Response.Status getStatus() {
+		return status;
+	}
+
+	public void setStatus(Response.Status status) {
+		this.status = status;
+	}
+
+	public String getMessage() {
+		return message;
+	}
+
+	public void setMessage(String message) {
+		this.message = message;
+	}
+
+	public String getStackTrace() {
+		return stackTrace;
+	}
+
+	public void setStackTrace(String stackTrace) {
+		this.stackTrace = stackTrace;
+	}
+}

hibernate-search/hsearch-elasticsearch-wikipedia/src/main/java/org/hibernate/search/demos/wikipedia/endpoint/error/ToJsonExceptionMapper.java

@@ -0,0 +1,41 @@
+/*
+ * Hibernate Search, full-text search for your domain model
+ *
+ * License: GNU Lesser General Public License (LGPL), version 2.1 or later
+ * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
+ */
+package org.hibernate.search.demos.wikipedia.endpoint.error;
+
+import java.io.PrintWriter;
+import java.io.StringWriter;
+import javax.ws.rs.WebApplicationException;
+import javax.ws.rs.core.MediaType;
+import javax.ws.rs.core.Response;
+import javax.ws.rs.ext.ExceptionMapper;
+
+public class ToJsonExceptionMapper implements ExceptionMapper<Throwable> {
+	@Override
+	public Response toResponse(Throwable ex) {
+		ErrorEntity errorEntity = new ErrorEntity();
+		Response.Status status;
+		if ( ex instanceof WebApplicationException ) {
+			Response originalResponse = ( (WebApplicationException) ex ).getResponse();
+			status = Response.Status.fromStatusCode( originalResponse.getStatus() );
+		}
+		else {
+			status = Response.Status.INTERNAL_SERVER_ERROR;
+		}
+
+		errorEntity.setStatus( status );
+		errorEntity.setMessage( ex.getMessage() );
+
+		StringWriter errorStackTrace = new StringWriter();
+		ex.printStackTrace( new PrintWriter( errorStackTrace ) );
+		errorEntity.setStackTrace( errorStackTrace.toString() );
+
+		return Response.status( status )
+				.entity( errorEntity )
+				.type( MediaType.APPLICATION_JSON )
+				.build();
+	}
+}