generated from zhangwei/Matrixai
84 lines
2.3 KiB
C++
84 lines
2.3 KiB
C++
#include "myshell.h"
|
||
|
||
namespace ai_matrix
|
||
{
|
||
const int TIME_SIZE = 32;
|
||
const int TIME_DIFF = 28800; // 8 hour
|
||
std::mutex MyShell::mutex;
|
||
uint32_t MyShell::shellLevel = SHELL_LEVEL_INFO;
|
||
std::vector<std::string> MyShell::levelString{"[Debug]", "[Info ]", "[Warn ]", "[Error]", "[Fatal]"};
|
||
|
||
MyShell::MyShell(std::string file, std::string function, int line, uint32_t level)
|
||
: myLevel_(level), file_(file), function_(function), line_(line)
|
||
{
|
||
}
|
||
|
||
MyShell::~MyShell()
|
||
{
|
||
if (myLevel_ >= shellLevel)
|
||
{
|
||
std::lock_guard<std::mutex> locker(mutex);
|
||
// cout to screen
|
||
std::cout << ss_.str() << std::endl;
|
||
}
|
||
};
|
||
|
||
std::ostringstream &MyShell::Stream()
|
||
{
|
||
if (myLevel_ >= shellLevel)
|
||
{
|
||
struct timeval time = {0, 0};
|
||
gettimeofday(&time, nullptr);
|
||
time_t timep = time.tv_sec + TIME_DIFF; //加上8小时
|
||
struct tm *ptm = gmtime(&timep); //返回tm结构的格林尼治时间（GMT）
|
||
char timeString[TIME_SIZE] = {0};
|
||
//%F 年-月-日 %X 标准的时间串
|
||
strftime(timeString, TIME_SIZE, "[%F %X:", ptm);
|
||
long int usValue = time.tv_usec;
|
||
|
||
date_ = timeString;
|
||
ss_.fill('0');
|
||
ss_ << levelString[myLevel_] << timeString << std::setw(3) << usValue / 1000 << "]";
|
||
|
||
std::string fileName = file_.substr(file_.rfind('/') + 1);
|
||
ss_ << "[" << fileName << " " << function_ << ":" << line_ << "] ";
|
||
}
|
||
return ss_;
|
||
}
|
||
|
||
void MyShell::ShellDebugOn()
|
||
{
|
||
shellLevel = SHELL_LEVEL_DEBUG;
|
||
return;
|
||
}
|
||
void MyShell::ShellInfoOn()
|
||
{
|
||
shellLevel = SHELL_LEVEL_INFO;
|
||
return;
|
||
}
|
||
void MyShell::ShellWarnOn()
|
||
{
|
||
shellLevel = SHELL_LEVEL_WARN;
|
||
return;
|
||
}
|
||
void MyShell::ShellErrorOn()
|
||
{
|
||
shellLevel = SHELL_LEVEL_ERROR;
|
||
return;
|
||
}
|
||
void MyShell::ShellFatalOn()
|
||
{
|
||
shellLevel = SHELL_LEVEL_FATAL;
|
||
return;
|
||
}
|
||
void MyShell::ShellAllOn()
|
||
{
|
||
shellLevel = SHELL_LEVEL_DEBUG;
|
||
return;
|
||
}
|
||
void MyShell::ShellAllOff()
|
||
{
|
||
shellLevel = SHELL_LEVEL_NONE;
|
||
return;
|
||
}
|
||
} // namespace matrix_common
|