io.undertow.predicate.PathTemplatePredicate Maven / Gradle / Ivy
/*
* JBoss, Home of Professional Open Source.
* Copyright 2014 Red Hat, Inc., and individual contributors
* as indicated by the @author tags.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.undertow.predicate;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;
import io.undertow.attribute.ExchangeAttribute;
import io.undertow.attribute.ExchangeAttributes;
import io.undertow.server.HttpServerExchange;
import io.undertow.util.PathTemplate;
import io.undertow.UndertowLogger;
/**
* @author Stuart Douglas
*/
public class PathTemplatePredicate implements Predicate {
private final ExchangeAttribute attribute;
private final String template;
private final PathTemplate value;
private static final boolean traceEnabled;
static {
traceEnabled = UndertowLogger.PREDICATE_LOGGER.isTraceEnabled();
}
public PathTemplatePredicate(final String template, final ExchangeAttribute attribute) {
this.attribute = attribute;
this.template = template;
this.value = PathTemplate.create(template);
}
@Override
public boolean resolve(final HttpServerExchange exchange) {
final Map params = new HashMap<>();
String path = attribute.readAttribute(exchange);
if(path == null) {
return false;
}
boolean result = this.value.matches(path, params);
if (traceEnabled) {
UndertowLogger.PREDICATE_LOGGER.tracef("Path template [%s] %s input [%s] for %s.", template, (result ? "MATCHES" : "DOES NOT MATCH" ), path, exchange);
}
if (result) {
Map context = exchange.getAttachment(PREDICATE_CONTEXT);
if(context == null) {
exchange.putAttachment(PREDICATE_CONTEXT, context = new TreeMap<>());
}
if (traceEnabled ) {
params.entrySet().forEach( param -> UndertowLogger.PREDICATE_LOGGER.tracef("Storing template match [%s=%s] for %s.", param.getKey(), param.getValue(), exchange) );
}
context.putAll(params);
}
return result;
}
public String toString() {
if( attribute == ExchangeAttributes.relativePath() ) {
return "path-template( '" + template + "' )";
} else {
return "path-template( value='" + template + "', match='" + attribute.toString() + "' )";
}
}
public static class Builder implements PredicateBuilder {
@Override
public String name() {
return "path-template";
}
@Override
public Map> parameters() {
final Map> params = new HashMap<>();
params.put("value", String.class);
params.put("match", ExchangeAttribute.class);
return params;
}
@Override
public Set requiredParameters() {
final Set params = new HashSet<>();
params.add("value");
return params;
}
@Override
public String defaultParameter() {
return "value";
}
@Override
public Predicate build(final Map config) {
ExchangeAttribute match = (ExchangeAttribute) config.get("match");
if (match == null) {
match = ExchangeAttributes.relativePath();
}
String value = (String) config.get("value");
return new PathTemplatePredicate(value, match);
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy