#include "pch.h"
|
#include "Log.h"
|
|
|
static const char* pszLevel[] = {" [Debug] ", " [Info] ", " [Warn] ", " [Error] "};
|
|
CLog::CLog()
|
{
|
m_nLevel = 0;
|
m_nOutputTarget = OT_TRACE;
|
m_bAutoAppendTime = TRUE;
|
m_strEquipmentId = _T("Unknown");
|
m_nDay = 0;
|
m_funOnLog = nullptr;
|
InitializeCriticalSection(&m_criticalSection);
|
}
|
|
|
CLog::~CLog()
|
{
|
DeleteCriticalSection(&m_criticalSection);
|
}
|
|
CLog *CLog::GetLog(void)
|
{
|
static CLog* pLog = NULL;
|
if (pLog == NULL) {
|
static CLog log;
|
pLog = &log;
|
}
|
|
return pLog;
|
}
|
|
void CLog::SetOnLogCallback(ONLOG funOnLog)
|
{
|
m_funOnLog = funOnLog;
|
}
|
|
void CLog::SetOutputTarget(int flag)
|
{
|
m_nOutputTarget = flag;
|
}
|
|
void CLog::SetEquipmentId(const char* pszEquipmentId)
|
{
|
m_strEquipmentId = pszEquipmentId;
|
}
|
|
void CLog::SetAutoAppendTimeString(BOOL bAutoAppendTime)
|
{
|
m_bAutoAppendTime = bAutoAppendTime;
|
}
|
|
void CLog::Batch()
|
{
|
if (m_file.m_hFile != CFile::hFileNull) {
|
m_file.Close();
|
}
|
}
|
|
BOOL CLog::BatchAndNew(int& nDay)
|
{
|
Batch();
|
if ( (m_nOutputTarget & OT_FILE) && m_file.m_hFile == CFile::hFileNull) {
|
CString strFilepath;
|
BOOL bRet = m_file.Open(MakeFilepathD(strFilepath, nDay), CFile::modeCreate | CFile::modeNoTruncate | CFile::modeWrite | CFile::shareDenyWrite);
|
if (bRet) {
|
m_file.SeekToEnd();
|
}
|
return bRet;
|
}
|
|
return FALSE;
|
}
|
|
#define BUFFERSIZE 1024*10
|
void CLog::LogFormat(int nLevel, const char* pszTag, char* szMessage, ...)
|
{
|
// ¼ì²éÈÕÆÚÊÇ·ñÓб仯£¬Óб仯Ôò½áÅú
|
Lock();
|
_SYSTEMTIME sysTime;
|
GetLocalTime(&sysTime);
|
if(m_nDay != sysTime.wDay) {
|
int nDay = 0;
|
if (BatchAndNew(nDay)) {
|
m_nDay = nDay;
|
}
|
}
|
Unlock();
|
|
if (nLevel < m_nLevel) {
|
return;
|
}
|
|
char szFullMessage[BUFFERSIZE];
|
char szFormatMessage[BUFFERSIZE];
|
|
// format message
|
va_list ap;
|
va_start(ap, szMessage);
|
_vsnprintf_s(szFormatMessage, BUFFERSIZE, szMessage, ap);
|
va_end(ap);
|
|
if (m_bAutoAppendTime) {
|
CString strTime;
|
strcpy_s(szFullMessage, BUFFERSIZE, (LPTSTR)(LPCTSTR)GetCurTime(strTime));
|
}
|
strcat_s(szFullMessage, BUFFERSIZE, pszLevel[nLevel]);
|
strcat_s(szFullMessage, szFormatMessage);
|
strcat_s(szFullMessage, BUFFERSIZE, "\n");
|
|
if (m_nOutputTarget & OT_FILE) {
|
Lock();
|
if (m_file.m_hFile != CFile::hFileNull) {
|
m_file.WriteString(szFullMessage);
|
}
|
Unlock();
|
}
|
if (m_nOutputTarget & OT_ODSTRING) {
|
OutputDebugStringA(szFullMessage);
|
}
|
else if(m_nOutputTarget & OT_TRACE) {
|
TRACE(szFormatMessage);
|
}
|
|
if (m_funOnLog != nullptr) {
|
m_funOnLog(nLevel, szFullMessage);
|
}
|
}
|
|
void CLog::Log(int nLevel, const char* pszTag, const char* szMessage)
|
{
|
// ¼ì²éÈÕÆÚÊÇ·ñÓб仯£¬Óб仯Ôò½áÅú
|
Lock();
|
_SYSTEMTIME sysTime;
|
GetLocalTime(&sysTime);
|
if (m_nDay != sysTime.wDay) {
|
int nDay = 0;
|
if (BatchAndNew(nDay)) {
|
m_nDay = nDay;
|
}
|
}
|
Unlock();
|
|
if (nLevel < m_nLevel) {
|
return;
|
}
|
|
CString strMsg;
|
if (m_bAutoAppendTime) {
|
CString strTime;
|
GetCurTime(strTime);
|
strMsg.Append(strTime);
|
}
|
strMsg.Append(pszTag);
|
strMsg.Append(szMessage);
|
strMsg.Append("\n");
|
|
if (m_nOutputTarget & OT_FILE) {
|
Lock();
|
if (m_file.m_hFile != CFile::hFileNull) {
|
m_file.WriteString(strMsg);
|
}
|
Unlock();
|
}
|
if (m_nOutputTarget & OT_ODSTRING) {
|
OutputDebugStringA(strMsg);
|
}
|
else if (m_nOutputTarget & OT_TRACE) {
|
TRACE(strMsg);
|
}
|
|
if (m_funOnLog != nullptr) {
|
m_funOnLog(nLevel, strMsg);
|
}
|
}
|
|
CString& CLog::GetCurTime(CString& strTime)
|
{
|
_SYSTEMTIME sysTime;
|
GetLocalTime(&sysTime);
|
strTime.Format(_T("%d/%02d/%02d %02d:%02d:%02d.%03d"), sysTime.wYear, sysTime.wMonth, sysTime.wDay,
|
sysTime.wHour, sysTime.wMinute, sysTime.wSecond, sysTime.wMilliseconds);
|
return strTime;
|
}
|
|
CString& CLog::MakeFilepath(CString& strFilepath)
|
{
|
_SYSTEMTIME sysTime;
|
GetLocalTime(&sysTime);
|
strFilepath.Format(_T("%s\\Log(%s)_%d_%02d_%02d.log"), (LPTSTR)(LPCTSTR)m_strLogsDir,
|
(LPTSTR)(LPCTSTR)m_strEquipmentId,
|
sysTime.wYear, sysTime.wMonth, sysTime.wDay);
|
|
return strFilepath;
|
}
|
|
CString& CLog::MakeFilepathD(CString& strFilepath, int& day)
|
{
|
_SYSTEMTIME sysTime;
|
GetLocalTime(&sysTime);
|
strFilepath.Format(_T("%s\\Log(%s)_%d_%02d_%02d.log"), (LPTSTR)(LPCTSTR)m_strLogsDir,
|
(LPTSTR)(LPCTSTR)m_strEquipmentId,
|
sysTime.wYear, sysTime.wMonth, sysTime.wDay);
|
day = sysTime.wDay;
|
|
return strFilepath;
|
}
|
|
void CLog::SetLogsDir(CString strDir)
|
{
|
m_strLogsDir = strDir;
|
}
|
