73 lines
1.6 KiB
C
73 lines
1.6 KiB
C
|
/*
|
||
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
||
|
|
* All rights reserved.
|
||
|
|
*
|
||
|
|
* This source code is licensed under both the BSD-style license (found in the
|
||
|
|
* LICENSE file in the root directory of this source tree) and the GPLv2 (found
|
||
|
|
* in the COPYING file in the root directory of this source tree).
|
||
|
|
*/
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <cstdio>
|
||
|
|
#include <mutex>
|
||
|
|
|
||
|
|
namespace pzstd {
|
||
|
|
|
||
|
|
constexpr int kLogError = 1;
|
||
|
|
constexpr int kLogInfo = 2;
|
||
|
|
constexpr int kLogDebug = 3;
|
||
|
|
constexpr int kLogVerbose = 4;
|
||
|
|
|
||
|
|
class Logger {
|
||
|
|
std::mutex mutex_;
|
||
|
|
FILE* out_;
|
||
|
|
const int level_;
|
||
|
|
|
||
|
|
using Clock = std::chrono::system_clock;
|
||
|
|
Clock::time_point lastUpdate_;
|
||
|
|
std::chrono::milliseconds refreshRate_;
|
||
|
|
|
||
|
|
public:
|
||
|
|
explicit Logger(int level, FILE* out = stderr)
|
||
|
|
: out_(out), level_(level), lastUpdate_(Clock::now()),
|
||
|
|
refreshRate_(150) {}
|
||
|
|
|
||
|
|
|
||
|
|
bool logsAt(int level) {
|
||
|
|
return level <= level_;
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename... Args>
|
||
|
|
void operator()(int level, const char *fmt, Args... args) {
|
||
|
|
if (level > level_) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
||
|
|
std::fprintf(out_, fmt, args...);
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename... Args>
|
||
|
|
void update(int level, const char *fmt, Args... args) {
|
||
|
|
if (level > level_) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
||
|
|
auto now = Clock::now();
|
||
|
|
if (now - lastUpdate_ > refreshRate_) {
|
||
|
|
lastUpdate_ = now;
|
||
|
|
std::fprintf(out_, "\r");
|
||
|
|
std::fprintf(out_, fmt, args...);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void clear(int level) {
|
||
|
|
if (level > level_) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
||
|
|
std::fprintf(out_, "\r%79s\r", "");
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
}
|