ClickHouse/src/DataStreams/InternalTextLogsRowOutputStream.cpp
alexey-milovidov b8e3655f0d
Merge pull request #10445 from azat/logger-source-coloring-client
Make logger name colored in client too
2020-04-23 00:35:40 +03:00

106 lines
3.8 KiB
C++

#include "InternalTextLogsRowOutputStream.h"
#include <Core/Block.h>
#include <Interpreters/InternalTextLogsQueue.h>
#include <Common/typeid_cast.h>
#include <Common/HashTable/Hash.h>
#include <DataTypes/IDataType.h>
#include <Columns/ColumnsNumber.h>
#include <Columns/ColumnString.h>
#include <IO/WriteHelpers.h>
#include <common/terminalColors.h>
namespace DB
{
Block InternalTextLogsRowOutputStream::getHeader() const
{
return InternalTextLogsQueue::getSampleBlock();
}
void InternalTextLogsRowOutputStream::write(const Block & block)
{
const auto & array_event_time = typeid_cast<const ColumnUInt32 &>(*block.getByName("event_time").column).getData();
const auto & array_microseconds = typeid_cast<const ColumnUInt32 &>(*block.getByName("event_time_microseconds").column).getData();
const auto & column_host_name = typeid_cast<const ColumnString &>(*block.getByName("host_name").column);
const auto & column_query_id = typeid_cast<const ColumnString &>(*block.getByName("query_id").column);
const auto & array_thread_id = typeid_cast<const ColumnUInt64 &>(*block.getByName("thread_id").column).getData();
const auto & array_priority = typeid_cast<const ColumnInt8 &>(*block.getByName("priority").column).getData();
const auto & column_source = typeid_cast<const ColumnString &>(*block.getByName("source").column);
const auto & column_text = typeid_cast<const ColumnString &>(*block.getByName("text").column);
for (size_t row_num = 0; row_num < block.rows(); ++row_num)
{
auto host_name = column_host_name.getDataAt(row_num);
if (host_name.size)
{
writeCString("[", wb);
if (color)
writeString(setColor(StringRefHash()(host_name)), wb);
writeString(host_name, wb);
if (color)
writeCString(resetColor(), wb);
writeCString("] ", wb);
}
auto event_time = array_event_time[row_num];
writeDateTimeText<'.', ':'>(event_time, wb);
auto microseconds = array_microseconds[row_num];
writeChar('.', wb);
writeChar('0' + ((microseconds / 100000) % 10), wb);
writeChar('0' + ((microseconds / 10000) % 10), wb);
writeChar('0' + ((microseconds / 1000) % 10), wb);
writeChar('0' + ((microseconds / 100) % 10), wb);
writeChar('0' + ((microseconds / 10) % 10), wb);
writeChar('0' + ((microseconds / 1) % 10), wb);
UInt64 thread_id = array_thread_id[row_num];
writeCString(" [ ", wb);
if (color)
writeString(setColor(intHash64(thread_id)), wb);
writeIntText(thread_id, wb);
if (color)
writeCString(resetColor(), wb);
writeCString(" ]", wb);
auto query_id = column_query_id.getDataAt(row_num);
if (query_id.size)
{
writeCString(" {", wb);
if (color)
writeString(setColor(StringRefHash()(query_id)), wb);
writeString(query_id, wb);
if (color)
writeCString(resetColor(), wb);
writeCString("}", wb);
}
Int8 priority = array_priority[row_num];
writeCString(" <", wb);
if (color)
writeCString(setColorForLogPriority(priority), wb);
writeString(InternalTextLogsQueue::getPriorityName(priority), wb);
if (color)
writeCString(resetColor(), wb);
writeCString("> ", wb);
auto source = column_source.getDataAt(row_num);
if (color)
writeString(setColor(StringRefHash()(source)), wb);
DB::writeString(source, wb);
if (color)
writeCString(resetColor(), wb);
writeCString(": ", wb);
auto text = column_text.getDataAt(row_num);
writeString(text, wb);
writeChar('\n', wb);
}
}
}