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

Loading…
Cancel
Save