io.undertow.server.handlers.proxy.HostTable 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.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;
}
}
}