You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.1 KiB
105 lines
3.1 KiB
#include "base/log.h"
|
|
|
|
DEFINE_int32(logtostderr, 0, "logging to stderr");
|
|
|
|
namespace ppspeech {
|
|
|
|
static char __progname[] = "paddlespeech";
|
|
|
|
namespace log {
|
|
|
|
std::mutex LogMessage::lock_;
|
|
std::string LogMessage::s_debug_logfile_("");
|
|
std::string LogMessage::s_info_logfile_("");
|
|
std::string LogMessage::s_warning_logfile_("");
|
|
std::string LogMessage::s_error_logfile_("");
|
|
std::string LogMessage::s_fatal_logfile_("");
|
|
|
|
void LogMessage::get_curr_proc_info(std::string* pid, std::string* proc_name) {
|
|
std::stringstream ss;
|
|
ss << getpid();
|
|
ss >> *pid;
|
|
*proc_name = ::ppspeech::__progname;
|
|
}
|
|
|
|
LogMessage::LogMessage(const char* file,
|
|
int line,
|
|
Severity level,
|
|
bool verbose,
|
|
bool out_to_file /* = false */)
|
|
: level_(level), verbose_(verbose), out_to_file_(out_to_file) {
|
|
if (FLAGS_logtostderr == 0) {
|
|
stream_ = static_cast<std::ostream*>(&std::cout);
|
|
} else if (FLAGS_logtostderr == 1) {
|
|
stream_ = static_cast<std::ostream*>(&std::cerr);
|
|
} else if (out_to_file_) {
|
|
// logfile
|
|
lock_.lock();
|
|
init(file, line);
|
|
}
|
|
}
|
|
|
|
LogMessage::~LogMessage() {
|
|
stream() << std::endl;
|
|
|
|
if (out_to_file_) {
|
|
lock_.unlock();
|
|
}
|
|
|
|
if (verbose_ && level_ == FATAL) {
|
|
std::abort();
|
|
}
|
|
}
|
|
|
|
std::ostream* LogMessage::nullstream() {
|
|
thread_local static std::ofstream os;
|
|
thread_local static bool flag_set = false;
|
|
if (!flag_set) {
|
|
os.setstate(std::ios_base::badbit);
|
|
flag_set = true;
|
|
}
|
|
return &os;
|
|
}
|
|
|
|
void LogMessage::init(const char* file, int line) {
|
|
time_t t = time(0);
|
|
char tmp[100];
|
|
strftime(tmp, sizeof(tmp), "%Y%m%d-%H%M%S", localtime(&t));
|
|
|
|
if (s_info_logfile_.empty()) {
|
|
std::string pid;
|
|
std::string proc_name;
|
|
get_curr_proc_info(&pid, &proc_name);
|
|
|
|
s_debug_logfile_ =
|
|
std::string("log." + proc_name + ".log.DEBUG." + tmp + "." + pid);
|
|
s_info_logfile_ =
|
|
std::string("log." + proc_name + ".log.INFO." + tmp + "." + pid);
|
|
s_warning_logfile_ =
|
|
std::string("log." + proc_name + ".log.WARNING." + tmp + "." + pid);
|
|
s_error_logfile_ =
|
|
std::string("log." + proc_name + ".log.ERROR." + tmp + "." + pid);
|
|
s_fatal_logfile_ =
|
|
std::string("log." + proc_name + ".log.FATAL." + tmp + "." + pid);
|
|
}
|
|
|
|
thread_local static std::ofstream ofs;
|
|
if (level_ == DEBUG) {
|
|
ofs.open(s_debug_logfile_.c_str(), std::ios::out | std::ios::app);
|
|
} else if (level_ == INFO) {
|
|
ofs.open(s_info_logfile_.c_str(), std::ios::out | std::ios::app);
|
|
} else if (level_ == WARNING) {
|
|
ofs.open(s_warning_logfile_.c_str(), std::ios::out | std::ios::app);
|
|
} else if (level_ == ERROR) {
|
|
ofs.open(s_error_logfile_.c_str(), std::ios::out | std::ios::app);
|
|
} else {
|
|
ofs.open(s_fatal_logfile_.c_str(), std::ios::out | std::ios::app);
|
|
}
|
|
|
|
stream_ = &ofs;
|
|
|
|
stream() << tmp << " " << file << " line " << line << "; ";
|
|
stream() << std::flush;
|
|
}
|
|
} // namespace log
|
|
} // namespace ppspeech
|