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

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

import java.util.ArrayDeque;
import java.util.List;

import io.quarkus.vertx.http.runtime.filters.OriginalRequestContext;
import io.vertx.ext.web.RoutingContext;

/**
 * Query parameter
 */
public class QueryParameterAttribute implements ExchangeAttribute {

    private final String parameter;
    private final boolean useOriginalRequest;

    public QueryParameterAttribute(String parameter, boolean useOriginalRequest) {
        this.parameter = parameter;
        this.useOriginalRequest = useOriginalRequest;
    }

    @Override
    public String readAttribute(final RoutingContext exchange) {
        List res = useOriginalRequest ? OriginalRequestContext.getAllQueryParams(exchange, parameter)
                : exchange.queryParams().getAll(parameter);
        if (res == null) {
            return null;
        } else if (res.isEmpty()) {
            return "";
        } else if (res.size() == 1) {
            return res.get(0);
        } else {
            StringBuilder sb = new StringBuilder("[");
            int i = 0;
            for (String s : res) {
                sb.append(s);
                if (++i != res.size()) {
                    sb.append(", ");
                }
            }
            sb.append("]");
            return sb.toString();
        }
    }

    @Override
    public void writeAttribute(final RoutingContext exchange, final String newValue) throws ReadOnlyAttributeException {
        final ArrayDeque value = new ArrayDeque<>();
        value.add(newValue);
        exchange.queryParams().set(parameter, value);
    }

    public static final class Builder implements ExchangeAttributeBuilder {

        @Override
        public String name() {
            return "Query Parameter";
        }

        @Override
        public ExchangeAttribute build(final String token) {
            if (token.startsWith("%{q,") && token.endsWith("}")) {
                final String qp = token.substring(4, token.length() - 1);
                return new QueryParameterAttribute(qp, false);
            }
            if (token.startsWith("%{