| <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN" "http://www.w3.org/TR/html4/loose.dtd"> |
| <html lang="en"> |
| <head> |
| <title>Source code</title> |
| <link rel="stylesheet" type="text/css" href="../../../../../../stylesheet.css" title="Style"> |
| </head> |
| <body> |
| <div class="sourceContainer"> |
| <pre><span class="sourceLineNo">001</span><a name="line.1">/*</a> |
| <span class="sourceLineNo">002</span><a name="line.2"> * Licensed to the Apache Software Foundation (ASF) under one or more</a> |
| <span class="sourceLineNo">003</span><a name="line.3"> * contributor license agreements. See the NOTICE file distributed with</a> |
| <span class="sourceLineNo">004</span><a name="line.4"> * this work for additional information regarding copyright ownership.</a> |
| <span class="sourceLineNo">005</span><a name="line.5"> * The ASF licenses this file to You under the Apache license, Version 2.0</a> |
| <span class="sourceLineNo">006</span><a name="line.6"> * (the "License"); you may not use this file except in compliance with</a> |
| <span class="sourceLineNo">007</span><a name="line.7"> * the License. You may obtain a copy of the License at</a> |
| <span class="sourceLineNo">008</span><a name="line.8"> *</a> |
| <span class="sourceLineNo">009</span><a name="line.9"> * http://www.apache.org/licenses/LICENSE-2.0</a> |
| <span class="sourceLineNo">010</span><a name="line.10"> *</a> |
| <span class="sourceLineNo">011</span><a name="line.11"> * Unless required by applicable law or agreed to in writing, software</a> |
| <span class="sourceLineNo">012</span><a name="line.12"> * distributed under the License is distributed on an "AS IS" BASIS,</a> |
| <span class="sourceLineNo">013</span><a name="line.13"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</a> |
| <span class="sourceLineNo">014</span><a name="line.14"> * See the license for the specific language governing permissions and</a> |
| <span class="sourceLineNo">015</span><a name="line.15"> * limitations under the license.</a> |
| <span class="sourceLineNo">016</span><a name="line.16"> */</a> |
| <span class="sourceLineNo">017</span><a name="line.17">package org.apache.logging.log4j.docker;</a> |
| <span class="sourceLineNo">018</span><a name="line.18"></a> |
| <span class="sourceLineNo">019</span><a name="line.19">import java.io.IOException;</a> |
| <span class="sourceLineNo">020</span><a name="line.20">import java.net.URL;</a> |
| <span class="sourceLineNo">021</span><a name="line.21">import java.util.List;</a> |
| <span class="sourceLineNo">022</span><a name="line.22">import java.util.Map;</a> |
| <span class="sourceLineNo">023</span><a name="line.23"></a> |
| <span class="sourceLineNo">024</span><a name="line.24">import org.apache.logging.log4j.Logger;</a> |
| <span class="sourceLineNo">025</span><a name="line.25">import org.apache.logging.log4j.core.LogEvent;</a> |
| <span class="sourceLineNo">026</span><a name="line.26">import org.apache.logging.log4j.core.config.plugins.Plugin;</a> |
| <span class="sourceLineNo">027</span><a name="line.27">import org.apache.logging.log4j.core.lookup.AbstractLookup;</a> |
| <span class="sourceLineNo">028</span><a name="line.28">import org.apache.logging.log4j.core.lookup.StrLookup;</a> |
| <span class="sourceLineNo">029</span><a name="line.29">import org.apache.logging.log4j.core.util.NetUtils;</a> |
| <span class="sourceLineNo">030</span><a name="line.30">import org.apache.logging.log4j.docker.model.Container;</a> |
| <span class="sourceLineNo">031</span><a name="line.31">import org.apache.logging.log4j.docker.model.Network;</a> |
| <span class="sourceLineNo">032</span><a name="line.32">import org.apache.logging.log4j.status.StatusLogger;</a> |
| <span class="sourceLineNo">033</span><a name="line.33">import org.apache.logging.log4j.util.PropertiesUtil;</a> |
| <span class="sourceLineNo">034</span><a name="line.34">import com.fasterxml.jackson.core.type.TypeReference;</a> |
| <span class="sourceLineNo">035</span><a name="line.35">import com.fasterxml.jackson.databind.ObjectMapper;</a> |
| <span class="sourceLineNo">036</span><a name="line.36"></a> |
| <span class="sourceLineNo">037</span><a name="line.37">/**</a> |
| <span class="sourceLineNo">038</span><a name="line.38"> *</a> |
| <span class="sourceLineNo">039</span><a name="line.39"> */</a> |
| <span class="sourceLineNo">040</span><a name="line.40">@Plugin(name = "docker", category = StrLookup.CATEGORY)</a> |
| <span class="sourceLineNo">041</span><a name="line.41">public class DockerLookup extends AbstractLookup {</a> |
| <span class="sourceLineNo">042</span><a name="line.42"></a> |
| <span class="sourceLineNo">043</span><a name="line.43"> private static final Logger LOGGER = StatusLogger.getLogger();</a> |
| <span class="sourceLineNo">044</span><a name="line.44"> private static final String DOCKER_URI = "DOCKER_URI";</a> |
| <span class="sourceLineNo">045</span><a name="line.45"> private static final String HTTP = "http";</a> |
| <span class="sourceLineNo">046</span><a name="line.46"> private final Container container;</a> |
| <span class="sourceLineNo">047</span><a name="line.47"></a> |
| <span class="sourceLineNo">048</span><a name="line.48"> public DockerLookup() {</a> |
| <span class="sourceLineNo">049</span><a name="line.49"> String baseUri = System.getenv(DOCKER_URI);</a> |
| <span class="sourceLineNo">050</span><a name="line.50"> if (baseUri == null) {</a> |
| <span class="sourceLineNo">051</span><a name="line.51"> PropertiesUtil props = PropertiesUtil.getProperties();</a> |
| <span class="sourceLineNo">052</span><a name="line.52"> baseUri = props.getStringProperty(DOCKER_URI);</a> |
| <span class="sourceLineNo">053</span><a name="line.53"> }</a> |
| <span class="sourceLineNo">054</span><a name="line.54"> if (baseUri == null) {</a> |
| <span class="sourceLineNo">055</span><a name="line.55"> LOGGER.warn("No Docker URI provided. Docker information is unavailable");</a> |
| <span class="sourceLineNo">056</span><a name="line.56"> }</a> |
| <span class="sourceLineNo">057</span><a name="line.57"> Container current = null;</a> |
| <span class="sourceLineNo">058</span><a name="line.58"> try {</a> |
| <span class="sourceLineNo">059</span><a name="line.59"> URL url= new URL(baseUri + "/containers/json");</a> |
| <span class="sourceLineNo">060</span><a name="line.60"> String hostName = NetUtils.getLocalHostname();</a> |
| <span class="sourceLineNo">061</span><a name="line.61"> String macAddr = NetUtils.getMacAddressString();</a> |
| <span class="sourceLineNo">062</span><a name="line.62"></a> |
| <span class="sourceLineNo">063</span><a name="line.63"> if (url.getProtocol().equals(HTTP)) {</a> |
| <span class="sourceLineNo">064</span><a name="line.64"> ObjectMapper objectMapper = new ObjectMapper();</a> |
| <span class="sourceLineNo">065</span><a name="line.65"> List<Container> containerList = objectMapper.readValue(url, new TypeReference<List<Container>>(){});</a> |
| <span class="sourceLineNo">066</span><a name="line.66"></a> |
| <span class="sourceLineNo">067</span><a name="line.67"> for (Container container : containerList) {</a> |
| <span class="sourceLineNo">068</span><a name="line.68"> if (macAddr != null && container.getNetworkSettings() != null) {</a> |
| <span class="sourceLineNo">069</span><a name="line.69"> Map<String, Network> networks = container.getNetworkSettings().getNetworks();</a> |
| <span class="sourceLineNo">070</span><a name="line.70"> if (networks != null) {</a> |
| <span class="sourceLineNo">071</span><a name="line.71"> for (Network network: networks.values()) {</a> |
| <span class="sourceLineNo">072</span><a name="line.72"> if (macAddr.equals(network.getMacAddress())) {</a> |
| <span class="sourceLineNo">073</span><a name="line.73"> current = container;</a> |
| <span class="sourceLineNo">074</span><a name="line.74"> break;</a> |
| <span class="sourceLineNo">075</span><a name="line.75"> }</a> |
| <span class="sourceLineNo">076</span><a name="line.76"> }</a> |
| <span class="sourceLineNo">077</span><a name="line.77"> }</a> |
| <span class="sourceLineNo">078</span><a name="line.78"> }</a> |
| <span class="sourceLineNo">079</span><a name="line.79"> if (current != null) {</a> |
| <span class="sourceLineNo">080</span><a name="line.80"> break;</a> |
| <span class="sourceLineNo">081</span><a name="line.81"> }</a> |
| <span class="sourceLineNo">082</span><a name="line.82"> }</a> |
| <span class="sourceLineNo">083</span><a name="line.83"> }</a> |
| <span class="sourceLineNo">084</span><a name="line.84"> if (current == null) {</a> |
| <span class="sourceLineNo">085</span><a name="line.85"> LOGGER.warn("Unable to determine current container");</a> |
| <span class="sourceLineNo">086</span><a name="line.86"> }</a> |
| <span class="sourceLineNo">087</span><a name="line.87"> } catch (IOException ioe) {</a> |
| <span class="sourceLineNo">088</span><a name="line.88"> LOGGER.warn("Unable to read container information: " + ioe.getMessage());</a> |
| <span class="sourceLineNo">089</span><a name="line.89"> }</a> |
| <span class="sourceLineNo">090</span><a name="line.90"> container = current;</a> |
| <span class="sourceLineNo">091</span><a name="line.91"> }</a> |
| <span class="sourceLineNo">092</span><a name="line.92"></a> |
| <span class="sourceLineNo">093</span><a name="line.93"> @Override</a> |
| <span class="sourceLineNo">094</span><a name="line.94"> public String lookup(LogEvent event, String key) {</a> |
| <span class="sourceLineNo">095</span><a name="line.95"> if (container == null) {</a> |
| <span class="sourceLineNo">096</span><a name="line.96"> return null;</a> |
| <span class="sourceLineNo">097</span><a name="line.97"> }</a> |
| <span class="sourceLineNo">098</span><a name="line.98"> switch (key) {</a> |
| <span class="sourceLineNo">099</span><a name="line.99"> case "shortContainerId": {</a> |
| <span class="sourceLineNo">100</span><a name="line.100"> return container.getId().substring(0, 12);</a> |
| <span class="sourceLineNo">101</span><a name="line.101"> }</a> |
| <span class="sourceLineNo">102</span><a name="line.102"> case "containerId": {</a> |
| <span class="sourceLineNo">103</span><a name="line.103"> return container.getId();</a> |
| <span class="sourceLineNo">104</span><a name="line.104"> }</a> |
| <span class="sourceLineNo">105</span><a name="line.105"> case "containerName": {</a> |
| <span class="sourceLineNo">106</span><a name="line.106"> if (container.getNames().size() > 1) {</a> |
| <span class="sourceLineNo">107</span><a name="line.107"> return container.getNames().toString();</a> |
| <span class="sourceLineNo">108</span><a name="line.108"> }</a> |
| <span class="sourceLineNo">109</span><a name="line.109"> return container.getNames().get(0);</a> |
| <span class="sourceLineNo">110</span><a name="line.110"> }</a> |
| <span class="sourceLineNo">111</span><a name="line.111"> case "shortImageId": {</a> |
| <span class="sourceLineNo">112</span><a name="line.112"> return container.getImageId().substring(0, 12);</a> |
| <span class="sourceLineNo">113</span><a name="line.113"> }</a> |
| <span class="sourceLineNo">114</span><a name="line.114"> case "imageId": {</a> |
| <span class="sourceLineNo">115</span><a name="line.115"> return container.getImageId();</a> |
| <span class="sourceLineNo">116</span><a name="line.116"> }</a> |
| <span class="sourceLineNo">117</span><a name="line.117"> case "imageName": {</a> |
| <span class="sourceLineNo">118</span><a name="line.118"> return container.getImage();</a> |
| <span class="sourceLineNo">119</span><a name="line.119"> }</a> |
| <span class="sourceLineNo">120</span><a name="line.120"> default:</a> |
| <span class="sourceLineNo">121</span><a name="line.121"> return null;</a> |
| <span class="sourceLineNo">122</span><a name="line.122"> }</a> |
| <span class="sourceLineNo">123</span><a name="line.123"> }</a> |
| <span class="sourceLineNo">124</span><a name="line.124">}</a> |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| </pre> |
| </div> |
| </body> |
| </html> |
