io.undertow.server.handlers.proxy.HostTable Maven / Gradle / Ivy
Go to download
This artifact provides a single jar that contains all classes required to use remote EJB and JMS, including
all dependencies. It is intended for use by those not using maven, maven users should just import the EJB and
JMS BOM's instead (shaded JAR's cause lots of problems with maven, as it is very easy to inadvertently end up
with different versions on classes on the class path).
/*
* 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.server.handlers.proxy;
import io.undertow.UndertowMessages;
import io.undertow.util.CopyOnWriteMap;
import io.undertow.util.PathMatcher;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArraySet;
/**
* Class that maintains a table of remote hosts that this proxy knows about.
*
* Basically this maps a virtual host + context path pair to a set of hosts.
*
* Note that this class does not have any knowledge of connection pooling
*
* @author Stuart Douglas
*/
public class HostTable {
private final Map> hosts = new CopyOnWriteMap<>();
private final Map>> targets = new CopyOnWriteMap<>();
public synchronized HostTable addHost(H host) {
if(hosts.containsKey(host)) {
throw UndertowMessages.MESSAGES.hostAlreadyRegistered(host);
}
hosts.put(host, new CopyOnWriteArraySet<>());
return this;
}
public synchronized HostTable removeHost(H host) {
Set targets = hosts.remove(host);
for(Target target : targets) {
removeRoute(host, target.virtualHost, target.contextPath);
}
return this;
}
public synchronized HostTable addRoute(H host, String virtualHost, String contextPath) {
Set hostData = hosts.get(host);
if(hostData == null) {
throw UndertowMessages.MESSAGES.hostHasNotBeenRegistered(host);
}
hostData.add(new Target(virtualHost, contextPath));
PathMatcher> paths = targets.get(virtualHost);
if(paths == null) {
paths = new PathMatcher<>();
targets.put(virtualHost, paths);
}
Set hostSet = paths.getPrefixPath(contextPath);
if(hostSet == null) {
hostSet = new CopyOnWriteArraySet<>();
paths.addPrefixPath(contextPath, hostSet);
}
hostSet.add(host);
return this;
}
public synchronized HostTable removeRoute(H host, String virtualHost, String contextPath) {
Set hostData = hosts.get(host);
if(hostData != null) {
hostData.remove(new Target(virtualHost, contextPath));
}
PathMatcher> paths = targets.get(virtualHost);
if(paths == null) {
return this;
}
Set hostSet = paths.getPrefixPath(contextPath);
if(hostSet == null) {
return this;
}
hostSet.remove(host);
if(hostSet.isEmpty()) {
paths.removePrefixPath(contextPath);
}
return this;
}
public Set getHostsForTarget(final String hostName, final String path) {
PathMatcher> matcher = targets.get(hostName);
if(matcher == null) {
return null;
}
return matcher.match(path).getValue();
}
private static final class Target {
final String virtualHost;
final String contextPath;
private Target(String virtualHost, String contextPath) {
this.virtualHost = virtualHost;
this.contextPath = contextPath;
}
@Override
public boolean equals(Object o) {
if (this == o) return true;
if (o == null || getClass() != o.getClass()) return false;
Target target = (Target) o;
if (contextPath != null ? !contextPath.equals(target.contextPath) : target.contextPath != null)
return false;
if (virtualHost != null ? !virtualHost.equals(target.virtualHost) : target.virtualHost != null)
return false;
return true;
}
@Override
public int hashCode() {
int result = virtualHost != null ? virtualHost.hashCode() : 0;
result = 31 * result + (contextPath != null ? contextPath.hashCode() : 0);
return result;
}
}
}