com.github.markusbernhardt.proxy.selector.misc.ProtocolDispatchSelector Maven / Gradle / Ivy
package com.github.markusbernhardt.proxy.selector.misc;
import java.io.IOException;
import java.net.Proxy;
import java.net.ProxySelector;
import java.net.SocketAddress;
import java.net.URI;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import com.github.markusbernhardt.proxy.selector.direct.NoProxySelector;
/*****************************************************************************
* This is a facade for a list of ProxySelecor objects. You can register
* different ProxySelectors per Protocol.
*
* @author Markus Bernhardt, Copyright 2016
* @author Bernd Rosstauscher, Copyright 2009
****************************************************************************/
public class ProtocolDispatchSelector extends ProxySelector {
private Map selectors;
private ProxySelector fallbackSelector;
/*************************************************************************
* Constructor
************************************************************************/
public ProtocolDispatchSelector() {
super();
this.selectors = new ConcurrentHashMap();
this.fallbackSelector = NoProxySelector.getInstance();
}
/*************************************************************************
* Sets a selector responsible for the given protocol.
*
* @param protocol
* the name of the protocol.
* @param selector
* the selector to use.
************************************************************************/
public void setSelector(String protocol, ProxySelector selector) {
if (protocol == null) {
throw new NullPointerException("Protocol must not be null.");
}
if (selector == null) {
throw new NullPointerException("Selector must not be null.");
}
this.selectors.put(protocol, selector);
}
/*************************************************************************
* Removes the selector installed for the given protocol.
*
* @param protocol
* the protocol name.
* @return the old selector that is removed.
************************************************************************/
public ProxySelector removeSelector(String protocol) {
return this.selectors.remove(protocol);
}
/*************************************************************************
* Gets the selector installed for the given protocol.
*
* @param protocol
* the protocol name.
* @return the selector for that protocol, null if none is currently set.
************************************************************************/
public ProxySelector getSelector(String protocol) {
return this.selectors.get(protocol);
}
/*************************************************************************
* Sets the fallback selector that is always called when no matching
* protocol selector was found..
*
* @param selector
* the selector to use.
************************************************************************/
public void setFallbackSelector(ProxySelector selector) {
if (selector == null) {
throw new NullPointerException("Selector must not be null.");
}
this.fallbackSelector = selector;
}
/*************************************************************************
* connectFailed
*
* @see java.net.ProxySelector#connectFailed(java.net.URI,
* java.net.SocketAddress, java.io.IOException)
************************************************************************/
@Override
public void connectFailed(URI uri, SocketAddress sa, IOException ioe) {
ProxySelector selector = this.fallbackSelector;
String protocol = uri.getScheme();
if (protocol != null && this.selectors.get(protocol) != null) {
selector = this.selectors.get(protocol);
}
selector.connectFailed(uri, sa, ioe);
}
/*************************************************************************
* select
*
* @see java.net.ProxySelector#select(java.net.URI)
************************************************************************/
@Override
public List select(URI uri) {
ProxySelector selector = this.fallbackSelector;
String protocol = uri.getScheme();
if (protocol != null && this.selectors.get(protocol) != null) {
selector = this.selectors.get(protocol);
}
return selector.select(uri);
}
/*************************************************************************
* Gets the size of the selector map.
*
* @return the size of the selector map.
************************************************************************/
public int size() {
return this.selectors.size();
}
@Override
public String toString() {
return "ProtocolDispatchSelector{" +
"selectors=" + selectors +
", fallbackSelector=" + fallbackSelector +
'}';
}
}