ClickHouse

1
#include "OwnPatternFormatter.h"
2

3
#include <functional>
4
#include <IO/WriteBufferFromString.h>
5
#include <IO/WriteHelpers.h>
6
#include <Common/HashTable/Hash.h>
7
#include <base/terminalColors.h>
8

9

10
OwnPatternFormatter::OwnPatternFormatter(bool color_)
11
    : Poco::PatternFormatter(""), color(color_)
12
{
13
}
14

15

16
void OwnPatternFormatter::formatExtended(const DB::ExtendedLogMessage & msg_ext, std::string & text) const
17
{
18
    DB::WriteBufferFromString wb(text);
19

20
    const Poco::Message & msg = msg_ext.base;
21

22
    /// Change delimiters in date for compatibility with old logs.
23
    DB::writeDateTimeText<'.', ':'>(msg_ext.time_seconds, wb, server_timezone);
24

25
    DB::writeChar('.', wb);
26
    DB::writeChar('0' + ((msg_ext.time_microseconds / 100000) % 10), wb);
27
    DB::writeChar('0' + ((msg_ext.time_microseconds / 10000) % 10), wb);
28
    DB::writeChar('0' + ((msg_ext.time_microseconds / 1000) % 10), wb);
29
    DB::writeChar('0' + ((msg_ext.time_microseconds / 100) % 10), wb);
30
    DB::writeChar('0' + ((msg_ext.time_microseconds / 10) % 10), wb);
31
    DB::writeChar('0' + ((msg_ext.time_microseconds / 1) % 10), wb);
32

33
    writeCString(" [ ", wb);
34
    if (color)
35
        writeString(setColor(intHash64(msg_ext.thread_id)), wb);
36
    DB::writeIntText(msg_ext.thread_id, wb);
37
    if (color)
38
        writeCString(resetColor(), wb);
39
    writeCString(" ] ", wb);
40

41
    /// We write query_id even in case when it is empty (no query context)
42
    /// just to be convenient for various log parsers.
43
    writeCString("{", wb);
44
    if (color)
45
        writeString(setColor(std::hash<std::string>()(msg_ext.query_id)), wb);
46
    DB::writeString(msg_ext.query_id, wb);
47
    if (color)
48
        writeCString(resetColor(), wb);
49
    writeCString("} ", wb);
50

51
    writeCString("<", wb);
52
    int priority = static_cast<int>(msg.getPriority());
53
    if (color)
54
        writeCString(setColorForLogPriority(priority), wb);
55
    DB::writeString(getPriorityName(priority), wb);
56
    if (color)
57
        writeCString(resetColor(), wb);
58
    writeCString("> ", wb);
59
    if (color)
60
        writeString(setColor(std::hash<std::string>()(msg.getSource())), wb);
61
    DB::writeString(msg.getSource(), wb);
62
    if (color)
63
        writeCString(resetColor(), wb);
64
    writeCString(": ", wb);
65
    DB::writeString(msg.getText(), wb);
66
}
67

68
void OwnPatternFormatter::format(const Poco::Message & msg, std::string & text)
69
{
70
    formatExtended(DB::ExtendedLogMessage::getFrom(msg), text);
71
}
72