package org.apache.logging.log4j.core.filter;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import org.apache.logging.log4j.Level;
import org.apache.logging.log4j.Marker;
import org.apache.logging.log4j.ThreadContext;
import org.apache.logging.log4j.core.Filter;
import org.apache.logging.log4j.core.LogEvent;
import org.apache.logging.log4j.core.Logger;
import org.apache.logging.log4j.core.config.Node;
import org.apache.logging.log4j.core.config.plugins.Plugin;
import org.apache.logging.log4j.core.config.plugins.PluginAttribute;
import org.apache.logging.log4j.core.config.plugins.PluginElement;
import org.apache.logging.log4j.core.config.plugins.PluginFactory;
import org.apache.logging.log4j.core.ContextDataInjector;
import org.apache.logging.log4j.core.impl.ContextDataInjectorFactory;
import org.apache.logging.log4j.core.util.KeyValuePair;
import org.apache.logging.log4j.message.Message;
import org.apache.logging.log4j.util.PerformanceSensitive;
import org.apache.logging.log4j.util.ReadOnlyStringMap;
@Plugin(name = "DynamicThresholdFilter", category = Node.CATEGORY, elementType = Filter.ELEMENT_TYPE, printObject = true)
@PerformanceSensitive("allocation")
public final class DynamicThresholdFilter extends AbstractFilter {
@PluginFactory
public static DynamicThresholdFilter createFilter(
@PluginAttribute("key") final String key,
@PluginElement("Pairs") final KeyValuePair[] pairs,
@PluginAttribute("defaultThreshold") final Level defaultThreshold,
@PluginAttribute("onMatch") final Result onMatch,
@PluginAttribute("onMismatch") final Result onMismatch) {
final Map<String, Level> map = new HashMap<>();
for (final KeyValuePair pair : pairs) {
map.put(pair.getKey(), Level.toLevel(pair.getValue()));
}
final Level level = defaultThreshold == null ? Level.ERROR : defaultThreshold;
return new DynamicThresholdFilter(key, map, level, onMatch, onMismatch);
}
private Level defaultThreshold = Level.ERROR;
private final String key;
private final ContextDataInjector injector = ContextDataInjectorFactory.createInjector();
private Map<String, Level> levelMap = new HashMap<>();
private DynamicThresholdFilter(final String key, final Map<String, Level> pairs, final Level defaultLevel,
final Result onMatch, final Result onMismatch) {
super(onMatch, onMismatch);
Objects.requireNonNull(key, "key cannot be null");
this.key = key;
this.levelMap = pairs;
this.defaultThreshold = defaultLevel;
}
@Override
public boolean equals(final Object obj) {
if (this == obj) {
return true;
}
if (!super.equalsImpl(obj)) {
return false;
}
if (getClass() != obj.getClass()) {
return false;
}
final DynamicThresholdFilter other = (DynamicThresholdFilter) obj;
if (defaultThreshold == null) {
if (other.defaultThreshold != null) {
return false;
}
} else if (!defaultThreshold.equals(other.defaultThreshold)) {
return false;
}
if (key == null) {
if (other.key != null) {
return false;
}
} else if (!key.equals(other.key)) {
return false;
}
if (levelMap == null) {
if (other.levelMap != null) {
return false;
}
} else if (!levelMap.equals(other.levelMap)) {
return false;
}
return true;
}
private Result filter(final Level level, final ReadOnlyStringMap contextMap) {
final String value = contextMap.getValue(key);
if (value != null) {
Level ctxLevel = levelMap.get(value);
if (ctxLevel == null) {
ctxLevel = defaultThreshold;
}
return level.isMoreSpecificThan(ctxLevel) ? onMatch : onMismatch;
}
return Result.NEUTRAL;
}
@Override
public Result filter(final LogEvent event) {
return filter(event.getLevel(), event.getContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final Message msg,
final Throwable t) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final Object msg,
final Throwable t) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object... params) {
return filter(level, currentContextData());
}
private ReadOnlyStringMap currentContextData() {
return injector.rawContextData();
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2, final Object p3) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2, final Object p3,
final Object p4) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2, final Object p3,
final Object p4, final Object p5) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2, final Object p3,
final Object p4, final Object p5, final Object p6) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2, final Object p3,
final Object p4, final Object p5, final Object p6,
final Object p7) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2, final Object p3,
final Object p4, final Object p5, final Object p6,
final Object p7, final Object p8) {
return filter(level, currentContextData());
}
@Override
public Result filter(final Logger logger, final Level level, final Marker marker, final String msg,
final Object p0, final Object p1, final Object p2, final Object p3,
final Object p4, final Object p5, final Object p6,
final Object p7, final Object p8, final Object p9) {
return filter(level, currentContextData());
}
public String getKey() {
return this.key;
}
public Map<String, Level> getLevelMap() {
return levelMap;
}
@Override
public int hashCode() {
final int prime = 31;
int result = super.hashCodeImpl();
result = prime * result + ((defaultThreshold == null) ? 0 : defaultThreshold.hashCode());
result = prime * result + ((key == null) ? 0 : key.hashCode());
result = prime * result + ((levelMap == null) ? 0 : levelMap.hashCode());
return result;
}
@Override
public String toString() {
final StringBuilder sb = new StringBuilder();
sb.append("key=").append(key);
sb.append(", default=").append(defaultThreshold);
if (levelMap.size() > 0) {
sb.append('{');
boolean first = true;
for (final Map.Entry<String, Level> entry : levelMap.entrySet()) {
if (!first) {
sb.append(", ");
first = false;
}
sb.append(entry.getKey()).append('=').append(entry.getValue());
}
sb.append('}');
}
return sb.toString();
}
}