Fix a segment fault of logger.

master
Chunting Gu 7 years ago
parent 8faa45defe
commit 2e2b45dd43

@ -121,38 +121,53 @@ static std::string GetThreadID() {
return ss.str(); return ss.str();
} }
static void WriteToFile(FILE* fd, int level, const char* file, int line, void LogWrite(int level, const char* file, int line, const char* format, ...) {
const char* format, va_list args) { assert(format != nullptr);
std::string timestamp = GetTimestamp();
std::string thread_id = GetThreadID();
if ((g_logger.modes & LOG_FILE) != 0 && g_logger.file != nullptr) {
std::lock_guard<std::mutex> lock(g_logger.mutex); std::lock_guard<std::mutex> lock(g_logger.mutex);
fprintf(fd, "%s, %s, %7s, %24s, %4d, ", va_list args;
GetTimestamp().c_str(), kLevelNames[level], GetThreadID().c_str(), va_start(args, format);
fprintf(g_logger.file, "%s, %s, %7s, %24s, %4d, ",
timestamp.c_str(), kLevelNames[level], thread_id.c_str(),
file, line); file, line);
vfprintf(fd, format, args); vfprintf(g_logger.file, format, args);
fprintf(fd, "\n"); fprintf(g_logger.file, "\n");
if ((g_logger.modes & LOG_FLUSH) != 0) { if ((g_logger.modes & LOG_FLUSH) != 0) {
fflush(fd); fflush(g_logger.file);
} }
}
void LogWrite(int level, const char* file, int line, const char* format, ...) { va_end(args);
assert(format != nullptr); }
if ((g_logger.modes & LOG_CONSOLE) != 0) {
std::lock_guard<std::mutex> lock(g_logger.mutex);
va_list args; va_list args;
va_start(args, format); va_start(args, format);
if ((g_logger.modes & LOG_FILE) != 0 && g_logger.file != nullptr) { fprintf(stderr, "%s, %s, %7s, %24s, %4d, ",
WriteToFile(g_logger.file, level, file, line, format, args); timestamp.c_str(), kLevelNames[level], thread_id.c_str(),
} file, line);
if ((g_logger.modes & LOG_CONSOLE) != 0) { vfprintf(stderr, format, args);
WriteToFile(stderr, level, file, line, format, args);
fprintf(stderr, "\n");
if ((g_logger.modes & LOG_FLUSH) != 0) {
fflush(stderr);
} }
va_end(args); va_end(args);
}
} }
} // namespace webcc } // namespace webcc

Loading…
Cancel
Save