#include <cstdarg>
#include <cstdio>
#include <mutex>
#include <unordered_map>
#include "Common/LogReporting.h"
namespace Reporting {
static std::unordered_map<const char *, int> logNTimes;
static std::mutex logNTimesLock;
AllowedCallback allowedCallback = nullptr;
MessageCallback messageCallback = nullptr;
bool ShouldLogNTimes(const char *identifier, int count) {
std::lock_guard<std::mutex> lock(logNTimesLock);
auto iter = logNTimes.find(identifier);
if (iter == logNTimes.end()) {
logNTimes.emplace(identifier, 1);
return true;
} else {
if (iter->second >= count) {
return false;
} else {
iter->second++;
return true;
}
}
}
void ResetCounts() {
std::lock_guard<std::mutex> lock(logNTimesLock);
logNTimes.clear();
}
void SetupCallbacks(AllowedCallback allowed, MessageCallback message) {
allowedCallback = allowed;
messageCallback = message;
}
void ReportMessage(const char *message, ...) {
if (!allowedCallback || !messageCallback) {
ERROR_LOG(Log::System, "Reporting not initialized, skipping: %s", message);
return;
}
if (!allowedCallback())
return;
const int MESSAGE_BUFFER_SIZE = 65536;
char *temp = new char [MESSAGE_BUFFER_SIZE];
va_list args;
va_start(args, message);
vsnprintf(temp, MESSAGE_BUFFER_SIZE - 1, message, args);
temp[MESSAGE_BUFFER_SIZE - 1] = '\0';
va_end(args);
messageCallback(message, temp);
delete[] temp;
}
void ReportMessageFormatted(const char *message, const char *formatted) {
if (!allowedCallback || !messageCallback) {
ERROR_LOG(Log::System, "Reporting not initialized, skipping: %s", message);
return;
}
if (!allowedCallback())
return;
messageCallback(message, formatted);
}
}