ClickHouse/libs/libdaemon/include/daemon/OwnPatternFormatter.h

#pragma once


#include <Poco/PatternFormatter.h>
#include <Poco/Ext/ThreadNumber.h>
#include <IO/WriteBufferFromString.h>
#include <IO/WriteHelpers.h>
#include <daemon/BaseDaemon.h>

#include <experimental/optional>
#include <functional>


/** Форматирует по своему.
  * Некоторые детали невозможно получить, используя только Poco::PatternFormatter.
  *
  * Во-первых, используется номер потока не среди потоков Poco::Thread,
  *  а среди всех потоков, для которых был получен номер (см. ThreadNumber.h)
  *
  * Во-вторых, корректно выводится локальная дата и время.
  * Poco::PatternFormatter плохо работает с локальным временем,
  *  в ситуациях, когда в ближайшем будущем намечается отмена или введение daylight saving time.
  *  - см. исходники Poco и http://thread.gmane.org/gmane.comp.time.tz/8883
  *
  * Также сделан чуть более эффективным (что имеет мало значения).
  */
class OwnPatternFormatter : public Poco::PatternFormatter
{
public:
    enum Options
    {
        ADD_NOTHING = 0,
        ADD_LAYER_TAG = 1 << 0
    };

    OwnPatternFormatter(const BaseDaemon * daemon_, Options options_ = ADD_NOTHING) : Poco::PatternFormatter(""), daemon(daemon_), options(options_) {}

    void format(const Poco::Message & msg, std::string & text) override
    {
        DB::WriteBufferFromString wb(text);

        /// For syslog: tag must be before message and first whitespace.
        if (options & ADD_LAYER_TAG && daemon)
        {
            auto layer = daemon->getLayer();
            if (layer)
            {
                writeCString("layer[", wb);
                DB::writeIntText(*layer, wb);
                writeCString("]: ", wb);
            }
        }

        /// Output time with microsecond resolution.
        timeval tv;
        if (0 != gettimeofday(&tv, nullptr))
            DB::throwFromErrno("Cannot gettimeofday");

        /// Change delimiters in date for compatibility with old logs.
        DB::writeDateTimeText<'.', ':'>(tv.tv_sec, wb);

        DB::writeChar('.', wb);
        DB::writeChar('0' + ((tv.tv_usec / 100000) % 10), wb);
        DB::writeChar('0' + ((tv.tv_usec / 10000) % 10), wb);
        DB::writeChar('0' + ((tv.tv_usec / 1000) % 10), wb);
        DB::writeChar('0' + ((tv.tv_usec / 100) % 10), wb);
        DB::writeChar('0' + ((tv.tv_usec / 10) % 10), wb);
        DB::writeChar('0' + ((tv.tv_usec / 1) % 10), wb);

        writeCString(" [ ", wb);
        DB::writeIntText(Poco::ThreadNumber::get(), wb);
        writeCString(" ] <", wb);
        DB::writeString(getPriorityName(static_cast<int>(msg.getPriority())), wb);
        writeCString("> ", wb);
        DB::writeString(msg.getSource(), wb);
        writeCString(": ", wb);
        DB::writeString(msg.getText(), wb);
    }

private:
    const BaseDaemon * daemon;
    Options options;
};