sink.cpp

/*
 *     .============.
 *    //  M A K E  / \
 *   //  C++ DEV  /   \
 *  //  E A S Y  /  \/ \
 * ++ ----------.  \/\  .
 *  \\     \     \ /\  /
 *   \\     \     \   /
 *    \\     \     \ /
 *     -============'
 *
 * Copyright (c) 2018 Hevake and contributors, all rights reserved.
 *
 * This file is part of cpp-tbox (https://github.com/cpp-main/cpp-tbox)
 * Use of this source code is governed by MIT license that can be found
 * in the LICENSE file in the root of the source tree. All contributing
 * project authors may be found in the CONTRIBUTORS.md file in the root
 * of the source tree.
 */
#include "sink.h"

#include <cstring>
#include <functional>
#include <iostream>

#define LOG_MAX_LEN (100 << 10)     //! 限定单条日志最大长度

namespace tbox {
namespace log {

Sink::~Sink()
{
    disable();
}

void Sink::setLevel(int level)
{
    std::lock_guard<std::mutex> _lk(lock_);
    default_level_ = level;
}

void Sink::setLevel(const std::string &module, int level)
{
    std::lock_guard<std::mutex> _lk(lock_);
    if (module.empty())
        default_level_ = level;
    else
        modules_level_[module] = level;
}

void Sink::unsetLevel(const std::string &module)
{
    modules_level_.erase(module);
}

void Sink::enableColor(bool enable)
{
    enable_color_ = enable;
}

bool Sink::enable()
{
    using namespace std::placeholders;
    if (output_id_ == 0) {
        onEnable();
        output_id_ = LogAddPrintfFunc(HandleLog, this);
        return true;
    }
    return false;
}

void Sink::disable()
{
    if (output_id_ != 0) {
        LogRemovePrintfFunc(output_id_);
        onDisable();
        output_id_ = 0;
    }
}

bool Sink::filter(int level, const std::string &module)
{
    std::lock_guard<std::mutex> _lk(lock_);
    auto iter = modules_level_.find(module);
    if (iter != modules_level_.end())
        return level <= iter->second;
    else
        return level <= default_level_;
}

void Sink::HandleLog(const LogContent *content, void *ptr)
{
    Sink *pthis = static_cast<Sink*>(ptr);
    pthis->handleLog(content);
}

void Sink::handleLog(const LogContent *content)
{
    if (!filter(content->level, content->module_id))
        return;

    onLogFrontEnd(content);
}

void Sink::updateTimestampStr(uint32_t sec)
{
    if (timestamp_sec_ != sec) {
        time_t ts_sec = sec;
        struct tm tm;
        localtime_r(&ts_sec, &tm);
        strftime(timestamp_str_, sizeof(timestamp_str_), "%F %H:%M:%S", &tm);
        timestamp_sec_ = sec;
    }
}

}
}