io.quarkus.vertx.http.runtime.attribute.RequestLineAttribute Maven / Gradle / Ivy

Go to download
package io.quarkus.vertx.http.runtime.attribute;

import io.vertx.ext.web.RoutingContext;

/**
 * The request line
 *
 */
public class RequestLineAttribute implements ExchangeAttribute {

    public static final String REQUEST_LINE_SHORT = "%r";
    public static final String REQUEST_LINE = "%{REQUEST_LINE}";

    public static final ExchangeAttribute INSTANCE = new RequestLineAttribute();

    private RequestLineAttribute() {

    }

    @Override
    public String readAttribute(final RoutingContext exchange) {
        StringBuilder sb = new StringBuilder()
                .append(exchange.request().method())
                .append(' ')
                .append(exchange.request().uri());
        sb.append(' ');
        String httpVersion = "-";
        switch (exchange.request().version()) {
            case HTTP_1_0:
                httpVersion = "HTTP/1.0";
                break;
            case HTTP_1_1:
                httpVersion = "HTTP/1.1";
                break;
            case HTTP_2:
                httpVersion = "HTTP/2";
                break;
            default:
                // best effort to try and infer the HTTP version from
                // any "unknown" enum value
                httpVersion = exchange.request().version().name()
                        .replace("HTTP_", "HTTP/")
                        .replace("_", ".");
                break;
        }
        sb.append(httpVersion);
        return sb.toString();
    }

    @Override
    public void writeAttribute(final RoutingContext exchange, final String newValue) throws ReadOnlyAttributeException {
        throw new ReadOnlyAttributeException("Request line", newValue);
    }

    public static final class Builder implements ExchangeAttributeBuilder {

        @Override
        public String name() {
            return "Request line";
        }

        @Override
        public ExchangeAttribute build(final String token) {
            if (token.equals(REQUEST_LINE) || token.equals(REQUEST_LINE_SHORT)) {
                return RequestLineAttribute.INSTANCE;
            }
            return null;
        }

        @Override
        public int priority() {
            return 0;
        }
    }
}