one.valuelogic.vertx.web.problem.ProblemHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertx-web-problem Show documentation

vertx-web integration with zalando problem

There is a newer version: 0.3.4

package one.valuelogic.vertx.web.problem;

import io.vertx.core.Handler;
import io.vertx.core.json.EncodeException;
import io.vertx.core.json.Json;
import io.vertx.core.logging.Logger;
import io.vertx.core.logging.LoggerFactory;
import io.vertx.rxjava.core.http.HttpServerResponse;
import io.vertx.rxjava.ext.web.RoutingContext;
import org.zalando.problem.Problem;

import javax.ws.rs.core.Response;

import static javax.ws.rs.core.Response.Status.OK;

public class ProblemHandler implements Handler {

    private static final Logger LOG = LoggerFactory.getLogger(ProblemHandler.class);
    private static final String INTERNAL_SERVER_ERROR = "{" +
        "\"title\":\"" +
        Response.Status.INTERNAL_SERVER_ERROR.getReasonPhrase() +
        "\"," +
        "\"status\":\"" +
        Response.Status.INTERNAL_SERVER_ERROR.getStatusCode() +
        "\"" +
        "}";

    @Override
    public void handle(RoutingContext context) {
        Problem problem = ProblemFactory.create(context.failure(), context.request().path());
        write(context.response(), problem);
    }

    public static void write(HttpServerResponse response, Problem problem) {
        int statusCode = problem.getStatus() != null ? problem.getStatus().getStatusCode() : OK.getStatusCode();

        response.setChunked(true).putHeader("Content-Type", "application/problem+json");

        try {
            response
                .setStatusCode(statusCode)
                .write(Json.encode(problem));
        } catch (EncodeException e) {
            LOG.error("Error while writing problem to JSON", e);
            response
                .setStatusCode(Response.Status.INTERNAL_SERVER_ERROR.getStatusCode())
                .write(INTERNAL_SERVER_ERROR);
        } finally {
            response.end();
        }
    }

    public static Handler create() {
        return new ProblemHandler();
    }
}