package org.apache.juli;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.Writer;
import java.lang.management.ManagementFactory;
import java.lang.management.ThreadInfo;
import java.lang.management.ThreadMXBean;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.logging.Formatter;
import java.util.logging.LogManager;
import java.util.logging.LogRecord;
public class OneLineFormatter extends Formatter {
private static final String UNKNOWN_THREAD_NAME = "Unknown thread with ID ";
private static final Object threadMxBeanLock = new Object();
private static volatile ThreadMXBean threadMxBean = null;
private static final int THREAD_NAME_CACHE_SIZE = 10000;
private static ThreadLocal<ThreadNameCache> threadNameCache =
ThreadLocal.withInitial(() -> new ThreadNameCache(THREAD_NAME_CACHE_SIZE));
private static final String DEFAULT_TIME_FORMAT = "dd-MMM-yyyy HH:mm:ss.SSS";
private static final int globalCacheSize = 30;
private static final int localCacheSize = 5;
private ThreadLocal<DateFormatCache> localDateCache;
private volatile MillisHandling millisHandling = MillisHandling.APPEND;
public OneLineFormatter() {
String timeFormat = LogManager.getLogManager().getProperty(
OneLineFormatter.class.getName() + ".timeFormat");
if (timeFormat == null) {
timeFormat = DEFAULT_TIME_FORMAT;
}
setTimeFormat(timeFormat);
}
public void setTimeFormat(final String timeFormat) {
final String cachedTimeFormat;
if (timeFormat.endsWith(".SSS")) {
cachedTimeFormat = timeFormat.substring(0, timeFormat.length() - 4);
millisHandling = MillisHandling.APPEND;
} else if (timeFormat.contains("SSS")) {
millisHandling = MillisHandling.REPLACE_SSS;
cachedTimeFormat = timeFormat;
} else if (timeFormat.contains("SS")) {
millisHandling = MillisHandling.REPLACE_SS;
cachedTimeFormat = timeFormat;
} else if (timeFormat.contains("S")) {
millisHandling = MillisHandling.REPLACE_S;
cachedTimeFormat = timeFormat;
} else {
millisHandling = MillisHandling.NONE;
cachedTimeFormat = timeFormat;
}
final DateFormatCache globalDateCache =
new DateFormatCache(globalCacheSize, cachedTimeFormat, null);
localDateCache = ThreadLocal.withInitial(() -> new DateFormatCache(localCacheSize, cachedTimeFormat, globalDateCache));
}
public String getTimeFormat() {
return localDateCache.get().getTimeFormat();
}
@Override
public String format(LogRecord record) {
StringBuilder sb = new StringBuilder();
addTimestamp(sb, record.getMillis());
sb.append(' ');
sb.append(record.getLevel().getLocalizedName());
sb.append(' ');
sb.append('[');
if (Thread.currentThread() instanceof AsyncFileHandler.LoggerThread) {
sb.append(getThreadName(record.getThreadID()));
} else {
sb.append(Thread.currentThread().getName());
}
sb.append(']');
sb.append(' ');
sb.append(record.getSourceClassName());
sb.append('.');
sb.append(record.getSourceMethodName());
sb.append(' ');
sb.append(formatMessage(record));
sb.append(System.lineSeparator());
if (record.getThrown() != null) {
StringWriter sw = new StringWriter();
PrintWriter pw = new IndentingPrintWriter(sw);
record.getThrown().printStackTrace(pw);
pw.close();
sb.append(sw.getBuffer());
}
return sb.toString();
}
protected void addTimestamp(StringBuilder buf, long timestamp) {
String cachedTimeStamp = localDateCache.get().getFormat(timestamp);
if (millisHandling == MillisHandling.NONE) {
buf.append(cachedTimeStamp);
} else if (millisHandling == MillisHandling.APPEND) {
buf.append(cachedTimeStamp);
long frac = timestamp % 1000;
buf.append('.');
if (frac < 100) {
if (frac < 10) {
buf.append('0');
buf.append('0');
} else {
buf.append('0');
}
}
buf.append(frac);
} else {
long frac = timestamp % 1000;
int insertStart = cachedTimeStamp.indexOf(DateFormatCache.MSEC_PATTERN);
buf.append(cachedTimeStamp.subSequence(0, insertStart));
if (frac < 100 && millisHandling == MillisHandling.REPLACE_SSS) {
buf.append('0');
if (frac < 10) {
buf.append('0');
}
} else if (frac < 10 && millisHandling == MillisHandling.REPLACE_SS) {
buf.append('0');
}
buf.append(frac);
if (millisHandling == MillisHandling.REPLACE_SSS) {
buf.append(cachedTimeStamp.substring(insertStart + 3));
} else if (millisHandling == MillisHandling.REPLACE_SS) {
buf.append(cachedTimeStamp.substring(insertStart + 2));
} else {
buf.append(cachedTimeStamp.substring(insertStart + 1));
}
}
}
private static String getThreadName(int logRecordThreadId) {
Map<Integer,String> cache = threadNameCache.get();
String result = null;
if (logRecordThreadId > (Integer.MAX_VALUE / 2)) {
result = cache.get(Integer.valueOf(logRecordThreadId));
}
if (result != null) {
return result;
}
if (logRecordThreadId > Integer.MAX_VALUE / 2) {
result = UNKNOWN_THREAD_NAME + logRecordThreadId;
} else {
if (threadMxBean == null) {
synchronized (threadMxBeanLock) {
if (threadMxBean == null) {
threadMxBean = ManagementFactory.getThreadMXBean();
}
}
}
ThreadInfo threadInfo =
threadMxBean.getThreadInfo(logRecordThreadId);
if (threadInfo == null) {
return Long.toString(logRecordThreadId);
}
result = threadInfo.getThreadName();
}
cache.put(Integer.valueOf(logRecordThreadId), result);
return result;
}
private static class ThreadNameCache extends LinkedHashMap<Integer,String> {
private static final long serialVersionUID = 1L;
private final int cacheSize;
public ThreadNameCache(int cacheSize) {
this.cacheSize = cacheSize;
}
@Override
protected boolean removeEldestEntry(Entry<Integer, String> eldest) {
return (size() > cacheSize);
}
}
private static class IndentingPrintWriter extends PrintWriter {
public IndentingPrintWriter(Writer out) {
super(out);
}
@Override
public void println(Object x) {
super.print('\t');
super.println(x);
}
}
private static enum MillisHandling {
NONE,
APPEND,
REPLACE_S,
REPLACE_SS,
REPLACE_SSS,
}
}