| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You 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 org.apache.log4j.rewrite; |
| |
| import org.apache.log4j.bridge.LogEventAdapter; |
| import org.apache.log4j.helpers.OptionConverter; |
| import org.apache.log4j.spi.LocationInfo; |
| import org.apache.log4j.spi.LoggingEvent; |
| import org.apache.logging.log4j.core.LogEvent; |
| import org.apache.logging.log4j.core.impl.Log4jLogEvent; |
| import org.apache.logging.log4j.message.SimpleMessage; |
| import org.apache.logging.log4j.util.SortedArrayStringMap; |
| |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Map; |
| import java.util.StringTokenizer; |
| |
| /** |
| * This policy rewrites events by adding |
| * a user-specified list of properties to the event. |
| * Existing properties are not modified. |
| * <p> |
| * The combination of the RewriteAppender and this policy |
| * performs the same actions as the PropertyFilter from log4j 1.3. |
| */ |
| |
| public class PropertyRewritePolicy implements RewritePolicy { |
| private Map<String, String> properties = Collections.EMPTY_MAP; |
| |
| public PropertyRewritePolicy() { |
| } |
| |
| /** |
| * Set a string representing the property name/value pairs. |
| * <p> |
| * Form: propname1=propvalue1,propname2=propvalue2 |
| * |
| * @param props The properties. |
| */ |
| public void setProperties(String props) { |
| Map hashTable = new HashMap(); |
| StringTokenizer pairs = new StringTokenizer(props, ","); |
| while (pairs.hasMoreTokens()) { |
| StringTokenizer entry = new StringTokenizer(pairs.nextToken(), "="); |
| hashTable.put(entry.nextElement().toString().trim(), entry.nextElement().toString().trim()); |
| } |
| synchronized (this) { |
| properties = hashTable; |
| } |
| } |
| |
| /** |
| * {@inheritDoc} |
| */ |
| public LoggingEvent rewrite(final LoggingEvent source) { |
| if (!properties.isEmpty()) { |
| Map<String, String> rewriteProps = source.getProperties() != null ? new HashMap<>(source.getProperties()) |
| : new HashMap<>(); |
| for (Map.Entry<String, String> entry : properties.entrySet()) { |
| if (!rewriteProps.containsKey(entry.getKey())) { |
| rewriteProps.put(entry.getKey(), entry.getValue()); |
| } |
| } |
| LogEvent event; |
| if (source instanceof LogEventAdapter) { |
| event = new Log4jLogEvent.Builder(((LogEventAdapter) source).getEvent()) |
| .setContextData(new SortedArrayStringMap(rewriteProps)) |
| .build(); |
| } else { |
| LocationInfo info = source.getLocationInformation(); |
| StackTraceElement element = new StackTraceElement(info.getClassName(), info.getMethodName(), |
| info.getFileName(), Integer.parseInt(info.getLineNumber())); |
| Thread thread = getThread(source.getThreadName()); |
| long threadId = thread != null ? thread.getId() : 0; |
| int threadPriority = thread != null ? thread.getPriority() : 0; |
| event = Log4jLogEvent.newBuilder() |
| .setContextData(new SortedArrayStringMap(rewriteProps)) |
| .setLevel(OptionConverter.convertLevel(source.getLevel())) |
| .setLoggerFqcn(source.getFQNOfLoggerClass()) |
| .setMarker(null) |
| .setMessage(new SimpleMessage(source.getRenderedMessage())) |
| .setSource(element) |
| .setLoggerName(source.getLoggerName()) |
| .setThreadName(source.getThreadName()) |
| .setThreadId(threadId) |
| .setThreadPriority(threadPriority) |
| .setThrown(source.getThrowableInformation().getThrowable()) |
| .setTimeMillis(source.getTimeStamp()) |
| .setNanoTime(0) |
| .setThrownProxy(null) |
| .build(); |
| } |
| return new LogEventAdapter(event); |
| } |
| return source; |
| } |
| |
| private Thread getThread(String name) { |
| for (Thread thread : Thread.getAllStackTraces().keySet()) { |
| if (thread.getName().equals(name)) { |
| return thread; |
| } |
| } |
| return null; |
| } |
| } |