/* 
Log File Library(WIN98/NT/2000) Compile by： BC++ 5; C++ BUILDER; VC++; VC.NET; copyright(c) 2004.5 - 2005.3 llbird wushaojian@21cn.com http://blog.csdn.net/wujian53 
*/
/* Use://这个代码我用工业现场24X7值守的程序纪录各种信息， 简单易用;//一般用一个全局日志对象， 有临界排斥可以多线程安全使用。//有两个类class LogFile;//用户定义日志文件名class LogFileEx;//有日志文件名自动生成功能 , 可分年月日频率生成文件名, 可指定日志存放的目录LogFile gLog("My.Log");gLog.Log("test", 4);//记录日志gLog.Log("系统启动");LogFileEx gLog(".", LogFileEx :: YEAR);//一年生成一个日志文件LogFileEx gLog(".\\Log", LogFileEx :: MONTH);//一月生成一个日志文件LogFileEx gLog(".\\Log", LogFileEx :: DAY);//一天生成一个日志文件//注意日志所属目录创建失败会自动退出， 请注意目录的合法性， 文件生成频率看情况掌握//24小时运行的程序可以每天生成一个日志文件， 以免内容过多 */
#ifndef _LOGFILE_H
#define
_LOGFILE_H
#include
<
assert.h
>
#include
<
time.h
>
#include
<
stdio.h
>
#include
<
windows.h
>
class
LogFile
{ protected : CRITICAL_SECTION _csLock; char * _szFileName; HANDLE _hFile; bool OpenFile() // 打开文件， 指针到文件尾 
{ if (IsOpen()) return true ; if ( ! _szFileName) return false ; _hFile = CreateFile( _szFileName, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL ); if ( ! IsOpen() && GetLastError() == 2 ) // 打开不成功， 且因为文件不存在， 创建文件 _hFile = CreateFile( _szFileName, GENERIC_WRITE, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL ); if (IsOpen()) SetFilePointer(_hFile, 0 , NULL, FILE_END); return IsOpen();
} DWORD Write(LPCVOID lpBuffer, DWORD dwLength) { DWORD dwWriteLength = 0 ; if (IsOpen()) WriteFile(_hFile, lpBuffer, dwLength, & dwWriteLength, NULL); return dwWriteLength; } virtual void WriteLog( LPCVOID lpBuffer, DWORD dwLength) // 写日志, 可以扩展修改 { time_t now; char temp[ 21 ]; DWORD dwWriteLength; if (IsOpen()) { time( & now); strftime(temp, 20 , " %Y-%m-%d %H:%M:%S " , localtime( & now)); WriteFile(_hFile, " \xd\xa#----------------------------- " , 32 , & dwWriteLength, NULL); WriteFile(_hFile, temp, 19 , & dwWriteLength, NULL); WriteFile(_hFile, " -----------------------------#\xd\xa " , 32 , & dwWriteLength, NULL); WriteFile(_hFile, lpBuffer, dwLength, & dwWriteLength, NULL); WriteFile(_hFile, " \xd\xa " , 2 , & dwWriteLength, NULL); FlushFileBuffers(_hFile); } } void Lock() { ::EnterCriticalSection( & _csLock); } void Unlock() { ::LeaveCriticalSection( & _csLock); } public : LogFile( const char * szFileName = " Log.log " ) // 设定日志文件名 { _szFileName = NULL; _hFile = INVALID_HANDLE_VALUE; ::InitializeCriticalSection( & _csLock); SetFileName(szFileName); } virtual ~ LogFile() { ::DeleteCriticalSection( & _csLock); Close(); if (_szFileName) delete []_szFileName; } const char * GetFileName() { return _szFileName; } void SetFileName( const char * szName) // 修改文件名， 同时关闭上一个日志文件 { assert(szName); if (_szFileName) delete []_szFileName; Close(); _szFileName = new char [strlen(szName) + 1 ]; assert(_szFileName); strcpy(_szFileName, szName); } bool IsOpen() { return _hFile != INVALID_HANDLE_VALUE; } void Close() { if (IsOpen()) { CloseHandle(_hFile); _hFile = INVALID_HANDLE_VALUE; } } void Log(LPCVOID lpBuffer, DWORD dwLength) // 追加日志内容 { assert(lpBuffer); __try { Lock(); if ( ! OpenFile()) return ; WriteLog(lpBuffer, dwLength); } __finally { Unlock(); } } void Log( const char * szText) { Log(szText, strlen(szText)); } private : // 屏蔽函数 LogFile( const LogFile & ); LogFile & operator = ( const LogFile & );}
;
class
LogFileEx :
public
LogFile
{ protected : char * _szPath; char _szLastDate[ 9 ]; int _iType; void SetPath( const char * szPath) { assert(szPath); WIN32_FIND_DATA wfd; char temp[MAX_PATH + 1 ] = { 0 } ; if (FindFirstFile(szPath, & wfd) == INVALID_HANDLE_VALUE && CreateDirectory(szPath, NULL) == 0 ) { strcat(strcpy(temp, szPath), " Create Fail. Exit Now! Error ID : " ); ltoa(GetLastError(), temp + strlen(temp), 10 ); MessageBox(NULL, temp, " Class LogFileEx " , MB_OK); exit( 1 ); } else { GetFullPathName(szPath, MAX_PATH, temp, NULL); _szPath = new char [strlen(temp) + 1 ]; assert(_szPath); strcpy(_szPath, temp); } } public : enum LOG_TYPE {YEAR = 0 , MONTH = 1 , DAY = 2 } ; LogFileEx( const char * szPath = " . " , LOG_TYPE iType = MONTH) { _szPath = NULL; SetPath(szPath); _iType = iType; memset(_szLastDate, 0 , 9 ); } ~ LogFileEx() { if (_szPath) delete []_szPath; } const char * GetPath() { return _szPath; } void Log(LPCVOID lpBuffer, DWORD dwLength) { assert(lpBuffer); char temp[ 10 ]; static const char format[ 3 ][ 10 ] = { " %Y " , " %Y-%m " , " %Y%m%d " } ; __try { Lock(); time_t now = time(NULL); strftime(temp, 9 , format[_iType], localtime( & now)); if (strcmp(_szLastDate, temp) != 0 ) // 更换文件名 { strcat(strcpy(_szFileName, _szPath), " \\ " ); strcat(strcat(_szFileName, temp), " .log " ); strcpy(_szLastDate, temp); Close(); } if ( ! OpenFile()) return ; WriteLog(lpBuffer, dwLength); } __finally { Unlock(); } } void Log( const char * szText) { Log(szText, strlen(szText)); } private : // 屏蔽函数 LogFileEx( const LogFileEx & ); LogFileEx & operator = ( const LogFileEx & );}
;
#endif
