Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
package io.fabric8.maven.docker.access;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import io.fabric8.maven.docker.model.Container;
import io.fabric8.maven.docker.util.EnvUtil;
/**
* Entity holding port mappings which can be set through the configuration.
*
* @author roland
* @since 04.04.14
*/
public class PortMapping {
// Pattern for splitting of the protocol
private static final Pattern PROTOCOL_SPLIT_PATTERN = Pattern.compile("(.*?)(?:/(tcp|udp))?$");
// Mapping between ports and the IP they should bind to
private final Map bindToHostMap = new HashMap<>();
// ports map (container port -> host port)
private final Map containerPortToHostPort = new HashMap<>();
// resolved dynamic properties
private final Properties dynamicProperties = new Properties();
// Mapping between property name and host ip (ip filled in after container creation)
private final Map hostIpVariableMap = new HashMap<>();
// Mapping between property name and host port (port filled in after container creation)
private final Map hostPortVariableMap = new HashMap<>();
// project properties
private final Properties projProperties;
// variables (container port spec -> host ip variable name)
private final Map specToHostIpVariableMap = new HashMap<>();
// variables (container port spec -> host port variable name)
private final Map specToHostPortVariableMap = new HashMap<>();
/**
* Create the mapping from a configuration. The configuration is list of port mapping specifications which has the
* format used by docker for port mapping (i.e. host_ip:host_port:container_port)
*
*
The "host_ip" part is optional. If not given, the all interfaces are used
*
If "host_port" is non numeric it is taken as a variable name. If this variable is given as value in
* variables, this number is used as host port. If no numeric value is given, it is considered to be filled with the
* real, dynamically created port value when {@link #updateProperties(Map)} is called
*
*
* @param portMappings a list of configuration strings where each string hast the format
* host_ip:host_port:container_port. If the host-port is non-numeric it is
* assumed to be a variable (which later might be filled in with the dynamically created port).
* @param projProperties project properties
* @throws IllegalArgumentException if the format doesn't fit
*/
public PortMapping(List portMappings, Properties projProperties) {
this.projProperties = projProperties;
for (String portMapping : portMappings) {
parsePortMapping(portMapping);
}
}
/**
* Check whether property needs updates for dynamically obtained host ports and ip adresses.
*
* @return true if any property are used which need to be filled in, false otherwise
*/
public boolean needsPropertiesUpdate() {
return !specToHostPortVariableMap.isEmpty() || !specToHostIpVariableMap.isEmpty();
}
/**
* @return Set of all mapped container ports
*/
public Set getContainerPorts() {
return containerPortToHostPort.keySet();
}
/**
* Update variable-to-port mappings with dynamically obtained ports and host ips.
* This should only be called once after this dynamically allocated parts has been be obtained.
*
* @param dockerObtainedDynamicBindings keys are the container ports, values are the dynamically mapped host ports and host ips.
*/
public void updateProperties(Map dockerObtainedDynamicBindings) {
for (Map.Entry entry : dockerObtainedDynamicBindings.entrySet()) {
String variable = entry.getKey();
Container.PortBinding portBinding = entry.getValue();
if (portBinding != null) {
update(hostPortVariableMap, specToHostPortVariableMap.get(variable), portBinding.getHostPort());
String hostIp = portBinding.getHostIp();
// Use the docker host if binding is on all interfaces
if ("0.0.0.0".equals(hostIp)) {
hostIp = projProperties.getProperty("docker.host.address");
}
update(hostIpVariableMap, specToHostIpVariableMap.get(variable), hostIp);
}
}
updateDynamicProperties(hostPortVariableMap);
updateDynamicProperties(hostIpVariableMap);
}
/**
* Create a JSON specification which can be used to in a Docker API request as the 'PortBindings' part
* for creating container.
*
* @return 'PortBindings' object or null if no port mappings are used.
*/
JsonObject toDockerPortBindingsJson() {
Map portMap = getContainerPortToHostPortMap();
if (!portMap.isEmpty()) {
JsonObject portBindings = new JsonObject();
Map bindToMap = getBindToHostMap();
for (Map.Entry entry : portMap.entrySet()) {
String containerPortSpec = entry.getKey();
Integer hostPort = entry.getValue();
JsonObject o = new JsonObject();
o.addProperty("HostPort", hostPort != null ? hostPort.toString() : "");
if (bindToMap.containsKey(containerPortSpec)) {
o.addProperty("HostIp", bindToMap.get(containerPortSpec));
}
JsonArray array = new JsonArray();
array.add(o);
portBindings.add(containerPortSpec, array);
}
return portBindings;
} else {
return null;
}
}
/**
* Return the content of the mapping as an array with all specifications as given
*
* @return port mappings as JSON array or null if no mappings exist
*/
public JsonArray toJson() {
Map portMap = getContainerPortToHostPortMap();
if (portMap.isEmpty()) {
return null;
}
JsonArray ret = new JsonArray();
Map bindToMap = getBindToHostMap();
for (Map.Entry entry : portMap.entrySet()) {
JsonObject mapping = new JsonObject();
String containerPortSpec = entry.getKey();
Matcher matcher = PROTOCOL_SPLIT_PATTERN.matcher(entry.getKey());
if (!matcher.matches()) {
throw new IllegalStateException("Internal error: " + entry.getKey() +
" doesn't contain protocol part and doesn't match "
+ PROTOCOL_SPLIT_PATTERN);
}
mapping.addProperty("containerPort", Integer.parseInt(matcher.group(1)));
if (matcher.group(2) != null) {
mapping.addProperty("protocol", matcher.group(2));
}
Integer hostPort = entry.getValue();
if (hostPort != null) {
mapping.addProperty("hostPort", hostPort);
}
if (bindToMap.containsKey(containerPortSpec)) {
mapping.addProperty("hostIP", bindToMap.get(containerPortSpec));
}
ret.add(mapping);
}
return ret;
}
// ==========================================================================================================
// visible for testing
Map getBindToHostMap() {
return bindToHostMap;
}
// visible for testing
Map getContainerPortToHostPortMap() {
return containerPortToHostPort;
}
// visible for testing
Map getHostIpVariableMap() {
return hostIpVariableMap;
}
// visible for testing
Map getHostPortVariableMap() {
return hostPortVariableMap;
}
// visible for testing
Map getPortsMap() {
return containerPortToHostPort;
}
private IllegalArgumentException createInvalidMappingError(String mapping, Exception exp) {
return new IllegalArgumentException("\nInvalid port mapping '" + mapping + "'\n" +
"Required format: '::(/tcp|udp)'\n" +
"See the reference manual for more details");
}
private void createMapping(String[] parts, String protocol) {
if (parts.length == 3) {
mapBindToAndHostPortSpec(parts[0], parts[1], createPortSpec(parts[2], protocol));
} else if (parts.length == 2) {
mapHostPortToSpec(parts[0], createPortSpec(parts[1], protocol));
} else {
mapHostPortToSpec(null, createPortSpec(parts[0], protocol));
}
}
private String createPortSpec(String port, String protocol) throws NumberFormatException {
return Integer.parseInt(port) + "/" + protocol;
}
private Integer getAsIntOrNull(String val) {
try {
return Integer.parseInt(val);
} catch (@SuppressWarnings("unused") NumberFormatException exp) {
return null;
}
}
// Check for a variable containing a port, return it as integer or null is not found or not a number
// First check system properties, then the variables given
private Integer getPortFromProjectOrSystemProperty(String var) {
String sysProp = System.getProperty(var);
if (sysProp != null) {
return getAsIntOrNull(sysProp);
}
if (projProperties.containsKey(var)) {
return getAsIntOrNull(projProperties.getProperty(var));
}
return null;
}
private String extractPortPropertyName(String name) {
String mavenPropName = EnvUtil.extractMavenPropertyName(name);
return mavenPropName != null ? mavenPropName : name;
}
private void mapBindToAndHostPortSpec(String bindTo, String hPort, String portSpec) {
mapHostPortToSpec(hPort, portSpec);
String hostPropName = extractHostPropertyName(bindTo);
if (hostPropName != null) {
String host = projProperties.getProperty(hostPropName);
if (host != null) {
// the container portSpec can never be null, so use that as the key
bindToHostMap.put(portSpec, resolveHostname(host));
}
specToHostIpVariableMap.put(portSpec, hostPropName);
} else {
// the container portSpec can never be null, so use that as the key
bindToHostMap.put(portSpec, resolveHostname(bindTo));
}
}
private String extractHostPropertyName(String name) {
if (name.startsWith("+")) {
return name.substring(1);
} else {
return EnvUtil.extractMavenPropertyName(name);
}
}
private void mapHostPortToSpec(String hPort, String portSpec) {
Integer hostPort;
if (hPort == null) {
hostPort = null;
} else {
try {
hostPort = Integer.parseInt(hPort);
} catch (@SuppressWarnings("unused") NumberFormatException exp) {
// Port should be dynamically assigned and set to the variable give in hPort
String portPropertyName = extractPortPropertyName(hPort);
hostPort = getPortFromProjectOrSystemProperty(portPropertyName);
if (hostPort != null) {
// portPropertyName: Prop name, hostPort: Port from a property value (prefilled)
hostPortVariableMap.put(portPropertyName, hostPort);
} else {
// portSpec: Port from container, portPropertyName: Variable name to be filled in later
specToHostPortVariableMap.put(portSpec, portPropertyName);
}
}
}
containerPortToHostPort.put(portSpec, hostPort);
}
private void parsePortMapping(String input) throws IllegalArgumentException {
try {
Matcher matcher = PROTOCOL_SPLIT_PATTERN.matcher(input);
// Matches always
matcher.matches();
String mapping = matcher.group(1);
String protocol = matcher.group(2);
if (protocol == null) {
protocol = "tcp";
}
createMapping(mapping.split(":", 3), protocol);
} catch (NullPointerException | NumberFormatException exp) {
throw createInvalidMappingError(input, exp);
}
}
private String resolveHostname(String bindToHost) {
try {
return InetAddress.getByName(bindToHost).getHostAddress();
} catch (@SuppressWarnings("unused") UnknownHostException e) {
throw new IllegalArgumentException("Host '" + bindToHost + "' to bind to cannot be resolved");
}
}
private void update(Map map, String key, T value) {
if (key != null) {
map.put(key, value);
}
}
private void updateDynamicProperties(Map dynamicPorts) {
for (Map.Entry entry : dynamicPorts.entrySet()) {
String var = entry.getKey();
String val = entry.getValue().toString();
projProperties.setProperty(var, val);
dynamicProperties.setProperty(var, val);
}
}
public static class PropertyWriteHelper {
private final Properties globalExport;
private final String globalFile;
private final Map toExport;
public PropertyWriteHelper(String globalFile) {
this.globalFile = globalFile;
this.toExport = new HashMap<>();
this.globalExport = new Properties();
}
public void add(PortMapping portMapping, String portPropertyFile) {
if (portPropertyFile != null) {
toExport.put(portPropertyFile, portMapping.dynamicProperties);
} else if (globalFile != null) {
globalExport.putAll(portMapping.dynamicProperties);
}
}
public void write() throws IOException {
for (Map.Entry entry : toExport.entrySet()) {
Properties props = entry.getValue();
writeProperties(props, entry.getKey());
globalExport.putAll(props);
}
if (globalFile != null && !globalExport.isEmpty()) {
writeProperties(globalExport, globalFile);
}
}
private void writeProperties(Properties props, String file) throws IOException {
File propFile = new File(file);
File parent = propFile.getParentFile();
if (!parent.exists()) {
parent.mkdirs();
}
try (OutputStream os = new FileOutputStream(propFile)) {
props.store(os, "Docker ports");
} catch (IOException e) {
throw new IOException("Cannot write properties to " + file + ": " + e, e);
}
}
}
}
