java/sql-dk/src/info/globalcode/sql/dk/logging/ColorfulConsoleFormatter.java
author František Kučera <franta-hg@frantovo.cz>
Wed, 25 Dec 2013 02:04:57 +0100
branchv_0
changeset 59 5f745ae795a8
parent 55 f5ed7c4efacc
child 155 eb3676c6929b
permissions -rw-r--r--
logging: print stacktraces if level is less than INFO
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
55
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
/**
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
 * SQL-DK
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     3
 * Copyright © 2013 František Kučera (frantovo.cz)
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     4
 *
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     5
 * This program is free software: you can redistribute it and/or modify
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     6
 * it under the terms of the GNU General Public License as published by
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     7
 * the Free Software Foundation, either version 3 of the License, or
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     8
 * (at your option) any later version.
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     9
 *
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
 * This program is distributed in the hope that it will be useful,
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 * GNU General Public License for more details.
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    14
 *
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    15
 * You should have received a copy of the GNU General Public License
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    16
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    17
 */
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    18
package info.globalcode.sql.dk.logging;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    19
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    20
import info.globalcode.sql.dk.ColorfulPrintWriter;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    21
import static info.globalcode.sql.dk.ColorfulPrintWriter.TerminalColor;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    22
import static info.globalcode.sql.dk.ColorfulPrintWriter.TerminalStyle;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    23
import static info.globalcode.sql.dk.Functions.rpad;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    24
import java.io.StringWriter;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    25
import java.util.logging.Formatter;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    26
import java.util.logging.Level;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    27
import java.util.logging.LogRecord;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    28
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    29
/**
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    30
 *
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    31
 * @author Ing. František Kučera (frantovo.cz)
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    32
 */
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    33
public class ColorfulConsoleFormatter extends Formatter {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    34
59
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    35
	private boolean printStacktrace = false;
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    36
55
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    37
	@Override
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    38
	public String format(LogRecord r) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    39
		StringWriter sw = new StringWriter();
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    40
		try (ColorfulPrintWriter out = new ColorfulPrintWriter(sw)) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    41
			printLevel(out, r.getLevel());
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    42
			printMessage(out, r);
59
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    43
			printThrowable(out, r);
55
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    44
			out.println();
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    45
		}
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    46
		return sw.toString();
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    47
	}
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    48
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    49
	private void printLevel(ColorfulPrintWriter out, Level l) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    50
		TerminalColor color = TerminalColor.Magenta;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    51
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    52
		if (l == Level.SEVERE) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    53
			color = TerminalColor.Red;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    54
		} else if (l == Level.WARNING) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    55
			color = TerminalColor.Yellow;
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    56
		}
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    57
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    58
		out.print(color, rpad(l.getLocalizedName() + ": ", 10));
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    59
	}
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    60
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    61
	private void printMessage(ColorfulPrintWriter out, LogRecord r) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    62
		out.print(formatMessage(r));
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    63
	}
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    64
59
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    65
	private void printThrowable(ColorfulPrintWriter out, LogRecord r) {
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    66
		Throwable t = r.getThrown();
55
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    67
		if (t != null) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    68
			out.print(": ");
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    69
			out.print(TerminalColor.Red, t.getClass().getSimpleName());
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
			String message = t.getLocalizedMessage();
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    71
			if (message != null) {
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    72
				out.print(": ");
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    73
				out.print(message);
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    74
			}
59
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    75
			if (printStacktrace) {
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    76
				out.println();
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    77
				out.setForegroundColor(TerminalColor.Yellow);
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    78
				out.setStyle(TerminalStyle.Dim);
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    79
				t.printStackTrace(out);
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    80
				out.resetAll();
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    81
			}
55
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    82
		}
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    83
	}
59
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    84
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    85
	public boolean isPrintStacktrace() {
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    86
		return printStacktrace;
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    87
	}
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    88
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    89
	public void setPrintStacktrace(boolean printStacktrace) {
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    90
		this.printStacktrace = printStacktrace;
5f745ae795a8 logging: print stacktraces if level is less than INFO
František Kučera <franta-hg@frantovo.cz>
parents: 55
diff changeset
    91
	}
55
f5ed7c4efacc colorful logging
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    92
}