#include "StdAfx.h"
|
#include "DisplayMessage.h"
|
|
CDisplayMessage::CDisplayMessage(void)
|
{
|
m_pDM2P = NULL;
|
m_pFileLog = NULL;
|
|
InitializeCriticalSection(&m_csLog);
|
}
|
|
CDisplayMessage::~CDisplayMessage(void)
|
{
|
if (m_pFileLog)
|
{
|
delete m_pFileLog;
|
m_pFileLog = NULL;
|
}
|
|
DeleteCriticalSection(&m_csLog);
|
}
|
|
void CDisplayMessage::SetPath(const CString& strPath)
|
{
|
m_strLogPath = strPath;
|
|
CreateDirectory(m_strLogPath, NULL);
|
|
m_FSLog.SetRemainDelete(m_strLogPath, TARGETTYPE_FILE, 300);
|
|
MakeLogFile();
|
}
|
|
BOOL CDisplayMessage::MakeLogFile()
|
{
|
// Log 贸府
|
if (m_strLogPath.IsEmpty())
|
return FALSE;
|
|
m_FSLog.CommitSchedule();
|
|
m_TimeLogFile = CTime::GetCurrentTime();
|
m_strLogFile.Format(_T("%s\\%04d-%02d-%02d.log"), m_strLogPath, m_TimeLogFile.GetYear(), m_TimeLogFile.GetMonth(), m_TimeLogFile.GetDay());
|
|
if (m_pFileLog)
|
{
|
delete m_pFileLog;
|
m_pFileLog = NULL;
|
}
|
|
m_pFileLog = new CStdioFile();
|
|
return TRUE;
|
}
|
|
void CDisplayMessage::DisplayMessage(const CString& strMessage)
|
{
|
CTime time = CTime::GetCurrentTime();
|
CString strValue = _T("");
|
strValue.Format(_T("[%02d:%02d:%02d] %s"), time.GetHour(), time.GetMinute(), time.GetSecond(), strMessage);
|
|
WriteToFile(strValue);
|
|
if(m_pDM2P)
|
{
|
m_pDM2P->DisplayMessage((TCHAR*)(LPCTSTR)strValue);
|
}
|
}
|
|
void CDisplayMessage::DisplayMessage(const TCHAR* lpstrFormat, ...)
|
{
|
va_list list;
|
TCHAR strText[2000] = {0};
|
|
va_start(list, lpstrFormat);
|
_vstprintf_s(strText, lpstrFormat, list);
|
va_end(list);
|
|
CTime time = CTime::GetCurrentTime();
|
CString strValue = _T("");
|
strValue.Format(_T("[%02d:%02d:%02d] %s"), time.GetHour(), time.GetMinute(), time.GetSecond(), strText);
|
|
WriteToFile(strValue);
|
|
if(m_pDM2P)
|
{
|
m_pDM2P->DisplayMessage((TCHAR*)(LPCTSTR)strValue);
|
}
|
}
|
|
BOOL CDisplayMessage::WriteToFile(const CString& strMessage)
|
{
|
if(m_pFileLog == NULL) return FALSE;
|
|
EnterCriticalSection(&m_csLog);
|
|
CTime time = CTime::GetCurrentTime();
|
|
if ((time.GetMonth() != m_TimeLogFile.GetMonth()) || (time.GetDay() != m_TimeLogFile.GetDay()) || !m_pFileLog)
|
{
|
MakeLogFile();
|
}
|
|
if ( !m_pFileLog->Open(m_strLogFile, CFile::modeCreate | CFile::modeWrite | CFile::modeNoTruncate | CFile::shareDenyNone, &m_fileException) )
|
{
|
// #ifdef _DEBUG
|
// TCHAR szCause[255] = {0};
|
// ex.GetErrorMessage(szCause, 255);
|
// TRACE(szCause);
|
// #endif
|
LeaveCriticalSection(&m_csLog);
|
return FALSE;
|
}
|
|
CString strText;
|
strText.Format(_T("%s\r\n"), strMessage);
|
m_pFileLog->SeekToEnd();
|
m_pFileLog->WriteString((TCHAR*)(LPCTSTR)strText);
|
m_pFileLog->Close();
|
|
LeaveCriticalSection(&m_csLog);
|
|
return TRUE;
|
}
|
