mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-10-08 01:20:55 +00:00
140 lines
4.1 KiB
C++
140 lines
4.1 KiB
C++
#include "OwnPatternFormatter.h"
|
|
|
|
#include <functional>
|
|
#include <optional>
|
|
#include <IO/WriteBufferFromString.h>
|
|
#include <IO/WriteHelpers.h>
|
|
#include <Common/HashTable/Hash.h>
|
|
#include <Interpreters/InternalTextLogsQueue.h>
|
|
#include <sys/time.h>
|
|
#include <Common/CurrentThread.h>
|
|
#include <common/getThreadId.h>
|
|
#include "Loggers.h"
|
|
|
|
|
|
static const char * setColor(UInt64 num)
|
|
{
|
|
/// ANSI escape sequences to set foreground font color in terminal.
|
|
|
|
static constexpr auto num_colors = 12;
|
|
static const char * colors[num_colors] =
|
|
{
|
|
/// Black on black is meaningless
|
|
"\033[0;31m",
|
|
"\033[0;32m",
|
|
"\033[0;33m",
|
|
/// Low intense blue on black is too dark.
|
|
"\033[0;35m",
|
|
"\033[0;36m",
|
|
"\033[1;31m",
|
|
"\033[1;32m",
|
|
"\033[1;33m",
|
|
"\033[1;34m",
|
|
"\033[1;35m",
|
|
"\033[1;36m",
|
|
"\033[1m", /// Not as white but just as high intense - for readability on white background.
|
|
};
|
|
|
|
return colors[num % num_colors];
|
|
}
|
|
|
|
static const char * setColorForLogPriority(int priority)
|
|
{
|
|
if (priority < 1 || priority > 8)
|
|
return "";
|
|
|
|
static const char * colors[] =
|
|
{
|
|
"",
|
|
"\033[1;41m", /// Fatal
|
|
"\033[7;31m", /// Critical
|
|
"\033[1;31m", /// Error
|
|
"\033[0;31m", /// Warning
|
|
"\033[0;33m", /// Notice
|
|
"\033[1m", /// Information
|
|
"", /// Debug
|
|
"\033[2m", /// Trace
|
|
};
|
|
|
|
return colors[priority];
|
|
}
|
|
|
|
static const char * resetColor()
|
|
{
|
|
return "\033[0m";
|
|
}
|
|
|
|
|
|
|
|
OwnPatternFormatter::OwnPatternFormatter(const Loggers * loggers_, OwnPatternFormatter::Options options_, bool color_)
|
|
: Poco::PatternFormatter(""), loggers(loggers_), options(options_), color(color_)
|
|
{
|
|
}
|
|
|
|
|
|
void OwnPatternFormatter::formatExtended(const DB::ExtendedLogMessage & msg_ext, std::string & text)
|
|
{
|
|
DB::WriteBufferFromString wb(text);
|
|
|
|
const Poco::Message & msg = msg_ext.base;
|
|
|
|
/// For syslog: tag must be before message and first whitespace.
|
|
/// This code is only used in Yandex.Metrika and unneeded in ClickHouse.
|
|
if ((options & ADD_LAYER_TAG) && loggers)
|
|
{
|
|
auto layer = loggers->getLayer();
|
|
if (layer)
|
|
{
|
|
writeCString("layer[", wb);
|
|
DB::writeIntText(*layer, wb);
|
|
writeCString("]: ", wb);
|
|
}
|
|
}
|
|
|
|
/// Change delimiters in date for compatibility with old logs.
|
|
DB::writeDateTimeText<'.', ':'>(msg_ext.time_seconds, wb);
|
|
|
|
DB::writeChar('.', wb);
|
|
DB::writeChar('0' + ((msg_ext.time_microseconds / 100000) % 10), wb);
|
|
DB::writeChar('0' + ((msg_ext.time_microseconds / 10000) % 10), wb);
|
|
DB::writeChar('0' + ((msg_ext.time_microseconds / 1000) % 10), wb);
|
|
DB::writeChar('0' + ((msg_ext.time_microseconds / 100) % 10), wb);
|
|
DB::writeChar('0' + ((msg_ext.time_microseconds / 10) % 10), wb);
|
|
DB::writeChar('0' + ((msg_ext.time_microseconds / 1) % 10), wb);
|
|
|
|
writeCString(" [ ", wb);
|
|
if (color)
|
|
writeCString(setColor(intHash64(msg_ext.thread_id)), wb);
|
|
DB::writeIntText(msg_ext.thread_id, wb);
|
|
if (color)
|
|
writeCString(resetColor(), wb);
|
|
writeCString(" ] ", wb);
|
|
|
|
/// We write query_id even in case when it is empty (no query context)
|
|
/// just to be convenient for various log parsers.
|
|
writeCString("{", wb);
|
|
if (color)
|
|
writeCString(setColor(std::hash<std::string>()(msg_ext.query_id)), wb);
|
|
DB::writeString(msg_ext.query_id, wb);
|
|
if (color)
|
|
writeCString(resetColor(), wb);
|
|
writeCString("} ", wb);
|
|
|
|
writeCString("<", wb);
|
|
int priority = static_cast<int>(msg.getPriority());
|
|
if (color)
|
|
writeCString(setColorForLogPriority(priority), wb);
|
|
DB::writeString(getPriorityName(priority), wb);
|
|
if (color)
|
|
writeCString(resetColor(), wb);
|
|
writeCString("> ", wb);
|
|
DB::writeString(msg.getSource(), wb);
|
|
writeCString(": ", wb);
|
|
DB::writeString(msg.getText(), wb);
|
|
}
|
|
|
|
void OwnPatternFormatter::format(const Poco::Message & msg, std::string & text)
|
|
{
|
|
formatExtended(DB::ExtendedLogMessage::getFrom(msg), text);
|
|
}
|