blob: 2d9cb4d8166929d7043e73e65efb04aac8795ce6 [file] [log] [blame]
<!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.spring.cloud.config.client;</a>
<span class="sourceLineNo">018</span><a name="line.18"></a>
<span class="sourceLineNo">019</span><a name="line.19">import org.apache.logging.log4j.Logger;</a>
<span class="sourceLineNo">020</span><a name="line.20">import org.apache.logging.log4j.core.LogEvent;</a>
<span class="sourceLineNo">021</span><a name="line.21">import org.apache.logging.log4j.core.config.plugins.Plugin;</a>
<span class="sourceLineNo">022</span><a name="line.22">import org.apache.logging.log4j.core.lookup.StrLookup;</a>
<span class="sourceLineNo">023</span><a name="line.23">import org.apache.logging.log4j.status.StatusLogger;</a>
<span class="sourceLineNo">024</span><a name="line.24">import org.springframework.core.env.Environment;</a>
<span class="sourceLineNo">025</span><a name="line.25"></a>
<span class="sourceLineNo">026</span><a name="line.26">import java.util.regex.Matcher;</a>
<span class="sourceLineNo">027</span><a name="line.27">import java.util.regex.Pattern;</a>
<span class="sourceLineNo">028</span><a name="line.28"></a>
<span class="sourceLineNo">029</span><a name="line.29">/**</a>
<span class="sourceLineNo">030</span><a name="line.30"> * Lookup for Spring properties.</a>
<span class="sourceLineNo">031</span><a name="line.31"> */</a>
<span class="sourceLineNo">032</span><a name="line.32">@Plugin(name = "spring", category = StrLookup.CATEGORY)</a>
<span class="sourceLineNo">033</span><a name="line.33">public class SpringLookup extends SpringEnvironmentHolder implements StrLookup {</a>
<span class="sourceLineNo">034</span><a name="line.34"></a>
<span class="sourceLineNo">035</span><a name="line.35"> private static final Logger LOGGER = StatusLogger.getLogger();</a>
<span class="sourceLineNo">036</span><a name="line.36"> private static final String ACTIVE = "profiles.active";</a>
<span class="sourceLineNo">037</span><a name="line.37"> private static final String DEFAULT = "profiles.default";</a>
<span class="sourceLineNo">038</span><a name="line.38"> private static final String PATTERN = "\\[(\\d+?)\\]";</a>
<span class="sourceLineNo">039</span><a name="line.39"> private static final Pattern ACTIVE_PATTERN = Pattern.compile(ACTIVE + PATTERN);</a>
<span class="sourceLineNo">040</span><a name="line.40"> private static final Pattern DEFAULT_PATTERN = Pattern.compile(DEFAULT + PATTERN);</a>
<span class="sourceLineNo">041</span><a name="line.41"></a>
<span class="sourceLineNo">042</span><a name="line.42"> public SpringLookup() {</a>
<span class="sourceLineNo">043</span><a name="line.43"> getEnvironment();</a>
<span class="sourceLineNo">044</span><a name="line.44"> }</a>
<span class="sourceLineNo">045</span><a name="line.45"></a>
<span class="sourceLineNo">046</span><a name="line.46"> @Override</a>
<span class="sourceLineNo">047</span><a name="line.47"> public String lookup(String key) {</a>
<span class="sourceLineNo">048</span><a name="line.48"> Environment env = getEnvironment();</a>
<span class="sourceLineNo">049</span><a name="line.49"> if (env != null) {</a>
<span class="sourceLineNo">050</span><a name="line.50"> String lowerKey = key.toLowerCase();</a>
<span class="sourceLineNo">051</span><a name="line.51"> if (lowerKey.startsWith(ACTIVE)) {</a>
<span class="sourceLineNo">052</span><a name="line.52"> switch (env.getActiveProfiles().length) {</a>
<span class="sourceLineNo">053</span><a name="line.53"> case 0: {</a>
<span class="sourceLineNo">054</span><a name="line.54"> return null;</a>
<span class="sourceLineNo">055</span><a name="line.55"> }</a>
<span class="sourceLineNo">056</span><a name="line.56"> case 1: {</a>
<span class="sourceLineNo">057</span><a name="line.57"> return env.getActiveProfiles()[0];</a>
<span class="sourceLineNo">058</span><a name="line.58"> }</a>
<span class="sourceLineNo">059</span><a name="line.59"> default: {</a>
<span class="sourceLineNo">060</span><a name="line.60"> Matcher matcher = ACTIVE_PATTERN.matcher(key);</a>
<span class="sourceLineNo">061</span><a name="line.61"> if (matcher.matches()) {</a>
<span class="sourceLineNo">062</span><a name="line.62"> try {</a>
<span class="sourceLineNo">063</span><a name="line.63"> int index = Integer.parseInt(matcher.group(1));</a>
<span class="sourceLineNo">064</span><a name="line.64"> if (index &lt; env.getActiveProfiles().length) {</a>
<span class="sourceLineNo">065</span><a name="line.65"> return env.getActiveProfiles()[index];</a>
<span class="sourceLineNo">066</span><a name="line.66"> } else {</a>
<span class="sourceLineNo">067</span><a name="line.67"> LOGGER.warn("Index out of bounds for Spring active profiles: {}", index);</a>
<span class="sourceLineNo">068</span><a name="line.68"> return null;</a>
<span class="sourceLineNo">069</span><a name="line.69"> }</a>
<span class="sourceLineNo">070</span><a name="line.70"> } catch (Exception ex) {</a>
<span class="sourceLineNo">071</span><a name="line.71"> LOGGER.warn("Unable to parse {} as integer value", matcher.group(1));</a>
<span class="sourceLineNo">072</span><a name="line.72"> return null;</a>
<span class="sourceLineNo">073</span><a name="line.73"> }</a>
<span class="sourceLineNo">074</span><a name="line.74"></a>
<span class="sourceLineNo">075</span><a name="line.75"> } else {</a>
<span class="sourceLineNo">076</span><a name="line.76"> StringBuilder sb = new StringBuilder();</a>
<span class="sourceLineNo">077</span><a name="line.77"> for (String profile : env.getActiveProfiles()) {</a>
<span class="sourceLineNo">078</span><a name="line.78"> if (sb.length() &gt; 0) {</a>
<span class="sourceLineNo">079</span><a name="line.79"> sb.append(",");</a>
<span class="sourceLineNo">080</span><a name="line.80"> }</a>
<span class="sourceLineNo">081</span><a name="line.81"> sb.append(profile);</a>
<span class="sourceLineNo">082</span><a name="line.82"> }</a>
<span class="sourceLineNo">083</span><a name="line.83"> return sb.toString();</a>
<span class="sourceLineNo">084</span><a name="line.84"> }</a>
<span class="sourceLineNo">085</span><a name="line.85"> }</a>
<span class="sourceLineNo">086</span><a name="line.86"> }</a>
<span class="sourceLineNo">087</span><a name="line.87"> } else if (lowerKey.startsWith(DEFAULT)) {</a>
<span class="sourceLineNo">088</span><a name="line.88"> switch (env.getDefaultProfiles().length) {</a>
<span class="sourceLineNo">089</span><a name="line.89"> case 0: {</a>
<span class="sourceLineNo">090</span><a name="line.90"> return null;</a>
<span class="sourceLineNo">091</span><a name="line.91"> }</a>
<span class="sourceLineNo">092</span><a name="line.92"> case 1: {</a>
<span class="sourceLineNo">093</span><a name="line.93"> return env.getDefaultProfiles()[0];</a>
<span class="sourceLineNo">094</span><a name="line.94"> }</a>
<span class="sourceLineNo">095</span><a name="line.95"> default: {</a>
<span class="sourceLineNo">096</span><a name="line.96"> Matcher matcher = DEFAULT_PATTERN.matcher(key);</a>
<span class="sourceLineNo">097</span><a name="line.97"> if (matcher.matches()) {</a>
<span class="sourceLineNo">098</span><a name="line.98"> try {</a>
<span class="sourceLineNo">099</span><a name="line.99"> int index = Integer.parseInt(matcher.group(1));</a>
<span class="sourceLineNo">100</span><a name="line.100"> if (index &lt; env.getDefaultProfiles().length) {</a>
<span class="sourceLineNo">101</span><a name="line.101"> return env.getDefaultProfiles()[index];</a>
<span class="sourceLineNo">102</span><a name="line.102"> } else {</a>
<span class="sourceLineNo">103</span><a name="line.103"> LOGGER.warn("Index out of bounds for Spring default profiles: {}", index);</a>
<span class="sourceLineNo">104</span><a name="line.104"> return null;</a>
<span class="sourceLineNo">105</span><a name="line.105"> }</a>
<span class="sourceLineNo">106</span><a name="line.106"> } catch (Exception ex) {</a>
<span class="sourceLineNo">107</span><a name="line.107"> LOGGER.warn("Unable to parse {} as integer value", matcher.group(1));</a>
<span class="sourceLineNo">108</span><a name="line.108"> return null;</a>
<span class="sourceLineNo">109</span><a name="line.109"> }</a>
<span class="sourceLineNo">110</span><a name="line.110"></a>
<span class="sourceLineNo">111</span><a name="line.111"> } else {</a>
<span class="sourceLineNo">112</span><a name="line.112"> StringBuilder sb = new StringBuilder();</a>
<span class="sourceLineNo">113</span><a name="line.113"> for (String profile : env.getDefaultProfiles()) {</a>
<span class="sourceLineNo">114</span><a name="line.114"> if (sb.length() &gt; 0) {</a>
<span class="sourceLineNo">115</span><a name="line.115"> sb.append(",");</a>
<span class="sourceLineNo">116</span><a name="line.116"> }</a>
<span class="sourceLineNo">117</span><a name="line.117"> sb.append(profile);</a>
<span class="sourceLineNo">118</span><a name="line.118"> }</a>
<span class="sourceLineNo">119</span><a name="line.119"> return sb.toString();</a>
<span class="sourceLineNo">120</span><a name="line.120"> }</a>
<span class="sourceLineNo">121</span><a name="line.121"> }</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>
<span class="sourceLineNo">125</span><a name="line.125"> return env.getProperty(key);</a>
<span class="sourceLineNo">126</span><a name="line.126"></a>
<span class="sourceLineNo">127</span><a name="line.127"> }</a>
<span class="sourceLineNo">128</span><a name="line.128"> return null;</a>
<span class="sourceLineNo">129</span><a name="line.129"> }</a>
<span class="sourceLineNo">130</span><a name="line.130"></a>
<span class="sourceLineNo">131</span><a name="line.131"> @Override</a>
<span class="sourceLineNo">132</span><a name="line.132"> public String lookup(LogEvent event, String key) {</a>
<span class="sourceLineNo">133</span><a name="line.133"> return lookup((key));</a>
<span class="sourceLineNo">134</span><a name="line.134"> }</a>
<span class="sourceLineNo">135</span><a name="line.135">}</a>
</pre>
</div>
</body>
</html>