wehub-resource-sync
127 lines
4.7 KiB
C++
127 lines
4.7 KiB
C++
//
|
|
// Created by MNN on 2024/12/18.
|
|
// Copyright (c) 2024 Alibaba Group Holding Limited All rights reserved.
|
|
//
|
|
|
|
#pragma once
|
|
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <memory>
|
|
|
|
namespace mnn::downloader {
|
|
|
|
// Terminal color constants for colored logging
|
|
namespace Colors {
|
|
const std::string RESET = "\033[0m";
|
|
const std::string RED = "\033[31m";
|
|
const std::string GREEN = "\033[32m";
|
|
const std::string YELLOW = "\033[33m";
|
|
const std::string BLUE = "\033[34m";
|
|
const std::string MAGENTA = "\033[35m";
|
|
const std::string CYAN = "\033[36m";
|
|
const std::string WHITE = "\033[37m";
|
|
const std::string BOLD = "\033[1m";
|
|
}
|
|
|
|
// Log levels enum
|
|
enum class LogLevel {
|
|
DEBUG_LEVEL,
|
|
INFO,
|
|
WARNING,
|
|
ERROR
|
|
};
|
|
|
|
// Log utility class for centralized logging
|
|
class LogUtils {
|
|
public:
|
|
// Set global verbose mode
|
|
static void SetVerbose(bool verbose) {
|
|
is_verbose_ = verbose;
|
|
}
|
|
|
|
// Check if verbose mode is enabled
|
|
static bool IsVerbose() {
|
|
return is_verbose_;
|
|
}
|
|
|
|
// Log methods with different levels
|
|
static void Debug(const std::string& message, const std::string& tag = "");
|
|
static void Info(const std::string& message, const std::string& tag = "");
|
|
static void Warning(const std::string& message, const std::string& tag = "");
|
|
static void Error(const std::string& message, const std::string& tag = "");
|
|
|
|
// Add formatted error method for printf-style formatting
|
|
static void ErrorFormatted(const char* format, ...);
|
|
|
|
// Overloaded error method for const char* (for backward compatibility)
|
|
static void Error(const char* message, const std::string& tag = "");
|
|
|
|
// Conditional debug logging (only outputs when verbose is enabled)
|
|
static void DebugIfVerbose(const std::string& message) {
|
|
if (is_verbose_) {
|
|
Debug(message);
|
|
}
|
|
}
|
|
|
|
// Template version for backward compatibility (deprecated)
|
|
template<typename... Args>
|
|
static void DebugIfVerbose(const std::string& format, Args... args) {
|
|
if (is_verbose_) {
|
|
// Format the message using sprintf-style formatting
|
|
char buffer[1024];
|
|
snprintf(buffer, sizeof(buffer), format.c_str(), args...);
|
|
Debug(std::string(buffer));
|
|
}
|
|
}
|
|
|
|
// Format file size for logging
|
|
static std::string FormatFileSize(int64_t bytes);
|
|
|
|
// Format progress percentage
|
|
static std::string FormatProgress(double progress);
|
|
|
|
// Get current timestamp string
|
|
static std::string GetTimestamp();
|
|
|
|
private:
|
|
static bool is_verbose_;
|
|
static std::string FormatMessage(LogLevel level, const std::string& message, const std::string& tag);
|
|
};
|
|
|
|
// Convenience macros for logging
|
|
#define LOG_DEBUG(...) mnn::downloader::LogUtils::DebugIfVerbose(__VA_ARGS__)
|
|
#define LOG_DEBUG_TAG(msg, tag) mnn::downloader::LogUtils::Debug(msg, tag)
|
|
#define LOG_INFO(msg) mnn::downloader::LogUtils::Info(msg)
|
|
#define LOG_WARNING(msg) mnn::downloader::LogUtils::Warning(msg)
|
|
|
|
// Smart LOG_ERROR macro that automatically chooses the right method based on argument count
|
|
#define LOG_ERROR_1(msg) mnn::downloader::LogUtils::Error(msg)
|
|
#define LOG_ERROR_2(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_3(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_4(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_5(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_6(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_7(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_8(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_9(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
#define LOG_ERROR_10(msg, ...) mnn::downloader::LogUtils::ErrorFormatted(msg, ##__VA_ARGS__)
|
|
|
|
#define GET_MACRO(_1, _2, _3, _4, _5, _6, _7, _8, _9, _10, NAME, ...) NAME
|
|
#define LOG_ERROR(...) GET_MACRO(__VA_ARGS__, LOG_ERROR_10, LOG_ERROR_9, LOG_ERROR_8, LOG_ERROR_7, LOG_ERROR_6, LOG_ERROR_5, LOG_ERROR_4, LOG_ERROR_3, LOG_ERROR_2, LOG_ERROR_1)(__VA_ARGS__)
|
|
|
|
// Special macro for string concatenation cases
|
|
#define LOG_ERROR_STR(msg) mnn::downloader::LogUtils::Error(msg)
|
|
|
|
// Conditional debug macro that only compiles when verbose is enabled
|
|
#ifdef DEBUG
|
|
#define VERBOSE_LOG(msg) mnn::downloader::LogUtils::Debug(msg)
|
|
#define VERBOSE_LOG_TAG(msg, tag) mnn::downloader::LogUtils::Debug(msg, tag)
|
|
#else
|
|
#define VERBOSE_LOG(msg) do {} while(0)
|
|
#define VERBOSE_LOG_TAG(msg, tag) do {} while(0)
|
|
#endif
|
|
|
|
} // namespace mnn::downloader
|