Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,106 @@
package org.tron.common.log.layout;

import ch.qos.logback.classic.Logger;
import ch.qos.logback.classic.encoder.PatternLayoutEncoder;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.Context;
import ch.qos.logback.core.Layout;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class MultiLayoutPattern extends PatternLayoutEncoder {
private List<Rule> rules = new ArrayList<>();
private Map<String, Layout<ILoggingEvent>> layoutMap = new HashMap<>();

public void addRule(Rule rule) {
this.rules.add(rule);
rule.start(context);
}

public byte[] encode(ILoggingEvent event) {
Layout<ILoggingEvent> layout = getLayout(event.getLoggerName());
String txt = layout.doLayout(event);
return convertToBytes(txt);
}

private byte[] convertToBytes(String s) {
Charset charset = getCharset();
if (charset == null) {
return s.getBytes();
} else {
return s.getBytes(charset);
}
}

private Layout<ILoggingEvent> getLayout(final String name) {
if (name == null) {
throw new IllegalArgumentException("name cannot be null");
}
if (Logger.ROOT_LOGGER_NAME.equalsIgnoreCase(name)) {
return this.getLayout();
}
if (layoutMap.containsKey(name)) {
return layoutMap.get(name);
}
Layout<ILoggingEvent> layout = getLayout();
for (Rule rule : rules) {
if (rule.logger.equals(name)) {
layout = rule.getPatternLayoutEncoder().getLayout();
break;
}
}
layoutMap.put(name, layout);
return layout;
}

@Override
public void start() {
super.start();
}

public static class Rule {
private String logger;
private String pattern;
private PatternLayoutEncoder patternLayoutEncoder;
private boolean outputPatternAsHeader = false;

public String getLogger() {
return logger;
}

public void setLogger(String logger) {
this.logger = logger;
}

public String getPattern() {
return pattern;
}

public void setPattern(String pattern) {
this.pattern = pattern;
}

public boolean isOutputPatternAsHeader() {
return outputPatternAsHeader;
}

public void setOutputPatternAsHeader(boolean outputPatternAsHeader) {
this.outputPatternAsHeader = outputPatternAsHeader;
}

public PatternLayoutEncoder getPatternLayoutEncoder() {
return patternLayoutEncoder;
}

public void start(Context context) {
patternLayoutEncoder = new PatternLayoutEncoder();
patternLayoutEncoder.setPattern(pattern);
patternLayoutEncoder.setContext(context);
patternLayoutEncoder.setOutputPatternAsHeader(outputPatternAsHeader);
patternLayoutEncoder.start();
}
}
}